From a77e7ab177f7582b1235ee315e8d7c31c5704d30 Mon Sep 17 00:00:00 2001 From: Joel Date: Tue, 27 May 2025 15:25:17 +0800 Subject: [PATCH] fix: tools icon --- web/app/components/workflow/hooks/use-workflow.ts | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/web/app/components/workflow/hooks/use-workflow.ts b/web/app/components/workflow/hooks/use-workflow.ts index ce47329f21..df60803d00 100644 --- a/web/app/components/workflow/hooks/use-workflow.ts +++ b/web/app/components/workflow/hooks/use-workflow.ts @@ -504,6 +504,8 @@ export const useToolIcon = (data: Node['data']) => { 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 toolIcon = useMemo(() => { if (data.type === BlockEnum.Tool) { let targetTools = buildInTools @@ -511,11 +513,13 @@ export const useToolIcon = (data: Node['data']) => { targetTools = buildInTools else if (data.provider_type === CollectionType.custom) targetTools = customTools + else if (data.provider_type === CollectionType.mcp) + targetTools = mcpTools else targetTools = workflowTools return targetTools.find(toolWithProvider => canFindTool(toolWithProvider.id, data.provider_id))?.icon } - }, [data, buildInTools, customTools, workflowTools]) + }, [data.type, data.provider_type, data.provider_id, buildInTools, customTools, mcpTools, workflowTools]) return toolIcon }