Procházet zdrojové kódy

fix: vision fail in complete app (#1933)

tags/0.4.4
takatost před 1 rokem
rodič
revize
2cfd5568e1
Žádný účet není propojen s e-mailovou adresou tvůrce revize
1 změnil soubory, kde provedl 12 přidání a 1 odebrání
  1. 12
    1
      api/core/prompt/prompt_transform.py

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

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

Načítá se…
Zrušit
Uložit