AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 16:05:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: anthropic system prompt not working (#2885)

Browse Source
This commit is contained in:
Su Yang 2024-03-19 15:50:02 +08:00 committed by GitHub
parent 507aa6d949
commit 2dee8a25d5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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