jack 1 month ago
parent
commit
f3640fd2a6
1 changed files with 10 additions and 10 deletions
  1. 10 10
      app/models/run.py

+ 10 - 10
app/models/run.py

@@ -73,11 +73,11 @@ class RunBase(BaseModel):
     usage: Optional[dict] = Field(default=None, sa_column=Column(JSON))  # 调用使用情况
     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[dict] = Field(default=None, sa_column=Column(JSON))  # audio
-    # modalities: Optional[List[Literal["text", "audio"]]] = Field(
-    #    default=None, sa_column=Column(JSON)
-    # )
+    parallel_tool_calls: bool = Field(default=False)  # parallel_tool_calls
+    audio: Optional[dict] = Field(default=None, sa_column=Column(JSON))  # audio
+    modalities: Optional[List[Literal["text", "audio"]]] = Field(
+        default=None, sa_column=Column(JSON)
+    )
 
 
 class Run(RunBase, PrimaryKeyMixin, TimeStampMixin, table=True):
@@ -114,11 +114,11 @@ class RunCreate(BaseModel):
     tool_choice: Optional[str] = Field(default=None)  # 工具选择
     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[dict] = Field(default=None, sa_column=Column(JSON))  # audio
-    # modalities: Optional[List[Literal["text", "audio"]]] = Field(
-    #    default=None, sa_column=Column(JSON)
-    # )
+    parallel_tool_calls: bool = Field(default=False)  # parallel_tool_calls
+    audio: Optional[dict] = Field(default=None, sa_column=Column(JSON))  # audio
+    modalities: Optional[List[Literal["text", "audio"]]] = Field(
+        default=None, sa_column=Column(JSON)
+    )
 
     @model_validator(mode="before")
     def model_validator(cls, data: Any):