From 8d2238d631cc1ebd2375ed3f1f0f68616d12d1b3 Mon Sep 17 00:00:00 2001
From: Joel <iamjoel007@gmail.com>
Date: Tue, 27 May 2025 15:09:44 +0800
Subject: [PATCH] feat: in tools fetch mcp

---
 web/app/components/workflow/hooks/use-workflow.ts        | 8 ++++++++
 web/app/components/workflow/index.tsx                    | 1 +
 web/app/components/workflow/nodes/tool/use-config.ts     | 3 +++
 web/app/components/workflow/store/workflow/tool-slice.ts | 4 ++++
 web/service/tools.ts                                     | 4 ++++
 5 files changed, 20 insertions(+)

diff --git a/web/app/components/workflow/hooks/use-workflow.ts b/web/app/components/workflow/hooks/use-workflow.ts
index 99dce4dc15..ce47329f21 100644
--- a/web/app/components/workflow/hooks/use-workflow.ts
+++ b/web/app/components/workflow/hooks/use-workflow.ts
@@ -40,6 +40,7 @@ import { useStore as useAppStore } from '@/app/components/app/store'
 import {
   fetchAllBuiltInTools,
   fetchAllCustomTools,
+  fetchAllMCPTools,
   fetchAllWorkflowTools,
 } from '@/service/tools'
 import { CollectionType } from '@/app/components/tools/types'
@@ -450,6 +451,13 @@ export const useFetchToolsData = () => {
         workflowTools: workflowTools || [],
       })
     }
+    if(type === 'mcp') {
+      const mcpTools = await fetchAllMCPTools()
+
+      workflowStore.setState({
+        mcpTools: mcpTools || [],
+      })
+    }
   }, [workflowStore])
 
   return {
diff --git a/web/app/components/workflow/index.tsx b/web/app/components/workflow/index.tsx
index 549117faf7..e23e1570a5 100644
--- a/web/app/components/workflow/index.tsx
+++ b/web/app/components/workflow/index.tsx
@@ -205,6 +205,7 @@ export const Workflow: FC<WorkflowProps> = memo(({
     handleFetchAllTools('builtin')
     handleFetchAllTools('custom')
     handleFetchAllTools('workflow')
+    handleFetchAllTools('mcp')
   }, [handleFetchAllTools])
 
   const {
diff --git a/web/app/components/workflow/nodes/tool/use-config.ts b/web/app/components/workflow/nodes/tool/use-config.ts
index 38ca5b5195..49a5e3b4d1 100644
--- a/web/app/components/workflow/nodes/tool/use-config.ts
+++ b/web/app/components/workflow/nodes/tool/use-config.ts
@@ -37,6 +37,7 @@ const useConfig = (id: string, payload: ToolNodeType) => {
   const buildInTools = useStore(s => s.buildInTools)
   const customTools = useStore(s => s.customTools)
   const workflowTools = useStore(s => s.workflowTools)
+  const mcpTools = useStore(s => s.mcpTools)
 
   const currentTools = (() => {
     switch (provider_type) {
@@ -46,6 +47,8 @@ const useConfig = (id: string, payload: ToolNodeType) => {
         return customTools
       case CollectionType.workflow:
         return workflowTools
+      case CollectionType.mcp:
+        return mcpTools
       default:
         return []
     }
diff --git a/web/app/components/workflow/store/workflow/tool-slice.ts b/web/app/components/workflow/store/workflow/tool-slice.ts
index 2d54bbd925..d6d89abcf0 100644
--- a/web/app/components/workflow/store/workflow/tool-slice.ts
+++ b/web/app/components/workflow/store/workflow/tool-slice.ts
@@ -10,6 +10,8 @@ export type ToolSliceShape = {
   setCustomTools: (tools: ToolWithProvider[]) => void
   workflowTools: ToolWithProvider[]
   setWorkflowTools: (tools: ToolWithProvider[]) => void
+  mcpTools: ToolWithProvider[]
+  setMcpTools: (tools: ToolWithProvider[]) => void
   toolPublished: boolean
   setToolPublished: (toolPublished: boolean) => void
 }
@@ -21,6 +23,8 @@ export const createToolSlice: StateCreator<ToolSliceShape> = set => ({
   setCustomTools: customTools => set(() => ({ customTools })),
   workflowTools: [],
   setWorkflowTools: workflowTools => set(() => ({ workflowTools })),
+  mcpTools: [],
+  setMcpTools: mcpTools => set(() => ({ mcpTools })),
   toolPublished: false,
   setToolPublished: toolPublished => set(() => ({ toolPublished })),
 })
diff --git a/web/service/tools.ts b/web/service/tools.ts
index 38dcf382e6..6a88d8d567 100644
--- a/web/service/tools.ts
+++ b/web/service/tools.ts
@@ -124,6 +124,10 @@ export const fetchAllWorkflowTools = () => {
   return get<ToolWithProvider[]>('/workspaces/current/tools/workflow')
 }
 
+export const fetchAllMCPTools = () => {
+  return get<ToolWithProvider[]>('/workspaces/current/tools/mcp')
+}
+
 export const fetchLabelList = () => {
   return get<Label[]>('/workspaces/current/tool-labels')
 }