fix: zhipu ai web_search not work (langgenius#9058)

AceDataCloud · Nov 9, 2024 · f4d2c31 · f4d2c31
1 parent 56a3742
commit f4d2c31
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/api/core/model_runtime/model_providers/zhipuai/llm/llm.py b/api/core/model_runtime/model_providers/zhipuai/llm/llm.py
@@ -223,6 +223,16 @@ def _generate(
                 else:
                     new_prompt_messages.append(copy_prompt_message)
 
+        # zhipuai moved web_search param to tools
+        if "web_search" in model_parameters:
+            enable_web_search = model_parameters.get("web_search")
+            model_parameters.pop("web_search")
+            web_search_params = {"type": "web_search", "web_search": {"enable": enable_web_search}}
+            if "tools" in model_parameters:
+                model_parameters["tools"].append(web_search_params)
+            else:
+                model_parameters["tools"] = [web_search_params]
+
         if model in {"glm-4v", "glm-4v-plus"}:
             params = self._construct_glm_4v_parameter(model, new_prompt_messages, model_parameters)
         else: