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