| @@ -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)) | |||