瀏覽代碼

fix: zhipu ai web_search not work (#9058)

tags/0.9.2
非法操作 1 年之前
父節點
當前提交
966e65bb66
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 10 行新增0 行删除
  1. 10
    0
      api/core/model_runtime/model_providers/zhipuai/llm/llm.py

+ 10
- 0
api/core/model_runtime/model_providers/zhipuai/llm/llm.py 查看文件

@@ -223,6 +223,16 @@ class ZhipuAILargeLanguageModel(_CommonZhipuaiAI, LargeLanguageModel):
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:

Loading…
取消
儲存