Skip to content

D
dify
Unverified Commit 237bb851 authored by Vikey Chen's avatar Vikey Chen Committed by GitHub
Browse files
Options

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

parent bd26c933
Show whitespace changes
Inline Side-by-side
Showing with 8 additions and 5 deletions
api/core/memory/token_buffer_memory.py
View file @ 237bb851
...@@ -47,6 +47,9 @@ class TokenBufferMemory: ...@@ -47,6 +47,9 @@ class TokenBufferMemory:
files, message.app_model_config files, message.app_model_config
) )
if not file_objs:
prompt_messages.append(UserPromptMessage(content=message.query))
else:
prompt_message_contents = [TextPromptMessageContent(data=message.query)] prompt_message_contents = [TextPromptMessageContent(data=message.query)]
for file_obj in file_objs: for file_obj in file_objs:
prompt_message_contents.append(file_obj.prompt_message_content) prompt_message_contents.append(file_obj.prompt_message_content)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment