Bläddra i källkod

replace message content type list to string when file_objs is empty .. (#2745)

tags/0.5.9
Vikey Chen 1 år sedan
förälder
incheckning
237bb8514e
Inget konto är kopplat till bidragsgivarens mejladress
1 ändrade filer med 8 tillägg och 5 borttagningar
  1. 8
    5
      api/core/memory/token_buffer_memory.py

+ 8
- 5
api/core/memory/token_buffer_memory.py Visa fil

@@ -47,11 +47,14 @@ class TokenBufferMemory:
files, message.app_model_config
)

prompt_message_contents = [TextPromptMessageContent(data=message.query)]
for file_obj in file_objs:
prompt_message_contents.append(file_obj.prompt_message_content)

prompt_messages.append(UserPromptMessage(content=prompt_message_contents))
if not file_objs:
prompt_messages.append(UserPromptMessage(content=message.query))
else:
prompt_message_contents = [TextPromptMessageContent(data=message.query)]
for file_obj in file_objs:
prompt_message_contents.append(file_obj.prompt_message_content)

prompt_messages.append(UserPromptMessage(content=prompt_message_contents))
else:
prompt_messages.append(UserPromptMessage(content=message.query))


Laddar…
Avbryt
Spara