From 48757e581ece25661a2a0827aefbb239ed2d0652 Mon Sep 17 00:00:00 2001 From: vccler Date: Sat, 22 Jun 2024 12:41:24 +0800 Subject: [PATCH] fix: zhipu tool calling, this PR fixes the bug described in issue #5496 (#5469) Co-authored-by: vccler Co-authored-by: -LAN- --- api/core/model_runtime/model_providers/zhipuai/llm/llm.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/api/core/model_runtime/model_providers/zhipuai/llm/llm.py b/api/core/model_runtime/model_providers/zhipuai/llm/llm.py index ee09b8cb74..ff971964a8 100644 --- a/api/core/model_runtime/model_providers/zhipuai/llm/llm.py +++ b/api/core/model_runtime/model_providers/zhipuai/llm/llm.py @@ -10,6 +10,7 @@ from core.model_runtime.entities.message_entities import ( PromptMessageRole, PromptMessageTool, SystemPromptMessage, + ToolPromptMessage, UserPromptMessage, ) from core.model_runtime.errors.validate import CredentialsValidateFailedError @@ -463,6 +464,8 @@ class ZhipuAILargeLanguageModel(_CommonZhipuaiAI, LargeLanguageModel): message_text = f"{ai_prompt} {content}" elif isinstance(message, SystemPromptMessage): message_text = content + elif isinstance(message, ToolPromptMessage): + message_text = content else: raise ValueError(f"Got unknown type {message}")