From 4d6f736be365b58c3e633c568048ac2d4d6a3a74 Mon Sep 17 00:00:00 2001
From: Archer <545436317@qq.com>
Date: Thu, 7 Nov 2024 15:04:46 +0800
Subject: [PATCH] fix: workflow file upload refresh (#3088)

---
 .../src/pages/app/detail/components/Plugin/index.tsx  | 11 ++++-------
 .../Flow/nodes/NodePluginIO/NodePluginConfig.tsx      |  4 ++--
 .../Flow/nodes/NodeSystemConfig.tsx                   |  4 ++--
 3 files changed, 8 insertions(+), 11 deletions(-)

diff --git a/projects/app/src/pages/app/detail/components/Plugin/index.tsx b/projects/app/src/pages/app/detail/components/Plugin/index.tsx
index bae51d83f..3a029267f 100644
--- a/projects/app/src/pages/app/detail/components/Plugin/index.tsx
+++ b/projects/app/src/pages/app/detail/components/Plugin/index.tsx
@@ -2,7 +2,7 @@ import React from 'react';
 import { pluginSystemModuleTemplates } from '@fastgpt/global/core/workflow/template/constants';
 import { useConfirm } from '@fastgpt/web/hooks/useConfirm';
 import { v1Workflow2V2 } from '@/web/core/workflow/adapt';
-import WorkflowContextProvider, { WorkflowContext } from '../WorkflowComponents/context';
+import { ReactFlowCustomProvider, WorkflowContext } from '../WorkflowComponents/context';
 import { useContextSelector } from 'use-context-selector';
 import { AppContext, TabEnum } from '../context';
 import { useMount } from 'ahooks';
@@ -13,7 +13,6 @@ import dynamic from 'next/dynamic';
 import { cloneDeep } from 'lodash';
 
 import Flow from '../WorkflowComponents/Flow';
-import { ReactFlowProvider } from 'reactflow';
 import { useTranslation } from 'next-i18next';
 
 const Logs = dynamic(() => import('../Logs/index'));
@@ -67,11 +66,9 @@ const WorkflowEdit = () => {
 
 const Render = () => {
   return (
-    <ReactFlowProvider>
-      <WorkflowContextProvider basicNodeTemplates={pluginSystemModuleTemplates}>
-        <WorkflowEdit />
-      </WorkflowContextProvider>
-    </ReactFlowProvider>
+    <ReactFlowCustomProvider templates={pluginSystemModuleTemplates}>
+      <WorkflowEdit />
+    </ReactFlowCustomProvider>
   );
 };
 
diff --git a/projects/app/src/pages/app/detail/components/WorkflowComponents/Flow/nodes/NodePluginIO/NodePluginConfig.tsx b/projects/app/src/pages/app/detail/components/WorkflowComponents/Flow/nodes/NodePluginIO/NodePluginConfig.tsx
index ce0bd249e..720f0c6d2 100644
--- a/projects/app/src/pages/app/detail/components/WorkflowComponents/Flow/nodes/NodePluginIO/NodePluginConfig.tsx
+++ b/projects/app/src/pages/app/detail/components/WorkflowComponents/Flow/nodes/NodePluginIO/NodePluginConfig.tsx
@@ -149,7 +149,7 @@ function FileSelectConfig({ chatConfig: { fileSelectConfig }, setAppDetail }: Co
           if (canUploadFiles) {
             !repeatKey &&
               onChangeNode({
-                nodeId: pluginInputNode.id,
+                nodeId: pluginInputNode.nodeId,
                 type: 'addOutput',
                 value: {
                   ...userFilesInput,
@@ -159,7 +159,7 @@ function FileSelectConfig({ chatConfig: { fileSelectConfig }, setAppDetail }: Co
           } else {
             repeatKey &&
               onChangeNode({
-                nodeId: pluginInputNode.id,
+                nodeId: pluginInputNode.nodeId,
                 type: 'delOutput',
                 key: userFilesInput.key
               });
diff --git a/projects/app/src/pages/app/detail/components/WorkflowComponents/Flow/nodes/NodeSystemConfig.tsx b/projects/app/src/pages/app/detail/components/WorkflowComponents/Flow/nodes/NodeSystemConfig.tsx
index 37b231b78..d0155bd7c 100644
--- a/projects/app/src/pages/app/detail/components/WorkflowComponents/Flow/nodes/NodeSystemConfig.tsx
+++ b/projects/app/src/pages/app/detail/components/WorkflowComponents/Flow/nodes/NodeSystemConfig.tsx
@@ -247,14 +247,14 @@ function FileSelectConfig({ chatConfig: { fileSelectConfig }, setAppDetail }: Co
         if (canUploadFiles) {
           !repeatKey &&
             onChangeNode({
-              nodeId: workflowStartNode.id,
+              nodeId: workflowStartNode.nodeId,
               type: 'addOutput',
               value: userFilesInput
             });
         } else {
           repeatKey &&
             onChangeNode({
-              nodeId: workflowStartNode.id,
+              nodeId: workflowStartNode.nodeId,
               type: 'delOutput',
               key: userFilesInput.key
             });