fix: plugin tool icon (#13918)

This commit is contained in:
zxhlyh 2025-02-18 16:54:14 +08:00 committed by GitHub
parent cb841e5cde
commit ed7851a4b3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

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