mirror of
https://github.com/langgenius/dify.git
synced 2025-11-10 08:21:29 +00:00
85cda47c70
Signed-off-by: -LAN- <laipz8200@outlook.com> Co-authored-by: twwu <twwu@dify.ai> Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com> Co-authored-by: jyong <718720800@qq.com> Co-authored-by: Wu Tianwei <30284043+WTW0313@users.noreply.github.com> Co-authored-by: QuantumGhost <obelisk.reg+git@gmail.com> Co-authored-by: lyzno1 <yuanyouhuilyz@gmail.com> Co-authored-by: quicksand <quicksandzn@gmail.com> Co-authored-by: Jyong <76649700+JohnJyong@users.noreply.github.com> Co-authored-by: lyzno1 <92089059+lyzno1@users.noreply.github.com> Co-authored-by: zxhlyh <jasonapring2015@outlook.com> Co-authored-by: Yongtao Huang <yongtaoh2022@gmail.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: Joel <iamjoel007@gmail.com> Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> Co-authored-by: nite-knite <nkCoding@gmail.com> Co-authored-by: Hanqing Zhao <sherry9277@gmail.com> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> Co-authored-by: Harry <xh001x@hotmail.com>
76 lines
2.4 KiB
TypeScript
76 lines
2.4 KiB
TypeScript
import {
|
|
useCallback,
|
|
useMemo,
|
|
} from 'react'
|
|
import type {
|
|
Node,
|
|
} from '../types'
|
|
import {
|
|
BlockEnum,
|
|
} from '../types'
|
|
import {
|
|
useStore,
|
|
useWorkflowStore,
|
|
} 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 mcpTools = useStore(s => s.mcpTools)
|
|
const dataSourceList = useStore(s => s.dataSourceList)
|
|
// const a = useStore(s => s.data)
|
|
const toolIcon = useMemo(() => {
|
|
if (!data)
|
|
return ''
|
|
if (data.type === BlockEnum.Tool) {
|
|
// eslint-disable-next-line sonarjs/no-dead-store
|
|
let targetTools = buildInTools
|
|
if (data.provider_type === CollectionType.builtIn)
|
|
targetTools = buildInTools
|
|
else if (data.provider_type === CollectionType.custom)
|
|
targetTools = customTools
|
|
else if (data.provider_type === CollectionType.mcp)
|
|
targetTools = mcpTools
|
|
else
|
|
targetTools = workflowTools
|
|
return targetTools.find(toolWithProvider => canFindTool(toolWithProvider.id, data.provider_id))?.icon
|
|
}
|
|
if (data.type === BlockEnum.DataSource)
|
|
return dataSourceList?.find(toolWithProvider => toolWithProvider.plugin_id === data.plugin_id)?.icon
|
|
}, [data, dataSourceList, buildInTools, customTools, mcpTools, workflowTools])
|
|
|
|
return toolIcon
|
|
}
|
|
|
|
export const useGetToolIcon = () => {
|
|
const workflowStore = useWorkflowStore()
|
|
const getToolIcon = useCallback((data: Node['data']) => {
|
|
const {
|
|
buildInTools,
|
|
customTools,
|
|
workflowTools,
|
|
dataSourceList,
|
|
} = workflowStore.getState()
|
|
|
|
if (data.type === BlockEnum.Tool) {
|
|
// eslint-disable-next-line sonarjs/no-dead-store
|
|
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
|
|
}
|
|
|
|
if (data.type === BlockEnum.DataSource)
|
|
return dataSourceList?.find(toolWithProvider => toolWithProvider.plugin_id === data.plugin_id)?.icon
|
|
}, [workflowStore])
|
|
|
|
return getToolIcon
|
|
}
|