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