mirror of
https://github.com/langgenius/dify.git
synced 2025-07-16 05:35:09 +00:00
35 lines
1003 B
TypeScript
35 lines
1003 B
TypeScript
import {
|
|
useMemo,
|
|
} from 'react'
|
|
import type {
|
|
Node,
|
|
} from '../types'
|
|
import {
|
|
BlockEnum,
|
|
} from '../types'
|
|
import {
|
|
useStore,
|
|
} from '../store'
|
|
import { CollectionType } from '@/app/components/tools/types'
|
|
import { canFindTool } from '@/utils'
|
|
|
|
export const useToolIcon = (data: Node['data']) => {
|
|
const buildInTools = useStore(s => s.buildInTools)
|
|
const customTools = useStore(s => s.customTools)
|
|
const workflowTools = useStore(s => s.workflowTools)
|
|
const toolIcon = useMemo(() => {
|
|
if (data.type === BlockEnum.Tool) {
|
|
let targetTools = buildInTools
|
|
if (data.provider_type === CollectionType.builtIn)
|
|
targetTools = buildInTools
|
|
else if (data.provider_type === CollectionType.custom)
|
|
targetTools = customTools
|
|
else
|
|
targetTools = workflowTools
|
|
return targetTools.find(toolWithProvider => canFindTool(toolWithProvider.id, data.provider_id))?.icon
|
|
}
|
|
}, [data, buildInTools, customTools, workflowTools])
|
|
|
|
return toolIcon
|
|
}
|