fix: plugin tool icon (#13918)

12 months ago · ed7851a4b3
parent cb841e5cde
commit ed7851a4b3
1 changed files with 5 additions and 1 deletions
--- a/web/app/components/workflow/hooks/use-workflow.ts
+++ b/web/app/components/workflow/hooks/use-workflow.ts
@ -608,7 +608,11 @@ export const useToolIcon = (data: Node['data']) => {
        targetTools = customTools
      else
        targetTools = workflowTools
-      return targetTools.find(toolWithProvider => toolWithProvider.id === data.provider_id)?.icon
+      return targetTools.find((toolWithProvider) => {
+        return toolWithProvider.id === data.provider_id
+        || toolWithProvider.id === `langgenius/${data.provider_id}/${data.provider_id}`
+        || toolWithProvider.id === `langgenius/${data.provider_id}_tool/${data.provider_id}`
+      })?.icon
    }
  }, [data, buildInTools, customTools, workflowTools])