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