🔧 Refactor Gemini payload generation: improve tool processing and add…

… conditional tool inclusion
yym68686 · Feb 1, 2025 · 93e2902 · 93e2902
1 parent 1b165d5
commit 93e2902
Showing 1 changed file with 13 additions and 11 deletions.
diff --git a/request.py b/request.py
@@ -287,18 +287,20 @@ async def get_gemini_payload(request, engine, provider):
                                 prop_value["description"] = f"{description}\nDefault: {default_value}"
                                 # 删除 default 字段
                                 del prop_value["default"]
-                    processed_tools.append({"function": function_def})
-
-                payload.update({
-                    "tools": [{
-                        "function_declarations": [tool["function"] for tool in processed_tools]
-                    }],
-                    "tool_config": {
-                        "function_calling_config": {
-                            "mode": "AUTO"
+                    if function_def["name"] != "googleSearch" and function_def["name"] != "googleSearch":
+                        processed_tools.append({"function": function_def})
+
+                if processed_tools:
+                    payload.update({
+                        "tools": [{
+                            "function_declarations": [tool["function"] for tool in processed_tools]
+                        }],
+                        "tool_config": {
+                            "function_calling_config": {
+                                "mode": "AUTO"
+                            }
                         }
-                    }
-                })
+                    })
             elif field == "temperature":
                 generation_config["temperature"] = value
             elif field == "max_tokens":