import { runAgent } from '@kevisual/ai/agent'
import { app, cnbManager } from '../../app.ts';
import z from 'zod';

app.route({
  path: 'cnb',
  key: 'chat',
  description: 'cnb智能对话接口',
  middleware: ['auth'],
  metadata: {
    args: {
      question: z.string().describe('用户输入的问题'),
      messages: z.array(z.object({
        role: z.enum(['user', 'assistant']).describe('消息角色，user表示用户输入，assistant表示助手回复'),
        content: z.string().describe('消息内容')
      })).describe('对话消息列表，按照时间顺序排列，包含用户和助手的历史消息'),
      model: z.string().optional().describe('默认auto')
    }
  }
}).define(async (ctx) => {
  // notCNBCheck(ctx);
  if (!ctx.args.question && !ctx.args.messages) {
    ctx.throw(400, '缺少必要参数，必须提供question或messages');
    return;
  }
  const model = ctx.args?.model || 'auto'
  const item = await cnbManager.getCNBItem(ctx);
  const cnbAi = item.cnbAi;
  const messages = ctx.args.messages || [{
    role: 'user',
    content: ctx.args.question
  }]
  const result = await runAgent({
    app,
    messages: messages,
    languageModel: cnbAi(model),
    token: '',
    // token: ctx.query.token as string,
  });
  ctx.body = result;
}).addTo(app);