jack 3 meses atrás
pai
commit
412a5cf22a
1 arquivos alterados com 13 adições e 2 exclusões
  1. 13 2
      app/models/run.py

+ 13 - 2
app/models/run.py

@@ -36,6 +36,17 @@ class ChatCompletionAudioParam(TypedDict, total=False):
 ChatCompletionModality: TypeAlias = Literal["text", "audio"]
 
 
+class StatusEnum(Enum):
+    cancelled = "cancelled"
+    cancelling = "cancelling"
+    completed = "completed"
+    expired = "expired"
+    failed = "failed"
+    in_progress = "in_progress"
+    queued = "queued"
+    requires_action = "requires_action"
+
+
 class RunBase(BaseModel):
     instructions: Optional[str] = Field(
         default=None, max_length=32768, sa_column=Column(TEXT)
@@ -94,7 +105,7 @@ class RunBase(BaseModel):
     temperature: Optional[float] = Field(default=None)  # 温度
     top_p: Optional[float] = Field(default=None)  # top_p
     # parallel_tool_calls: bool = Field(default=False)  # parallel_tool_calls
-    # audio: Optional[ChatCompletionAudioParam] = Field(default=None)  # audio
+    # audio: Optional[ChatCompletionAudioParam] = Field(default=None, sa_column=Column(JSON))  # audio
     # modalities: Optional[List[ChatCompletionModality]] = Field(default=None)
 
 
@@ -133,7 +144,7 @@ class RunCreate(BaseModel):
     temperature: Optional[float] = Field(default=None)  # 温度
     top_p: Optional[float] = Field(default=None)  # top_p
     # parallel_tool_calls: bool = Field(default=False)  # parallel_tool_calls
-    # audio: Optional[ChatCompletionAudioParam] = Field(default=None)  # audio
+    # audio: Optional[ChatCompletionAudioParam] = Field(default=None, sa_column=Column(JSON))  # audio
     # modalities: Optional[List[ChatCompletionModality]] = Field(default=None)
 
     @model_validator(mode="before")