Преглед на файлове

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

Co-authored-by: vccler <vccler@163.com>
Co-authored-by: -LAN- <laipz8200@outlook.com>
vccler преди 10 месеца
родител
ревизия
48757e581e
променени са 1 файла, в които са добавени 3 реда и са изтрити 0 реда
  1. 3 0
      api/core/model_runtime/model_providers/zhipuai/llm/llm.py

+ 3 - 0
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}")