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