From 2886255c8bcf965fe0da611f72fb958495a3daf4 Mon Sep 17 00:00:00 2001
From: Joel <iamjoel007@gmail.com>
Date: Fri, 15 Mar 2024 15:10:41 +0800
Subject: [PATCH] fix: can not get first var type

---
 .../nodes/_base/components/variable/var-reference-picker.tsx   | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/web/app/components/workflow/nodes/_base/components/variable/var-reference-picker.tsx b/web/app/components/workflow/nodes/_base/components/variable/var-reference-picker.tsx
index 60263fa540..7149e570e3 100644
--- a/web/app/components/workflow/nodes/_base/components/variable/var-reference-picker.tsx
+++ b/web/app/components/workflow/nodes/_base/components/variable/var-reference-picker.tsx
@@ -67,6 +67,7 @@ const VarReferencePicker: FC<Props> = ({
   const isConstant = isSupportConstantValue && varKindType === VarKindType.static
   const { getTreeLeafNodes, getBeforeNodesInSameBranch } = useWorkflow()
   const availableNodes = onlyLeafNodeVar ? getTreeLeafNodes(nodeId) : getBeforeNodesInSameBranch(nodeId)
+  const allOutputVars = toNodeOutputVars(availableNodes, isChatMode)
   const outputVars = toNodeOutputVars(availableNodes, isChatMode, filterVar)
   const [open, setOpen] = useState(false)
   useEffect(() => {
@@ -82,7 +83,7 @@ const VarReferencePicker: FC<Props> = ({
     if (isConstant)
       return 'undefined'
 
-    const targetVar = outputVars.find(v => v.nodeId === outputVarNodeId)
+    const targetVar = allOutputVars.find(v => v.nodeId === outputVarNodeId)
     if (!targetVar)
       return 'undefined'