import type {NodeProps} from '@xyflow/react'
import AmisNode, {outputsFormColumns} from './AmisNode.tsx'

const LlmNode = (props: NodeProps) => AmisNode({
  nodeProps: props,
  type: 'normal',
  defaultNodeName: '大模型',
  defaultNodeDescription: '使用大模型对话',
  columnSchema: [
    {
      type: 'select',
      name: 'model',
      label: '大模型',
      required: true,
      selectFirst: true,
      options: [
        {
          label: 'Qwen3',
          value: 'qwen3',
        },
        {
          label: 'Deepseek',
          value: 'deepseek',
        },
      ],
    },
    {
      type: 'textarea',
      name: 'systemPrompt',
      label: '系统提示词',
      required: true,
    },
    {
      type: 'divider',
    },
    ...outputsFormColumns(false, true, {text: {type: 'string'}}),
  ],
})

export default LlmNode