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