jack 1 月之前
父節點
當前提交
d880626ec0
共有 2 個文件被更改,包括 8 次插入2 次删除
  1. 1 1
      app/core/runner/pub_handler.py
  2. 7 1
      app/models/run.py

+ 1 - 1
app/core/runner/pub_handler.py

@@ -110,7 +110,7 @@ def _data_adjust(obj):
         "--------------------------------====================================11221212212121212121"
         "--------------------------------====================================11221212212121212121"
     )
     )
     print(data)
     print(data)
-    data["parallel_tool_calls"] = True
+    # data["parallel_tool_calls"] = True
     return data
     return data
 
 
 
 

+ 7 - 1
app/models/run.py

@@ -116,7 +116,9 @@ class RunCreate(BaseModel):
     tool_choice: Optional[str] = Field(default=None)  # 工具选择
     tool_choice: Optional[str] = Field(default=None)  # 工具选择
     temperature: Optional[float] = Field(default=None)  # 温度
     temperature: Optional[float] = Field(default=None)  # 温度
     top_p: Optional[float] = Field(default=None)  # top_p
     top_p: Optional[float] = Field(default=None)  # top_p
-    parallel_tool_calls: Optional[bool] = False  # parallel_tool_calls
+    parallel_tool_calls: bool = Field(
+        default=False, nullable=False
+    )  # parallel_tool_calls
     audio: Optional[dict] = Field(default=None, sa_column=Column(JSON))  # audio
     audio: Optional[dict] = Field(default=None, sa_column=Column(JSON))  # audio
     modalities: Optional[List[Literal["text", "audio"]]] = Field(
     modalities: Optional[List[Literal["text", "audio"]]] = Field(
         default=None, sa_column=Column(JSON)
         default=None, sa_column=Column(JSON)
@@ -126,6 +128,10 @@ class RunCreate(BaseModel):
     def model_validator(cls, data: Any):
     def model_validator(cls, data: Any):
         extra_body = data.get("extra_body")
         extra_body = data.get("extra_body")
         if extra_body:
         if extra_body:
+            # 取出并赋值
+            for key in ["parallel_tool_calls", "audio", "modalities"]:
+                if key in extra_body and extra_body[key] is not None:
+                    data[key] = extra_body[key]
             action_authentications = extra_body.get("action_authentications")
             action_authentications = extra_body.get("action_authentications")
             if action_authentications:
             if action_authentications:
                 res = action_authentications.values()
                 res = action_authentications.values()