mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-05-13 15:08:17 +08:00

Signed-off-by: -LAN- <laipz8200@outlook.com> Co-authored-by: Joel <iamjoel007@gmail.com> Co-authored-by: -LAN- <laipz8200@outlook.com>
89 lines
3.6 KiB
TypeScript
89 lines
3.6 KiB
TypeScript
import type { FC } from 'react'
|
|
import React from 'react'
|
|
import cn from 'classnames'
|
|
import type { EndNodeType } from './types'
|
|
import type { NodeProps, Variable } from '@/app/components/workflow/types'
|
|
import { isConversationVar, isENV, isSystemVar } from '@/app/components/workflow/nodes/_base/components/variable/utils'
|
|
import {
|
|
useIsChatMode,
|
|
useWorkflow,
|
|
useWorkflowVariables,
|
|
} from '@/app/components/workflow/hooks'
|
|
import { VarBlockIcon } from '@/app/components/workflow/block-icon'
|
|
import { Line3 } from '@/app/components/base/icons/src/public/common'
|
|
import { Variable02 } from '@/app/components/base/icons/src/vender/solid/development'
|
|
import { BubbleX, Env } from '@/app/components/base/icons/src/vender/line/others'
|
|
import { BlockEnum } from '@/app/components/workflow/types'
|
|
|
|
const Node: FC<NodeProps<EndNodeType>> = ({
|
|
id,
|
|
data,
|
|
}) => {
|
|
const { getBeforeNodesInSameBranch } = useWorkflow()
|
|
const availableNodes = getBeforeNodesInSameBranch(id)
|
|
const { getCurrentVariableType } = useWorkflowVariables()
|
|
const isChatMode = useIsChatMode()
|
|
|
|
const startNode = availableNodes.find((node: any) => {
|
|
return node.data.type === BlockEnum.Start
|
|
})
|
|
|
|
const getNode = (id: string) => {
|
|
return availableNodes.find(node => node.id === id) || startNode
|
|
}
|
|
|
|
const { outputs } = data
|
|
const filteredOutputs = (outputs as Variable[]).filter(({ value_selector }) => value_selector.length > 0)
|
|
|
|
if (!filteredOutputs.length)
|
|
return null
|
|
|
|
return (
|
|
<div className='mb-1 px-3 py-1 space-y-0.5'>
|
|
{filteredOutputs.map(({ value_selector }, index) => {
|
|
const node = getNode(value_selector[0])
|
|
const isSystem = isSystemVar(value_selector)
|
|
const isEnv = isENV(value_selector)
|
|
const isChatVar = isConversationVar(value_selector)
|
|
const varName = isSystem ? `sys.${value_selector[value_selector.length - 1]}` : value_selector[value_selector.length - 1]
|
|
const varType = getCurrentVariableType({
|
|
valueSelector: value_selector,
|
|
availableNodes,
|
|
isChatMode,
|
|
})
|
|
return (
|
|
<div key={index} className='flex items-center h-6 justify-between bg-gray-100 rounded-md px-1 space-x-1 text-xs font-normal text-gray-700'>
|
|
<div className='flex items-center text-xs font-medium text-gray-500'>
|
|
{!isEnv && !isChatVar && (
|
|
<>
|
|
<div className='p-[1px]'>
|
|
<VarBlockIcon
|
|
className='!text-gray-900'
|
|
type={node?.data.type || BlockEnum.Start}
|
|
/>
|
|
</div>
|
|
<div className='max-w-[75px] truncate'>{node?.data.title}</div>
|
|
<Line3 className='mr-0.5'></Line3>
|
|
</>
|
|
)}
|
|
<div className='flex items-center text-primary-600'>
|
|
{!isEnv && !isChatVar && <Variable02 className='shrink-0 w-3.5 h-3.5 text-primary-500' />}
|
|
{isEnv && <Env className='shrink-0 w-3.5 h-3.5 text-util-colors-violet-violet-600' />}
|
|
{isChatVar && <BubbleX className='w-3.5 h-3.5 text-util-colors-teal-teal-700' />}
|
|
|
|
<div className={cn('max-w-[50px] ml-0.5 text-xs font-medium truncate', (isEnv || isChatVar) && '!max-w-[70px] text-gray-900')}>{varName}</div>
|
|
</div>
|
|
</div>
|
|
<div className='text-xs font-normal text-gray-700'>
|
|
<div className='max-w-[42px] ml-0.5 text-xs font-normal text-gray-500 capitalize truncate' title={varType}>{varType}</div>
|
|
</div>
|
|
</div>
|
|
)
|
|
})}
|
|
|
|
</div>
|
|
)
|
|
}
|
|
|
|
export default React.memo(Node)
|