youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-04 09:28:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

block icon

Browse Source
This commit is contained in:
StyleZhang
2024-03-18 21:02:40 +08:00
parent 202492e5ac
commit a89287bf20
10 changed files with 109 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
web/app/components/workflow/nodes/_base/node.tsx
View File

@ -5,12 +5,14 @@ import type {
import {
cloneElement,
memo,
useMemo,
} from 'react'
import type { NodeProps } from '../../types'
import {
BlockEnum,
NodeRunningStatus,
} from '../../types'
import { useStore } from '../../store'
import {
NodeSourceHandle,
NodeTargetHandle,
@ -32,6 +34,12 @@ const BaseNode: FC<BaseNodeProps> = ({
data,
children,
}) => {
const toolsets = useStore(s => s.toolsets)
const toolIcon = useMemo(() => {
if (data.type === BlockEnum.Tool)
return toolsets.find(toolset => toolset.id === data.provider_id)?.icon
}, [data, toolsets])
return (
<div
className={`
@ -84,7 +92,7 @@ const BaseNode: FC<BaseNodeProps> = ({
className='shrink-0 mr-2'
type={data.type}
size='md'
toolProviderId={data.provider_id}
toolIcon={toolIcon}
/>
<div
title={data.title}