feat: automatically patch collections for >python 3.10 support (#21)

Co-authored-by: Bo Liu <benjaminliu.eecs@gmail.com>
This commit is contained in:
Nicola Dall'Asen
2024-03-13 15:34:10 +01:00
committed by GitHub
parent 601426030d
commit 717ed7639c
5 changed files with 28 additions and 9 deletions

View File

@@ -50,9 +50,11 @@ def convert_conversation_to_prompts(conversation: Conversation):
for i in range(0, len(messages), 2):
prompt = {
"role": messages[i][0],
"content": messages[i][1][0]
if isinstance(messages[i][1], tuple)
else messages[i][1],
"content": (
messages[i][1][0]
if isinstance(messages[i][1], tuple)
else messages[i][1]
),
"images": [messages[i][1][1]] if isinstance(messages[i][1], tuple) else [],
}
response = {"role": messages[i + 1][0], "content": messages[i + 1][1]}