doc gpt V0.2

src/pages/api/chat/chatGpt.ts

@@ -0,0 +1,110 @@
+// Next.js API route support: https://nextjs.org/docs/api-routes/introduction
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { connectToDatabase, Chat, ChatWindow } from '@/service/mongo';
+import type { ModelType } from '@/types/model';
+import { getOpenAIApi, authChat } from '@/service/utils/chat';
+import { openaiProxy } from '@/service/utils/tools';
+import { ChatCompletionRequestMessage, ChatCompletionRequestMessageRoleEnum } from 'openai';
+import { ChatItemType } from '@/types/chat';
+
+/* 发送提示词 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  res.writeHead(200, {
+    Connection: 'keep-alive',
+    'Content-Encoding': 'none',
+    'Cache-Control': 'no-cache',
+    'Content-Type': 'text/event-stream'
+  });
+  const { chatId, windowId } = req.query as { chatId: string; windowId: string };
+
+  try {
+    if (!windowId || !chatId) {
+      throw new Error('缺少参数');
+    }
+
+    await connectToDatabase();
+
+    const { chat, userApiKey } = await authChat(chatId);
+
+    const model: ModelType = chat.modelId;
+
+    const map = {
+      Human: ChatCompletionRequestMessageRoleEnum.User,
+      AI: ChatCompletionRequestMessageRoleEnum.Assistant,
+      SYSTEM: ChatCompletionRequestMessageRoleEnum.System
+    };
+    // 读取对话内容
+    const prompts: ChatItemType[] = (await ChatWindow.findById(windowId)).content;
+
+    // 长度过滤
+    const maxContext = model.security.contextMaxLen;
+    const filterPrompts =
+      prompts.length > maxContext + 2
+        ? [prompts[0], ...prompts.slice(prompts.length - maxContext)]
+        : prompts.slice(0, prompts.length);
+
+    // 格式化文本内容
+    const formatPrompts: ChatCompletionRequestMessage[] = filterPrompts.map(
+      (item: ChatItemType) => ({
+        role: map[item.obj],
+        content: item.value
+      })
+    );
+
+    // 获取 chatAPI
+    const chatAPI = getOpenAIApi(userApiKey);
+    const chatResponse = await chatAPI.createChatCompletion(
+      {
+        model: model.service.chatModel,
+        temperature: 1,
+        // max_tokens: model.security.contentMaxLen,
+        messages: formatPrompts,
+        stream: true
+      },
+      openaiProxy
+    );
+
+    // 截取字符串内容
+    const reg = /{"content"(.*)"}/g;
+    // @ts-ignore
+    const match = chatResponse.data.match(reg);
+    let AIResponse = '';
+    if (match) {
+      match.forEach((item: string, i: number) => {
+        try {
+          const json = JSON.parse(item);
+          // 开头的换行忽略
+          if (i === 0 && json.content?.startsWith('\n')) return;
+          AIResponse += json.content;
+          const content = json.content.replace(/\n/g, '<br/>'); // 无法直接传输\n
+          content && res.write(`data: ${content}\n\n`);
+        } catch (err) {
+          err;
+        }
+      });
+    }
+    res.write(`data: [DONE]\n\n`);
+
+    // 存入库
+    await ChatWindow.findByIdAndUpdate(windowId, {
+      $push: {
+        content: {
+          obj: 'AI',
+          value: AIResponse
+        }
+      },
+      updateTime: Date.now()
+    });
+
+    res.end();
+  } catch (err: any) {
+    console.log(err?.response?.data || err);
+    // 删除最一条数据库记录, 也就是预发送的那一条
+    await ChatWindow.findByIdAndUpdate(windowId, {
+      $pop: { content: 1 },
+      updateTime: Date.now()
+    });
+
+    res.end();
+  }
+}