zxhlyh 2 месяцев назад
Родитель
Сommit
ed7851a4b3
1 измененных файлов с 5 добавлено и 1 удалено
  1. 5 1
      web/app/components/workflow/hooks/use-workflow.ts

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