import { useContextKey } from '@kevisual/use-config/context';
import { ProviderManager } from '@kevisual/ai';
import { App } from '@kevisual/router';
import { assistantConfig } from '@/app.ts';
import { Query } from '@kevisual/query';
const app = useContextKey<App>('app');
const query = useContextKey<Query>('assistantQuery');

app
  .route({
    path: 'ai',
    key: 'chat',
    description: '与 AI 进行对话, 调用 GPT 的AI 服务，生成结果，并返回。',
    middleware: ['auth-admin'],
    metadata: {
      admin: true,
    }
  })
  .define(async (ctx) => {
    const { messages = [], username, group, question, chatOpts = {} } = ctx.query;
    if (messages.length === 0 && question) {
      messages.push({
        role: 'user',
        content: question,
      });
    } else {
      ctx.throw(4001, 'messages or question is required');
    }

    const as = assistantConfig.getCacheAssistantConfig();
    const { provider, apiKey, model } = as.ai || {};
    if (!provider) {
      const response = await query.post(ctx.query);
      if (response.code === 200) {
        ctx.body = response.data;
        return;
      } else {
        return ctx.throw(response.code, response.message);
      }
    }
    const pm = new ProviderManager({
      provider: provider,
      apiKey: apiKey,
      model: model,
    });
    const result = await pm.provider.chat(messages, chatOpts);
    ctx.body = result;
  })
  .addTo(app);