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