Procházet zdrojové kódy

fix: plugin tool icon (#13918)

zxhlyh před 2 měsíci
rodič
revize
ed7851a4b3

+ 5 - 1
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])