BedrockLLMService: pull initial system frame from messages

This commit is contained in:
Aleix Conchillo Flaqué
2025-05-06 13:42:41 -07:00
parent b4de98cfb7
commit bed2e894a2

View File

@@ -382,15 +382,8 @@ class BedrockLLMContext(OpenAILLMContext):
# See if we should pull the system message out of our context.messages list. (For
# compatibility with Open AI messages format.)
if self.messages and self.messages[0]["role"] == "system":
if len(self.messages) == 1:
# If we have only have a system message in the list, all we can really do
# without introducing too much magic is change the role to "user".
self.messages[0]["role"] = "user"
else:
# If we have more than one message, we'll pull the system message out of the
# list.
self.system = self.messages[0]["content"]
self.messages.pop(0)
self.system = self.messages[0]["content"]
self.messages.pop(0)
# Merge consecutive messages with the same role.
i = 0