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