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

237bb851 · Vikey Chen · GitHub · bd26c933 · 237bb851
Unverified Commit 237bb851 authored Mar 08, 2024 by Vikey Chen Committed by GitHub Mar 08, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 5 deletions

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

No files found.
--- a/api/core/memory/token_buffer_memory.py
+++ b/api/core/memory/token_buffer_memory.py
@@ -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))