From ab56e6b6aff70942d6426c9860913c4ca4aefa46 Mon Sep 17 00:00:00 2001 From: Joel Date: Wed, 3 Apr 2024 17:42:21 +0800 Subject: [PATCH] fix: enable memory add sys query var input --- web/app/components/workflow/nodes/llm/use-config.ts | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/web/app/components/workflow/nodes/llm/use-config.ts b/web/app/components/workflow/nodes/llm/use-config.ts index 3b594b426f..72cd22f764 100644 --- a/web/app/components/workflow/nodes/llm/use-config.ts +++ b/web/app/components/workflow/nodes/llm/use-config.ts @@ -263,7 +263,15 @@ const useConfig = (id: string, payload: LLMNodeType) => { }) }, [runInputData, setRunInputData]) - const varInputs = getInputVars(isChatModel ? (inputs.prompt_template as PromptItem[]).map(item => item.text) : [(inputs.prompt_template as PromptItem).text]) + const allVarStrArr = (() => { + const arr = isChatModel ? (inputs.prompt_template as PromptItem[]).map(item => item.text) : [(inputs.prompt_template as PromptItem).text] + if (isChatMode && isChatModel && !!inputs.memory) + arr.push('{{#sys.query#}}') + + return arr + })() + + const varInputs = getInputVars(allVarStrArr) return { readOnly,