AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 00:39:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: vision fail in complete app (#1933)

Browse Source
This commit is contained in:
takatost 2024-01-05 04:23:12 +08:00 committed by GitHub
parent faf40a42bc
commit 2cfd5568e1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
api/core/prompt/prompt_transform.py
View File

@ -334,7 +334,18 @@ class PromptTransform:
prompt = re.sub(r'<\|.*?\|>', '', prompt) 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: def _set_context_variable(self, context: str, prompt_template: PromptTemplateParser, prompt_inputs: dict) -> None:
if '#context#' in prompt_template.variable_keys: if '#context#' in prompt_template.variable_keys: