jandres/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-01-08 07:14:14 +00:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

fix: image text when retrieve chat histories (#3220)

Browse Source
This commit is contained in:
takatost
2024-04-09 15:20:45 +08:00
committed by GitHub
parent 29918c498c
commit 4ad3f2cdc2
4 changed files with 54 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
api/core/memory/token_buffer_memory.py
View File

@@ -3,6 +3,7 @@ from core.file.message_file_parser import MessageFileParser
from core.model_manager import ModelInstance
from core.model_runtime.entities.message_entities import (
AssistantPromptMessage,
ImagePromptMessageContent,
PromptMessage,
PromptMessageRole,
TextPromptMessageContent,
@@ -124,7 +125,17 @@ class TokenBufferMemory:
else:
continue
message = f"{role}: {m.content}"
string_messages.append(message)
if isinstance(m.content, list):
inner_msg = ""
for content in m.content:
if isinstance(content, TextPromptMessageContent):
inner_msg += f"{content.data}\n"
elif isinstance(content, ImagePromptMessageContent):
inner_msg += "[image]\n"
string_messages.append(f"{role}: {inner_msg.strip()}")
else:
message = f"{role}: {m.content}"
string_messages.append(message)
return "\n".join(string_messages)