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