From 58fe02fb3f60f18811ee0d1b7938fe1a9f8644e8 Mon Sep 17 00:00:00 2001 From: zxhlyh Date: Mon, 15 Sep 2025 15:56:08 +0800 Subject: [PATCH] fix: variable assigner in node metadata --- .../rag-pipeline/hooks/use-available-nodes-meta-data.ts | 6 +++++- .../workflow-app/hooks/use-available-nodes-meta-data.ts | 6 +++++- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/web/app/components/rag-pipeline/hooks/use-available-nodes-meta-data.ts b/web/app/components/rag-pipeline/hooks/use-available-nodes-meta-data.ts index 6ce4524c8e..66b9214214 100644 --- a/web/app/components/rag-pipeline/hooks/use-available-nodes-meta-data.ts +++ b/web/app/components/rag-pipeline/hooks/use-available-nodes-meta-data.ts @@ -6,6 +6,7 @@ import dataSourceDefault from '@/app/components/workflow/nodes/data-source/defau import dataSourceEmptyDefault from '@/app/components/workflow/nodes/data-source-empty/default' import { WORKFLOW_COMMON_NODES } from '@/app/components/workflow/constants/node' import type { AvailableNodesMetaData } from '@/app/components/workflow/hooks-store/store' +import { BlockEnum } from '@/app/components/workflow/types' export const useAvailableNodesMetaData = () => { const { t } = useTranslation() @@ -59,7 +60,10 @@ export const useAvailableNodesMetaData = () => { return useMemo(() => { return { nodes: availableNodesMetaData, - nodesMap: availableNodesMetaDataMap, + nodesMap: { + ...availableNodesMetaDataMap, + [BlockEnum.VariableAssigner]: availableNodesMetaDataMap?.[BlockEnum.VariableAggregator], + }, } }, [availableNodesMetaData, availableNodesMetaDataMap]) } diff --git a/web/app/components/workflow-app/hooks/use-available-nodes-meta-data.ts b/web/app/components/workflow-app/hooks/use-available-nodes-meta-data.ts index f590c1a8ce..ba51b06401 100644 --- a/web/app/components/workflow-app/hooks/use-available-nodes-meta-data.ts +++ b/web/app/components/workflow-app/hooks/use-available-nodes-meta-data.ts @@ -7,6 +7,7 @@ import AnswerDefault from '@/app/components/workflow/nodes/answer/default' import { WORKFLOW_COMMON_NODES } from '@/app/components/workflow/constants/node' import type { AvailableNodesMetaData } from '@/app/components/workflow/hooks-store/store' import { useIsChatMode } from './use-is-chat-mode' +import { BlockEnum } from '@/app/components/workflow/types' export const useAvailableNodesMetaData = () => { const { t } = useTranslation() @@ -58,7 +59,10 @@ export const useAvailableNodesMetaData = () => { return useMemo(() => { return { nodes: availableNodesMetaData, - nodesMap: availableNodesMetaDataMap, + nodesMap: { + ...availableNodesMetaDataMap, + [BlockEnum.VariableAssigner]: availableNodesMetaDataMap?.[BlockEnum.VariableAggregator], + }, } }, [availableNodesMetaData, availableNodesMetaDataMap]) }