|  | @@ -334,7 +334,18 @@ class PromptTransform:
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |          prompt = re.sub(r'<\|.*?\|>', '', prompt)
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -        return [UserPromptMessage(content=prompt)]
 | 
	
		
			
				|  |  | +        model_mode = ModelMode.value_of(model_config.mode)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +        if model_mode == ModelMode.CHAT and files:
 | 
	
		
			
				|  |  | +            prompt_message_contents = [TextPromptMessageContent(data=prompt)]
 | 
	
		
			
				|  |  | +            for file in files:
 | 
	
		
			
				|  |  | +                prompt_message_contents.append(file.prompt_message_content)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +            prompt_message = UserPromptMessage(content=prompt_message_contents)
 | 
	
		
			
				|  |  | +        else:
 | 
	
		
			
				|  |  | +            prompt_message = UserPromptMessage(content=prompt)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +        return [prompt_message]
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      def _set_context_variable(self, context: str, prompt_template: PromptTemplateParser, prompt_inputs: dict) -> None:
 | 
	
		
			
				|  |  |          if '#context#' in prompt_template.variable_keys:
 |