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}")