All files / src/components/WorkflowBuilder/nodes LlmNode.tsx

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
/**
 * LLM Node Component
 *
 * Node for LLM generation actions.
 */
 
import { memo } from 'react';
import { Handle, Position, NodeProps, Node } from '@xyflow/react';
import type { LlmNodeData } from '../../../lib/workflow-builder/types';
 
export const LlmNode = memo(({ data, selected }: NodeProps<Node<LlmNodeData>>) => {
  const templatePreview = data.template.length > 50
    ? data.template.slice(0, 50) + '...'
    : data.template || 'No template';
 
  return (
    <div
      className={`
        px-4 py-3 rounded-lg border-2 min-w-[200px]
        bg-violet-50 border-violet-300
        ${selected ? 'border-violet-500 shadow-lg shadow-violet-200' : ''}
      `}
    >
      {/* Input Handle */}
      <Handle
        type="target"
        position={Position.Left}
        className="w-3 h-3 bg-violet-400 border-2 border-white"
      />
 
      {/* Output Handle */}
      <Handle
        type="source"
        position={Position.Right}
        className="w-3 h-3 bg-violet-500 border-2 border-white"
      />
 
      {/* Header */}
      <div className="flex items-center gap-2 mb-2">
        <span className="text-lg">🤖</span>
        <span className="font-medium text-violet-800">{data.label}</span>
        {data.jsonMode && (
          <span className="text-xs bg-violet-200 text-violet-700 px-1.5 py-0.5 rounded">
            JSON
          </span>
        )}
      </div>
 
      {/* Template Preview */}
      <div className="text-sm text-violet-600 bg-violet-100 rounded px-2 py-1 font-mono">
        {data.isTemplateFile ? '📄 ' : ''}
        {templatePreview}
      </div>
 
      {/* Model Info */}
      {(data.model || data.temperature !== undefined) && (
        <div className="flex gap-2 mt-2 text-xs text-violet-500">
          {data.model && <span>Model: {data.model}</span>}
          {data.temperature !== undefined && (
            <span>Temp: {data.temperature}</span>
          )}
        </div>
      )}
    </div>
  );
});
 
LlmNode.displayName = 'LlmNode';
 
export default LlmNode;