fix: pydantic schema issue

This commit is contained in:
Christopher Gondek 2025-10-09 10:54:11 +02:00
parent 478e139730
commit 8538821d0c

View file

@ -1,6 +1,7 @@
"""Chat models: ChatWorkflow, ChatMessage, ChatLog, ChatStat, ChatDocument."""
from typing import List, Dict, Any, Optional
from enum import Enum
from pydantic import BaseModel, Field
from modules.shared.attributeUtils import register_model_labels, ModelMixin
from modules.shared.timezoneUtils import get_utc_timestamp
@ -574,7 +575,7 @@ register_model_labels(
)
class TaskStatus(str):
class TaskStatus(str, Enum):
PENDING = "pending"
RUNNING = "running"
COMPLETED = "completed"