Spaces:
Sleeping
Sleeping
| import { Handle, Position, type NodeProps } from '@xyflow/react' | |
| import { Bot, CircleDot, Flag, GitBranch, Hand, Wrench } from 'lucide-react' | |
| import type { WorkflowNode as WNode } from './types' | |
| const icons = { Input: CircleDot, 'AI Worker': Bot, 'Decision Rules': GitBranch, 'Human Review': Hand, 'Simulated Tool': Wrench, Output: Flag } | |
| export default function WorkflowNode({ data, selected }: NodeProps<WNode>) { | |
| const Icon = icons[data.mode] | |
| const receivesFlow = data.mode !== 'Input' && data.mode !== 'Simulated Tool' | |
| const sendsFlow = data.mode !== 'Output' && data.mode !== 'Simulated Tool' | |
| return <div className={`flow-node ${data.mode.toLowerCase().replaceAll(' ', '-')} ${selected ? 'selected' : ''}`}> | |
| {receivesFlow && <Handle id="flow-in" type="target" position={Position.Left} />} | |
| {data.mode === 'Simulated Tool' && <Handle id="tool-in" className="tool-handle" type="target" position={Position.Top} />} | |
| <div className="node-icon"><Icon size={16} /></div> | |
| <div><small>{data.mode}</small><strong>{data.label}</strong></div> | |
| {sendsFlow && <Handle id="flow-out" type="source" position={Position.Right} />} | |
| {data.mode === 'AI Worker' && <Handle id="tool-out" className="tool-handle" type="source" position={Position.Bottom} />} | |
| {data.mode === 'AI Worker' && <span className="tool-port-label">TOOLS</span>} | |
| </div> | |
| } | |