refac/fix: system prompt duplication

This commit is contained in:
Timothy Jaeryang Baek
2025-10-02 02:57:54 -05:00
parent 6c4deed37a
commit a1fc99c66f
3 changed files with 24 additions and 5 deletions

View File

@@ -136,6 +136,14 @@ def update_message_content(message: dict, content: str, append: bool = True) ->
return message
def replace_system_message_content(content: str, messages: list[dict]) -> dict:
for message in messages:
if message["role"] == "system":
message["content"] = content
break
return messages
def add_or_update_system_message(
content: str, messages: list[dict], append: bool = False
):