From 097a6fc1e0135e1de4c6f9dec5e1ba68dbd7df3d Mon Sep 17 00:00:00 2001 From: zxhlyh Date: Thu, 7 Aug 2025 17:22:29 +0800 Subject: [PATCH] fix: variable --- .../workflow-variable-block-replacement-block.tsx | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/web/app/components/base/prompt-editor/plugins/workflow-variable-block/workflow-variable-block-replacement-block.tsx b/web/app/components/base/prompt-editor/plugins/workflow-variable-block/workflow-variable-block-replacement-block.tsx index 6a73e5bb02..62b9cccb3f 100644 --- a/web/app/components/base/prompt-editor/plugins/workflow-variable-block/workflow-variable-block-replacement-block.tsx +++ b/web/app/components/base/prompt-editor/plugins/workflow-variable-block/workflow-variable-block-replacement-block.tsx @@ -21,6 +21,13 @@ const WorkflowVariableBlockReplacementBlock = ({ variables, }: WorkflowVariableBlockType) => { const [editor] = useLexicalComposerContext() + const ragVariables = variables?.reduce((acc, curr) => { + if (curr.nodeId === 'rag') + acc.push(...curr.vars) + else + acc.push(...curr.vars.filter(v => v.isRagVariable)) + return acc + }, []) useEffect(() => { if (!editor.hasNodes([WorkflowVariableBlockNode])) @@ -32,7 +39,7 @@ const WorkflowVariableBlockReplacementBlock = ({ onInsert() const nodePathString = textNode.getTextContent().slice(3, -3) - return $applyNodeReplacement($createWorkflowVariableBlockNode(nodePathString.split('.'), workflowNodesMap, getVarType, variables?.find(o => o.nodeId === 'env')?.vars || [], variables?.find(o => o.nodeId === 'conversation')?.vars || [], variables?.find(o => o.nodeId === 'rag')?.vars || [])) + return $applyNodeReplacement($createWorkflowVariableBlockNode(nodePathString.split('.'), workflowNodesMap, getVarType, variables?.find(o => o.nodeId === 'env')?.vars || [], variables?.find(o => o.nodeId === 'conversation')?.vars || [], ragVariables)) }, [onInsert, workflowNodesMap, getVarType, variables]) const getMatch = useCallback((text: string) => {