Skip to content

Commit 966e65b

Browse files
authored
fix: zhipu ai web_search not work (#9058)
1 parent 896998e commit 966e65b

File tree

1 file changed

+10
-0
lines changed
  • api/core/model_runtime/model_providers/zhipuai/llm

1 file changed

+10
-0
lines changed

Diff for: api/core/model_runtime/model_providers/zhipuai/llm/llm.py

+10
Original file line numberDiff line numberDiff line change
@@ -223,6 +223,16 @@ def _generate(
223223
else:
224224
new_prompt_messages.append(copy_prompt_message)
225225

226+
# zhipuai moved web_search param to tools
227+
if "web_search" in model_parameters:
228+
enable_web_search = model_parameters.get("web_search")
229+
model_parameters.pop("web_search")
230+
web_search_params = {"type": "web_search", "web_search": {"enable": enable_web_search}}
231+
if "tools" in model_parameters:
232+
model_parameters["tools"].append(web_search_params)
233+
else:
234+
model_parameters["tools"] = [web_search_params]
235+
226236
if model in {"glm-4v", "glm-4v-plus"}:
227237
params = self._construct_glm_4v_parameter(model, new_prompt_messages, model_parameters)
228238
else:

0 commit comments

Comments
 (0)