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