fix: pydantic schema issue
This commit is contained in:
parent
478e139730
commit
8538821d0c
1 changed files with 2 additions and 1 deletions
|
|
@ -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"
|
||||
|
|
|
|||
Loading…
Reference in a new issue