From 3ec6f96722d03c7a4073dd2d962ae5412c236421 Mon Sep 17 00:00:00 2001
From: Joel <iamjoel007@gmail.com>
Date: Fri, 23 May 2025 10:18:27 +0800
Subject: [PATCH] feat: after run start node fetch sys vars

---
 web/app/components/workflow/hooks/use-inspect-vars-crud.ts  | 1 +
 .../workflow/nodes/_base/hooks/use-one-step-run.ts          | 6 +++++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/web/app/components/workflow/hooks/use-inspect-vars-crud.ts b/web/app/components/workflow/hooks/use-inspect-vars-crud.ts
index e97113c0f4..8f347b9139 100644
--- a/web/app/components/workflow/hooks/use-inspect-vars-crud.ts
+++ b/web/app/components/workflow/hooks/use-inspect-vars-crud.ts
@@ -228,6 +228,7 @@ const useInspectVarsCrud = () => {
     deleteAllInspectorVars,
     isInspectVarEdited,
     resetToLastRunVar,
+    invalidateSysVarValues,
   }
 }
 
diff --git a/web/app/components/workflow/nodes/_base/hooks/use-one-step-run.ts b/web/app/components/workflow/nodes/_base/hooks/use-one-step-run.ts
index b84400c4da..a21fbf4949 100644
--- a/web/app/components/workflow/nodes/_base/hooks/use-one-step-run.ts
+++ b/web/app/components/workflow/nodes/_base/hooks/use-one-step-run.ts
@@ -116,6 +116,7 @@ const useOneStepRun = <T>({
   const isChatMode = useIsChatMode()
   const isIteration = data.type === BlockEnum.Iteration
   const isLoop = data.type === BlockEnum.Loop
+  const isStartNode = data.type === BlockEnum.Start
 
   const availableNodes = getBeforeNodesInSameBranch(id)
   const availableNodesIncludeParent = getBeforeNodesInSameBranchIncludeParent(id)
@@ -166,6 +167,7 @@ const useOneStepRun = <T>({
   const [runResult, doSetRunResult] = useState<NodeRunResult | null>(null)
   const {
     appendNodeInspectVars,
+    invalidateSysVarValues,
   } = useInspectVarsCrud()
   const setRunResult = useCallback(async (data: NodeRunResult | null) => {
     doSetRunResult(data)
@@ -175,8 +177,10 @@ const useOneStepRun = <T>({
       const { getNodes } = store.getState()
       const nodes = getNodes()
       appendNodeInspectVars(id, vars, nodes)
+      if(isStartNode)
+        invalidateSysVarValues()
     }
-  }, [invalidLastRun, appId, store, appendNodeInspectVars, id])
+  }, [invalidLastRun, appId, id, store, appendNodeInspectVars, isStartNode, invalidateSysVarValues])
 
   const { handleNodeDataUpdate }: { handleNodeDataUpdate: (data: any) => void } = useNodeDataUpdate()
   const [canShowSingleRun, setCanShowSingleRun] = useState(false)