소스 검색

fix: vision fail in complete app (#1933)

takatost 1 년 전
부모
커밋
2cfd5568e1
1개의 변경된 파일12개의 추가작업 그리고 1개의 파일을 삭제
  1. 12 1
      api/core/prompt/prompt_transform.py

+ 12 - 1
api/core/prompt/prompt_transform.py

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