|
@@ -2,6 +2,7 @@ from typing import Optional
|
|
|
|
|
|
from core.app.app_config.features.file_upload.manager import FileUploadConfigManager
|
|
|
from core.file import file_manager
|
|
|
+from core.file.models import FileType
|
|
|
from core.model_manager import ModelInstance
|
|
|
from core.model_runtime.entities import (
|
|
|
AssistantPromptMessage,
|
|
@@ -98,8 +99,9 @@ class TokenBufferMemory:
|
|
|
prompt_message_contents: list[PromptMessageContent] = []
|
|
|
prompt_message_contents.append(TextPromptMessageContent(data=message.query))
|
|
|
for file_obj in file_objs:
|
|
|
- prompt_message = file_manager.to_prompt_message_content(file_obj)
|
|
|
- prompt_message_contents.append(prompt_message)
|
|
|
+ if file_obj.type in {FileType.IMAGE, FileType.AUDIO}:
|
|
|
+ prompt_message = file_manager.to_prompt_message_content(file_obj)
|
|
|
+ prompt_message_contents.append(prompt_message)
|
|
|
|
|
|
prompt_messages.append(UserPromptMessage(content=prompt_message_contents))
|
|
|
else:
|