youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-02 00:18:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix knowledge single retrieve when function call response is none

Browse Source
This commit is contained in:
jyong
2024-03-19 14:17:22 +08:00
parent 1c7573a686
commit 4ec14d8d91
2 changed files with 71 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
api/core/model_runtime/model_providers/anthropic/llm/llm.py
View File

@ -342,12 +342,21 @@ class AnthropicLargeLanguageModel(LargeLanguageModel):
Convert prompt messages to dict list and system
"""
system = ""
first_loop = True
for message in prompt_messages:
if isinstance(message, SystemPromptMessage):
message.content = message.content.strip()
if first_loop:
system = message.content
first_loop = False
else:
system += "\n"
system += message.content
prompt_message_dicts = []
for message in prompt_messages:
if isinstance(message, SystemPromptMessage):
system += message.content + ("\n" if not system else "")
else:
if not isinstance(message, SystemPromptMessage):
prompt_message_dicts.append(self._convert_prompt_message_to_dict(message))
return system, prompt_message_dicts