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> = ({ 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 (
{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 (
{!isEnv && !isChatVar && ( <>
{node?.data.title}
)}
{!isEnv && !isChatVar && } {isEnv && } {isChatVar && }
{varName}
{varType}
) })}
) } export default React.memo(Node)