fix: avoid got at least one message error when there was no user message for Claude llm

This commit is contained in:
郭卫东 2024-09-26 13:50:00 +08:00
parent a0b0809b1c
commit 3fd1f844ec

View File

@ -538,6 +538,10 @@ class AnthropicLargeLanguageModel(LargeLanguageModel):
else: else:
raise ValueError(f"Got unknown type {message}") raise ValueError(f"Got unknown type {message}")
# if no user message, add a system message as the user message to avoid at least one message error
if not prompt_message_dicts:
prompt_message_dicts.append({"role": "user", "content": system})
return system, prompt_message_dicts return system, prompt_message_dicts
def _convert_one_message_to_text(self, message: PromptMessage) -> str: def _convert_one_message_to_text(self, message: PromptMessage) -> str: