fix: zhipu tool calling, this PR fixes the bug described in issue #5496…

… (#5469) Co-authored-by: vccler <[email protected]> Co-authored-by: -LAN- <[email protected]>
langgenius · Jun 22, 2024 · 3c4075b · 3c4075b
1 parent 5c2b4d4
commit 3c4075b
Showing 1 changed file with 3 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
@@ -10,6 +10,7 @@
     PromptMessageRole,
     PromptMessageTool,
     SystemPromptMessage,
+    ToolPromptMessage,
     UserPromptMessage,
 )
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
@@ -463,6 +464,8 @@ def _convert_one_message_to_text(self, message: PromptMessage) -> str:
             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}")