BedrockLLMService: pull initial system frame from messages
This commit is contained in:
@@ -382,15 +382,8 @@ class BedrockLLMContext(OpenAILLMContext):
|
|||||||
# See if we should pull the system message out of our context.messages list. (For
|
# See if we should pull the system message out of our context.messages list. (For
|
||||||
# compatibility with Open AI messages format.)
|
# compatibility with Open AI messages format.)
|
||||||
if self.messages and self.messages[0]["role"] == "system":
|
if self.messages and self.messages[0]["role"] == "system":
|
||||||
if len(self.messages) == 1:
|
self.system = self.messages[0]["content"]
|
||||||
# If we have only have a system message in the list, all we can really do
|
self.messages.pop(0)
|
||||||
# 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)
|
|
||||||
|
|
||||||
# Merge consecutive messages with the same role.
|
# Merge consecutive messages with the same role.
|
||||||
i = 0
|
i = 0
|
||||||
|
|||||||
Reference in New Issue
Block a user