update

assistant/src/routes/ai/index.ts

@@ -1,29 +1,40 @@
 import { useContextKey } from '@kevisual/use-config/context';
-// import { appDefine } from '@/query/query-ai/defines/ai.ts';
 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'],
   })
   .define(async (ctx) => {
     const { messages = [], username, group, question, chatOpts = {} } = ctx.query;
-    // if (!username || !model || !group) {
-    //   return ctx.throw(4001, 'username, model, group is required');
-    // }
     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,