Merge pull request #4924 from Peter-De-Ath/fix-duplicate-system-prompt

fix: stop system prompt being duplicated
This commit is contained in:
Timothy Jaeryang Baek 2024-08-27 13:07:26 +02:00 committed by GitHub
commit 7c841c9f63
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -82,7 +82,7 @@ def add_or_update_system_message(content: str, messages: list[dict]):
"""
if messages and messages[0].get("role") == "system":
messages[0]["content"] += f"{content}\n{messages[0]['content']}"
messages[0]["content"] = f"{content}\n{messages[0]['content']}"
else:
# Insert at the beginning
messages.insert(0, {"role": "system", "content": content})