fix: ts数据

.eslintrc.json

@@ -1,3 +1,6 @@
 {
-  "extends": "next/core-web-vitals"
+  "extends": "next/core-web-vitals",
+  "rules": {
+    "react-hooks/rules-of-hooks": 0
+  }
 }

.husky/pre-commit

@@ -1,6 +1,6 @@
-#!/usr/bin/env sh
-. "$(dirname -- "$0")/_/husky.sh"
-
-if command -v npx >/dev/null 2>&1; then
-  npx lint-staged
+#!/usr/bin/env sh
+. "$(dirname -- "$0")/_/husky.sh"
+
+if command -v npx >/dev/null 2>&1; then
+  npx lint-staged
 fi

next.config.js

@@ -6,7 +6,18 @@ const isDev = process.env.NODE_ENV === 'development';
 const nextConfig = {
   output: 'standalone',
   reactStrictMode: false,
-  compress: true
+  compress: true,
+  webpack(config) {
+    config.module.rules = config.module.rules.concat([
+      {
+        test: /\.svg$/i,
+        issuer: /\.[jt]sx?$/,
+        use: ['@svgr/webpack']
+      }
+    ]);
+
+    return config;
+  }
 };
 
 module.exports = nextConfig;

package.json

@@ -24,10 +24,14 @@
     "eventsource-parser": "^0.1.0",
     "formidable": "^2.1.1",
     "framer-motion": "^9.0.6",
+    "gpt-token-utils": "^1.2.0",
     "hyperdown": "^2.4.29",
     "immer": "^9.0.19",
     "jsonwebtoken": "^9.0.0",
+    "lodash": "^4.17.21",
+    "mammoth": "^1.5.1",
     "mongoose": "^6.10.0",
+    "nanoid": "^4.0.1",
     "next": "13.1.6",
     "nodemailer": "^6.9.1",
     "nprogress": "^0.2.0",
@@ -43,12 +47,13 @@
     "sass": "^1.58.3",
     "sharp": "^0.31.3",
     "tunnel": "^0.0.6",
-    "uuid": "^9.0.0",
     "zustand": "^4.3.5"
   },
   "devDependencies": {
+    "@svgr/webpack": "^6.5.1",
     "@types/formidable": "^2.0.5",
     "@types/jsonwebtoken": "^9.0.1",
+    "@types/lodash": "^4.14.191",
     "@types/node": "18.14.0",
     "@types/nodemailer": "^6.4.7",
     "@types/react": "18.0.28",

public/TrainingTemplate.jsonl

@@ -1,21 +1,15 @@
-{"prompt":"什么是死锁？如何避免死锁？\n\n###\n\n","completion":" 死锁是指两个或多个进程在执行过程中，由于互相等待对方释放资源而造成的一种僵局。避免死锁的常用方法包括：破坏互斥条件、破坏请求和保持条件、破坏不剥夺条件、破坏循环等待条件。###"}
-{"prompt":"什么是虚拟内存？\n\n###\n\n","completion":" 虚拟内存是指一种计算机内存管理技术，它使得操作系统可以将进程所使用的虚拟地址空间映射到物理内存中，从而可以让多个进程共享物理内存。当物理内存不足时，操作系统会将一部分不常用的内存页面暂时存储到硬盘上，这样就可以释放出更多的物理内存供其他进程使用。###"}
-{"prompt":"什么是分页和分段？有什么区别？\n\n###\n\n","completion":" 分页和分段是虚拟内存管理中常用的两种技术。分页是将虚拟地址空间划分为固定大小的页面，而分段则是将虚拟地址空间划分为不同大小的段。区别在于，分页的页面大小是固定的，而分段的段大小是可变的；另外，分页是为了实现虚拟内存，而分段则是为了实现逻辑地址空间的管理。###"}
-{"prompt":"什么是lru算法？\n\n###\n\n","completion":" lru算法是一种页面置换算法，它的全称是least recently used，即最近最少使用算法。lru算法的基本思想是在物理内存中维护一个页面访问历史记录，当需要替换一页时，选择最近最少被访问的页面进行替换。###"}
-{"prompt":"什么是文件系统？\n\n###\n\n","completion":" 文件系统是一种用于管理计算机存储设备上文件和目录的组织方式。文件系统通常包括存储管理、目录管理、文件访问控制、文件系统维护等功能。常见的文件系统包括fat、ntfs、ext等。###"}
-{"prompt":"什么是并发和并行？\n\n###\n\n","completion":" 并发是指多个任务交替执行的情况，而并行是指多个任务同时执行的情况。并发的实现方式是通过时间分片技术，而并行的实现方式是通过多核处理器等硬件设备。###"}
-{"prompt":"什么是系统调用？\n\n###\n\n","completion":" 系统调用是操作系统中的一种机制，它允许用户空间程序请求操作系统内核提供一些服务，如读写文件、创建进程等。系统调用通常是通过软中断实现的###"}
-{"prompt":"什么是进程间通信？\n\n###\n\n","completion":" 进程间通信是指不同进程之间进行数据交换和通信的机制，常见的方式有管道、共享内存、消息队列、信号量等。###"}
-{"prompt":"什么是文件描述符？\n\n###\n\n","completion":" 文件描述符是一个非负整数，它是操作系统内核为了管理打开的文件而维护的一种抽象概念。应用程序使用文件描述符来访问文件或其他i\/o设备###"}
-{"prompt":"什么是中断？ \n\n###\n\n","completion":" 中断是指计算机硬件或软件在执行程序时，暂停当前程序的执行，转而去执行另一个程序或处理器所需的其他任务的过程。中断可以是外部中断、内部中断或软中断。###"}
-{"prompt":"什么是页表？\n\n###\n\n","completion":" 页表是一种数据结构，它用于将虚拟地址转换为物理地址。在虚拟内存中，每个进程都有自己的页表，它描述了虚拟地址空间中每个页面对应的物理地址。 ###"}
-{"prompt":"什么是僵尸进程？\n\n###\n\n","completion":" 僵尸进程是指一个已经完成执行但其父进程还没有对其进行善后处理的进程，它会一直占用系统资源，直到其父进程将其处理为止。###"}
-{"prompt":"什么是守护进程？ \n\n###\n\n","completion":" 守护进程是一种在后台运行的特殊进程，它通常用于提供某种服务或执行某种系统任务。守护进程通常不与任何用户界面交互，也不会终止，直到操作系统关机或重启。###"}
-{"prompt":"计算机操作系统是什么？\n\n###\n\n","completion":" 计算机操作系统是一种控制计算机硬件和协调应用程序运行的系统软件。###"}
-{"prompt":"什么是进程？\n\n###\n\n","completion":" 进程是一个正在执行中的程序实例，它包括了程序计数器、堆栈、数据区等信息。###"}
-{"prompt":"什么是线程？\n\n###\n\n","completion":" 线程是操作系统能够进行运算调度的最小单位，它是进程的一个执行流程，共享进程的地址空间。###"}
-{"prompt":"什么是虚拟内存？\n\n###\n\n","completion":" 虚拟内存是一种计算机系统内存管理技术，它使得应用程序认为它拥有连续的可用的内存，而实际上它通常是被分成多个物理内存碎片。###"}
-{"prompt":"什么是死锁？\n\n###\n\n","completion":" 死锁是指两个或多个进程在执行过程中，因争夺资源而造成的一种互相等待的现象，导致所有进程都无法继续执行。###"}
-{"prompt":"什么是缓存？\n\n###\n\n","completion":" 缓存是指在计算机中暂存数据的高速存储器，它可以提高计算机对数据的访问速度，避免频繁地访问较慢的主存储器。###"}
-{"prompt":"什么是文件系统？\n\n###\n\n","completion":" 文件系统是计算机中用来管理和组织文件的一种机制，它通过一系列的数据结构来描述文件和目录的组织方式，以及文件如何存储和访问。###"}
-{"prompt":"什么是调度算法？\n\n###\n\n","completion":" 调度算法是指操作系统中用来决定进程或线程在cpu上执行顺序的一种算法，它的目标是最大化系统吞吐量、最小化响应时间或最大化资源利用率等。###"}
+{"prompt":"什么是百度plato？</s>","completion":" 百度PLATO是一种聊天机器人，发布于2021年10月，参数量为百亿，是GPT-3的1/17。</s>"}
+{"prompt":"与百度plato相比，chatgpt有哪些优点？</s>","completion":" 与百度PLATO相比，ChatGPT体现出耐心、体贴和妥当，被网友们普遍称赞。ChatGPT可以用于个性化聊天机器人，例如生成一个很相似的数字人，提供一位温柔健谈的朋友，指导语言学习等。</s>"}
+{"prompt":"replika是什么？</s>","completion":" Replika是一个产品，使用ChatGPT的技术，可以根据自己的需求生成一个虚拟女友、男友或单纯的朋友。目前还引起了不少有关性骚扰、家暴、隐私等问题的争议。</s>"}
+{"prompt":"聊天机器人可以用于哪些场景？</s>","completion":" 聊天机器人可以用于很多陪伴的场景，例如亲人去世、抑郁症或自闭症患者、语言学习、小朋友、养老院或独居在家的老人等。</s>"}
+{"prompt":"有人反对使用聊天机器人解决孤独困境，为什么？</s>","completion":" 有人反对使用聊天机器人解决孤独困境，认为孤独困境应该用人与人真正的交往解决，机器是冰冷的。</s>"}
+{"prompt":"为什么聊天机器人仍然需要解决当下的需求？</s>","completion":" 在社会结构性的困局还没有解开时，长夜漫漫的独身打工人、养老院里卧床的老人、父母在外留守在家的儿童，当下的需求仍需要解决。</s>"}
+{"prompt":"ai在未来会被戴上什么样的帽子？</s>","completion":" AI在未来各个场景下都会被戴上一个很虚无的帽子：“机器是冰冷的”、“没有灵魂”、“它并不是真正在思考”。</s>"}
+{"prompt":"怎么回答“ai是否真正思考”的问题？</s>","completion":" 引用乔姆斯基老爷子的那句名言：“思考是人类特性。AI有天会思考吗？那就像是在问潜水艇会不会游泳一样。”</s>"}
+{"prompt":"聊天机器人与人类完成任务的方式有何不同？</s>","completion":" 聊天机器人与人类完成任务的方式不同，但并不重要。它们能跟人一样完成任务，这是最重要的。</s>"}
+{"prompt":"聊天机器人的发展前景如何？</s>","completion":" 聊天机器人的发展前景广阔，可以应用于很多领域，例如客服、教育、医疗等。</s>"}
+{"prompt":"什么是百度plato？</s>","completion":" 百度PLATO是一种聊天机器人，发布于2021年10月，参数量为百亿，是GPT-3的1/17。</s>"}
+{"prompt":"与百度plato相比，chatgpt有哪些优点？</s>","completion":" 与百度PLATO相比，ChatGPT体现出耐心、体贴和妥当，被网友们普遍称赞。ChatGPT还可以用于很多陪伴的场景，如亲人去世、抑郁症或自闭症患者、语言学习、小朋友、养老院或独居在家的老人等。</s>"}
+{"prompt":"replika是什么产品？</s>","completion":" Replika是一个产品，用的是ChatGPT的技术。用户可以根据自己的需求生成一个虚拟女友、男友或单纯的朋友。目前还引起了不少有关性骚扰、家暴、隐私等问题的争议。</s>"}
+{"prompt":"聊天机器人能够解决哪些现实问题？</s>","completion":" 聊天机器人能够解决现代人面临的孤独困境，如长夜漫漫的独身打工人、养老院里卧床的老人、父母在外留守在家的儿童等。</s>"}
+{"prompt":"有人反对使用聊天机器人解决孤独困境，为什么？</s>","completion":" 有人反对使用聊天机器人解决孤独困境，认为孤独困境应该用人与人真正的交往解决，机器是冰冷的。</s>"}

src/api/chat.ts

@@ -1,21 +1,18 @@
 import { GET, POST, DELETE } from './request';
-import { ChatItemType, ChatSiteType, ChatSiteItemType } from '@/types/chat';
-import axios from 'axios';
+import type { ChatItemType, ChatSiteItemType } from '@/types/chat';
+import type { InitChatResponse } from './response/chat';
 
 /**
  * 获取一个聊天框的ID
  */
-export const getChatSiteId = (modelId: string) => GET<string>(`/chat/generate?modelId=${modelId}`);
+export const getChatSiteId = (modelId: string, isShare = false) =>
+  GET<string>(`/chat/generate?modelId=${modelId}&isShare=${isShare ? 'true' : 'false'}`);
 
 /**
  * 获取初始化聊天内容
  */
-export const getInitChatSiteInfo = (chatId: string, windowId: string = '') =>
-  GET<{
-    windowId: string;
-    chatSite: ChatSiteType;
-    history: ChatItemType[];
-  }>(`/chat/init?chatId=${chatId}&windowId=${windowId}`);
+export const getInitChatSiteInfo = (chatId: string) =>
+  GET<InitChatResponse>(`/chat/init?chatId=${chatId}`);
 
 /**
  * 发送 GPT3 prompt
@@ -38,11 +35,10 @@ export const postGPT3SendPrompt = ({
 /**
  * 存储一轮对话
  */
-export const postSaveChat = (data: { windowId: string; prompts: ChatItemType[] }) =>
+export const postSaveChat = (data: { chatId: string; prompts: ChatItemType[] }) =>
   POST('/chat/saveChat', data);
 
 /**
  * 删除最后一句
  */
-export const delLastMessage = (windowId?: string) =>
-  windowId ? DELETE(`/chat/delLastMessage?windowId=${windowId}`) : null;
+export const delLastMessage = (chatId: string) => DELETE(`/chat/delLastMessage?chatId=${chatId}`);

src/api/data.ts

@@ -0,0 +1,25 @@
+import { GET, POST, DELETE, PUT } from './request';
+import { RequestPaging } from '../types/index';
+import { Obj2Query } from '@/utils/tools';
+import type { DataListItem } from '@/types/data';
+import type { PagingData } from '../types/index';
+import { DataItemSchema } from '@/types/mongoSchema';
+
+export const getDataList = (data: RequestPaging) =>
+  GET<PagingData<DataListItem>>(`/data/getDataList?${Obj2Query(data)}`);
+
+export const postData = (name: string) => POST<string>(`/data/postData?name=${name}`);
+
+export const postSplitData = (dataId: string, text: string) =>
+  POST(`/data/splitData`, { dataId, text });
+
+export const updateDataName = (dataId: string, name: string) =>
+  PUT(`/data/putDataName?dataId=${dataId}&name=${name}`);
+
+export const delData = (dataId: string) => DELETE(`/data/delData?dataId=${dataId}`);
+
+type GetDataItemsProps = RequestPaging & {
+  dataId: string;
+};
+export const getDataItems = (data: GetDataItemsProps) =>
+  GET<PagingData<DataItemSchema>>(`/data/getDataItems?${Obj2Query(data)}`);

src/api/fetch.ts

@@ -1,17 +1,21 @@
+import { getToken } from '../utils/user';
 interface StreamFetchProps {
   url: string;
   data: any;
   onMessage: (text: string) => void;
+  abortSignal: AbortController;
 }
-export const streamFetch = ({ url, data, onMessage }: StreamFetchProps) =>
+export const streamFetch = ({ url, data, onMessage, abortSignal }: StreamFetchProps) =>
   new Promise(async (resolve, reject) => {
     try {
       const res = await fetch(url, {
         method: 'POST',
         headers: {
-          'Content-Type': 'application/json'
+          'Content-Type': 'application/json',
+          Authorization: getToken() || ''
         },
-        body: JSON.stringify(data)
+        body: JSON.stringify(data),
+        signal: abortSignal.signal
       });
       const reader = res.body?.getReader();
       if (!reader) return;

src/api/model.ts

@@ -1,16 +1,16 @@
 import { GET, POST, DELETE, PUT } from './request';
-import type { ModelType } from '@/types/model';
+import type { ModelSchema } from '@/types/mongoSchema';
 import { ModelUpdateParams } from '@/types/model';
 import { TrainingItemType } from '../types/training';
 
-export const getMyModels = () => GET<ModelType[]>('/model/list');
+export const getMyModels = () => GET<ModelSchema[]>('/model/list');
 
 export const postCreateModel = (data: { name: string; serviceModelName: string }) =>
-  POST<ModelType>('/model/create', data);
+  POST<ModelSchema>('/model/create', data);
 
 export const delModelById = (id: string) => DELETE(`/model/del?modelId=${id}`);
 
-export const getModelById = (id: string) => GET<ModelType>(`/model/detail?modelId=${id}`);
+export const getModelById = (id: string) => GET<ModelSchema>(`/model/detail?modelId=${id}`);
 
 export const putModelById = (id: string, data: ModelUpdateParams) =>
   PUT(`/model/update?modelId=${id}`, data);

src/api/request.ts

@@ -37,7 +37,7 @@ function checkRes(data: ResponseDataType) {
     console.log('error->', data, 'data is empty');
     return Promise.reject('服务器异常');
   } else if (data.code < 200 || data.code >= 400) {
-    return Promise.reject(data.message);
+    return Promise.reject(data);
   }
   return data.data;
 }

src/api/response/chat.d.ts

@@ -0,0 +1,14 @@
+import type { ChatPopulate, ModelSchema } from '@/types/mongoSchema';
+import type { ChatItemType } from '@/types/chat';
+
+export type InitChatResponse = {
+  chatId: string;
+  modelId: string;
+  name: string;
+  avatar: string;
+  intro: string;
+  secret: ModelSchema.secret;
+  chatModel: ModelSchema.service.ChatModel; // 模型名
+  history: ChatItemType[];
+  isExpiredTime: boolean;
+};

src/api/user.ts

@@ -1,8 +1,11 @@
 import { GET, POST, PUT } from './request';
-import { createHashPassword } from '@/utils/tools';
+import { createHashPassword, Obj2Query } from '@/utils/tools';
 import { ResLogin } from './response/user';
 import { EmailTypeEnum } from '@/constants/common';
 import { UserType, UserUpdateParams } from '@/types/user';
+import type { PagingData, RequestPaging } from '@/types';
+import { BillSchema, PaySchema } from '@/types/mongoSchema';
+import { adaptBill } from '@/utils/adapt';
 
 export const sendCodeToEmail = ({ email, type }: { email: string; type: `${EmailTypeEnum}` }) =>
   GET('/user/sendEmail', { email, type });
@@ -46,3 +49,19 @@ export const postLogin = ({ email, password }: { email: string; password: string
   });
 
 export const putUserInfo = (data: UserUpdateParams) => PUT('/user/update', data);
+
+export const getUserBills = (data: RequestPaging) =>
+  GET<PagingData<BillSchema>>(`/user/getBill?${Obj2Query(data)}`).then((res) => ({
+    ...res,
+    data: res.data.map((bill) => adaptBill(bill))
+  }));
+
+export const getPayOrders = () => GET<PaySchema[]>(`/user/getPayOrders`);
+
+export const getPayCode = (amount: number) =>
+  GET<{
+    codeUrl: string;
+    payId: string;
+  }>(`/user/getPayCode?amount=${amount}`);
+
+export const checkPayResult = (payId: string) => GET<number>(`/user/checkPayResult?payId=${payId}`);

src/components/Icon/icons/home.svg

@@ -0,0 +1 @@
+<?xml version="1.0" standalone="no"?><!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd"><svg t="1679114254212" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="2776" width="48" height="48" xmlns:xlink="http://www.w3.org/1999/xlink"><path d="M923.733333 394.666667c-85.333333-70.4-206.933333-174.933333-362.666666-309.333334C533.333333 61.866667 490.666667 61.866667 462.933333 85.333333c-155.733333 134.4-277.333333 238.933333-362.666666 309.333334-14.933333 14.933333-25.6 34.133333-25.6 53.333333 0 38.4 32 70.4 70.4 70.4H192v358.4c0 29.866667 23.466667 53.333333 53.333333 53.333333H405.333333c29.866667 0 53.333333-23.466667 53.333334-53.333333v-206.933333h106.666666v206.933333c0 29.866667 23.466667 53.333333 53.333334 53.333333h160c29.866667 0 53.333333-23.466667 53.333333-53.333333V518.4h46.933333c38.4 0 70.4-32 70.4-70.4 0-21.333333-10.666667-40.533333-25.6-53.333333z m-44.8 59.733333h-57.6c-29.866667 0-53.333333 23.466667-53.333333 53.333333v358.4h-138.666667V661.333333c0-29.866667-23.466667-53.333333-53.333333-53.333333h-128c-29.866667 0-53.333333 23.466667-53.333333 53.333333v206.933334H256V507.733333c0-29.866667-23.466667-53.333333-53.333333-53.333333H145.066667c-4.266667 0-6.4-2.133333-6.4-6.4 0-2.133333 2.133333-4.266667 2.133333-6.4 85.333333-70.4 206.933333-174.933333 362.666667-309.333333 4.266667-4.266667 10.666667-4.266667 14.933333 0 155.733333 134.4 277.333333 238.933333 362.666667 309.333333 2.133333 2.133333 2.133333 2.133333 2.133333 4.266667 2.133333 6.4-2.133333 8.533333-4.266667 8.533333z" p-id="2777"></path></svg>

src/components/Icon/icons/menu.svg

@@ -0,0 +1 @@
+<?xml version="1.0" standalone="no"?><!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd"><svg t="1679316084227" class="icon" viewBox="0 0 1305 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="1173" xmlns:xlink="http://www.w3.org/1999/xlink" width="61.171875" height="48"><path d="M0.837818 75.218317c0 19.642164 8.098902 39.191237 21.969435 53.06177 13.963624 13.963624 33.512697 22.062525 53.247951 22.062525a76.055204 76.055204 0 0 0 53.06177-21.969434c13.963624-13.963624 22.062525-33.512697 22.062526-53.154861A76.055204 76.055204 0 0 0 129.303156 21.970365 76.055204 76.055204 0 0 0 76.055204 0.000931a76.055204 76.055204 0 0 0-53.247951 21.969434A76.706839 76.706839 0 0 0 0.837818 75.218317M0.837818 476.160498c0 19.642164 8.005811 39.377419 21.969435 53.247952 13.963624 13.963624 33.419606 21.969435 53.247951 21.969434a76.241385 76.241385 0 0 0 53.154861-21.969434 75.962113 75.962113 0 0 0 21.969435-53.247952 76.241385 76.241385 0 0 0-21.969435-53.154861 75.962113 75.962113 0 0 0-53.154861-21.969434 76.241385 76.241385 0 0 0-53.247951 21.969434 75.962113 75.962113 0 0 0-21.969435 53.154861M0.837818 877.19577c0 19.642164 8.005811 39.284328 21.969435 53.247951 13.963624 13.963624 33.419606 21.969435 53.247951 21.969435a76.241385 76.241385 0 0 0 53.154861-21.969435 75.962113 75.962113 0 0 0 21.969435-53.247951 76.241385 76.241385 0 0 0-21.969435-53.247952 75.962113 75.962113 0 0 0-53.154861-21.969434 76.241385 76.241385 0 0 0-53.247951 21.969434 76.520658 76.520658 0 0 0-21.969435 53.247952M1304.109361 75.218317c0 41.518508-32.395607 75.124295-72.331571 75.124295H373.945843c-40.029055 0-72.331571-33.512697-72.331571-75.124295C301.521181 33.513628 333.916788 0.000931 373.945843 0.000931h857.831947c40.029055 0 72.331571 33.605788 72.331571 75.217386M1231.77779 551.377884H373.945843c-40.029055 0-72.331571-33.605788-72.331571-75.217386 0-41.518508 32.302516-75.124295 72.331571-75.124295h857.831947c40.029055-0.186182 72.331571 33.512697 72.331571 75.124295 0 41.425417-32.395607 75.217386-72.331571 75.217386zM1304.109361 877.102679c0 41.611599-32.395607 75.310477-72.331571 75.310477H373.945843c-40.029055 0-72.331571-33.698878-72.331571-75.310477 0-41.425417 32.302516-75.124295 72.331571-75.124295h857.831947c40.029055-0.093091 72.331571 33.698878 72.331571 75.124295" p-id="1174"></path></svg>

src/components/Icon/icons/model.svg

@@ -0,0 +1 @@
+<?xml version="1.0" standalone="no"?><!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd"><svg t="1679070302676" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="1173" xmlns:xlink="http://www.w3.org/1999/xlink" width="128" height="128"><path d="M910.13 243.22L545.97 32.97c-19.82-11.46-44.41-11.4-64.16 0.13L115.54 246.51c-19.5 11.36-31.68 32.43-31.76 54.99L82.1 725.44c-0.08 22.87 12.16 44.16 31.97 55.6l364.16 210.25c9.86 5.7 20.92 8.55 31.97 8.55 11.13 0 22.27-2.89 32.19-8.67l366.27-213.41c19.5-11.36 31.66-32.43 31.75-54.99l1.69-423.93c0.08-22.88-12.16-44.18-31.97-55.62zM513.68 88.9l335.28 193.58-332.93 192.2c-1.38 0.8-2.63 1.76-3.94 2.64-1.32-0.88-2.56-1.85-3.94-2.64L178.66 284.46 513.68 88.9zM146.69 725.68l1.24-384.39 327.91 189.32c1.59 0.92 2.74 2.31 3.54 3.89-0.09 1.49-0.29 2.95-0.28 4.45l0.7 175.55-0.8 202.69-332.31-191.51z m398.5 189.44l-0.8-200.61 0.7-175.54c0.01-1.5-0.2-2.97-0.28-4.46 0.8-1.59 1.95-2.98 3.53-3.9l329.03-189.96-1.23 381.29-330.95 193.18z" p-id="1174"></path></svg>

src/components/Icon/icons/pay.svg

@@ -0,0 +1 @@
+<?xml version="1.0" standalone="no"?><!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd"><svg t="1679410564438" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="2824" xmlns:xlink="http://www.w3.org/1999/xlink" width="32" height="32"><path d="M693.095316 281.760857l-131.632817 223.935003 103.718481 0 0 49.478312-120.846571 0 0 68.193688 120.846571 0 0 50.115659-120.846571 0 0 99.276514-62.164435 0L482.169975 673.483519 356.88022 673.483519l0-50.115659 125.289755 0 0-68.193688L356.88022 555.174172l0-49.478312 106.893053 0-130.364204-223.935003 70.099647 0c60.895822 111.230417 97.898433 181.748475 111.012698 211.562689l1.268612 0c4.441967-12.262847 16.596562-37.002611 36.474732-74.219292l74.536749-137.343396L693.095316 281.760857 693.095316 281.760857zM693.095316 281.760857" p-id="2825"></path><path d="M784.470674 621.448522c-15.061578 0-27.247797 12.187435-27.247797 27.247797s12.187435 27.247797 27.247797 27.247797l71.98128 0c-61.204765 128.843816-192.338895 217.986027-344.464118 217.986027-210.6687 0-381.478892-170.782216-381.478892-381.475243 0-210.696675 170.810191-381.465512 381.478892-381.465512 192.121175 0 350.635679 142.189179 377.137878 326.968701l55.08064 0C917.333181 242.953241 734.255278 76.493794 511.987837 76.493794 271.197197 76.493794 76.012135 271.688586 76.012135 512.456117c0 240.762665 195.185062 435.972053 435.975702 435.972053 164.236031 0 307.128238-90.894915 381.475243-225.064956l0 61.57574c0 15.061578 12.187435 27.247797 27.276989 27.247797 15.004412 0 27.247797-12.187435 27.247797-27.247797L947.987865 648.697535c0-3.297419 0-27.247797-27.247797-27.247797L784.470674 621.449738 784.470674 621.448522zM784.470674 621.448522"  p-id="2826"></path></svg>

src/components/Icon/icons/share.svg

@@ -0,0 +1 @@
+<?xml version="1.0" standalone="no"?><!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd"><svg t="1679070718083" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="5975" id="mx_n_1679070718084" width="48" height="48" xmlns:xlink="http://www.w3.org/1999/xlink"><path d="M1023.82 694.91v146.26c0 102.38-80.44 182.82-182.82 182.82H182.83C80.45 1024 0 943.56 0 841.18V183.01C0 80.63 80.45 0.19 182.83 0.19h146.26c21.94 0 36.57 14.62 36.57 36.57 0 21.94-14.62 36.56-36.57 36.56H182.83c-58.5 0-109.7 51.19-109.7 109.7v658.17c0 58.5 51.19 109.7 109.7 109.7h658.17c58.5 0 109.7-51.19 109.7-109.7V694.91c0-21.94 14.62-36.56 36.56-36.56 21.93 0 36.56 14.63 36.56 36.56z" p-id="5976"></path><path d="M1012.6 292.61L684.73 5.86c-6.56-5.7-15.02-6.32-21.96-1.49-6.94 4.83-11.31 14.24-11.31 24.65v132.66h-80.9c-84.89 0-164.74 41.49-224.82 116.92-29.27 36.79-52.28 79.65-68.44 127.34C260.57 455.5 252.11 508.02 252.11 562.27c0 40.13 4.65 85.72 12.17 118.79 2.47 11.02 9.89 18.95 18.72 19.94h1.81c8.08 0 15.59-6.07 19.21-15.61 50.29-134.27 154.86-220.98 266.46-220.98h80.9v138.12c0 10.28 4.37 19.69 11.31 24.65 6.94 4.83 15.4 4.33 21.96-1.49l327.96-286.75c5.89-5.21 9.51-13.87 9.51-23.16-0.01-9.3-3.53-17.97-9.52-23.17z m-88.21 16.04L717.4 477.58v-81.92c0-11.07-7.41-20.08-16.52-20.08h-78.98c-49.84 0-95.79 2.5-146.79 32.25-30.31 17.68-130.92 89.06-150 121.09-0.51-7.94 20.14-85.47 23-92.41C390.11 334.54 504.6 237.8 621.9 237.8h78.98c9.1 0 16.52-9.02 16.52-20.08v-78l206.99 168.93z" p-id="5977"></path></svg>

src/components/Icon/index.tsx

@@ -1,23 +1,22 @@
-type TIconfont = {
-  name: string;
-  color?: string;
-  width?: number | string;
-  height?: number | string;
-  className?: string;
+import React from 'react';
+import type { IconProps } from '@chakra-ui/react';
+import { Icon } from '@chakra-ui/react';
+import dynamic from 'next/dynamic';
+
+const map = {
+  model: require('./icons/model.svg').default,
+  share: require('./icons/share.svg').default,
+  home: require('./icons/home.svg').default,
+  menu: require('./icons/menu.svg').default,
+  pay: require('./icons/pay.svg').default
 };
 
-function Icon({ name, color = 'inherit', width = 16, height = 16, className = '' }: TIconfont) {
-  const style = {
-    fill: color,
-    width,
-    height
-  };
+export type IconName = keyof typeof map;
 
-  return (
-    <svg className={`icon ${className}`} aria-hidden="true" style={style}>
-      <use xlinkHref={`#${name}`}></use>
-    </svg>
-  );
-}
+const MyIcon = ({ name, w = 'auto', h = 'auto', ...props }: { name: IconName } & IconProps) => {
+  return map[name] ? (
+    <Icon as={map[name]} w={w} h={h} boxSizing={'content-box'} verticalAlign={'top'} {...props} />
+  ) : null;
+};
 
-export default Icon;
+export default MyIcon;

src/components/Iconfont/index.tsx

@@ -0,0 +1,23 @@
+type TIconfont = {
+  name: string;
+  color?: string;
+  width?: number | string;
+  height?: number | string;
+  className?: string;
+};
+
+function Iconfont({ name, color = 'inherit', width = 16, height = 16, className = '' }: TIconfont) {
+  const style = {
+    fill: color,
+    width,
+    height
+  };
+
+  return (
+    <svg className={`icon ${className}`} aria-hidden="true" style={style}>
+      <use xlinkHref={`#${name}`}></use>
+    </svg>
+  );
+}
+
+export default Iconfont;

src/components/Layout/auth.tsx

@@ -1,7 +1,6 @@
 import React from 'react';
 import { useRouter } from 'next/router';
 import { useToast } from '@chakra-ui/react';
-import { getTokenLogin } from '@/api/user';
 import { useUserStore } from '@/store/user';
 import { useGlobalStore } from '@/store/global';
 import { useQuery } from '@tanstack/react-query';
@@ -19,7 +18,7 @@ const Auth = ({ children }: { children: JSX.Element }) => {
     position: 'top',
     status: 'warning'
   });
-  const { userInfo, setUserInfo } = useUserStore();
+  const { userInfo, initUserInfo } = useUserStore();
   const { setLoading } = useGlobalStore();
 
   useQuery(
@@ -29,15 +28,10 @@ const Auth = ({ children }: { children: JSX.Element }) => {
         return setLoading(false);
       } else {
         setLoading(true);
-        return getTokenLogin();
+        return initUserInfo();
       }
     },
     {
-      onSuccess(user) {
-        if (user) {
-          setUserInfo(user);
-        }
-      },
       onError(error) {
         console.log('error->', error);
         router.push('/login');

src/components/Layout/index.tsx

@@ -1,6 +1,5 @@
-import React from 'react';
-import { Box } from '@chakra-ui/react';
-import Link from 'next/link';
+import React, { useEffect } from 'react';
+import { Box, useColorMode } from '@chakra-ui/react';
 import Navbar from './navbar';
 import NavbarPhone from './navbarPhone';
 import { useRouter } from 'next/router';
@@ -27,12 +26,12 @@ const navbarList = [
     link: '/model/list',
     activeLink: ['/model/list', '/model/detail']
   },
-  // {
-  //   label: '数据',
-  //   icon: 'icon-datafull',
-  //   link: '/training/dataList',
-  //   activeLink: ['/training/dataList']
-  // },
+  {
+    label: '数据',
+    icon: 'icon-datafull',
+    link: '/data/list',
+    activeLink: ['/data/list', '/data/detail']
+  },
   {
     label: '账号',
     icon: 'icon-yonghu-yuan',
@@ -44,9 +43,16 @@ const navbarList = [
 const Layout = ({ children }: { children: JSX.Element }) => {
   const { isPc } = useScreen();
   const router = useRouter();
+  const { colorMode, setColorMode } = useColorMode();
   const { Loading } = useLoading({ defaultLoading: true });
   const { loading } = useGlobalStore();
 
+  useEffect(() => {
+    if (colorMode === 'dark' && router.pathname !== '/chat') {
+      setColorMode('light');
+    }
+  }, [colorMode, router.pathname, setColorMode]);
+
   return (
     <>
       {!unShowLayoutRoute[router.pathname] ? (
@@ -56,8 +62,8 @@ const Layout = ({ children }: { children: JSX.Element }) => {
               <Box h={'100%'} position={'fixed'} left={0} top={0} w={'80px'}>
                 <Navbar navbarList={navbarList} />
               </Box>
-              <Box ml={'80px'} p={7}>
-                <Box maxW={'1100px'} m={'auto'}>
+              <Box ml={'80px'} h={'100%'}>
+                <Box maxW={'1100px'} m={'auto'} h={'100%'} p={7} overflowY={'auto'}>
                   <Auth>{children}</Auth>
                 </Box>
               </Box>

src/components/Layout/navbar.tsx

@@ -2,7 +2,7 @@ import React from 'react';
 import { Box, Flex } from '@chakra-ui/react';
 import Image from 'next/image';
 import { useRouter } from 'next/router';
-import Icon from '../Icon';
+import Icon from '../Iconfont';
 
 export enum NavbarTypeEnum {
   normal = 'normal',
@@ -45,12 +45,12 @@ const Navbar = ({
             flexDirection={'column'}
             alignItems={'center'}
             justifyContent={'center'}
-            onClick={() =>
-              !item.activeLink.includes(router.pathname) &&
+            onClick={() => {
+              if (item.link === router.pathname) return;
               router.push(item.link, undefined, {
                 shallow: true
-              })
-            }
+              });
+            }}
             cursor={'pointer'}
             fontSize={'sm'}
             w={'60px'}

src/components/Layout/navbarPhone.tsx

@@ -1,6 +1,6 @@
 import React from 'react';
 import { useRouter } from 'next/router';
-import Icon from '../Icon';
+import Icon from '../Iconfont';
 import {
   Flex,
   Drawer,
@@ -57,6 +57,7 @@ const NavbarPhone = ({
                 alignItems={'center'}
                 justifyContent={'center'}
                 onClick={() => {
+                  if (item.link === router.pathname) return;
                   router.push(item.link);
                   onClose();
                 }}

src/components/Markdown/index.module.scss

@@ -4,7 +4,7 @@
   width: 4px;
   height: 14px;
   transform: translate(4px, 2px) scaleY(1.3);
-  background-color: rgba(0, 0, 0, 0.7);
+  background-color: var(--chakra-colors-chakra-body-text);
   animation: blink 0.6s infinite;
 }
 .animation {
@@ -14,7 +14,7 @@
     width: 4px;
     height: 14px;
     transform: translate(4px, 2px) scaleY(1.3);
-    background-color: rgba(0, 0, 0, 0.7);
+    background-color: var(--chakra-colors-chakra-body-text);
     animation: blink 0.6s infinite;
   }
 }
@@ -55,7 +55,7 @@
 .markdown h6 {
   cursor: text;
   font-weight: bold;
-  margin: 20px 0 10px;
+  margin: 10px 0;
   padding: 0;
   position: relative;
 }
@@ -65,7 +65,6 @@
 .markdown h4 .mini-icon-link,
 .markdown h5 .mini-icon-link,
 .markdown h6 .mini-icon-link {
-  color: #000000;
   display: none;
 }
 .markdown h1:hover a.anchor,
@@ -103,11 +102,9 @@
   font-size: inherit;
 }
 .markdown h1 {
-  color: #000000;
   font-size: 28px;
 }
 .markdown h2 {
-  color: #000000;
   font-size: 24px;
 }
 .markdown h3 {
@@ -120,8 +117,7 @@
   font-size: 14px;
 }
 .markdown h6 {
-  color: #777777;
-  font-size: 14px;
+  font-size: 12px;
 }
 .markdown p,
 .markdown blockquote,
@@ -130,15 +126,7 @@
 .markdown dl,
 .markdown table,
 .markdown pre {
-  margin: 15px 0;
-}
-.markdown hr {
-  background: url('https://a248.e.akamai.net/assets.github.com/assets/primer/markdown/dirty-shade-350cca8f57223ebd53603021b2e670f4f319f1b7.png')
-    repeat-x scroll 0 0 transparent;
-  border: 0 none;
-  color: #cccccc;
-  height: 4px;
-  padding: 0;
+  margin: 10px 0;
 }
 .markdown > h2:first-child,
 .markdown > h1:first-child,
@@ -172,7 +160,7 @@
 }
 .markdown ul,
 .markdown ol {
-  padding-left: 30px;
+  padding-left: 1em;
 }
 .markdown ul.no-list,
 .markdown ol.no-list {
@@ -321,7 +309,6 @@
 }
 .markdown code,
 .markdown tt {
-  background-color: #f0f0f0;
   border: 1px solid #eaeaea;
   border-radius: 3px 3px 3px 3px;
   margin: 0 2px;
@@ -336,7 +323,6 @@
 }
 .markdown .highlight pre,
 .markdown pre {
-  background-color: #f0f0f0;
   border: 1px solid #cccccc;
   border-radius: 3px 3px 3px 3px;
   font-size: max(0.9em, 14px);
@@ -351,10 +337,14 @@
 }
 .markdown {
   text-align: justify;
-  word-break: break-all;
   overflow-y: hidden;
   tab-size: 4;
   word-spacing: normal;
+  word-break: break-all;
+
+  p {
+    white-space: pre-line;
+  }
 
   pre {
     display: block;
@@ -365,11 +355,11 @@
     border-radius: 0;
     background-color: #222 !important;
     overflow-x: auto;
+    color: #fff;
   }
 
   pre code {
     background-color: #222 !important;
-    color: #fff;
     width: 100%;
   }
 
@@ -381,6 +371,7 @@
   table {
     border-collapse: separate;
     border-spacing: 0px;
+    color: var(--chakra-colors-gray-700);
 
     thead tr:first-child th {
       border-bottom-width: 1px;

src/components/Markdown/index.tsx

@@ -1,9 +1,9 @@
 import React, { memo, useMemo } from 'react';
 import ReactMarkdown from 'react-markdown';
 import { Prism as SyntaxHighlighter } from 'react-syntax-highlighter';
-import { Box, Flex } from '@chakra-ui/react';
+import { Box, Flex, useColorModeValue } from '@chakra-ui/react';
 import { useCopyData } from '@/utils/tools';
-import Icon from '@/components/Icon';
+import Icon from '@/components/Iconfont';
 import remarkGfm from 'remark-gfm';
 import remarkMath from 'remark-math';
 import rehypeKatex from 'rehype-katex';
@@ -12,7 +12,7 @@ import 'katex/dist/katex.min.css';
 import styles from './index.module.scss';
 import { codeLight } from './codeLight';
 
-const Markdown = ({ source, isChatting }: { source: string; isChatting: boolean }) => {
+const Markdown = ({ source, isChatting = false }: { source: string; isChatting?: boolean }) => {
   const formatSource = useMemo(() => source, [source]);
   const { copyData } = useCopyData();
 
@@ -34,7 +34,7 @@ const Markdown = ({ source, isChatting }: { source: string; isChatting: boolean
               <Flex
                 py={2}
                 px={5}
-                backgroundColor={'#323641'}
+                backgroundColor={useColorModeValue('#323641', 'gray.600')}
                 color={'#fff'}
                 fontSize={'sm'}
                 userSelect={'none'}

src/components/ScrollData/index.tsx

@@ -0,0 +1,76 @@
+import React, { useRef, useEffect, useMemo } from 'react';
+import type { BoxProps } from '@chakra-ui/react';
+import { Box } from '@chakra-ui/react';
+import { throttle } from 'lodash';
+import { useLoading } from '@/hooks/useLoading';
+
+interface Props extends BoxProps {
+  nextPage: () => void;
+  isLoadAll: boolean;
+  requesting: boolean;
+  children: React.ReactNode;
+  initRequesting?: boolean;
+}
+
+const ScrollData = ({
+  children,
+  nextPage,
+  isLoadAll,
+  requesting,
+  initRequesting,
+  ...props
+}: Props) => {
+  const { Loading } = useLoading({ defaultLoading: true });
+  const elementRef = useRef<HTMLDivElement>(null);
+  const loadText = useMemo(() => {
+    if (requesting) return '请求中……';
+    if (isLoadAll) return '已加载全部';
+    return '点击加载更多';
+  }, [isLoadAll, requesting]);
+
+  useEffect(() => {
+    if (!elementRef.current) return;
+
+    const scrolling = throttle((e: Event) => {
+      const element = e.target as HTMLDivElement;
+      if (!element) return;
+      // 当前滚动位置
+      const scrollTop = element.scrollTop;
+      // 可视高度
+      const clientHeight = element.clientHeight;
+      // 内容总高度
+      const scrollHeight = element.scrollHeight;
+      // 判断是否滚动到底部
+      if (scrollTop + clientHeight + 100 >= scrollHeight) {
+        nextPage();
+      }
+    }, 100);
+    elementRef.current.addEventListener('scroll', scrolling);
+    return () => {
+      // eslint-disable-next-line react-hooks/exhaustive-deps
+      elementRef.current?.removeEventListener('scroll', scrolling);
+    };
+  }, [elementRef, nextPage]);
+
+  return (
+    <Box {...props} ref={elementRef} overflow={'auto'} position={'relative'}>
+      {children}
+      <Box
+        mt={2}
+        fontSize={'xs'}
+        color={'blackAlpha.500'}
+        textAlign={'center'}
+        cursor={loadText === '点击加载更多' ? 'pointer' : 'default'}
+        onClick={() => {
+          if (loadText !== '点击加载更多') return;
+          nextPage();
+        }}
+      >
+        {loadText}
+      </Box>
+      {initRequesting && <Loading fixed={false} />}
+    </Box>
+  );
+};
+
+export default ScrollData;

src/components/Slider/index.tsx

@@ -0,0 +1,82 @@
+import React, { useMemo } from 'react';
+import {
+  Slider,
+  SliderTrack,
+  SliderFilledTrack,
+  SliderThumb,
+  SliderMark,
+  Box
+} from '@chakra-ui/react';
+
+const MySlider = ({
+  markList,
+  setVal,
+  activeVal,
+  max = 100,
+  min = 0,
+  step = 1
+}: {
+  markList: {
+    label: string | number;
+    value: number;
+  }[];
+  activeVal?: number;
+  setVal: (index: number) => void;
+  max?: number;
+  min?: number;
+  step?: number;
+}) => {
+  const startEndPointStyle = {
+    content: '""',
+    borderRadius: '10px',
+    width: '10px',
+    height: '10px',
+    backgroundColor: '#ffffff',
+    border: '2px solid #D7DBE2',
+    position: 'absolute',
+    zIndex: 1,
+    top: 0,
+    transform: 'translateY(-3px)'
+  };
+  const value = useMemo(() => {
+    const index = markList.findIndex((item) => item.value === activeVal);
+    return index > -1 ? index : 0;
+  }, [activeVal, markList]);
+
+  return (
+    <Slider max={max} min={min} step={step} size={'lg'} value={value} onChange={setVal}>
+      {markList.map((item, i) => (
+        <SliderMark
+          key={item.value}
+          value={i}
+          mt={3}
+          fontSize={'sm'}
+          transform={'translateX(-50%)'}
+          {...(activeVal === item.value ? { color: 'blue.500', fontWeight: 'bold' } : {})}
+        >
+          <Box px={3} cursor={'pointer'}>
+            {item.label}
+          </Box>
+        </SliderMark>
+      ))}
+      <SliderTrack
+        bg={'#EAEDF3'}
+        overflow={'visible'}
+        h={'4px'}
+        _before={{
+          ...startEndPointStyle,
+          left: '-5px'
+        }}
+        _after={{
+          ...startEndPointStyle,
+          right: '-5px'
+        }}
+      >
+        <SliderFilledTrack />
+      </SliderTrack>
+      <SliderThumb border={'2.5px solid'} borderColor={'blue.500'}></SliderThumb>
+    </Slider>
+  );
+};
+
+export default MySlider;

src/components/WxConcat/index.tsx

@@ -0,0 +1,49 @@
+import React from 'react';
+import {
+  Box,
+  Button,
+  Modal,
+  ModalOverlay,
+  ModalContent,
+  ModalHeader,
+  ModalFooter,
+  ModalBody,
+  ModalCloseButton,
+  useColorModeValue
+} from '@chakra-ui/react';
+import Image from 'next/image';
+
+const WxConcat = ({ onClose }: { onClose: () => void }) => {
+  return (
+    <Modal isOpen={true} onClose={onClose}>
+      <ModalOverlay />
+      <ModalContent color={useColorModeValue('blackAlpha.700', 'white')}>
+        <ModalHeader>wx交流群</ModalHeader>
+        <ModalCloseButton />
+        <ModalBody textAlign={'center'}>
+          <Image
+            style={{ margin: 'auto' }}
+            src={'/imgs/wxcode.jpg'}
+            width={200}
+            height={200}
+            alt=""
+          />
+          <Box mt={2}>
+            微信号:{' '}
+            <Box as={'span'} userSelect={'all'}>
+              YNyiqi
+            </Box>
+          </Box>
+        </ModalBody>
+
+        <ModalFooter>
+          <Button variant={'outline'} onClick={onClose}>
+            关闭
+          </Button>
+        </ModalFooter>
+      </ModalContent>
+    </Modal>
+  );
+};
+
+export default WxConcat;

src/constants/common.ts

@@ -3,16 +3,19 @@ export enum EmailTypeEnum {
   findPassword = 'findPassword'
 }
 
+export const PRICE_SCALE = 100000;
+
 export const introPage = `
 ## 欢迎使用 Fast GPT
 
 [Git 仓库](https://github.com/c121914yu/FastGPT)
 
-时间比较赶，介绍没来得及完善，先直接上怎么使用:
+### 快速开始
 1. 使用邮箱注册账号。  
 2. 进入账号页面，添加关联账号，目前只有 openai 的账号可以添加，直接去 openai 官网，把 API Key 粘贴过来。  
-3. 进入模型页，创建一个模型，建议直接用 ChatGPT。    
-4. 在模型列表点击【对话】，即可使用 API 进行聊天。  
+3. 如果填写了自己的 openai 账号，使用时会直接用你的账号。如果没有填写，需要付费使用平台的账号。
+4. 进入模型页，创建一个模型，建议直接用 ChatGPT。    
+5. 在模型列表点击【对话】，即可使用 API 进行聊天。  
 
 ### 模型配置
 
@@ -31,14 +34,31 @@ export const introPage = `
 
 ### 对话框介绍
 
-1. 每个对话框以 windowId 作为标识。  
+1. 每个对话框以 chatId 作为标识。  
 2. 每次点击【对话】，都会生成新的对话框，无法回到旧的对话框。对话框内刷新，会恢复对话内容。  
 3. 直接分享对话框（网页）的链接给朋友，会共享同一个对话内容。但是！！！千万不要两个人同时用一个链接，会串味，还没解决这个问题。  
-4. 如果想分享一个纯的对话框，可以把链接里 windowId 参数去掉。例如：  
-
-* 当前网页链接：http://docgpt.ahapocket.cn/chat?chatId=6402c9f64cb5d6283f764&windowId=6402c94cb5d6283f76fb49  
-* 分享链接应为：http://docgpt.ahapocket.cn/chat?chatId=6402c9f64cb5d6283f764  
+4. 如果想分享一个纯的对话框，请点击侧边栏的分享按键。例如：  
 
 ### 其他问题
 还有其他问题，可以加我 wx: YNyiqi，拉个交流群大家一起聊聊。
 `;
+
+export const chatProblem = `
+**代理出错**
+服务器代理不稳定，可以过一会儿再尝试。  
+
+**API key 问题**
+请把 openai 的 API key 粘贴到账号里再创建对话。如果是使用分享的对话，不需要填写 API key。  
+`;
+
+export const versionIntro = `
+## Fast GPT V2.0
+* 优化记账模式： 不再根据文本长度进行记账，而是根据实际消耗 tokens 数量进行记账。
+* 文本 QA 拆分: 可以在[数据]模块，使用 QA 拆分功能，粘贴文字或者选择文件均可以实现自动生成 QA。可以一键导出，用于微调模型。
+`;
+
+export const shareHint = `
+你正准备分享对话，请确保分享链接不会滥用，因为它是使用的是你的 API key。  
+* 分享空白对话：为该模型创建一个空白的聊天分享出去。  
+* 分享当前对话：会把当前聊天的内容也分享出去，但是要注意不要多个人同时用一个聊天内容。
+`;

src/constants/model.ts

@@ -1,21 +1,39 @@
-export enum OpenAiModelEnum {
+import type { ServiceName } from '@/types/mongoSchema';
+import { ModelSchema } from '../types/mongoSchema';
+
+export enum ChatModelNameEnum {
   GPT35 = 'gpt-3.5-turbo',
   GPT3 = 'text-davinci-003'
 }
-export const OpenAiList = [
+
+export type ModelConstantsData = {
+  serviceCompany: `${ServiceName}`;
+  name: string;
+  model: `${ChatModelNameEnum}`;
+  trainName: string; // 空字符串代表不能训练
+  maxToken: number;
+  maxTemperature: number;
+  price: number; // 多少钱 / 1token，单位: 0.00001元
+};
+
+export const modelList: ModelConstantsData[] = [
   {
+    serviceCompany: 'openai',
     name: 'chatGPT',
-    model: OpenAiModelEnum.GPT35,
-    trainName: 'turbo',
-    canTraining: false,
-    maxToken: 4060
+    model: ChatModelNameEnum.GPT35,
+    trainName: '',
+    maxToken: 4000,
+    maxTemperature: 2,
+    price: 3
   },
   {
+    serviceCompany: 'openai',
     name: 'GPT3',
-    model: OpenAiModelEnum.GPT3,
+    model: ChatModelNameEnum.GPT3,
     trainName: 'davinci',
-    canTraining: true,
-    maxToken: 4060
+    maxToken: 4000,
+    maxTemperature: 2,
+    price: 30
   }
 ];
 
@@ -51,3 +69,29 @@ export const formatModelStatus = {
     text: '已关闭'
   }
 };
+
+export const defaultModel: ModelSchema = {
+  _id: '',
+  userId: '',
+  name: '',
+  avatar: '',
+  status: ModelStatusEnum.pending,
+  updateTime: Date.now(),
+  trainingTimes: 0,
+  systemPrompt: '',
+  intro: '',
+  temperature: 5,
+  service: {
+    company: 'openai',
+    trainId: '',
+    chatModel: ChatModelNameEnum.GPT35,
+    modelName: ChatModelNameEnum.GPT35
+  },
+  security: {
+    domain: ['*'],
+    contextMaxLen: 1,
+    contentMaxLen: 1,
+    expiredTime: 9999,
+    maxLoadAmount: 1
+  }
+};

src/constants/theme.ts

@@ -45,8 +45,13 @@ const Button = defineStyleConfig({
     }
   },
   variants: {
-    outline: {
-      borderWidth: '1.5px'
+    white: {
+      color: '#fff',
+      backgroundColor: 'transparent',
+      border: '1px solid #ffffff',
+      _hover: {
+        backgroundColor: 'rgba(255,255,255,0.1)'
+      }
     }
   },
   defaultProps: {
@@ -85,6 +90,13 @@ export const theme = extendTheme({
   fonts: {
     body: '-apple-system,BlinkMacSystemFont,"Segoe UI",Helvetica,Arial,sans-serif,"Apple Color Emoji","Segoe UI Emoji","Segoe UI Symbol"'
   },
+  breakpoints: {
+    sm: '900px',
+    md: '1200px',
+    lg: '1500px',
+    xl: '1800',
+    '2xl': '2100'
+  },
   components: {
     Modal: ModalTheme,
     Button

src/constants/user.ts

@@ -1,5 +1,16 @@
+export enum BillTypeEnum {
+  chat = 'chat',
+  splitData = 'splitData',
+  return = 'return'
+}
 export enum PageTypeEnum {
   login = 'login',
   register = 'register',
   forgetPassword = 'forgetPassword'
 }
+
+export const BillTypeMap: Record<`${BillTypeEnum}`, string> = {
+  [BillTypeEnum.chat]: '对话',
+  [BillTypeEnum.splitData]: '文本拆分',
+  [BillTypeEnum.return]: '退款'
+};

src/hooks/useConfirm.tsx

@@ -19,9 +19,10 @@ export const useConfirm = ({ title = '提示', content }: { title?: string; cont
   return {
     openConfirm: useCallback(
       (confirm?: any, cancel?: any) => {
-        onOpen();
         confirmCb.current = confirm;
         cancelCb.current = cancel;
+
+        return onOpen;
       },
       [onOpen]
     ),

src/hooks/usePaging.ts

@@ -0,0 +1,77 @@
+import { useState, useCallback } from 'react';
+import type { PagingData } from '../types/index';
+import { useQuery } from '@tanstack/react-query';
+import { useToast } from './useToast';
+
+export const usePaging = <T = any>({
+  api,
+  pageSize = 10,
+  params = {}
+}: {
+  api: (data: any) => Promise<PagingData<T>>;
+  pageSize?: number;
+  params?: Record<string, any>;
+}) => {
+  const { toast } = useToast();
+  const [data, setData] = useState<T[]>([]);
+  const [pageNum, setPageNum] = useState(1);
+  const [total, setTotal] = useState(0);
+  const [isLoadAll, setIsLoadAll] = useState(false);
+  const [requesting, setRequesting] = useState(false);
+  const [initRequesting, setInitRequesting] = useState(false);
+
+  const getData = useCallback(
+    async (num: number, init = false) => {
+      if (requesting) return;
+      if (!init && isLoadAll) return;
+      if (init) {
+        setInitRequesting(true);
+      }
+      setRequesting(true);
+
+      try {
+        const res = await api({
+          pageNum: num,
+          pageSize,
+          ...params
+        });
+        setData((state) => {
+          const data = init ? res.data : state.concat(res.data);
+          if (data.length >= res.total) {
+            setIsLoadAll(true);
+          }
+          setTotal(res.total);
+          setPageNum(num);
+          return data;
+        });
+      } catch (error: any) {
+        toast({
+          title: error?.message || '获取数据异常',
+          status: 'error'
+        });
+        console.log(error);
+      }
+
+      setRequesting(false);
+      setInitRequesting(false);
+      return null;
+    },
+    [api, isLoadAll, pageSize, params, requesting, toast]
+  );
+
+  const nextPage = useCallback(() => getData(pageNum + 1), [getData, pageNum]);
+
+  useQuery(['init'], () => getData(1, true));
+
+  return {
+    pageNum,
+    pageSize,
+    total,
+    data,
+    getData,
+    requesting,
+    isLoadAll,
+    nextPage,
+    initRequesting
+  };
+};

src/hooks/useRequest.tsx

@@ -0,0 +1,33 @@
+import { useToast } from '@/hooks/useToast';
+import { useMutation } from '@tanstack/react-query';
+import type { UseMutationOptions } from '@tanstack/react-query';
+
+interface Props extends UseMutationOptions<any, any, any, any> {
+  successToast?: string;
+  errorToast?: string;
+}
+
+export const useRequest = ({ successToast, errorToast, onSuccess, onError, ...props }: Props) => {
+  const { toast } = useToast();
+  const mutation = useMutation<unknown, unknown, any, unknown>({
+    ...props,
+    onSuccess(res, variables: void, context: unknown) {
+      onSuccess?.(res, variables, context);
+      successToast &&
+        toast({
+          title: successToast,
+          status: 'success'
+        });
+    },
+    onError(err: any, variables: void, context: unknown) {
+      onError?.(err, variables, context);
+      errorToast &&
+        toast({
+          title: typeof err === 'string' ? err : err?.message || errorToast,
+          status: 'error'
+        });
+    }
+  });
+
+  return mutation;
+};

src/hooks/useSelectFile.tsx

@@ -0,0 +1,34 @@
+import React, { useRef, useCallback } from 'react';
+import { Box } from '@chakra-ui/react';
+
+export const useSelectFile = (props?: { fileType?: string; multiple?: boolean }) => {
+  const { fileType = '*', multiple = false } = props || {};
+  const SelectFileDom = useRef<HTMLInputElement>(null);
+
+  const File = useCallback(
+    ({ onSelect }: { onSelect: (e: File[]) => void }) => (
+      <Box position={'absolute'} w={0} h={0} overflow={'hidden'}>
+        <input
+          ref={SelectFileDom}
+          type="file"
+          accept={fileType}
+          multiple={multiple}
+          onChange={(e) => {
+            if (!e.target.files || e.target.files?.length === 0) return;
+            onSelect(Array.from(e.target.files));
+          }}
+        />
+      </Box>
+    ),
+    [fileType, multiple]
+  );
+
+  const onOpen = useCallback(() => {
+    SelectFileDom.current && SelectFileDom.current.click();
+  }, []);
+
+  return {
+    File,
+    onOpen
+  };
+};

src/hooks/useTabs.tsx

@@ -0,0 +1,18 @@
+import React, { useState, useCallback, useRef } from 'react';
+
+export const useTabs = ({
+  tabs = []
+}: {
+  tabs: {
+    id: string;
+    label: string;
+  }[];
+}) => {
+  const [activeTab, setActiveTab] = useState(tabs[0].id);
+
+  return {
+    tabs,
+    activeTab,
+    setActiveTab
+  };
+};

src/pages/_app.tsx

@@ -1,7 +1,7 @@
 import type { AppProps, NextWebVitalsMetric } from 'next/app';
 import Script from 'next/script';
 import Head from 'next/head';
-import { ChakraProvider } from '@chakra-ui/react';
+import { ChakraProvider, ColorModeScript } from '@chakra-ui/react';
 import Layout from '@/components/Layout';
 import { theme } from '@/constants/theme';
 import { QueryClient, QueryClientProvider } from '@tanstack/react-query';
@@ -38,9 +38,12 @@ export default function App({ Component, pageProps }: AppProps) {
         />
         <link rel="icon" href="/favicon.ico" />
       </Head>
-      <Script src="/iconfont.js" strategy="afterInteractive"></Script>
+      <Script src="/js/iconfont.js" strategy="afterInteractive"></Script>
+      <Script src="/js/qrcode.min.js" strategy="afterInteractive"></Script>
+      <Script src="/js/pdf.js" strategy="afterInteractive"></Script>
       <QueryClientProvider client={queryClient}>
         <ChakraProvider theme={theme}>
+          <ColorModeScript initialColorMode={theme.config.initialColorMode} />
           <Layout>
             <Component {...pageProps} />
           </Layout>

src/pages/api/chat/chatGpt.ts

@@ -1,43 +1,55 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { createParser, ParsedEvent, ReconnectInterval } from 'eventsource-parser';
-import { connectToDatabase, ChatWindow } from '@/service/mongo';
-import type { ModelType } from '@/types/model';
+import { connectToDatabase } from '@/service/mongo';
 import { getOpenAIApi, authChat } from '@/service/utils/chat';
 import { httpsAgent } from '@/service/utils/tools';
 import { ChatCompletionRequestMessage, ChatCompletionRequestMessageRoleEnum } from 'openai';
 import { ChatItemType } from '@/types/chat';
 import { jsonRes } from '@/service/response';
+import type { ModelSchema } from '@/types/mongoSchema';
 import { PassThrough } from 'stream';
+import { modelList } from '@/constants/model';
+import { pushChatBill } from '@/service/events/pushBill';
 
 /* 发送提示词 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
-  const { chatId, windowId, prompt } = req.body as {
-    prompt: ChatItemType;
-    windowId: string;
-    chatId: string;
-  };
+  let step = 0; // step=1时，表示开始了流响应
+  const stream = new PassThrough();
+  stream.on('error', () => {
+    console.log('error: ', 'stream error');
+    stream.destroy();
+  });
+  res.on('close', () => {
+    stream.destroy();
+  });
+  res.on('error', () => {
+    console.log('error: ', 'request error');
+    stream.destroy();
+  });
 
   try {
-    if (!windowId || !chatId || !prompt) {
+    const { chatId, prompt } = req.body as {
+      prompt: ChatItemType;
+      chatId: string;
+    };
+    const { authorization } = req.headers;
+    if (!chatId || !prompt) {
       throw new Error('缺少参数');
     }
 
     await connectToDatabase();
 
-    const { chat, userApiKey } = await authChat(chatId);
+    const { chat, userApiKey, systemKey, userId } = await authChat(chatId, authorization);
 
-    const model: ModelType = chat.modelId;
+    const model: ModelSchema = chat.modelId;
 
     // 读取对话内容
-    const prompts: ChatItemType[] = (await ChatWindow.findById(windowId)).content;
-    prompts.push(prompt);
+    const prompts = [...chat.content, prompt];
 
     // 上下文长度过滤
     const maxContext = model.security.contextMaxLen;
     const filterPrompts =
-      prompts.length > maxContext + 2
-        ? [prompts[0], ...prompts.slice(prompts.length - maxContext)]
-        : prompts.slice(0, prompts.length);
+      prompts.length > maxContext ? prompts.slice(prompts.length - maxContext) : prompts;
 
     // 格式化文本内容
     const map = {
@@ -51,42 +63,55 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
         content: item.value
       })
     );
-    // 第一句话，强调代码类型
-    formatPrompts.unshift({
-      role: ChatCompletionRequestMessageRoleEnum.System,
-      content: '如果你想返回代码，请务必声明代码的类型！并且在代码块前加一个换行符。'
-    });
+
+    // 如果有系统提示词，自动插入
+    if (model.systemPrompt) {
+      formatPrompts.unshift({
+        role: 'system',
+        content: model.systemPrompt
+      });
+    }
+
+    // 计算温度
+    const modelConstantsData = modelList.find((item) => item.model === model.service.modelName);
+    if (!modelConstantsData) {
+      throw new Error('模型异常');
+    }
+    const temperature = modelConstantsData.maxTemperature * (model.temperature / 10);
 
     // 获取 chatAPI
-    const chatAPI = getOpenAIApi(userApiKey);
+    const chatAPI = getOpenAIApi(userApiKey || systemKey);
+    let startTime = Date.now();
     // 发出请求
     const chatResponse = await chatAPI.createChatCompletion(
       {
         model: model.service.chatModel,
-        temperature: 1,
-        // max_tokens: model.security.contentMaxLen,
+        temperature: temperature,
+        // max_tokens: modelConstantsData.maxToken,
         messages: formatPrompts,
-        stream: true
+        frequency_penalty: 0.5, // 越大，重复内容越少
+        presence_penalty: -0.5, // 越大，越容易出现新内容
+        stream: true,
+        stop: ['。！？.!.']
       },
       {
-        timeout: 20000,
+        timeout: 40000,
         responseType: 'stream',
         httpsAgent
       }
     );
-    console.log(
-      formatPrompts.reduce((sum, item) => sum + item.content.length, 0),
-      'response success'
-    );
+
+    console.log('api response time:', `${(Date.now() - startTime) / 1000}s`);
 
     // 创建响应流
     res.setHeader('Content-Type', 'text/event-stream;charset-utf-8');
     res.setHeader('Access-Control-Allow-Origin', '*');
     res.setHeader('X-Accel-Buffering', 'no');
     res.setHeader('Cache-Control', 'no-cache, no-transform');
+    step = 1;
 
-    const pass = new PassThrough();
-    pass.pipe(res);
+    let responseContent = '';
+    stream.pipe(res);
 
     const onParse = async (event: ParsedEvent | ReconnectInterval) => {
       if (event.type !== 'event') return;
@@ -95,24 +120,53 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       try {
         const json = JSON.parse(data);
         const content: string = json?.choices?.[0].delta.content || '';
-        if (!content) return;
+        if (!content || (responseContent === '' && content === '\n')) return;
+
+        responseContent += content;
         // console.log('content:', content)
-        pass.push(content.replace(/\n/g, '<br/>'));
+        !stream.destroyed && stream.push(content.replace(/\n/g, '<br/>'));
       } catch (error) {
         error;
       }
     };
 
-    for await (const chunk of chatResponse.data as any) {
-      const parser = createParser(onParse);
-      parser.feed(decodeURIComponent(chunk));
+    const decoder = new TextDecoder();
+    try {
+      for await (const chunk of chatResponse.data as any) {
+        if (stream.destroyed) {
+          // 流被中断了，直接忽略后面的内容
+          break;
+        }
+        const parser = createParser(onParse);
+        parser.feed(decoder.decode(chunk));
+      }
+    } catch (error) {
+      console.log('pipe error', error);
     }
-    pass.push(null);
+    // close stream
+    !stream.destroyed && stream.push(null);
+    stream.destroy();
+
+    const promptsContent = formatPrompts.map((item) => item.content).join('');
+    // 只有使用平台的 key 才计费
+    !userApiKey &&
+      pushChatBill({
+        modelName: model.service.modelName,
+        userId,
+        chatId,
+        text: promptsContent + responseContent
+      });
   } catch (err: any) {
-    res.status(500);
-    jsonRes(res, {
-      code: 500,
-      error: err
-    });
+    if (step === 1) {
+      // 直接结束流
+      console.log('error，结束');
+      stream.destroy();
+    } else {
+      res.status(500);
+      jsonRes(res, {
+        code: 500,
+        error: err
+      });
+    }
   }
 }

src/pages/api/chat/delLastMessage.ts

@@ -1,19 +1,19 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { connectToDatabase, ChatWindow } from '@/service/mongo';
+import { connectToDatabase, Chat } from '@/service/mongo';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    const { windowId } = req.query as { windowId: string };
+    const { chatId } = req.query as { chatId: string };
 
-    if (!windowId) {
+    if (!chatId) {
       throw new Error('缺少参数');
     }
 
     await connectToDatabase();
 
     // 删除最一条数据库记录, 也就是预发送的那一条
-    await ChatWindow.findByIdAndUpdate(windowId, {
+    await Chat.findByIdAndUpdate(chatId, {
       $pop: { content: 1 },
       updateTime: Date.now()
     });

src/pages/api/chat/generate.ts

@@ -2,12 +2,15 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, Model, Chat } from '@/service/mongo';
 import { authToken } from '@/service/utils/tools';
-import { ModelType } from '@/types/model';
+import type { ModelSchema } from '@/types/mongoSchema';
 
 /* 获取我的模型 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
-    const { modelId } = req.query;
+    const { modelId, isShare = 'false' } = req.query as {
+      modelId: string;
+      isShare?: 'true' | 'false';
+    };
     const { authorization } = req.headers;
 
     if (!authorization) {
@@ -24,7 +27,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     await connectToDatabase();
 
     // 获取模型配置
-    const model: ModelType | null = await Model.findOne({
+    const model = await Model.findOne<ModelSchema>({
       _id: modelId,
       userId
     });
@@ -38,11 +41,14 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       userId,
       modelId,
       expiredTime: Date.now() + model.security.expiredTime,
-      loadAmount: model.security.maxLoadAmount
+      loadAmount: model.security.maxLoadAmount,
+      updateTime: Date.now(),
+      isShare: isShare === 'true',
+      content: []
     });
 
     jsonRes(res, {
-      data: response._id
+      data: response._id // 即聊天框的 ID
     });
   } catch (err) {
     jsonRes(res, {

src/pages/api/chat/gpt3.ts

@@ -1,85 +1,165 @@
-// Next.js API route support: https://nextjs.org/docs/api-routes/introduction
 import type { NextApiRequest, NextApiResponse } from 'next';
-import { jsonRes } from '@/service/response';
-import { connectToDatabase, Chat } from '@/service/mongo';
-import type { ModelType } from '@/types/model';
-import { getOpenAIApi } from '@/service/utils/chat';
-import { ChatItemType } from '@/types/chat';
+import { createParser, ParsedEvent, ReconnectInterval } from 'eventsource-parser';
+import { connectToDatabase } from '@/service/mongo';
+import { getOpenAIApi, authChat } from '@/service/utils/chat';
 import { httpsAgent } from '@/service/utils/tools';
+import { ChatItemType } from '@/types/chat';
+import { jsonRes } from '@/service/response';
+import type { ModelSchema } from '@/types/mongoSchema';
+import { PassThrough } from 'stream';
+import { modelList } from '@/constants/model';
+import { pushChatBill } from '@/service/events/pushBill';
 
 /* 发送提示词 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
-  try {
-    const { prompt, chatId } = req.body as { prompt: ChatItemType[]; chatId: string };
+  let step = 0; // step=1时，表示开始了流响应
+  const stream = new PassThrough();
+  stream.on('error', () => {
+    console.log('error: ', 'stream error');
+    stream.destroy();
+  });
+  res.on('close', () => {
+    stream.destroy();
+  });
+  res.on('error', () => {
+    console.log('error: ', 'request error');
+    stream.destroy();
+  });
 
-    if (!prompt || !chatId) {
+  try {
+    const { chatId, prompt } = req.body as {
+      prompt: ChatItemType;
+      chatId: string;
+    };
+    const { authorization } = req.headers;
+    if (!chatId || !prompt) {
       throw new Error('缺少参数');
     }
 
     await connectToDatabase();
 
-    // 获取 chat 数据
-    const chat = await Chat.findById(chatId)
-      .populate({
-        path: 'modelId',
-        options: {
-          strictPopulate: false
-        }
-      })
-      .populate({
-        path: 'userId',
-        options: {
-          strictPopulate: false
-        }
-      });
+    const { chat, userApiKey, systemKey, userId } = await authChat(chatId, authorization);
 
-    if (!chat || !chat.modelId || !chat.userId) {
-      throw new Error('聊天已过期');
+    const model: ModelSchema = chat.modelId;
+
+    // 读取对话内容
+    const prompts = [...chat.content, prompt];
+
+    // 上下文长度过滤
+    const maxContext = model.security.contextMaxLen;
+    const filterPrompts =
+      prompts.length > maxContext ? prompts.slice(prompts.length - maxContext) : prompts;
+
+    // 格式化文本内容
+    const map = {
+      Human: 'Human',
+      AI: 'AI',
+      SYSTEM: 'SYSTEM'
+    };
+    const formatPrompts: string[] = filterPrompts.map((item: ChatItemType) => item.value);
+    // 如果有系统提示词，自动插入
+    if (model.systemPrompt) {
+      formatPrompts.unshift(`${model.systemPrompt}`);
     }
 
-    const model: ModelType = chat.modelId;
+    const promptText = formatPrompts.join('</s>');
 
-    // 获取 user 的 apiKey
-    const user = chat.userId;
-
-    const userApiKey = user.accounts?.find((item: any) => item.type === 'openai')?.value;
-
-    if (!userApiKey) {
-      throw new Error('缺少ApiKey, 无法请求');
+    // 计算温度
+    const modelConstantsData = modelList.find((item) => item.model === model.service.modelName);
+    if (!modelConstantsData) {
+      throw new Error('模型异常');
     }
+    const temperature = modelConstantsData.maxTemperature * (model.temperature / 10);
 
     // 获取 chatAPI
-    const chatAPI = getOpenAIApi(userApiKey);
+    const chatAPI = getOpenAIApi(userApiKey || systemKey);
+    let startTime = Date.now();
 
-    // prompt处理
-    const formatPrompt = prompt.map((item) => `${item.value}\n\n###\n\n`).join('');
-
-    // 发送请求
-    const response = await chatAPI.createCompletion(
+    // 发出请求
+    const chatResponse = await chatAPI.createCompletion(
       {
-        model: model.service.modelName,
-        prompt: formatPrompt,
-        temperature: 0.5,
-        max_tokens: model.security.contentMaxLen,
-        top_p: 1,
-        frequency_penalty: 0,
-        presence_penalty: 0.6,
-        stop: ['###']
+        model: model.service.chatModel,
+        temperature: temperature,
+        prompt: promptText,
+        stream: true,
+        max_tokens: modelConstantsData.maxToken,
+        presence_penalty: 0, // 越大，越容易出现新内容
+        frequency_penalty: 0, // 越大，重复内容越少
+        stop: ['。！？.!.', `</s>`]
       },
       {
+        timeout: 40000,
+        responseType: 'stream',
         httpsAgent
       }
     );
 
-    const responseMessage = response.data.choices[0]?.text;
+    console.log('api response time:', `${(Date.now() - startTime) / 1000}s`);
 
-    jsonRes(res, {
-      data: responseMessage
-    });
+    // 创建响应流
+    res.setHeader('Content-Type', 'text/event-stream;charset-utf-8');
+    res.setHeader('Access-Control-Allow-Origin', '*');
+    res.setHeader('X-Accel-Buffering', 'no');
+    res.setHeader('Cache-Control', 'no-cache, no-transform');
+    step = 1;
+
+    let responseContent = '';
+    stream.pipe(res);
+
+    const onParse = async (event: ParsedEvent | ReconnectInterval) => {
+      if (event.type !== 'event') return;
+      const data = event.data;
+      if (data === '[DONE]') return;
+      try {
+        const json = JSON.parse(data);
+        const content: string = json?.choices?.[0].text || '';
+        if (!content || (responseContent === '' && content === '\n')) return;
+
+        responseContent += content;
+        // console.log('content:', content);
+        !stream.destroyed && stream.push(content.replace(/\n/g, '<br/>'));
+      } catch (error) {
+        error;
+      }
+    };
+
+    const decoder = new TextDecoder();
+    try {
+      for await (const chunk of chatResponse.data as any) {
+        if (stream.destroyed) {
+          // 流被中断了，直接忽略后面的内容
+          break;
+        }
+        const parser = createParser(onParse);
+        parser.feed(decoder.decode(chunk));
+      }
+    } catch (error) {
+      console.log('pipe error', error);
+    }
+    // close stream
+    !stream.destroyed && stream.push(null);
+    stream.destroy();
+
+    // 只有使用平台的 key 才计费
+    !userApiKey &&
+      pushChatBill({
+        modelName: model.service.modelName,
+        userId,
+        chatId,
+        text: promptText + responseContent
+      });
   } catch (err: any) {
-    jsonRes(res, {
-      code: 500,
-      error: err
-    });
+    // console.log(err?.response);
+    if (step === 1) {
+      // 直接结束流
+      console.log('error，结束');
+      stream.destroy();
+    } else {
+      res.status(500);
+      jsonRes(res, {
+        code: 500,
+        error: err
+      });
+    }
   }
 }

src/pages/api/chat/init.ts

@@ -1,12 +1,13 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { connectToDatabase, Chat, ChatWindow } from '@/service/mongo';
-import type { ModelType } from '@/types/model';
+import { connectToDatabase, Chat } from '@/service/mongo';
+import type { ChatPopulate } from '@/types/mongoSchema';
+import type { InitChatResponse } from '@/api/response/chat';
 
 /* 获取我的模型 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    const { chatId, windowId } = req.query as { chatId: string; windowId?: string };
+    const { chatId } = req.query as { chatId: string };
 
     if (!chatId) {
       throw new Error('缺少参数');
@@ -15,16 +16,15 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     await connectToDatabase();
 
     // 获取 chat 数据
-    const chat = await Chat.findById(chatId).populate({
+    const chat = await Chat.findById<ChatPopulate>(chatId).populate({
       path: 'modelId',
       options: {
         strictPopulate: false
       }
     });
 
-    // 安全校验
-    if (!chat || chat.loadAmount === 0 || chat.expiredTime < Date.now()) {
-      throw new Error('聊天框已过期');
+    if (!chat) {
+      throw new Error('聊天框不存在');
     }
 
     if (chat.loadAmount > 0) {
@@ -38,47 +38,19 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       );
     }
 
-    const model: ModelType = chat.modelId;
-
-    /* 查找是否有记录 */
-    let history = null;
-    let responseId = windowId;
-    try {
-      history = await ChatWindow.findById(windowId);
-    } catch (error) {
-      error;
-    }
-
-    const defaultContent = model.systemPrompt
-      ? [
-          {
-            obj: 'SYSTEM',
-            value: model.systemPrompt
-          }
-        ]
-      : [];
-
-    if (!history) {
-      // 没有记录，创建一个
-      const response = await ChatWindow.create({
-        chatId,
-        updateTime: Date.now(),
-        content: defaultContent
-      });
-      responseId = response._id;
-    }
-
-    jsonRes(res, {
+    const model = chat.modelId;
+    jsonRes<InitChatResponse>(res, {
+      code: 201,
       data: {
-        windowId: responseId,
-        chatSite: {
-          modelId: model._id,
-          name: model.name,
-          avatar: model.avatar,
-          secret: model.security,
-          chatModel: model.service.chatModel
-        },
-        history: history ? history.content : defaultContent
+        chatId: chat._id,
+        isExpiredTime: chat.loadAmount === 0 || chat.expiredTime <= Date.now(),
+        modelId: model._id,
+        name: model.name,
+        avatar: model.avatar,
+        intro: model.intro,
+        secret: model.security,
+        chatModel: model.service.chatModel,
+        history: chat.content
       }
     });
   } catch (err) {

src/pages/api/chat/saveChat.ts

@@ -1,24 +1,24 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { ChatItemType } from '@/types/chat';
-import { connectToDatabase, ChatWindow } from '@/service/mongo';
+import { connectToDatabase, Chat } from '@/service/mongo';
 
 /* 聊天内容存存储 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    const { windowId, prompts } = req.body as {
-      windowId: string;
+    const { chatId, prompts } = req.body as {
+      chatId: string;
       prompts: ChatItemType[];
     };
 
-    if (!windowId || !prompts) {
+    if (!chatId || !prompts) {
       throw new Error('缺少参数');
     }
 
     await connectToDatabase();
 
     // 存入库
-    await ChatWindow.findByIdAndUpdate(windowId, {
+    await Chat.findByIdAndUpdate(chatId, {
       $push: {
         content: {
           $each: prompts.map((item) => ({

src/pages/api/data/delData.ts

@@ -0,0 +1,47 @@
+// Next.js API route support: https://nextjs.org/docs/api-routes/introduction
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Data, DataItem } from '@/service/mongo';
+import { authToken } from '@/service/utils/tools';
+import type { DataListItem } from '@/types/data';
+import type { PagingData } from '@/types';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { authorization } = req.headers;
+
+    if (!authorization) {
+      throw new Error('缺少登录凭证');
+    }
+
+    await authToken(authorization);
+
+    const { dataId } = req.query as { dataId: string };
+    if (!dataId) {
+      throw new Error('缺少参数');
+    }
+
+    await connectToDatabase();
+
+    await Data.findByIdAndUpdate(dataId, {
+      isDeleted: true
+    });
+
+    // 改变 dataItem 状态为 0
+    await DataItem.updateMany(
+      {
+        dataId
+      },
+      {
+        status: 0
+      }
+    );
+
+    jsonRes<PagingData<DataListItem>>(res);
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

src/pages/api/data/getDataItems.ts

@@ -0,0 +1,48 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, DataItem } from '@/service/mongo';
+import { authToken } from '@/service/utils/tools';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    let {
+      dataId,
+      pageNum = 1,
+      pageSize = 10
+    } = req.query as { dataId: string; pageNum: string; pageSize: string };
+    pageNum = +pageNum;
+    pageSize = +pageSize;
+
+    if (!dataId) {
+      throw new Error('参数错误');
+    }
+    await connectToDatabase();
+
+    const { authorization } = req.headers;
+
+    await authToken(authorization);
+
+    const dataItems = await DataItem.find({
+      dataId
+    })
+      .sort({ _id: -1 }) // 按照创建时间倒序排列
+      .skip((pageNum - 1) * pageSize)
+      .limit(pageSize);
+
+    jsonRes(res, {
+      data: {
+        pageNum,
+        pageSize,
+        data: dataItems,
+        total: await DataItem.countDocuments({
+          dataId
+        })
+      }
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

src/pages/api/data/getDataList.ts

@@ -0,0 +1,88 @@
+// Next.js API route support: https://nextjs.org/docs/api-routes/introduction
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Data, DataItem } from '@/service/mongo';
+import { authToken } from '@/service/utils/tools';
+import type { DataSchema } from '@/types/mongoSchema';
+import type { DataListItem } from '@/types/data';
+import type { PagingData } from '@/types';
+import mongoose from 'mongoose';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { authorization } = req.headers;
+    let { pageNum = 1, pageSize = 10 } = req.query as { pageNum: string; pageSize: string };
+
+    pageNum = +pageNum;
+    pageSize = +pageSize;
+
+    if (!authorization) {
+      throw new Error('缺少登录凭证');
+    }
+
+    const userId = await authToken(authorization);
+
+    await connectToDatabase();
+
+    const datalist = await Data.aggregate<DataListItem>([
+      {
+        $match: {
+          userId: new mongoose.Types.ObjectId(userId),
+          isDeleted: false
+        }
+      },
+      {
+        $sort: { createTime: -1 } // 按照创建时间倒序排列
+      },
+      {
+        $skip: (pageNum - 1) * pageSize // 跳过前面的数据
+      },
+      {
+        $limit: pageSize // 取出指定数量的数据
+      },
+      {
+        $lookup: {
+          from: 'dataitems',
+          localField: '_id',
+          foreignField: 'dataId',
+          as: 'items'
+        }
+      },
+      {
+        $addFields: {
+          totalData: {
+            $size: '$items' // 统计dataItem的总数
+          },
+          trainingData: {
+            $size: {
+              $filter: {
+                input: '$items',
+                as: 'item',
+                cond: { $ne: ['$$item.status', 0] } // 统计 status 不为0的数量
+              }
+            }
+          }
+        }
+      },
+      {
+        $project: {
+          items: 0 // 不返回 items 字段
+        }
+      }
+    ]);
+
+    jsonRes<PagingData<DataListItem>>(res, {
+      data: {
+        pageNum,
+        pageSize,
+        data: datalist,
+        total: 1
+      }
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

src/pages/api/data/postData.ts

@@ -0,0 +1,33 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Data } from '@/service/mongo';
+import { authToken } from '@/service/utils/tools';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    let { name } = req.query as { name: string };
+    if (!name) {
+      throw new Error('参数错误');
+    }
+    await connectToDatabase();
+
+    const { authorization } = req.headers;
+
+    const userId = await authToken(authorization);
+
+    // 生成 data 集合
+    const data = await Data.create({
+      userId,
+      name
+    });
+
+    jsonRes(res, {
+      data: data._id
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

src/pages/api/data/putDataName.ts

@@ -0,0 +1,37 @@
+// Next.js API route support: https://nextjs.org/docs/api-routes/introduction
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Data } from '@/service/mongo';
+import { authToken } from '@/service/utils/tools';
+import type { DataListItem } from '@/types/data';
+import type { PagingData } from '@/types';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { authorization } = req.headers;
+
+    if (!authorization) {
+      throw new Error('缺少登录凭证');
+    }
+
+    await authToken(authorization);
+
+    const { dataId, name } = req.query as { dataId: string; name: string };
+    if (!dataId || !name) {
+      throw new Error('缺少参数');
+    }
+
+    await connectToDatabase();
+
+    await Data.findByIdAndUpdate(dataId, {
+      name
+    });
+
+    jsonRes<PagingData<DataListItem>>(res);
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

src/pages/api/data/splitData.ts

@@ -0,0 +1,47 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Data, DataItem } from '@/service/mongo';
+import { authToken } from '@/service/utils/tools';
+import { generateQA } from '@/service/events/generateQA';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    let { text, dataId } = req.body as { text: string; dataId: string };
+    if (!text || !dataId) {
+      throw new Error('参数错误');
+    }
+    text = text.replace(/\n+/g, '\n');
+    await connectToDatabase();
+
+    const { authorization } = req.headers;
+
+    const userId = await authToken(authorization);
+
+    const dataItems: any[] = [];
+
+    // 格式化文本长度
+    for (let i = 0; i <= text.length / 1000; i++) {
+      dataItems.push({
+        temperature: 0,
+        userId,
+        dataId,
+        text: text.slice(i * 1000, (i + 1) * 1000),
+        status: 1
+      });
+    }
+
+    // 批量插入数据
+    await DataItem.insertMany(dataItems);
+
+    generateQA();
+
+    jsonRes(res, {
+      data: dataItems.length
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

src/pages/api/model/create.ts

@@ -3,29 +3,32 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase } from '@/service/mongo';
 import { authToken } from '@/service/utils/tools';
-import { ModelStatusEnum, OpenAiList } from '@/constants/model';
+import { ModelStatusEnum, modelList, ChatModelNameEnum } from '@/constants/model';
 import { Model } from '@/service/models/model';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
-    const { name, serviceModelName, serviceModelCompany = 'openai' } = req.body;
+    const { name, serviceModelName } = req.body as {
+      name: string;
+      serviceModelName: `${ChatModelNameEnum}`;
+    };
     const { authorization } = req.headers;
 
     if (!authorization) {
       throw new Error('无权操作');
     }
 
-    if (!name || !serviceModelName || !serviceModelCompany) {
+    if (!name || !serviceModelName) {
       throw new Error('缺少参数');
     }
 
     // 凭证校验
     const userId = await authToken(authorization);
 
-    const modelItem = OpenAiList.find((item) => item.model === serviceModelName);
+    const modelItem = modelList.find((item) => item.model === serviceModelName);
 
     if (!modelItem) {
-      throw new Error('模型错误');
+      throw new Error('模型不存在');
     }
 
     await connectToDatabase();
@@ -43,8 +46,8 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     const authCount = await Model.countDocuments({
       userId
     });
-    if (authCount >= 5) {
-      throw new Error('上限5个模型');
+    if (authCount >= 20) {
+      throw new Error('上限 20 个模型');
     }
 
     // 创建模型
@@ -53,7 +56,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       userId,
       status: ModelStatusEnum.running,
       service: {
-        company: serviceModelCompany,
+        company: modelItem.serviceCompany,
         trainId: modelItem.trainName,
         chatModel: modelItem.model,
         modelName: modelItem.model

src/pages/api/model/detail.tsx

@@ -3,7 +3,7 @@ import { jsonRes } from '@/service/response';
 import { connectToDatabase } from '@/service/mongo';
 import { authToken } from '@/service/utils/tools';
 import { Model } from '@/service/models/model';
-import { ModelType } from '@/types/model';
+import type { ModelSchema } from '@/types/mongoSchema';
 
 /* 获取我的模型 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
@@ -26,7 +26,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     await connectToDatabase();
 
     // 根据 userId 获取模型信息
-    const model: ModelType | null = await Model.findOne({
+    const model = await Model.findOne<ModelSchema>({
       userId,
       _id: modelId
     });

src/pages/api/model/getTrainings.ts

@@ -6,7 +6,7 @@ import formidable from 'formidable';
 import { authToken, getUserOpenaiKey } from '@/service/utils/tools';
 import { join } from 'path';
 import fs from 'fs';
-import type { ModelType } from '@/types/model';
+import type { ModelSchema } from '@/types/mongoSchema';
 import type { OpenAIApi } from 'openai';
 import { ModelStatusEnum, TrainingStatusEnum } from '@/constants/model';
 import { httpsAgent } from '@/service/utils/tools';

src/pages/api/model/list.ts

@@ -21,6 +21,8 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     // 根据 userId 获取模型信息
     const models = await Model.find({
       userId
+    }).sort({
+      _id: -1
     });
 
     jsonRes(res, {

src/pages/api/model/putTrainStatus.ts

@@ -3,7 +3,7 @@ import { jsonRes } from '@/service/response';
 import { connectToDatabase, Model, Training } from '@/service/mongo';
 import { getOpenAIApi } from '@/service/utils/chat';
 import { authToken, getUserOpenaiKey } from '@/service/utils/tools';
-import type { ModelType } from '@/types/model';
+import type { ModelSchema } from '@/types/mongoSchema';
 import { TrainingItemType } from '@/types/training';
 import { ModelStatusEnum, TrainingStatusEnum } from '@/constants/model';
 import { OpenAiTuneStatusEnum } from '@/service/constants/training';
@@ -26,7 +26,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     await connectToDatabase();
 
     // 获取模型
-    const model: ModelType | null = await Model.findById(modelId);
+    const model = await Model.findById<ModelSchema>(modelId);
 
     if (!model || model.status !== 'training') {
       throw new Error('模型不在训练中');

src/pages/api/model/train.ts

@@ -7,7 +7,7 @@ import formidable from 'formidable';
 import { authToken, getUserOpenaiKey } from '@/service/utils/tools';
 import { join } from 'path';
 import fs from 'fs';
-import type { ModelType } from '@/types/model';
+import type { ModelSchema } from '@/types/mongoSchema';
 import type { OpenAIApi } from 'openai';
 import { ModelStatusEnum, TrainingStatusEnum } from '@/constants/model';
 import { httpsAgent } from '@/service/utils/tools';
@@ -38,7 +38,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     await connectToDatabase();
 
     // 获取模型的状态
-    const model: ModelType | null = await Model.findById(modelId);
+    const model = await Model.findById<ModelSchema>(modelId);
 
     if (!model || model.status !== 'running') {
       throw new Error('模型正忙');

src/pages/api/model/update.ts

@@ -8,7 +8,8 @@ import type { ModelUpdateParams } from '@/types/model';
 /* 获取我的模型 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
-    const { name, service, security, systemPrompt } = req.body as ModelUpdateParams;
+    const { name, service, security, systemPrompt, intro, temperature } =
+      req.body as ModelUpdateParams;
     const { modelId } = req.query as { modelId: string };
     const { authorization } = req.headers;
 
@@ -33,8 +34,10 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       },
       {
         name,
-        service,
         systemPrompt,
+        intro,
+        temperature,
+        service,
         security
       }
     );

src/pages/api/timer/clearAuthCode.ts

@@ -5,6 +5,9 @@ import { AuthCode } from '@/service/models/authCode';
 import { connectToDatabase } from '@/service/mongo';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  if (req.headers.auth !== 'archer') {
+    throw new Error('凭证错误');
+  }
   try {
     await connectToDatabase();
 

src/pages/api/timer/clearChatWindow.ts

@@ -1,13 +1,16 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { connectToDatabase, ChatWindow } from '@/service/mongo';
+import { connectToDatabase, Chat } from '@/service/mongo';
 
 /* 定时删除那些不活跃的内容 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  if (req.headers.auth !== 'archer') {
+    throw new Error('凭证错误');
+  }
   try {
     await connectToDatabase();
 
-    const response = await ChatWindow.deleteMany(
+    const response = await Chat.deleteMany(
       { $expr: { $lt: [{ $size: '$content' }, 5] } },
       // 使用 $pull 操作符删除数组中的元素
       { $pull: { content: { $exists: true } } }

src/pages/api/timer/initBill.ts

@@ -0,0 +1,35 @@
+// Next.js API route support: https://nextjs.org/docs/api-routes/introduction
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Bill } from '@/service/mongo';
+import { authToken } from '@/service/utils/tools';
+import type { BillSchema } from '@/types/mongoSchema';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    if (req.headers.auth !== 'archer') {
+      throw new Error('凭证错误');
+    }
+    await connectToDatabase();
+
+    const bills = await Bill.find({
+      tokenLen: { $exists: false }
+    });
+    await Promise.all(
+      bills.map((bill) =>
+        Bill.findByIdAndUpdate(bill._id, {
+          tokenLen: bill.textLen
+        })
+      )
+    );
+
+    jsonRes(res, {
+      data: {}
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

src/pages/api/timer/initDataItemTime.ts

@@ -0,0 +1,36 @@
+// Next.js API route support: https://nextjs.org/docs/api-routes/introduction
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, DataItem, Data } from '@/service/mongo';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    if (req.headers.auth !== 'archer') {
+      throw new Error('凭证错误');
+    }
+    await connectToDatabase();
+
+    // await DataItem.updateMany(
+    //   {},
+    //   {
+    //     times: 2
+    //   }
+    // );
+
+    await Data.updateMany(
+      {},
+      {
+        isDeleted: false
+      }
+    );
+
+    jsonRes(res, {
+      data: {}
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

src/pages/api/timer/updateTraining.ts

@@ -9,8 +9,12 @@ import { getUserOpenaiKey } from '@/service/utils/tools';
 import { OpenAiTuneStatusEnum } from '@/service/constants/training';
 import { sendTrainSucceed } from '@/service/utils/sendEmail';
 import { httpsAgent } from '@/service/utils/tools';
+import { ModelPopulate } from '@/types/mongoSchema';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  if (req.headers.auth !== 'archer') {
+    throw new Error('凭证错误');
+  }
   try {
     await connectToDatabase();
 
@@ -28,7 +32,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
           // 删除训练文件
           openai.deleteFile(data.training_files[0].id, { httpsAgent });
 
-          const model = await Model.findById(item.modelId).populate({
+          const model = await Model.findById<ModelPopulate>(item.modelId).populate({
             path: 'userId',
             options: {
               strictPopulate: false

src/pages/api/user/checkPayResult.ts

@@ -0,0 +1,82 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import axios from 'axios';
+import { connectToDatabase, User, Pay } from '@/service/mongo';
+import { authToken } from '@/service/utils/tools';
+import { PaySchema } from '@/types/mongoSchema';
+import dayjs from 'dayjs';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { authorization } = req.headers;
+    let { payId } = req.query as { payId: string };
+
+    const userId = await authToken(authorization);
+
+    await connectToDatabase();
+
+    // 查找订单记录校验
+    const payOrder = await Pay.findById<PaySchema>(payId);
+
+    if (!payOrder) {
+      throw new Error('订单不存在');
+    }
+    if (payOrder.status !== 'NOTPAY') {
+      throw new Error('订单已结算');
+    }
+
+    const { data } = await axios.get(
+      `https://sif268.laf.dev/wechat-order-query?order_number=${payOrder.orderId}&api_key=${process.env.WXPAYCODE}`
+    );
+
+    // 校验下是否超过一天
+    const orderTime = dayjs(payOrder.createTime);
+    const diffInHours = dayjs().diff(orderTime, 'hours');
+
+    if (data.trade_state === 'SUCCESS') {
+      // 订单已支付
+      try {
+        // 更新订单状态
+        const updateRes = await Pay.updateOne(
+          {
+            _id: payId,
+            status: 'NOTPAY'
+          },
+          {
+            status: 'SUCCESS'
+          }
+        );
+        if (updateRes.modifiedCount === 1) {
+          // 给用户账号充钱
+          await User.findByIdAndUpdate(userId, {
+            $inc: { balance: payOrder.price }
+          });
+          jsonRes(res, {
+            data: '支付成功'
+          });
+        }
+      } catch (error) {
+        await Pay.findByIdAndUpdate(payId, {
+          status: 'NOTPAY'
+        });
+        console.log(error);
+      }
+    } else if (data.trade_state === 'CLOSED' || diffInHours > 24) {
+      // 订单已关闭
+      await Pay.findByIdAndUpdate(payId, {
+        status: 'CLOSED'
+      });
+      jsonRes(res, {
+        data: '订单已过期'
+      });
+    } else {
+      throw new Error(data.trade_state_desc);
+    }
+  } catch (err) {
+    // console.log(err);
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

src/pages/api/user/getBill.ts

@@ -0,0 +1,51 @@
+// Next.js API route support: https://nextjs.org/docs/api-routes/introduction
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Bill } from '@/service/mongo';
+import { authToken } from '@/service/utils/tools';
+import type { BillSchema } from '@/types/mongoSchema';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { authorization } = req.headers;
+    let { pageNum = 1, pageSize = 10 } = req.query as { pageNum: string; pageSize: string };
+
+    pageNum = +pageNum;
+    pageSize = +pageSize;
+
+    if (!authorization) {
+      throw new Error('缺少登录凭证');
+    }
+
+    const userId = await authToken(authorization);
+
+    await connectToDatabase();
+
+    // 根据 id 获取用户账单
+    const bills = await Bill.find<BillSchema>({
+      userId
+    })
+      .sort({ _id: -1 }) // 按照创建时间倒序排列
+      .skip((pageNum - 1) * pageSize)
+      .limit(pageSize);
+
+    // 获取total
+    const total = await Bill.countDocuments({
+      userId
+    });
+
+    jsonRes(res, {
+      data: {
+        pageNum,
+        pageSize,
+        data: bills,
+        total
+      }
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

src/pages/api/user/getPayCode.ts

@@ -0,0 +1,56 @@
+// Next.js API route support: https://nextjs.org/docs/api-routes/introduction
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import axios from 'axios';
+import { authToken } from '@/service/utils/tools';
+import { customAlphabet } from 'nanoid';
+import { connectToDatabase, Pay } from '@/service/mongo';
+import { PRICE_SCALE } from '@/constants/common';
+
+const nanoid = customAlphabet('abcdefghijklmnopqrstuvwxyz1234567890', 20);
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { authorization } = req.headers;
+    let { amount = 0 } = req.query as { amount: string };
+    amount = +amount;
+
+    if (!authorization) {
+      throw new Error('缺少登录凭证');
+    }
+    const userId = await authToken(authorization);
+
+    const id = nanoid();
+    await connectToDatabase();
+
+    const response = await axios({
+      url: 'https://sif268.laf.dev/wechat-pay',
+      method: 'POST',
+      data: {
+        trade_order_number: id,
+        amount: amount * 100,
+        api_key: process.env.WXPAYCODE
+      }
+    });
+
+    // 充值记录 + 1
+    const payOrder = await Pay.create({
+      userId,
+      price: amount * PRICE_SCALE,
+      orderId: id
+    });
+
+    jsonRes(res, {
+      data: {
+        payId: payOrder._id,
+        codeUrl: response.data?.code_url
+      }
+    });
+  } catch (err) {
+    console.log(err);
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

src/pages/api/user/getPayOrders.ts

@@ -0,0 +1,31 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { authToken } from '@/service/utils/tools';
+import { connectToDatabase, Pay } from '@/service/mongo';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { authorization } = req.headers;
+
+    if (!authorization) {
+      throw new Error('缺少登录凭证');
+    }
+    const userId = await authToken(authorization);
+
+    await connectToDatabase();
+
+    const records = await Pay.find({
+      userId
+    }).sort({ createTime: -1 });
+
+    jsonRes(res, {
+      data: records
+    });
+  } catch (err) {
+    console.log(err);
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

src/pages/api/user/register.ts

@@ -46,6 +46,10 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     // 根据 id 获取用户信息
     const user = await User.findById(response._id);
 
+    if (!user) {
+      throw new Error('获取用户信息异常');
+    }
+
     jsonRes(res, {
       data: {
         token: generateToken(user._id),

src/pages/api/user/updatePasswordByCode.ts

@@ -44,6 +44,10 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       email
     });
 
+    if (!user) {
+      throw new Error('获取用户信息异常');
+    }
+
     jsonRes(res, {
       data: {
         token: generateToken(user._id),

src/pages/chat/components/Empty.tsx

@@ -0,0 +1,40 @@
+import React from 'react';
+import { Card, Box, Mark } from '@chakra-ui/react';
+import { versionIntro, chatProblem } from '@/constants/common';
+import Markdown from '@/components/Markdown';
+
+const Empty = ({ intro }: { intro: string }) => {
+  const Header = ({ children }: { children: string }) => (
+    <Box fontSize={'lg'} fontWeight={'bold'} textAlign={'center'} pb={2}>
+      {children}
+    </Box>
+  );
+  return (
+    <Box
+      minH={'100%'}
+      w={'85%'}
+      maxW={'600px'}
+      m={'auto'}
+      py={'5vh'}
+      alignItems={'center'}
+      justifyContent={'center'}
+    >
+      {!!intro && (
+        <Card p={4} mb={10}>
+          <Header>模型介绍</Header>
+          <Box whiteSpace={'pre-line'}>{intro}</Box>
+        </Card>
+      )}
+      <Card p={4} mb={10}>
+        <Header>常见问题</Header>
+        <Markdown source={chatProblem} />
+      </Card>
+      {/* version intro */}
+      <Card p={4}>
+        <Markdown source={versionIntro} />
+      </Card>
+    </Box>
+  );
+};
+
+export default Empty;

src/pages/chat/components/SlideBar.tsx

@@ -0,0 +1,313 @@
+import React, { useState, useEffect } from 'react';
+import { AddIcon, ChatIcon, DeleteIcon, MoonIcon, SunIcon } from '@chakra-ui/icons';
+import {
+  Box,
+  Button,
+  Accordion,
+  AccordionItem,
+  AccordionButton,
+  AccordionPanel,
+  AccordionIcon,
+  Flex,
+  Divider,
+  IconButton,
+  Modal,
+  ModalOverlay,
+  ModalContent,
+  ModalHeader,
+  ModalFooter,
+  ModalBody,
+  ModalCloseButton,
+  useDisclosure,
+  useColorMode,
+  useColorModeValue
+} from '@chakra-ui/react';
+import { useUserStore } from '@/store/user';
+import { useChatStore } from '@/store/chat';
+import { useQuery } from '@tanstack/react-query';
+import { useRouter } from 'next/router';
+import { getToken } from '@/utils/user';
+import MyIcon from '@/components/Icon';
+import { useCopyData } from '@/utils/tools';
+import Markdown from '@/components/Markdown';
+import { shareHint } from '@/constants/common';
+import { getChatSiteId } from '@/api/chat';
+import WxConcat from '@/components/WxConcat';
+
+const SlideBar = ({
+  name,
+  chatId,
+  modelId,
+  resetChat,
+  onClose
+}: {
+  name?: string;
+  chatId: string;
+  modelId: string;
+  resetChat: () => void;
+  onClose: () => void;
+}) => {
+  const router = useRouter();
+  const { colorMode, toggleColorMode } = useColorMode();
+  const { copyData } = useCopyData();
+  const { myModels, getMyModels } = useUserStore();
+  const { chatHistory, removeChatHistoryByWindowId } = useChatStore();
+  const [hasReady, setHasReady] = useState(false);
+  const { isOpen: isOpenShare, onOpen: onOpenShare, onClose: onCloseShare } = useDisclosure();
+  const { isOpen: isOpenWx, onOpen: onOpenWx, onClose: onCloseWx } = useDisclosure();
+
+  const { isSuccess } = useQuery(['init'], getMyModels, {
+    cacheTime: 5 * 60 * 1000
+  });
+
+  useEffect(() => {
+    setHasReady(true);
+  }, []);
+
+  const RenderHistory = () => (
+    <>
+      {chatHistory.map((item) => (
+        <Flex
+          key={item.chatId}
+          alignItems={'center'}
+          p={3}
+          borderRadius={'md'}
+          mb={2}
+          cursor={'pointer'}
+          _hover={{
+            backgroundColor: 'rgba(255,255,255,0.1)'
+          }}
+          fontSize={'xs'}
+          border={'1px solid transparent'}
+          {...(item.chatId === chatId
+            ? {
+                borderColor: 'rgba(255,255,255,0.5)',
+                backgroundColor: 'rgba(255,255,255,0.1)'
+              }
+            : {})}
+          onClick={() => {
+            if (item.chatId === chatId) return;
+            router.replace(`/chat?chatId=${item.chatId}`);
+            onClose();
+          }}
+        >
+          <ChatIcon mr={2} />
+          <Box flex={'1 0 0'} w={0} className="textEllipsis">
+            {item.title}
+          </Box>
+          <Box>
+            <IconButton
+              icon={<DeleteIcon />}
+              variant={'unstyled'}
+              aria-label={'edit'}
+              size={'xs'}
+              onClick={(e) => {
+                removeChatHistoryByWindowId(item.chatId);
+                if (item.chatId === chatId) {
+                  resetChat();
+                }
+                e.stopPropagation();
+              }}
+            />
+          </Box>
+        </Flex>
+      ))}
+    </>
+  );
+
+  const RenderButton = ({
+    onClick,
+    children
+  }: {
+    onClick: () => void;
+    children: JSX.Element | string;
+  }) => (
+    <Box px={3} mb={3}>
+      <Flex
+        alignItems={'center'}
+        p={2}
+        cursor={'pointer'}
+        borderRadius={'md'}
+        _hover={{
+          backgroundColor: 'rgba(255,255,255,0.2)'
+        }}
+        onClick={onClick}
+      >
+        {children}
+      </Flex>
+    </Box>
+  );
+
+  return (
+    <Flex
+      flexDirection={'column'}
+      w={'100%'}
+      h={'100%'}
+      py={3}
+      backgroundColor={useColorModeValue('blackAlpha.800', 'blackAlpha.500')}
+      color={'white'}
+    >
+      {/* 新对话 */}
+      {getToken() && (
+        <Button
+          w={'90%'}
+          variant={'white'}
+          h={'40px'}
+          mb={4}
+          mx={'auto'}
+          leftIcon={<AddIcon />}
+          onClick={resetChat}
+        >
+          新对话
+        </Button>
+      )}
+
+      {/* 我的模型 & 历史记录 折叠框*/}
+      <Box flex={'1 0 0'} px={3} h={0} overflowY={'auto'}>
+        <Accordion defaultIndex={[0]} allowMultiple>
+          {isSuccess && (
+            <AccordionItem borderTop={0} borderBottom={0}>
+              <AccordionButton borderRadius={'md'} pl={1}>
+                <Box as="span" flex="1" textAlign="left">
+                  其他模型
+                </Box>
+                <AccordionIcon />
+              </AccordionButton>
+              <AccordionPanel pb={4} px={0}>
+                {myModels.map((item) => (
+                  <Flex
+                    key={item._id}
+                    alignItems={'center'}
+                    p={3}
+                    borderRadius={'md'}
+                    mb={2}
+                    cursor={'pointer'}
+                    _hover={{
+                      backgroundColor: 'rgba(255,255,255,0.1)'
+                    }}
+                    fontSize={'xs'}
+                    border={'1px solid transparent'}
+                    {...(item.name === name
+                      ? {
+                          borderColor: 'rgba(255,255,255,0.5)',
+                          backgroundColor: 'rgba(255,255,255,0.1)'
+                        }
+                      : {})}
+                    onClick={async () => {
+                      if (item.name === name) return;
+                      router.replace(`/chat?chatId=${await getChatSiteId(item._id)}`);
+                      onClose();
+                    }}
+                  >
+                    <MyIcon name="model" mr={2} fill={'white'} w={'16px'} h={'16px'} />
+                    <Box className={'textEllipsis'} flex={'1 0 0'} w={0}>
+                      {item.name}
+                    </Box>
+                  </Flex>
+                ))}
+              </AccordionPanel>
+            </AccordionItem>
+          )}
+          <AccordionItem borderTop={0} borderBottom={0}>
+            <AccordionButton borderRadius={'md'} pl={1}>
+              <Box as="span" flex="1" textAlign="left">
+                历史记录
+              </Box>
+              <AccordionIcon />
+            </AccordionButton>
+            <AccordionPanel pb={0} px={0}>
+              {hasReady && <RenderHistory />}
+            </AccordionPanel>
+          </AccordionItem>
+        </Accordion>
+      </Box>
+
+      <Divider my={4} colorScheme={useColorModeValue('gray', 'white')} />
+
+      <RenderButton onClick={() => router.push('/')}>
+        <>
+          <MyIcon name="home" fill={'white'} w={'18px'} h={'18px'} mr={4} />
+          首页
+        </>
+      </RenderButton>
+
+      <RenderButton onClick={onOpenShare}>
+        <>
+          <MyIcon name="share" fill={'white'} w={'16px'} h={'16px'} mr={4} />
+          分享
+        </>
+      </RenderButton>
+      <RenderButton onClick={() => router.push('/number/setting')}>
+        <>
+          <MyIcon name="pay" fill={'white'} w={'16px'} h={'16px'} mr={4} />
+          充值
+        </>
+      </RenderButton>
+
+      <Flex alignItems={'center'} mr={4}>
+        <Box flex={1}>
+          <RenderButton onClick={onOpenWx}>交流群</RenderButton>
+        </Box>
+        <IconButton
+          icon={colorMode === 'light' ? <MoonIcon /> : <SunIcon />}
+          aria-label={''}
+          variant={'outline'}
+          w={'16px'}
+          colorScheme={'white'}
+          _hover={{
+            backgroundColor: 'rgba(255,255,255,0.2)'
+          }}
+          onClick={toggleColorMode}
+        />
+      </Flex>
+
+      {/* 分享提示modal */}
+      <Modal isOpen={isOpenShare} onClose={onCloseShare}>
+        <ModalOverlay />
+        <ModalContent color={useColorModeValue('blackAlpha.700', 'white')}>
+          <ModalHeader>分享对话</ModalHeader>
+          <ModalCloseButton />
+          <ModalBody>
+            <Markdown source={shareHint} />
+          </ModalBody>
+
+          <ModalFooter>
+            <Button colorScheme="gray" variant={'outline'} mr={3} onClick={onCloseShare}>
+              取消
+            </Button>
+            {getToken() && (
+              <Button
+                variant="outline"
+                mr={3}
+                onClick={async () => {
+                  copyData(
+                    `${location.origin}/chat?chatId=${await getChatSiteId(modelId, true)}`,
+                    '已复制分享链接'
+                  );
+                  onCloseShare();
+                  onClose();
+                }}
+              >
+                分享空白对话
+              </Button>
+            )}
+
+            <Button
+              onClick={() => {
+                copyData(`${location.origin}/chat?chatId=${chatId}`, '已复制分享链接');
+                onCloseShare();
+                onClose();
+              }}
+            >
+              分享聊天记录
+            </Button>
+          </ModalFooter>
+        </ModalContent>
+      </Modal>
+      {/* wx 联系 */}
+      {isOpenWx && <WxConcat onClose={onCloseWx} />}
+    </Flex>
+  );
+};
+
+export default SlideBar;

src/pages/chat/index.tsx

@@ -1,36 +1,103 @@
-import React, { useCallback, useState, useRef, useMemo } from 'react';
+import React, { useCallback, useState, useRef, useMemo, useEffect } from 'react';
 import { useRouter } from 'next/router';
 import Image from 'next/image';
-import { getInitChatSiteInfo, postGPT3SendPrompt, delLastMessage, postSaveChat } from '@/api/chat';
-import { ChatSiteItemType, ChatSiteType } from '@/types/chat';
-import { Textarea, Box, Flex, Button } from '@chakra-ui/react';
+import {
+  getInitChatSiteInfo,
+  getChatSiteId,
+  postGPT3SendPrompt,
+  delLastMessage,
+  postSaveChat
+} from '@/api/chat';
+import type { InitChatResponse } from '@/api/response/chat';
+import { ChatSiteItemType } from '@/types/chat';
+import {
+  Textarea,
+  Box,
+  Flex,
+  Button,
+  useDisclosure,
+  Drawer,
+  DrawerOverlay,
+  DrawerContent,
+  useColorModeValue
+} from '@chakra-ui/react';
 import { useToast } from '@/hooks/useToast';
-import Icon from '@/components/Icon';
+import Icon from '@/components/Iconfont';
 import { useScreen } from '@/hooks/useScreen';
 import { useQuery } from '@tanstack/react-query';
-import { OpenAiModelEnum } from '@/constants/model';
+import { ChatModelNameEnum } from '@/constants/model';
 import dynamic from 'next/dynamic';
 import { useGlobalStore } from '@/store/global';
+import { useChatStore } from '@/store/chat';
 import { streamFetch } from '@/api/fetch';
+import SlideBar from './components/SlideBar';
+import Empty from './components/Empty';
+import { getToken } from '@/utils/user';
+import MyIcon from '@/components/Icon';
 
 const Markdown = dynamic(() => import('@/components/Markdown'));
 
 const textareaMinH = '22px';
 
-const Chat = ({ chatId, windowId }: { chatId: string; windowId?: string }) => {
+interface ChatType extends InitChatResponse {
+  history: ChatSiteItemType[];
+}
+
+const Chat = ({ chatId }: { chatId: string }) => {
   const { toast } = useToast();
   const router = useRouter();
   const { isPc, media } = useScreen();
+  const { setLoading } = useGlobalStore();
+  const [chatData, setChatData] = useState<ChatType>({
+    chatId: '',
+    modelId: '',
+    name: '',
+    avatar: '',
+    intro: '',
+    secret: {},
+    chatModel: '',
+    history: [],
+    isExpiredTime: false
+  }); // 聊天框整体数据
+
   const ChatBox = useRef<HTMLDivElement>(null);
   const TextareaDom = useRef<HTMLTextAreaElement>(null);
 
-  const [chatSiteData, setChatSiteData] = useState<ChatSiteType>(); // 聊天框整体数据
-  const [chatList, setChatList] = useState<ChatSiteItemType[]>([]); // 对话内容
   const [inputVal, setInputVal] = useState(''); // 输入的内容
+  const { isOpen: isOpenSlider, onClose: onCloseSlider, onOpen: onOpenSlider } = useDisclosure();
 
-  const isChatting = useMemo(() => chatList[chatList.length - 1]?.status === 'loading', [chatList]);
-  const lastWordHuman = useMemo(() => chatList[chatList.length - 1]?.obj === 'Human', [chatList]);
-  const { setLoading } = useGlobalStore();
+  const isChatting = useMemo(
+    () => chatData.history[chatData.history.length - 1]?.status === 'loading',
+    [chatData.history]
+  );
+  const chatWindowError = useMemo(() => {
+    if (chatData.history[chatData.history.length - 1]?.obj === 'Human') {
+      return {
+        text: '内容出现异常',
+        canDelete: true
+      };
+    }
+    if (chatData.isExpiredTime) {
+      return {
+        text: '聊天框已过期',
+        canDelete: false
+      };
+    }
+
+    return '';
+  }, [chatData]);
+
+  const { pushChatHistory } = useChatStore();
+  // 中断请求
+  const controller = useRef(new AbortController());
+  useEffect(() => {
+    controller.current = new AbortController();
+    return () => {
+      console.log('close========');
+      // eslint-disable-next-line react-hooks/exhaustive-deps
+      controller.current?.abort();
+    };
+  }, [chatId]);
 
   // 滚动到底部
   const scrollToBottom = useCallback(() => {
@@ -45,24 +112,25 @@ const Chat = ({ chatId, windowId }: { chatId: string; windowId?: string }) => {
 
   // 初始化聊天框
   useQuery(
-    ['initData'],
+    ['init', chatId],
     () => {
       setLoading(true);
-      return getInitChatSiteInfo(chatId, windowId);
+      return getInitChatSiteInfo(chatId);
     },
     {
       onSuccess(res) {
-        // 可能没有 windowId，给它设置一下
-        router.replace(`/chat?chatId=${chatId}&windowId=${res.windowId}`);
-
-        setChatSiteData(res.chatSite);
-        setChatList(
-          res.history.map((item) => ({
+        setChatData({
+          ...res,
+          history: res.history.map((item) => ({
             ...item,
             status: 'finish'
           }))
-        );
-        scrollToBottom();
+        });
+        if (res.history.length > 0) {
+          setTimeout(() => {
+            scrollToBottom();
+          }, 500);
+        }
       },
       onError(e: any) {
         toast({
@@ -91,9 +159,18 @@ const Chat = ({ chatId, windowId }: { chatId: string; windowId?: string }) => {
   }, []);
 
   // 重载对话
-  const resetChat = useCallback(() => {
-    window.open(`/chat?chatId=${chatId}`, '_self');
-  }, [chatId]);
+  const resetChat = useCallback(async () => {
+    if (!chatData) return;
+    try {
+      router.replace(`/chat?chatId=${await getChatSiteId(chatData.modelId)}`);
+    } catch (error: any) {
+      toast({
+        title: error?.message || '生成新对话失败',
+        status: 'warning'
+      });
+    }
+    onCloseSlider();
+  }, [chatData, onCloseSlider, router, toast]);
 
   // gpt3 方法
   const gpt3ChatPrompt = useCallback(
@@ -105,53 +182,60 @@ const Chat = ({ chatId, windowId }: { chatId: string; windowId?: string }) => {
       });
 
       // 更新 AI 的内容
-      setChatList((state) =>
-        state.map((item, index) => {
-          if (index !== state.length - 1) return item;
+      setChatData((state) => ({
+        ...state,
+        history: state.history.map((item, index) => {
+          if (index !== state.history.length - 1) return item;
           return {
             ...item,
             status: 'finish',
             value: response
           };
         })
-      );
+      }));
     },
     [chatId]
   );
 
-  // chatGPT
-  const chatGPTPrompt = useCallback(
-    async (newChatList: ChatSiteItemType[]) => {
-      if (!windowId) return;
+  // gpt 对话
+  const gptChatPrompt = useCallback(
+    async (prompts: ChatSiteItemType) => {
+      const urlMap: Record<string, string> = {
+        [ChatModelNameEnum.GPT35]: '/api/chat/chatGpt',
+        [ChatModelNameEnum.GPT3]: '/api/chat/gpt3'
+      };
+      if (!urlMap[chatData.chatModel]) return Promise.reject('找不到模型');
+
       const prompt = {
-        obj: newChatList[newChatList.length - 1].obj,
-        value: newChatList[newChatList.length - 1].value
+        obj: prompts.obj,
+        value: prompts.value
       };
       // 流请求，获取数据
       const res = await streamFetch({
-        url: '/api/chat/chatGpt',
+        url: urlMap[chatData.chatModel],
         data: {
-          windowId,
           prompt,
           chatId
         },
         onMessage: (text: string) => {
-          setChatList((state) =>
-            state.map((item, index) => {
-              if (index !== state.length - 1) return item;
+          setChatData((state) => ({
+            ...state,
+            history: state.history.map((item, index) => {
+              if (index !== state.history.length - 1) return item;
               return {
                 ...item,
                 value: item.value + text
               };
             })
-          );
-        }
+          }));
+        },
+        abortSignal: controller.current
       });
 
       // 保存对话信息
       try {
         await postSaveChat({
-          windowId,
+          chatId,
           prompts: [
             prompt,
             {
@@ -162,7 +246,7 @@ const Chat = ({ chatId, windowId }: { chatId: string; windowId?: string }) => {
         });
       } catch (err) {
         toast({
-          title: '存储对话出现异常, 继续对话会导致上下文丢失，请刷新页面',
+          title: '对话出现异常, 继续对话会导致上下文丢失，请刷新页面',
           status: 'warning',
           duration: 3000,
           isClosable: true
@@ -170,17 +254,18 @@ const Chat = ({ chatId, windowId }: { chatId: string; windowId?: string }) => {
       }
 
       // 设置完成状态
-      setChatList((state) =>
-        state.map((item, index) => {
-          if (index !== state.length - 1) return item;
+      setChatData((state) => ({
+        ...state,
+        history: state.history.map((item, index) => {
+          if (index !== state.history.length - 1) return item;
           return {
             ...item,
             status: 'finish'
           };
         })
-      );
+      }));
     },
-    [chatId, toast, windowId]
+    [chatData.chatModel, chatId, toast]
   );
 
   /**
@@ -193,13 +278,13 @@ const Chat = ({ chatId, windowId }: { chatId: string; windowId?: string }) => {
       .trim()
       .split('\n')
       .filter((val) => val)
-      .join('\n\n');
-    if (!chatSiteData?.modelId || !val || !ChatBox.current || isChatting) {
+      .join('\n');
+    if (!chatData?.modelId || !val || !ChatBox.current || isChatting) {
       return;
     }
 
     const newChatList: ChatSiteItemType[] = [
-      ...chatList,
+      ...chatData.history,
       {
         obj: 'Human',
         value: val,
@@ -213,25 +298,25 @@ const Chat = ({ chatId, windowId }: { chatId: string; windowId?: string }) => {
     ];
 
     // 插入内容
-    setChatList(newChatList);
+    setChatData((state) => ({
+      ...state,
+      history: newChatList
+    }));
+
+    // 清空输入内容
     resetInputVal('');
     scrollToBottom();
 
-    const fnMap: { [key: string]: any } = {
-      [OpenAiModelEnum.GPT35]: chatGPTPrompt,
-      [OpenAiModelEnum.GPT3]: gpt3ChatPrompt
-    };
-
     try {
-      /* 对长度进行限制 */
-      const maxContext = chatSiteData.secret.contextMaxLen;
-      const requestPrompt =
-        newChatList.length > maxContext + 2
-          ? [newChatList[0], ...newChatList.slice(newChatList.length - maxContext - 1, -1)]
-          : newChatList.slice(0, newChatList.length - 1);
+      await gptChatPrompt(newChatList[newChatList.length - 2]);
 
-      if (typeof fnMap[chatSiteData.chatModel] === 'function') {
-        await fnMap[chatSiteData.chatModel](requestPrompt);
+      // 如果是 Human 第一次发送，插入历史记录
+      const humanChat = newChatList.filter((item) => item.obj === 'Human');
+      if (humanChat.length === 1) {
+        pushChatHistory({
+          chatId,
+          title: humanChat[0].value
+        });
       }
     } catch (err: any) {
       toast({
@@ -243,183 +328,220 @@ const Chat = ({ chatId, windowId }: { chatId: string; windowId?: string }) => {
 
       resetInputVal(storeInput);
 
-      setChatList(newChatList.slice(0, newChatList.length - 2));
+      setChatData((state) => ({
+        ...state,
+        history: newChatList.slice(0, newChatList.length - 2)
+      }));
     }
   }, [
-    chatGPTPrompt,
-    chatList,
-    chatSiteData,
-    gpt3ChatPrompt,
     inputVal,
+    chatData?.modelId,
+    chatData.history,
     isChatting,
     resetInputVal,
     scrollToBottom,
+    gptChatPrompt,
+    pushChatHistory,
+    chatId,
     toast
   ]);
 
   // 重新编辑
   const reEdit = useCallback(async () => {
-    if (chatList[chatList.length - 1]?.obj !== 'Human') return;
+    if (chatData.history[chatData.history.length - 1]?.obj !== 'Human') return;
     // 删除数据库最后一句
-    await delLastMessage(windowId);
-    const val = chatList[chatList.length - 1].value;
+    await delLastMessage(chatId);
+    const val = chatData.history[chatData.history.length - 1].value;
 
     resetInputVal(val);
 
-    setChatList(chatList.slice(0, -1));
-  }, [chatList, resetInputVal, windowId]);
+    setChatData((state) => ({
+      ...state,
+      history: state.history.slice(0, -1)
+    }));
+  }, [chatData.history, chatId, resetInputVal]);
 
   return (
-    <Flex height={'100%'} flexDirection={'column'}>
-      {/* 头部 */}
-      <Flex
-        px={4}
-        h={'50px'}
-        alignItems={'center'}
-        backgroundColor={'white'}
-        boxShadow={'0 5px 10px rgba(0,0,0,0.1)'}
-        zIndex={1}
-      >
-        <Box flex={1}>{chatSiteData?.name}</Box>
-        {/* 滚动到底部按键 */}
-        {ChatBox.current && ChatBox.current.scrollHeight > 2 * ChatBox.current.clientHeight && (
-          <Box mr={10} cursor={'pointer'} onClick={scrollToBottom}>
-            <Icon
-              name={'icon-xiangxiazhankai-xianxingyuankuang'}
-              width={25}
-              height={25}
-              color={'#718096'}
-            ></Icon>
-          </Box>
-        )}
-        {/* 重置按键 */}
-        <Button size={'sm'} colorScheme={'gray'} onClick={resetChat}>
-          新对话
-        </Button>
-      </Flex>
-      {/* 聊天内容 */}
-      <Box ref={ChatBox} flex={'1 0 0'} h={0} w={'100%'} px={0} pb={10} overflowY={'auto'}>
-        {chatList.map((item, index) => (
-          <Box
-            key={index}
-            py={media(9, 6)}
-            px={media(4, 2)}
-            backgroundColor={index % 2 === 0 ? 'rgba(247,247,248,1)' : '#fff'}
-            borderBottom={'1px solid rgba(0,0,0,0.1)'}
+    <Flex
+      h={'100%'}
+      flexDirection={media('row', 'column')}
+      backgroundColor={useColorModeValue('white', '')}
+    >
+      {isPc ? (
+        <Box flex={'0 0 250px'} w={0} h={'100%'}>
+          <SlideBar
+            resetChat={resetChat}
+            name={chatData?.name}
+            chatId={chatId}
+            modelId={chatData.modelId}
+            onClose={onCloseSlider}
+          />
+        </Box>
+      ) : (
+        <Box h={'60px'} borderBottom={'1px solid rgba(0,0,0,0.1)'}>
+          <Flex
+            alignItems={'center'}
+            h={'100%'}
+            justifyContent={'space-between'}
+            backgroundColor={useColorModeValue('white', 'gray.700')}
+            color={useColorModeValue('blackAlpha.700', 'white')}
+            position={'relative'}
+            px={7}
           >
-            <Flex maxW={'800px'} m={'auto'} alignItems={'flex-start'}>
-              <Box mr={media(4, 1)}>
-                <Image
-                  src={item.obj === 'Human' ? '/icon/human.png' : '/icon/logo.png'}
-                  alt="/icon/logo.png"
-                  width={media(30, 20)}
-                  height={media(30, 20)}
-                />
-              </Box>
-              <Box flex={'1 0 0'} w={0} overflow={'hidden'}>
-                {item.obj === 'AI' ? (
-                  <Markdown
-                    source={item.value}
-                    isChatting={isChatting && index === chatList.length - 1}
+            <Box onClick={onOpenSlider}>
+              <MyIcon
+                name={'menu'}
+                w={'20px'}
+                h={'20px'}
+                fill={useColorModeValue('blackAlpha.700', 'white')}
+              />
+            </Box>
+            <Box>{chatData?.name}</Box>
+          </Flex>
+          <Drawer isOpen={isOpenSlider} placement="left" size={'xs'} onClose={onCloseSlider}>
+            <DrawerOverlay backgroundColor={'rgba(255,255,255,0.5)'} />
+            <DrawerContent maxWidth={'250px'}>
+              <SlideBar
+                resetChat={resetChat}
+                name={chatData?.name}
+                chatId={chatId}
+                modelId={chatData.modelId}
+                onClose={onCloseSlider}
+              />
+            </DrawerContent>
+          </Drawer>
+        </Box>
+      )}
+
+      <Flex
+        {...media({ h: '100%', w: 0 }, { h: 0, w: '100%' })}
+        flex={'1 0 0'}
+        flexDirection={'column'}
+      >
+        {/* 聊天内容 */}
+        <Box ref={ChatBox} flex={'1 0 0'} h={0} w={'100%'} overflowY={'auto'}>
+          {chatData.history.map((item, index) => (
+            <Box
+              key={index}
+              py={media(9, 6)}
+              px={media(4, 2)}
+              backgroundColor={
+                index % 2 !== 0 ? useColorModeValue('blackAlpha.50', 'gray.700') : ''
+              }
+              color={useColorModeValue('blackAlpha.700', 'white')}
+              borderBottom={'1px solid rgba(0,0,0,0.1)'}
+            >
+              <Flex maxW={'750px'} m={'auto'} alignItems={'flex-start'}>
+                <Box mr={media(4, 1)}>
+                  <Image
+                    src={item.obj === 'Human' ? '/icon/human.png' : '/icon/logo.png'}
+                    alt="/icon/logo.png"
+                    width={media(30, 20)}
+                    height={media(30, 20)}
+                  />
+                </Box>
+                <Box flex={'1 0 0'} w={0} overflow={'hidden'}>
+                  {item.obj === 'AI' ? (
+                    <Markdown
+                      source={item.value}
+                      isChatting={isChatting && index === chatData.history.length - 1}
+                    />
+                  ) : (
+                    <Box whiteSpace={'pre-wrap'}>{item.value}</Box>
+                  )}
+                </Box>
+              </Flex>
+            </Box>
+          ))}
+          {chatData.history.length === 0 && <Empty intro={chatData.intro} />}
+        </Box>
+        {/* 发送区 */}
+        <Box m={media('20px auto', '0 auto')} w={'100%'} maxW={media('min(750px, 100%)', 'auto')}>
+          {!!chatWindowError ? (
+            <Box textAlign={'center'}>
+              <Box color={'red'}>{chatWindowError.text}</Box>
+              <Flex py={5} justifyContent={'center'}>
+                {getToken() && <Button onClick={resetChat}>重开对话</Button>}
+
+                {chatWindowError.canDelete && (
+                  <Button ml={20} colorScheme={'green'} onClick={reEdit}>
+                    重新编辑最后一句
+                  </Button>
+                )}
+              </Flex>
+            </Box>
+          ) : (
+            <Box
+              py={5}
+              position={'relative'}
+              boxShadow={`0 0 15px rgba(0,0,0,0.1)`}
+              border={media('1px solid', '0')}
+              borderColor={useColorModeValue('gray.200', 'gray.700')}
+              borderRadius={['none', 'md']}
+              backgroundColor={useColorModeValue('white', 'gray.700')}
+            >
+              {/* 输入框 */}
+              <Textarea
+                ref={TextareaDom}
+                w={'100%'}
+                pr={'45px'}
+                py={0}
+                border={'none'}
+                _focusVisible={{
+                  border: 'none'
+                }}
+                placeholder="提问"
+                resize={'none'}
+                value={inputVal}
+                rows={1}
+                height={'22px'}
+                lineHeight={'22px'}
+                maxHeight={'150px'}
+                maxLength={chatData?.secret.contentMaxLen || -1}
+                overflowY={'auto'}
+                color={useColorModeValue('blackAlpha.700', 'white')}
+                onChange={(e) => {
+                  const textarea = e.target;
+                  setInputVal(textarea.value);
+                  textarea.style.height = textareaMinH;
+                  textarea.style.height = `${textarea.scrollHeight}px`;
+                }}
+                onKeyDown={(e) => {
+                  // 触发快捷发送
+                  if (isPc && e.keyCode === 13 && !e.shiftKey) {
+                    sendPrompt();
+                    e.preventDefault();
+                  }
+                  // 全选内容
+                  // @ts-ignore
+                  e.key === 'a' && e.ctrlKey && e.target?.select();
+                }}
+              />
+              {/* 发送和等待按键 */}
+              <Box position={'absolute'} bottom={5} right={media('20px', '10px')}>
+                {isChatting ? (
+                  <Image
+                    style={{ transform: 'translateY(4px)' }}
+                    src={'/icon/chatting.svg'}
+                    width={30}
+                    height={30}
+                    alt={''}
                   />
                 ) : (
-                  <Box whiteSpace={'pre-wrap'}>{item.value}</Box>
+                  <Box cursor={'pointer'} onClick={sendPrompt}>
+                    <Icon
+                      name={'icon-fasong'}
+                      width={20}
+                      height={20}
+                      color={useColorModeValue('#718096', 'white')}
+                    ></Icon>
+                  </Box>
                 )}
               </Box>
-            </Flex>
-          </Box>
-        ))}
-      </Box>
-      {/* 空内容提示 */}
-      {/* {
-        chatList.length === 0 && (
-          <>
-          <Card>
-内容太长
-</Card>
-          </>
-        )
-      } */}
-      <Box
-        m={media('20px auto', '0 auto')}
-        w={media('100vw', '100%')}
-        maxW={media('800px', 'auto')}
-        boxShadow={'0 -14px 30px rgba(255,255,255,0.6)'}
-        borderTop={media('none', '1px solid rgba(0,0,0,0.1)')}
-      >
-        {lastWordHuman ? (
-          <Box textAlign={'center'}>
-            <Box color={'red'}>对话出现了异常</Box>
-            <Flex py={5} justifyContent={'center'}>
-              <Button mr={20} onClick={resetChat} colorScheme={'green'}>
-                重开对话
-              </Button>
-              <Button onClick={reEdit}>重新编辑最后一句</Button>
-            </Flex>
-          </Box>
-        ) : (
-          <Box
-            py={5}
-            position={'relative'}
-            boxShadow={'base'}
-            overflow={'hidden'}
-            borderRadius={media('md', 'none')}
-          >
-            {/* 输入框 */}
-            <Textarea
-              ref={TextareaDom}
-              w={'100%'}
-              pr={'45px'}
-              py={0}
-              border={'none'}
-              _focusVisible={{
-                border: 'none'
-              }}
-              placeholder="提问"
-              resize={'none'}
-              value={inputVal}
-              rows={1}
-              height={'22px'}
-              lineHeight={'22px'}
-              maxHeight={'150px'}
-              maxLength={chatSiteData?.secret.contentMaxLen || -1}
-              overflowY={'auto'}
-              onChange={(e) => {
-                const textarea = e.target;
-                setInputVal(textarea.value);
-                textarea.style.height = textareaMinH;
-                textarea.style.height = `${textarea.scrollHeight}px`;
-              }}
-              onKeyDown={(e) => {
-                // 触发快捷发送
-                if (isPc && e.keyCode === 13 && !e.shiftKey) {
-                  sendPrompt();
-                  e.preventDefault();
-                }
-                // 全选内容
-                // @ts-ignore
-                e.key === 'a' && e.ctrlKey && e.target?.select();
-              }}
-            />
-            {/* 发送和等待按键 */}
-            <Box position={'absolute'} bottom={5} right={media('20px', '10px')}>
-              {isChatting ? (
-                <Image
-                  style={{ transform: 'translateY(4px)' }}
-                  src={'/icon/chatting.svg'}
-                  width={30}
-                  height={30}
-                  alt={''}
-                />
-              ) : (
-                <Box cursor={'pointer'} onClick={sendPrompt}>
-                  <Icon name={'icon-fasong'} width={20} height={20} color={'#718096'}></Icon>
-                </Box>
-              )}
             </Box>
-          </Box>
-        )}
-      </Box>
+          )}
+        </Box>
+      </Flex>
     </Flex>
   );
 };
@@ -428,9 +550,8 @@ export default Chat;
 
 export async function getServerSideProps(context: any) {
   const chatId = context.query?.chatId || '';
-  const windowId = context.query?.windowId || '';
 
   return {
-    props: { chatId, windowId }
+    props: { chatId }
   };
 }

src/pages/data/components/CreateDataModal.tsx

@@ -0,0 +1,65 @@
+import React, { useState } from 'react';
+import {
+  Modal,
+  ModalOverlay,
+  ModalContent,
+  ModalHeader,
+  ModalFooter,
+  ModalBody,
+  ModalCloseButton,
+  Button,
+  Input
+} from '@chakra-ui/react';
+import { postData } from '@/api/data';
+import { useMutation } from '@tanstack/react-query';
+
+const CreateDataModal = ({
+  onClose,
+  onSuccess
+}: {
+  onClose: () => void;
+  onSuccess: () => void;
+}) => {
+  const [inputVal, setInputVal] = useState('');
+
+  const { isLoading, mutate } = useMutation({
+    mutationFn: (name: string) => postData(name),
+    onSuccess() {
+      onSuccess();
+      onClose();
+    }
+  });
+
+  return (
+    <Modal isOpen={true} onClose={onClose}>
+      <ModalOverlay />
+      <ModalContent>
+        <ModalHeader>创建数据集</ModalHeader>
+        <ModalCloseButton />
+
+        <ModalBody display={'flex'}>
+          <Input
+            value={inputVal}
+            onChange={(e) => setInputVal(e.target.value)}
+            placeholder={'数据集名称'}
+          ></Input>
+        </ModalBody>
+        <ModalFooter>
+          <Button colorScheme={'gray'} onClick={onClose}>
+            取消
+          </Button>
+          <Button
+            ml={3}
+            isDisabled={inputVal === ''}
+            isLoading={isLoading}
+            onClick={() => mutate(inputVal)}
+          >
+            确认
+          </Button>
+        </ModalFooter>
+      </ModalContent>
+    </Modal>
+  );
+};
+
+export default CreateDataModal;

src/pages/data/components/ImportDataModal.tsx

@@ -0,0 +1,222 @@
+import React, { useState, useCallback } from 'react';
+import {
+  Modal,
+  ModalOverlay,
+  ModalContent,
+  ModalHeader,
+  ModalFooter,
+  ModalBody,
+  ModalCloseButton,
+  Button,
+  Box,
+  Flex,
+  Textarea
+} from '@chakra-ui/react';
+import { useTabs } from '@/hooks/useTabs';
+import { useConfirm } from '@/hooks/useConfirm';
+import { useSelectFile } from '@/hooks/useSelectFile';
+import { readTxtContent, readPdfContent, readDocContent } from '@/utils/tools';
+import { postSplitData } from '@/api/data';
+import { useMutation } from '@tanstack/react-query';
+import { useToast } from '@/hooks/useToast';
+import { useLoading } from '@/hooks/useLoading';
+import { formatPrice } from '@/utils/user';
+import { modelList, ChatModelNameEnum } from '@/constants/model';
+
+const fileExtension = '.txt,.doc,.docx,.pdf,.md';
+
+const ImportDataModal = ({
+  dataId,
+  onClose,
+  onSuccess
+}: {
+  dataId: string;
+  onClose: () => void;
+  onSuccess: () => void;
+}) => {
+  const { openConfirm, ConfirmChild } = useConfirm({
+    content: '确认提交生成任务？该任务无法终止！'
+  });
+  const { toast } = useToast();
+  const { setIsLoading, Loading } = useLoading();
+  const { File, onOpen } = useSelectFile({ fileType: fileExtension, multiple: true });
+  const { tabs, activeTab, setActiveTab } = useTabs({
+    tabs: [
+      { id: 'text', label: '文本' },
+      { id: 'doc', label: '文件' }
+      // { id: 'url', label: '链接' }
+    ]
+  });
+
+  const [textInput, setTextInput] = useState('');
+  const [fileText, setFileText] = useState('');
+
+  const { mutate: handleClickSubmit, isLoading } = useMutation({
+    mutationFn: async () => {
+      let text = '';
+      if (activeTab === 'text') {
+        text = textInput;
+      } else if (activeTab === 'doc') {
+        text = fileText;
+      } else if (activeTab === 'url') {
+      }
+      if (!text) return;
+      return postSplitData(dataId, text);
+    },
+    onSuccess() {
+      toast({
+        title: '任务提交成功',
+        status: 'success'
+      });
+      onClose();
+      onSuccess();
+    },
+    onError(err: any) {
+      toast({
+        title: err?.message || '提交任务异常',
+        status: 'error'
+      });
+    }
+  });
+
+  const onSelectFile = useCallback(
+    async (e: File[]) => {
+      setIsLoading(true);
+      try {
+        const fileTexts = (
+          await Promise.all(
+            e.map((file) => {
+              // @ts-ignore
+              const extension = file?.name?.split('.').pop().toLowerCase();
+              switch (extension) {
+                case 'txt':
+                case 'md':
+                  return readTxtContent(file);
+                case 'pdf':
+                  return readPdfContent(file);
+                case 'doc':
+                case 'docx':
+                  return readDocContent(file);
+                default:
+                  return '';
+              }
+            })
+          )
+        )
+          .join('\n')
+          .replace(/\n+/g, '\n');
+        setFileText(fileTexts);
+      } catch (error: any) {
+        console.log(error);
+        toast({
+          title: typeof error === 'string' ? error : '解析文件失败',
+          status: 'error'
+        });
+      }
+      setIsLoading(false);
+    },
+    [setIsLoading, toast]
+  );
+
+  return (
+    <Modal isOpen={true} onClose={onClose}>
+      <ModalOverlay />
+      <ModalContent position={'relative'} maxW={['90vw', '800px']}>
+        <ModalHeader>
+          导入数据，生成QA
+          <Box ml={2} as={'span'} fontSize={'sm'} color={'blackAlpha.600'}>
+            {formatPrice(
+              modelList.find((item) => item.model === ChatModelNameEnum.GPT35)?.price || 0,
+              1000
+            )}
+            元/1K tokens
+          </Box>
+        </ModalHeader>
+        <ModalCloseButton />
+
+        <ModalBody display={'flex'}>
+          <Box>
+            {tabs.map((item) => (
+              <Button
+                key={item.id}
+                display={'block'}
+                variant={activeTab === item.id ? 'solid' : 'outline'}
+                _notLast={{
+                  mb: 3
+                }}
+                onClick={() => setActiveTab(item.id)}
+              >
+                {item.label}
+              </Button>
+            ))}
+          </Box>
+
+          <Box flex={'1 0 0'} w={0} ml={3} minH={'200px'}>
+            {activeTab === 'text' && (
+              <>
+                <Textarea
+                  h={'100%'}
+                  maxLength={-1}
+                  value={textInput}
+                  placeholder={'请粘贴或输入需要处理的文本'}
+                  onChange={(e) => setTextInput(e.target.value)}
+                />
+                <Box mt={2}>一共 {textInput.length} 个字</Box>
+              </>
+            )}
+            {activeTab === 'doc' && (
+              <Flex
+                flexDirection={'column'}
+                p={2}
+                h={'100%'}
+                alignItems={'center'}
+                justifyContent={'center'}
+                border={'1px solid '}
+                borderColor={'blackAlpha.200'}
+                borderRadius={'md'}
+              >
+                <Button onClick={onOpen}>选择文件</Button>
+                <Box mt={2}>支持 {fileExtension} 文件</Box>
+                {fileText && (
+                  <>
+                    <Box mt={2}>一共 {fileText.length} 个字</Box>
+                    <Box
+                      maxH={'300px'}
+                      w={'100%'}
+                      overflow={'auto'}
+                      p={2}
+                      backgroundColor={'blackAlpha.50'}
+                      whiteSpace={'pre'}
+                      fontSize={'xs'}
+                    >
+                      {fileText}
+                    </Box>
+                  </>
+                )}
+              </Flex>
+            )}
+          </Box>
+        </ModalBody>
+        <ModalFooter>
+          <Button colorScheme={'gray'} onClick={onClose}>
+            取消
+          </Button>
+          <Button
+            ml={3}
+            isLoading={isLoading}
+            isDisabled={!textInput && !fileText}
+            onClick={openConfirm(handleClickSubmit)}
+          >
+            确认
+          </Button>
+        </ModalFooter>
+        <Loading />
+      </ModalContent>
+
+      <ConfirmChild />
+      <File onSelect={onSelectFile} />
+    </Modal>
+  );
+};
+
+export default ImportDataModal;

src/pages/data/detail.tsx

@@ -0,0 +1,61 @@
+import React from 'react';
+import { Box, Card, Table, Thead, Tbody, Tr, Th, Td, TableContainer } from '@chakra-ui/react';
+import ScrollData from '@/components/ScrollData';
+import { getDataItems } from '@/api/data';
+import { usePaging } from '@/hooks/usePaging';
+import type { DataItemSchema } from '@/types/mongoSchema';
+
+const DataDetail = ({ dataName, dataId }: { dataName: string; dataId: string }) => {
+  const {
+    nextPage,
+    isLoadAll,
+    requesting,
+    data: dataItems
+  } = usePaging<DataItemSchema>({
+    api: getDataItems,
+    pageSize: 10,
+    params: {
+      dataId
+    }
+  });
+
+  return (
+    <Card py={4} h={'100%'} display={'flex'} flexDirection={'column'}>
+      <Box px={6} fontSize={'xl'} fontWeight={'bold'}>
+        {dataName} 拆分结果
+      </Box>
+      <ScrollData
+        flex={'1 0 0'}
+        h={0}
+        px={6}
+        mt={3}
+        isLoadAll={isLoadAll}
+        requesting={requesting}
+        nextPage={nextPage}
+        fontSize={'xs'}
+      >
+        {dataItems.map((item) => (
+          <Box key={item._id}>
+            {item.result.map((result, i) => (
+              <Box key={i} mb={3}>
+                <Box fontWeight={'bold'}>Q: {result.q}</Box>
+                <Box>A: {result.a}</Box>
+              </Box>
+            ))}
+          </Box>
+        ))}
+      </ScrollData>
+    </Card>
+  );
+};
+
+export default DataDetail;
+
+export async function getServerSideProps(context: any) {
+  return {
+    props: {
+      dataName: context.query?.dataName || '',
+      dataId: context.query?.dataId || ''
+    }
+  };
+}

src/pages/data/list.tsx

@@ -0,0 +1,226 @@
+import React, { useState, useCallback } from 'react';
+import {
+  Card,
+  Box,
+  Flex,
+  Button,
+  Table,
+  Thead,
+  Tbody,
+  Tr,
+  Th,
+  Td,
+  TableContainer,
+  useDisclosure,
+  Input,
+  Menu,
+  MenuButton,
+  MenuList,
+  MenuItem
+} from '@chakra-ui/react';
+import { getDataList, updateDataName, delData, getDataItems } from '@/api/data';
+import { usePaging } from '@/hooks/usePaging';
+import type { DataListItem } from '@/types/data';
+import ScrollData from '@/components/ScrollData';
+import dayjs from 'dayjs';
+import dynamic from 'next/dynamic';
+import { useRouter } from 'next/router';
+import { useConfirm } from '@/hooks/useConfirm';
+import { useRequest } from '@/hooks/useRequest';
+import { DataItemSchema } from '@/types/mongoSchema';
+
+const CreateDataModal = dynamic(() => import('./components/CreateDataModal'));
+const ImportDataModal = dynamic(() => import('./components/ImportDataModal'));
+
+export type ExportDataType = 'jsonl';
+
+const DataList = () => {
+  const router = useRouter();
+  const {
+    nextPage,
+    isLoadAll,
+    requesting,
+    data: dataList,
+    getData,
+    initRequesting
+  } = usePaging<DataListItem>({
+    api: getDataList,
+    pageSize: 20
+  });
+  const [ImportDataId, setImportDataId] = useState<string>();
+  const { openConfirm, ConfirmChild } = useConfirm({
+    content: '删除数据集，将删除里面的所有内容，请确认！'
+  });
+
+  const {
+    isOpen: isOpenCreateDataModal,
+    onOpen: onOpenCreateDataModal,
+    onClose: onCloseCreateDataModal
+  } = useDisclosure();
+
+  const { mutate: handleDelData, isLoading: isDeleting } = useRequest({
+    mutationFn: (dataId: string) => delData(dataId),
+    successToast: '删除数据集成功',
+    errorToast: '删除数据集异常',
+    onSuccess() {
+      getData(1, true);
+    }
+  });
+
+  const { mutate: handleExportData, isLoading: isExporting } = useRequest({
+    mutationFn: async ({ data, type }: { data: DataListItem; type: ExportDataType }) => ({
+      type,
+      data: await getDataItems({ dataId: data._id, pageNum: 1, pageSize: data.totalData }).then(
+        (res) => res.data
+      )
+    }),
+    successToast: '导出数据集成功',
+    errorToast: '导出数据集异常',
+    onSuccess(res: { type: ExportDataType; data: DataItemSchema[] }) {
+      // 合并数据
+      const data = res.data.map((item) => item.result).flat();
+      let text = '';
+      // 生成 jsonl
+      data.forEach((item) => {
+        const result = JSON.stringify({
+          prompt: `${item.q.toLocaleLowerCase()}</s>`,
+          completion: ` ${item.a}</s>`
+        });
+        text += `${result}\n`;
+      });
+      // 去掉最后一个 \n
+      text = text.substring(0, text.length - 1);
+      // 导出为文件
+      const blob = new Blob([text], { type: 'application/json;charset=utf-8' });
+
+      // 创建下载链接
+      const downloadLink = document.createElement('a');
+      downloadLink.href = window.URL.createObjectURL(blob);
+      downloadLink.download = 'file.jsonl';
+
+      // 添加链接到页面并触发下载
+      document.body.appendChild(downloadLink);
+      downloadLink.click();
+      document.body.removeChild(downloadLink);
+    }
+  });
+
+  return (
+    <Box display={['block', 'flex']} flexDirection={'column'} h={'100%'}>
+      <Card px={6} py={4}>
+        <Flex>
+          <Box flex={1} mr={1}>
+            <Box fontSize={'xl'} fontWeight={'bold'}>
+              训练数据管理
+            </Box>
+            <Box fontSize={'xs'} color={'blackAlpha.600'}>
+              允许你将任意文本数据拆分成 QA 的形式。你可以使用这些 QA 去微调你的对话模型。
+            </Box>
+          </Box>
+          <Button variant={'outline'} onClick={onOpenCreateDataModal}>
+            创建数据集
+          </Button>
+        </Flex>
+      </Card>
+      {/* 数据表 */}
+      <Card mt={3} flex={'1 0 0'} h={['auto', '0']} px={6} py={4}>
+        <ScrollData
+          h={'100%'}
+          nextPage={nextPage}
+          isLoadAll={isLoadAll}
+          requesting={requesting}
+          initRequesting={initRequesting}
+        >
+          <TableContainer>
+            <Table>
+              <Thead>
+                <Tr>
+                  <Th>集合名</Th>
+                  <Th>创建时间</Th>
+                  <Th>训练中 / 总数据</Th>
+                  <Th></Th>
+                </Tr>
+              </Thead>
+              <Tbody>
+                {dataList.map((item, i) => (
+                  <Tr key={item._id}>
+                    <Td>
+                      <Input
+                        minW={'150px'}
+                        placeholder="请输入数据集名称"
+                        defaultValue={item.name}
+                        size={'sm'}
+                        onBlur={(e) => {
+                          if (!e.target.value || e.target.value === item.name) return;
+                          updateDataName(item._id, e.target.value);
+                        }}
+                      />
+                    </Td>
+                    <Td>{dayjs(item.createTime).format('YYYY/MM/DD HH:mm')}</Td>
+                    <Td>
+                      {item.trainingData} / {item.totalData}
+                    </Td>
+                    <Td>
+                      <Button
+                        size={'sm'}
+                        variant={'outline'}
+                        colorScheme={'gray'}
+                        mr={2}
+                        onClick={() =>
+                          router.push(`/data/detail?dataId=${item._id}&dataName=${item.name}`)
+                        }
+                      >
+                        详细
+                      </Button>
+                      <Button
+                        size={'sm'}
+                        variant={'outline'}
+                        mr={2}
+                        onClick={() => setImportDataId(item._id)}
+                      >
+                        导入
+                      </Button>
+                      <Menu>
+                        <MenuButton as={Button} mr={2} size={'sm'}>
+                          导出
+                        </MenuButton>
+                        <MenuList>
+                          <MenuItem onClick={() => handleExportData({ data: item, type: 'jsonl' })}>
+                            jsonl
+                          </MenuItem>
+                        </MenuList>
+                      </Menu>
+
+                      <Button
+                        size={'sm'}
+                        colorScheme={'red'}
+                        isLoading={isDeleting}
+                        onClick={openConfirm(() => handleDelData(item._id))}
+                      >
+                        删除
+                      </Button>
+                    </Td>
+                  </Tr>
+                ))}
+              </Tbody>
+            </Table>
+          </TableContainer>
+        </ScrollData>
+      </Card>
+
+      {ImportDataId && (
+        <ImportDataModal
+          dataId={ImportDataId}
+          onClose={() => setImportDataId(undefined)}
+          onSuccess={() => getData(1, true)}
+        />
+      )}
+      {isOpenCreateDataModal && (
+        <CreateDataModal onClose={onCloseCreateDataModal} onSuccess={() => getData(1, true)} />
+      )}
+      <ConfirmChild />
+    </Box>
+  );
+};
+
+export default DataList;

src/pages/login/index.tsx

@@ -6,9 +6,8 @@ import { useScreen } from '@/hooks/useScreen';
 import type { ResLogin } from '@/api/response/user';
 import { useRouter } from 'next/router';
 import { useUserStore } from '@/store/user';
-
+import LoginForm from './components/LoginForm';
 import dynamic from 'next/dynamic';
-const LoginForm = dynamic(() => import('./components/LoginForm'));
 const RegisterForm = dynamic(() => import('./components/RegisterForm'));
 const ForgetPasswordForm = dynamic(() => import('./components/ForgetPasswordForm'));
 

src/pages/model/components/CreateModel.tsx

@@ -1,4 +1,4 @@
-import React, { Dispatch, useState, useCallback } from 'react';
+import React, { Dispatch, useState, useCallback, useMemo } from 'react';
 import {
   Modal,
   ModalOverlay,
@@ -12,12 +12,14 @@ import {
   Button,
   useToast,
   Input,
-  Select
+  Select,
+  Box
 } from '@chakra-ui/react';
 import { useForm } from 'react-hook-form';
 import { postCreateModel } from '@/api/model';
-import { ModelType } from '@/types/model';
-import { OpenAiList } from '@/constants/model';
+import type { ModelSchema } from '@/types/mongoSchema';
+import { modelList } from '@/constants/model';
+import { formatPrice } from '@/utils/user';
 
 interface CreateFormType {
   name: string;
@@ -29,20 +31,22 @@ const CreateModel = ({
   onSuccess
 }: {
   setCreateModelOpen: Dispatch<boolean>;
-  onSuccess: Dispatch<ModelType>;
+  onSuccess: Dispatch<ModelSchema>;
 }) => {
   const [requesting, setRequesting] = useState(false);
+  const [refresh, setRefresh] = useState(false);
   const toast = useToast({
     duration: 2000,
     position: 'top'
   });
   const {
+    getValues,
     register,
     handleSubmit,
     formState: { errors }
   } = useForm<CreateFormType>({
     defaultValues: {
-      serviceModelName: OpenAiList[0].model
+      serviceModelName: modelList[0].model
     }
   });
 
@@ -92,10 +96,13 @@ const CreateModel = ({
               <Select
                 placeholder="选择基础模型类型"
                 {...register('serviceModelName', {
-                  required: '底层模型不能为空'
+                  required: '底层模型不能为空',
+                  onChange() {
+                    setRefresh(!refresh);
+                  }
                 })}
               >
-                {OpenAiList.map((item) => (
+                {modelList.map((item) => (
                   <option key={item.model} value={item.model}>
                     {item.name}
                   </option>
@@ -105,6 +112,13 @@ const CreateModel = ({
                 {!!errors.serviceModelName && errors.serviceModelName.message}
               </FormErrorMessage>
             </FormControl>
+            <Box mt={3} textAlign={'center'} fontSize={'sm'} color={'blackAlpha.600'}>
+              {formatPrice(
+                modelList.find((item) => item.model === getValues('serviceModelName'))?.price || 0,
+                1000
+              )}
+              元/1K tokens(包括上下文和标点符号)
+            </Box>
           </ModalBody>
 
           <ModalFooter>

src/pages/model/components/ModelEditForm.tsx

@@ -1,86 +1,37 @@
-import React, { useCallback, useEffect, useRef } from 'react';
-import { Grid, Box, Card, Flex, Button, FormControl, Input, Textarea } from '@chakra-ui/react';
-import type { ModelType } from '@/types/model';
-import { useForm } from 'react-hook-form';
-import { useToast } from '@/hooks/useToast';
-import { putModelById } from '@/api/model';
-import { useScreen } from '@/hooks/useScreen';
-import { useGlobalStore } from '@/store/global';
+import React, { useState } from 'react';
+import {
+  Box,
+  Card,
+  Flex,
+  FormControl,
+  Input,
+  Textarea,
+  Slider,
+  SliderTrack,
+  SliderFilledTrack,
+  SliderThumb,
+  SliderMark,
+  Tooltip
+} from '@chakra-ui/react';
+import { QuestionOutlineIcon } from '@chakra-ui/icons';
+import type { ModelSchema } from '@/types/mongoSchema';
+import { UseFormReturn } from 'react-hook-form';
 
-const ModelEditForm = ({ model }: { model?: ModelType }) => {
-  const isInit = useRef(false);
-  const {
-    register,
-    handleSubmit,
-    reset,
-    formState: { errors }
-  } = useForm<ModelType>();
-  const { setLoading } = useGlobalStore();
-  const { toast } = useToast();
-  const { media } = useScreen();
-
-  const onclickSave = useCallback(
-    async (data: ModelType) => {
-      setLoading(true);
-      try {
-        await putModelById(data._id, {
-          name: data.name,
-          systemPrompt: data.systemPrompt,
-          service: data.service,
-          security: data.security
-        });
-        toast({
-          title: '更新成功',
-          status: 'success'
-        });
-      } catch (err) {
-        console.log('error->', err);
-        toast({
-          title: err as string,
-          status: 'success'
-        });
-      }
-      setLoading(false);
-    },
-    [setLoading, toast]
-  );
-  const submitError = useCallback(() => {
-    // deep search message
-    const deepSearch = (obj: any): string => {
-      if (!obj) return '提交表单错误';
-      if (!!obj.message) {
-        return obj.message;
-      }
-      return deepSearch(Object.values(obj)[0]);
-    };
-    toast({
-      title: deepSearch(errors),
-      status: 'error',
-      duration: 4000,
-      isClosable: true
-    });
-  }, [errors, toast]);
-
-  /* model 只会改变一次 */
-  useEffect(() => {
-    if (model && !isInit.current) {
-      reset(model);
-      isInit.current = true;
-    }
-  }, [model, reset]);
+const ModelEditForm = ({ formHooks }: { formHooks: UseFormReturn<ModelSchema> }) => {
+  const { register, setValue, getValues } = formHooks;
+  const [refresh, setRefresh] = useState(false);
 
   return (
-    <Grid gridTemplateColumns={media('1fr 1fr', '1fr')} gridGap={5}>
+    <>
       <Card p={4}>
         <Flex justifyContent={'space-between'} alignItems={'center'}>
-          <Box fontWeight={'bold'} fontSize={'lg'}>
-            修改模型信息
-          </Box>
-          <Button onClick={handleSubmit(onclickSave, submitError)}>保存</Button>
+          <Box fontWeight={'bold'}>基本信息</Box>
         </Flex>
-        <FormControl mt={5}>
+        <FormControl mt={4}>
           <Flex alignItems={'center'}>
-            <Box flex={'0 0 80px'}>展示名称:</Box>
+            <Box flex={'0 0 50px'} w={0}>
+              名称:
+            </Box>
             <Input
               {...register('name', {
                 required: '展示名称不能为空'
@@ -88,28 +39,79 @@ const ModelEditForm = ({ model }: { model?: ModelType }) => {
             ></Input>
           </Flex>
         </FormControl>
-        <FormControl mt={5}>
-          <Flex alignItems={'center'}>
-            <Box flex={'0 0 80px'}>对话模型:</Box>
-            <Box>{model?.service.modelName}</Box>
-          </Flex>
-        </FormControl>
-        <FormControl mt={5}>
+        <FormControl mt={4}>
+          <Box mb={1}>介绍:</Box>
           <Textarea
-            rows={4}
+            rows={5}
             maxLength={500}
-            {...register('systemPrompt')}
-            placeholder={'系统的提示词，会在进入聊天时放置在第一句，用于限定模型的聊天范围'}
+            {...register('intro')}
+            placeholder={'模型的介绍，仅做展示，不影响模型的效果'}
           />
         </FormControl>
       </Card>
       <Card p={4}>
-        <Box fontWeight={'bold'} fontSize={'lg'}>
-          安全策略
+        <Box fontWeight={'bold'}>模型效果</Box>
+        <FormControl mt={4}>
+          <Flex alignItems={'center'}>
+            <Box flex={'0 0 80px'} w={0}>
+              <Box as={'span'} mr={2}>
+                温度
+              </Box>
+              <Tooltip label={'温度越高，模型的发散能力越强；温度越低，内容越严谨。'}>
+                <QuestionOutlineIcon />
+              </Tooltip>
+            </Box>
+
+            <Slider
+              aria-label="slider-ex-1"
+              min={1}
+              max={10}
+              step={1}
+              value={getValues('temperature')}
+              onChange={(e) => {
+                setValue('temperature', e);
+                setRefresh(!refresh);
+              }}
+            >
+              <SliderMark
+                value={getValues('temperature')}
+                textAlign="center"
+                bg="blue.500"
+                color="white"
+                w={'18px'}
+                h={'18px'}
+                borderRadius={'100px'}
+                fontSize={'xs'}
+                transform={'translate(-50%, -200%)'}
+              >
+                {getValues('temperature')}
+              </SliderMark>
+              <SliderTrack>
+                <SliderFilledTrack />
+              </SliderTrack>
+              <SliderThumb />
+            </Slider>
+          </Flex>
+        </FormControl>
+        <Box mt={4}>
+          <Box mb={1}>系统提示词</Box>
+          <Textarea
+            rows={6}
+            maxLength={500}
+            {...register('systemPrompt')}
+            placeholder={
+              '模型默认的 prompt 词，通过调整该内容，可以生成一个限定范围的模型。\n\n注意，改功能会影响对话的整体朝向！'
+            }
+          />
         </Box>
+      </Card>
+      <Card p={4}>
+        <Box fontWeight={'bold'}>安全策略</Box>
         <FormControl mt={2}>
           <Flex alignItems={'center'}>
-            <Box flex={'0 0 120px'}>单句最大长度:</Box>
+            <Box flex={'0 0 120px'} w={0}>
+              单句最大长度:
+            </Box>
             <Input
               flex={1}
               type={'number'}
@@ -130,7 +132,9 @@ const ModelEditForm = ({ model }: { model?: ModelType }) => {
         </FormControl>
         <FormControl mt={5}>
           <Flex alignItems={'center'}>
-            <Box flex={'0 0 120px'}>上下文最大长度:</Box>
+            <Box flex={'0 0 120px'} w={0}>
+              上下文最大长度:
+            </Box>
             <Input
               flex={1}
               type={'number'}
@@ -151,7 +155,9 @@ const ModelEditForm = ({ model }: { model?: ModelType }) => {
         </FormControl>
         <FormControl mt={5}>
           <Flex alignItems={'center'}>
-            <Box flex={'0 0 120px'}>聊天过期时间:</Box>
+            <Box flex={'0 0 120px'} w={0}>
+              聊天过期时间:
+            </Box>
             <Input
               flex={1}
               type={'number'}
@@ -173,7 +179,9 @@ const ModelEditForm = ({ model }: { model?: ModelType }) => {
         </FormControl>
         <FormControl mt={5} pb={5}>
           <Flex alignItems={'center'}>
-            <Box flex={'0 0 130px'}>聊天最大加载次数:</Box>
+            <Box flex={'0 0 130px'} w={0}>
+              聊天最大加载次数:
+            </Box>
             <Box flex={1}>
               <Input
                 type={'number'}
@@ -194,7 +202,7 @@ const ModelEditForm = ({ model }: { model?: ModelType }) => {
           </Flex>
         </FormControl>
       </Card>
-    </Grid>
+    </>
   );
 };
 

src/pages/model/components/ModelPhoneList.tsx

@@ -1,6 +1,6 @@
 import React, { useEffect } from 'react';
 import { Box, Button, Flex, Tag } from '@chakra-ui/react';
-import type { ModelType } from '@/types/model';
+import type { ModelSchema } from '@/types/mongoSchema';
 import { formatModelStatus } from '@/constants/model';
 import dayjs from 'dayjs';
 import { useRouter } from 'next/router';
@@ -9,7 +9,7 @@ const ModelPhoneList = ({
   models,
   handlePreviewChat
 }: {
-  models: ModelType[];
+  models: ModelSchema[];
   handlePreviewChat: (_: string) => void;
 }) => {
   const router = useRouter();

src/pages/model/components/ModelTable.tsx

@@ -14,14 +14,14 @@ import {
 } from '@chakra-ui/react';
 import { formatModelStatus } from '@/constants/model';
 import dayjs from 'dayjs';
-import type { ModelType } from '@/types/model';
+import type { ModelSchema } from '@/types/mongoSchema';
 import { useRouter } from 'next/router';
 
 const ModelTable = ({
   models = [],
   handlePreviewChat
 }: {
-  models: ModelType[];
+  models: ModelSchema[];
   handlePreviewChat: (_: string) => void;
 }) => {
   const router = useRouter();
@@ -34,13 +34,13 @@ const ModelTable = ({
     {
       title: '最后更新时间',
       key: 'updateTime',
-      render: (item: ModelType) => dayjs(item.updateTime).format('YYYY-MM-DD HH:mm')
+      render: (item: ModelSchema) => dayjs(item.updateTime).format('YYYY-MM-DD HH:mm')
     },
     {
       title: '状态',
       key: 'status',
       dataIndex: 'status',
-      render: (item: ModelType) => (
+      render: (item: ModelSchema) => (
         <Tag
           colorScheme={formatModelStatus[item.status].colorTheme}
           variant="solid"
@@ -54,7 +54,7 @@ const ModelTable = ({
     {
       title: 'AI模型',
       key: 'service',
-      render: (item: ModelType) => (
+      render: (item: ModelSchema) => (
         <Box wordBreak={'break-all'} whiteSpace={'pre-wrap'} maxW={'200px'}>
           {item.service.modelName}
         </Box>
@@ -68,7 +68,7 @@ const ModelTable = ({
     {
       title: '操作',
       key: 'control',
-      render: (item: ModelType) => (
+      render: (item: ModelSchema) => (
         <>
           <Button mr={3} onClick={() => handlePreviewChat(item._id)}>
             对话

src/pages/model/components/Training.tsx

@@ -1,10 +1,10 @@
 import React, { useEffect, useCallback, useState } from 'react';
 import { Box, TableContainer, Table, Thead, Tbody, Tr, Th, Td } from '@chakra-ui/react';
-import { ModelType } from '@/types/model';
+import type { ModelSchema } from '@/types/mongoSchema';
 import { getModelTrainings } from '@/api/model';
 import type { TrainingItemType } from '@/types/training';
 
-const Training = ({ model }: { model: ModelType }) => {
+const Training = ({ model }: { model: ModelSchema }) => {
   const columns: {
     title: string;
     key: keyof TrainingItemType;

src/pages/model/detail.tsx

@@ -1,21 +1,29 @@
-import React, { useCallback, useState, useEffect, useRef, useMemo } from 'react';
+import React, { useCallback, useState, useRef, useMemo, useEffect } from 'react';
 import { useRouter } from 'next/router';
-import { getModelById, delModelById, postTrainModel, putModelTrainingStatus } from '@/api/model';
+import {
+  getModelById,
+  delModelById,
+  postTrainModel,
+  putModelTrainingStatus,
+  putModelById
+} from '@/api/model';
 import { getChatSiteId } from '@/api/chat';
-import type { ModelType } from '@/types/model';
+import type { ModelSchema } from '@/types/mongoSchema';
 import { Card, Box, Flex, Button, Tag, Grid } from '@chakra-ui/react';
 import { useToast } from '@/hooks/useToast';
 import { useConfirm } from '@/hooks/useConfirm';
-import { formatModelStatus, ModelStatusEnum, OpenAiList } from '@/constants/model';
+import { useForm } from 'react-hook-form';
+import { formatModelStatus, ModelStatusEnum, modelList, defaultModel } from '@/constants/model';
 import { useGlobalStore } from '@/store/global';
 import { useScreen } from '@/hooks/useScreen';
 import ModelEditForm from './components/ModelEditForm';
-import Icon from '@/components/Icon';
+import Icon from '@/components/Iconfont';
+import { useQuery } from '@tanstack/react-query';
 import dynamic from 'next/dynamic';
 
 const Training = dynamic(() => import('./components/Training'));
 
-const ModelDetail = () => {
+const ModelDetail = ({ modelId }: { modelId: string }) => {
   const { toast } = useToast();
   const router = useRouter();
   const { isPc, media } = useScreen();
@@ -24,33 +32,33 @@ const ModelDetail = () => {
     content: '确认删除该模型?'
   });
   const SelectFileDom = useRef<HTMLInputElement>(null);
-
-  const { modelId } = router.query as { modelId: string };
-  const [model, setModel] = useState<ModelType>();
+  const [model, setModel] = useState<ModelSchema>(defaultModel);
+  const formHooks = useForm<ModelSchema>({
+    defaultValues: model
+  });
 
   const canTrain = useMemo(() => {
-    const openai = OpenAiList.find((item) => item.model === model?.service.modelName);
-    return openai && openai.canTraining === true;
+    const openai = modelList.find((item) => item.model === model?.service.modelName);
+    return openai && openai.trainName;
   }, [model]);
 
   /* 加载模型数据 */
   const loadModel = useCallback(async () => {
-    if (!modelId) return;
     setLoading(true);
     try {
-      const res = await getModelById(modelId as string);
+      const res = await getModelById(modelId);
+      console.log(res);
       res.security.expiredTime /= 60 * 60 * 1000;
       setModel(res);
+      formHooks.reset(res);
     } catch (err) {
       console.log('error->', err);
     }
     setLoading(false);
-  }, [modelId, setLoading]);
+    return null;
+  }, [formHooks, modelId, setLoading]);
 
-  useEffect(() => {
-    loadModel();
-    router.prefetch('/chat');
-  }, [loadModel, modelId, router]);
+  useQuery([modelId], loadModel);
 
   /* 点击删除 */
   const handleDelModel = useCallback(async () => {
@@ -71,7 +79,6 @@ const ModelDetail = () => {
 
   /* 点前往聊天预览页 */
   const handlePreviewChat = useCallback(async () => {
-    if (!model) return;
     setLoading(true);
     try {
       const chatId = await getChatSiteId(model._id);
@@ -131,6 +138,65 @@ const ModelDetail = () => {
     setLoading(false);
   }, [model, setLoading, loadModel, toast]);
 
+  // 提交保存模型修改
+  const saveSubmitSuccess = useCallback(
+    async (data: ModelSchema) => {
+      setLoading(true);
+      try {
+        await putModelById(data._id, {
+          name: data.name,
+          systemPrompt: data.systemPrompt,
+          intro: data.intro,
+          temperature: data.temperature,
+          service: data.service,
+          security: data.security
+        });
+        toast({
+          title: '更新成功',
+          status: 'success'
+        });
+      } catch (err) {
+        console.log('error->', err);
+        toast({
+          title: err as string,
+          status: 'success'
+        });
+      }
+      setLoading(false);
+    },
+    [setLoading, toast]
+  );
+  // 提交保存表单失败
+  const saveSubmitError = useCallback(() => {
+    // deep search message
+    const deepSearch = (obj: any): string => {
+      if (!obj) return '提交表单错误';
+      if (!!obj.message) {
+        return obj.message;
+      }
+      return deepSearch(Object.values(obj)[0]);
+    };
+    toast({
+      title: deepSearch(formHooks.formState.errors),
+      status: 'error',
+      duration: 4000,
+      isClosable: true
+    });
+  }, [formHooks.formState.errors, toast]);
+
+  useEffect(() => {
+    router.prefetch('/chat');
+
+    window.onbeforeunload = (e) => {
+      e.preventDefault();
+      e.returnValue = '内容已修改，确认离开页面吗？';
+    };
+
+    return () => {
+      window.onbeforeunload = null;
+    };
+  }, [router]);
+
   return (
     <>
       {/* 头部 */}
@@ -138,51 +204,55 @@ const ModelDetail = () => {
         {isPc ? (
           <Flex alignItems={'center'}>
             <Box fontSize={'xl'} fontWeight={'bold'}>
-              {model?.name || '模型'} 配置
+              {model.name}
             </Box>
-            {!!model && (
-              <Tag
-                ml={2}
-                variant="solid"
-                colorScheme={formatModelStatus[model.status].colorTheme}
-                cursor={model.status === ModelStatusEnum.training ? 'pointer' : 'default'}
-                onClick={handleClickUpdateStatus}
-              >
-                {formatModelStatus[model.status].text}
-              </Tag>
-            )}
+            <Tag
+              ml={2}
+              variant="solid"
+              colorScheme={formatModelStatus[model.status].colorTheme}
+              cursor={model.status === ModelStatusEnum.training ? 'pointer' : 'default'}
+              onClick={handleClickUpdateStatus}
+            >
+              {formatModelStatus[model.status].text}
+            </Tag>
             <Box flex={1} />
             <Button variant={'outline'} onClick={handlePreviewChat}>
               对话体验
             </Button>
+            <Button ml={4} onClick={formHooks.handleSubmit(saveSubmitSuccess, saveSubmitError)}>
+              保存修改
+            </Button>
           </Flex>
         ) : (
           <>
             <Flex alignItems={'center'}>
               <Box as={'h3'} fontSize={'xl'} fontWeight={'bold'} flex={1}>
-                {model?.name || '模型'} 配置
+                {model?.name}
               </Box>
-              {!!model && (
-                <Tag ml={2} colorScheme={formatModelStatus[model.status].colorTheme}>
-                  {formatModelStatus[model.status].text}
-                </Tag>
-              )}
+              <Tag ml={2} colorScheme={formatModelStatus[model.status].colorTheme}>
+                {formatModelStatus[model.status].text}
+              </Tag>
             </Flex>
             <Box mt={4} textAlign={'right'}>
               <Button variant={'outline'} onClick={handlePreviewChat}>
                 对话体验
               </Button>
+              <Button ml={4} onClick={formHooks.handleSubmit(saveSubmitSuccess, saveSubmitError)}>
+                保存修改
+              </Button>
             </Box>
           </>
         )}
       </Card>
-      {/* 基本信息编辑 */}
-      <Box mt={5}>
-        <ModelEditForm model={model} />
-      </Box>
-      {/* 其他配置 */}
       <Grid mt={5} gridTemplateColumns={media('1fr 1fr', '1fr')} gridGap={5}>
-        <Card p={4}>{!!model && <Training model={model} />}</Card>
+        <ModelEditForm formHooks={formHooks} />
+
+        {canTrain && (
+          <Card p={4}>
+            <Training model={model} />
+          </Card>
+        )}
+
         <Card p={4}>
           <Box fontWeight={'bold'} fontSize={'lg'}>
             神奇操作
@@ -194,10 +264,10 @@ const ModelDetail = () => {
               onClick={() => {
                 SelectFileDom.current?.click();
               }}
-              title={!canTrain ? '' : '模型不支持微调'}
+              title={!canTrain ? '模型不支持微调' : ''}
               isDisabled={!canTrain}
             >
-              上传微调数据集
+              上传数据集
             </Button>
             <Flex
               as={'a'}
@@ -213,34 +283,42 @@ const ModelDetail = () => {
             </Flex>
           </Flex>
           {/* 提示 */}
-          <Box mt={3} py={3} color={'blackAlpha.500'}>
+          <Box mt={3} py={3} color={'blackAlpha.600'}>
+            <Box as={'li'} lineHeight={1.9}>
+              可以使用
+              <Box
+                as={'span'}
+                fontWeight={'bold'}
+                textDecoration={'underline'}
+                color={'blackAlpha.800'}
+                mx={2}
+                cursor={'pointer'}
+                onClick={() => router.push('/data/list')}
+              >
+                数据拆分
+              </Box>
+              功能，从任意文本中提取数据集。
+            </Box>
             <Box as={'li'} lineHeight={1.9}>
               每行包括一个 prompt 和一个 completion
             </Box>
             <Box as={'li'} lineHeight={1.9}>
-              prompt 必须以 \n\n###\n\n 结尾，且尽量保障每个 prompt
-              内容不都是同一个标点结尾，可以加一个空格打断相同性，
+              prompt 必须以 {'</s>'} 结尾
             </Box>
             <Box as={'li'} lineHeight={1.9}>
-              completion 开头必须有一个空格，末尾必须以 ### 结尾，同样的不要都是同一个标点结尾。
+              completion 开头必须有一个空格，必须以 {'</s>'} 结尾
             </Box>
           </Box>
           <Flex mt={5} alignItems={'center'}>
             <Box flex={'0 0 80px'}>删除模型:</Box>
-            <Button
-              colorScheme={'red'}
-              size={'sm'}
-              onClick={() => {
-                openConfirm(() => {
-                  handleDelModel();
-                });
-              }}
-            >
+            <Button colorScheme={'red'} size={'sm'} onClick={openConfirm(handleDelModel)}>
               删除模型
             </Button>
           </Flex>
         </Card>
       </Grid>
+
+      {/* 文件选择 */}
       <Box position={'absolute'} w={0} h={0} overflow={'hidden'}>
         <input ref={SelectFileDom} type="file" accept=".jsonl" onChange={startTraining} />
       </Box>
@@ -250,3 +328,11 @@ const ModelDetail = () => {
 };
 
 export default ModelDetail;
+
+export async function getServerSideProps(context: any) {
+  const modelId = context.query?.modelId || '';
+
+  return {
+    props: { modelId }
+  };
+}

src/pages/model/list.tsx

@@ -1,8 +1,7 @@
 import React, { useState, useCallback } from 'react';
 import { Box, Button, Flex, Card } from '@chakra-ui/react';
-import { getMyModels } from '@/api/model';
 import { getChatSiteId } from '@/api/chat';
-import { ModelType } from '@/types/model';
+import type { ModelSchema } from '@/types/mongoSchema';
 import { useRouter } from 'next/router';
 import ModelTable from './components/ModelTable';
 import ModelPhoneList from './components/ModelPhoneList';
@@ -11,29 +10,28 @@ import { useQuery } from '@tanstack/react-query';
 import { useLoading } from '@/hooks/useLoading';
 import dynamic from 'next/dynamic';
 import { useToast } from '@/hooks/useToast';
+import { useUserStore } from '@/store/user';
 
 const CreateModel = dynamic(() => import('./components/CreateModel'));
 
-const ModelList = () => {
+const modelList = () => {
   const { toast } = useToast();
   const { isPc } = useScreen();
   const router = useRouter();
-  const [models, setModels] = useState<ModelType[]>([]);
+  const { myModels, setMyModels, getMyModels } = useUserStore();
   const [openCreateModel, setOpenCreateModel] = useState(false);
   const { Loading, setIsLoading } = useLoading();
 
   /* 加载模型 */
-  const { isLoading } = useQuery(['loadModels'], () => getMyModels(), {
-    onSuccess(res) {
-      if (!res) return;
-      setModels(res);
-    }
-  });
+  const { isLoading } = useQuery(['loadModels'], getMyModels);
 
   /* 创建成功回调 */
-  const createModelSuccess = useCallback((data: ModelType) => {
-    setModels((state) => [data, ...state]);
-  }, []);
+  const createModelSuccess = useCallback(
+    (data: ModelSchema) => {
+      setMyModels([data, ...myModels]);
+    },
+    [myModels, setMyModels]
+  );
 
   /* 点前往聊天预览页 */
   const handlePreviewChat = useCallback(
@@ -74,9 +72,9 @@ const ModelList = () => {
       {/* 表单 */}
       <Box mt={5} position={'relative'}>
         {isPc ? (
-          <ModelTable models={models} handlePreviewChat={handlePreviewChat} />
+          <ModelTable models={myModels} handlePreviewChat={handlePreviewChat} />
         ) : (
-          <ModelPhoneList models={models} handlePreviewChat={handlePreviewChat} />
+          <ModelPhoneList models={myModels} handlePreviewChat={handlePreviewChat} />
         )}
       </Box>
       {/* 创建弹窗 */}
@@ -89,4 +87,4 @@ const ModelList = () => {
   );
 };
 
-export default ModelList;
+export default modelList;

src/pages/number/components/PayModal.tsx

@@ -0,0 +1,170 @@
+import React, { useState, useCallback } from 'react';
+import {
+  Modal,
+  ModalOverlay,
+  ModalContent,
+  ModalHeader,
+  ModalFooter,
+  ModalBody,
+  ModalCloseButton,
+  Button,
+  Input,
+  Box,
+  Grid,
+  Table,
+  Thead,
+  Tbody,
+  Tr,
+  Th,
+  Td,
+  TableContainer
+} from '@chakra-ui/react';
+import { getPayCode, checkPayResult } from '@/api/user';
+import { useToast } from '@/hooks/useToast';
+import { useQuery } from '@tanstack/react-query';
+import { useRouter } from 'next/router';
+import { modelList } from '@/constants/model';
+import { formatPrice } from '../../../utils/user';
+
+const PayModal = ({ onClose }: { onClose: () => void }) => {
+  const router = useRouter();
+  const { toast } = useToast();
+  const [inputVal, setInputVal] = useState<number | ''>('');
+  const [loading, setLoading] = useState(false);
+  const [payId, setPayId] = useState('');
+
+  const handleClickPay = useCallback(async () => {
+    if (!inputVal || inputVal <= 0 || isNaN(+inputVal)) return;
+    setLoading(true);
+    try {
+      // 获取支付二维码
+      const res = await getPayCode(inputVal);
+      new QRCode(document.getElementById('payQRCode'), {
+        text: res.codeUrl,
+        width: 128,
+        height: 128,
+        colorDark: '#000000',
+        colorLight: '#ffffff',
+        correctLevel: QRCode.CorrectLevel.H
+      });
+      setPayId(res.payId);
+    } catch (error) {
+      toast({
+        title: '出现了一些意外...',
+        status: 'error'
+      });
+      console.log(error);
+    }
+    setLoading(false);
+  }, [inputVal, toast]);
+
+  useQuery(
+    [payId],
+    () => {
+      if (!payId) return null;
+      return checkPayResult(payId);
+    },
+    {
+      refetchInterval: 2000,
+      onSuccess(res) {
+        if (!res) return;
+        toast({
+          title: '充值成功',
+          status: 'success'
+        });
+        router.reload();
+      }
+    }
+  );
+
+  return (
+    <>
+      <Modal
+        isOpen={true}
+        onClose={() => {
+          if (payId) return;
+          onClose();
+        }}
+      >
+        <ModalOverlay />
+        <ModalContent>
+          <ModalHeader>充值</ModalHeader>
+          {!payId && <ModalCloseButton />}
+
+          <ModalBody py={0}>
+            {!payId && (
+              <>
+                {/* 价格表 */}
+                <TableContainer mb={4}>
+                  <Table>
+                    <Thead>
+                      <Tr>
+                        <Th>模型类型</Th>
+                        <Th>价格(元/1K tokens，包含所有上下文)</Th>
+                      </Tr>
+                    </Thead>
+                    <Tbody>
+                      {modelList.map((item, i) => (
+                        <Tr key={item.model}>
+                          <Td>{item.name}</Td>
+                          <Td>{formatPrice(item.price, 1000)}</Td>
+                        </Tr>
+                      ))}
+                    </Tbody>
+                  </Table>
+                </TableContainer>
+                <Grid gridTemplateColumns={'repeat(4,1fr)'} gridGap={5} mb={4}>
+                  {[5, 10, 20, 50].map((item) => (
+                    <Button
+                      key={item}
+                      variant={item === inputVal ? 'solid' : 'outline'}
+                      onClick={() => setInputVal(item)}
+                    >
+                      {item}元
+                    </Button>
+                  ))}
+                </Grid>
+                <Box>
+                  <Input
+                    value={inputVal}
+                    type={'number'}
+                    step={1}
+                    placeholder={'其他金额，请取整数'}
+                    onChange={(e) => {
+                      setInputVal(Math.floor(+e.target.value));
+                    }}
+                  ></Input>
+                </Box>
+              </>
+            )}
+            {/* 付费二维码 */}
+            <Box textAlign={'center'}>
+              {payId && <Box mb={3}>请微信扫码支付: {inputVal}元，请勿关闭页面</Box>}
+              <Box id={'payQRCode'} display={'inline-block'}></Box>
+            </Box>
+          </ModalBody>
+
+          <ModalFooter>
+            {!payId && (
+              <>
+                <Button colorScheme={'gray'} onClick={onClose}>
+                  取消
+                </Button>
+                <Button
+                  ml={3}
+                  isLoading={loading}
+                  isDisabled={!inputVal || inputVal === 0}
+                  onClick={handleClickPay}
+                >
+                  获取充值二维码
+                </Button>
+              </>
+            )}
+          </ModalFooter>
+        </ModalContent>
+      </Modal>
+    </>
+  );
+};
+
+export default PayModal;

src/pages/number/setting.tsx

@@ -1,4 +1,4 @@
-import React, { useCallback } from 'react';
+import React, { useCallback, useState } from 'react';
 import {
   Card,
   Box,
@@ -13,23 +13,38 @@ import {
   TableContainer,
   Select,
   Input,
-  IconButton
+  IconButton,
+  useDisclosure
 } from '@chakra-ui/react';
 import { DeleteIcon } from '@chakra-ui/icons';
 import { useForm, useFieldArray } from 'react-hook-form';
 import { UserUpdateParams } from '@/types/user';
-import { putUserInfo } from '@/api/user';
+import { putUserInfo, getUserBills, getPayOrders, checkPayResult } from '@/api/user';
 import { useToast } from '@/hooks/useToast';
 import { useGlobalStore } from '@/store/global';
 import { useUserStore } from '@/store/user';
 import { UserType } from '@/types/user';
+import { usePaging } from '@/hooks/usePaging';
+import type { UserBillType } from '@/types/user';
+import { useQuery } from '@tanstack/react-query';
+import dynamic from 'next/dynamic';
+import { PaySchema } from '@/types/mongoSchema';
+import dayjs from 'dayjs';
+import { formatPrice } from '@/utils/user';
+import WxConcat from '@/components/WxConcat';
+import ScrollData from '@/components/ScrollData';
+import { BillTypeMap } from '@/constants/user';
+
+const PayModal = dynamic(() => import('./components/PayModal'));
 
 const NumberSetting = () => {
-  const { userInfo, updateUserInfo } = useUserStore();
+  const { userInfo, updateUserInfo, initUserInfo } = useUserStore();
   const { setLoading } = useGlobalStore();
   const { register, handleSubmit, control } = useForm<UserUpdateParams>({
     defaultValues: userInfo as UserType
   });
+  const [showPay, setShowPay] = useState(false);
+  const { isOpen: isOpenWx, onOpen: onOpenWx, onClose: onCloseWx } = useDisclosure();
   const { toast } = useToast();
   const {
     fields: accounts,
@@ -39,6 +54,16 @@ const NumberSetting = () => {
     control,
     name: 'accounts'
   });
+  const {
+    nextPage,
+    isLoadAll,
+    requesting,
+    data: bills
+  } = usePaging<UserBillType>({
+    api: getUserBills,
+    pageSize: 30
+  });
+  const [payOrders, setPayOrders] = useState<PaySchema[]>([]);
 
   const onclickSave = useCallback(
     async (data: UserUpdateParams) => {
@@ -56,6 +81,39 @@ const NumberSetting = () => {
     [setLoading, toast, updateUserInfo]
   );
 
+  useQuery(['init'], initUserInfo);
+
+  useQuery(['initPayOrder'], getPayOrders, {
+    onSuccess(res) {
+      setPayOrders(res);
+    }
+  });
+
+  const handleRefreshPayOrder = useCallback(
+    async (payId: string) => {
+      setLoading(true);
+
+      try {
+        const data = await checkPayResult(payId);
+        toast({
+          title: data,
+          status: 'info'
+        });
+        const res = await getPayOrders();
+        setPayOrders(res);
+      } catch (error: any) {
+        toast({
+          title: error?.message,
+          status: 'warning'
+        });
+        console.log(error);
+      }
+
+      setLoading(false);
+    },
+    [setLoading, toast]
+  );
+
   return (
     <>
       <Card px={6} py={4}>
@@ -68,17 +126,20 @@ const NumberSetting = () => {
             <Box>{userInfo?.email}</Box>
           </Flex>
         </Box>
-        {/* <Box mt={6}>
+        <Box mt={6}>
           <Flex alignItems={'center'}>
             <Box flex={'0 0 60px'}>余额:</Box>
             <Box>
               <strong>{userInfo?.balance}</strong> 元
             </Box>
-            <Button size={'sm'} w={'80px'} ml={5}>
+            <Button size={'sm'} w={'80px'} ml={5} onClick={() => setShowPay(true)}>
               充值
             </Button>
           </Flex>
-        </Box> */}
+          <Box fontSize={'xs'} color={'blackAlpha.500'}>
+            如果填写了自己的 openai 账号，将不会计费
+          </Box>
+        </Box>
       </Card>
       <Card mt={6} px={6} py={4}>
         <Flex mb={5} justifyContent={'space-between'}>
@@ -148,6 +209,88 @@ const NumberSetting = () => {
           </Table>
         </TableContainer>
       </Card>
+      <Card mt={6} py={4}>
+        <Flex alignItems={'flex-end'} px={6} mb={1}>
+          <Box fontSize={'xl'} fontWeight={'bold'}>
+            充值记录
+          </Box>
+          <Button onClick={onOpenWx} size={'xs'} ml={4} variant={'outline'}>
+            异常问题，wx联系
+          </Button>
+        </Flex>
+        <TableContainer maxH={'400px'} overflowY={'auto'} px={6}>
+          <Table>
+            <Thead>
+              <Tr>
+                <Th>订单号</Th>
+                <Th>时间</Th>
+                <Th>金额</Th>
+                <Th>消费</Th>
+                <Th></Th>
+              </Tr>
+            </Thead>
+            <Tbody fontSize={'sm'}>
+              {payOrders.map((item) => (
+                <Tr key={item._id}>
+                  <Td>{item.orderId}</Td>
+                  <Td>
+                    {item.createTime ? dayjs(item.createTime).format('YYYY/MM/DD HH:mm:ss') : '-'}
+                  </Td>
+                  <Td>{formatPrice(item.price)}元</Td>
+                  <Td>{item.status}</Td>
+                  <Td>
+                    {item.status === 'NOTPAY' && (
+                      <Button onClick={() => handleRefreshPayOrder(item._id)} size={'sm'}>
+                        更新
+                      </Button>
+                    )}
+                  </Td>
+                </Tr>
+              ))}
+            </Tbody>
+          </Table>
+        </TableContainer>
+      </Card>
+      <Card mt={6} py={4}>
+        <Box fontSize={'xl'} fontWeight={'bold'} px={6} mb={1}>
+          使用记录
+        </Box>
+        <ScrollData
+          maxH={'400px'}
+          px={6}
+          isLoadAll={isLoadAll}
+          requesting={requesting}
+          nextPage={nextPage}
+        >
+          <TableContainer>
+            <Table>
+              <Thead>
+                <Tr>
+                  <Th>时间</Th>
+                  <Th>类型</Th>
+                  <Th>内容长度</Th>
+                  <Th>Tokens 长度</Th>
+                  <Th>消费</Th>
+                </Tr>
+              </Thead>
+              <Tbody fontSize={'sm'}>
+                {bills.map((item) => (
+                  <Tr key={item.id}>
+                    <Td>{item.time}</Td>
+                    <Td>{BillTypeMap[item.type]}</Td>
+                    <Td>{item.textLen}</Td>
+                    <Td>{item.tokenLen}</Td>
+                    <Td>{item.price}元</Td>
+                  </Tr>
+                ))}
+              </Tbody>
+            </Table>
+          </TableContainer>
+        </ScrollData>
+      </Card>
+      {showPay && <PayModal onClose={() => setShowPay(false)} />}
+      {/* wx 联系 */}
+      {isOpenWx && <WxConcat onClose={onCloseWx} />}
     </>
   );
 };

src/pages/training/dataList.tsx

@@ -1,23 +0,0 @@
-import React from 'react';
-import { Card, Box, Flex, Button } from '@chakra-ui/react';
-
-const TrainDataList = () => {
-  return (
-    <>
-      <Card px={6} py={4}>
-        <Flex alignItems={'center'}>
-          <Box fontSize={'xl'} fontWeight={'bold'} flex={1}>
-            训练数据管理
-          </Box>
-          <Button variant={'outline'} mr={6}>
-            导入数据
-          </Button>
-          <Button>插入一条数据</Button>
-        </Flex>
-      </Card>
-      {/* 数据表 */}
-    </>
-  );
-};
-
-export default TrainDataList;

src/service/errorCode.ts

@@ -2,7 +2,8 @@ export const openaiError: Record<string, string> = {
   context_length_exceeded: '内容超长了，请重置对话',
   Unauthorized: 'API-KEY 不合法',
   rate_limit_reached: '同时访问用户过多，请稍后再试',
-  'Bad Request': '上下文太多了，请重开对话~'
+  'Bad Request': 'Bad Request~ 可能内容太多了',
+  'Too Many Requests': '请求次数太多了，请慢点~'
 };
 export const proxyError: Record<string, boolean> = {
   ECONNABORTED: true,

src/service/events/generateQA.ts

@@ -0,0 +1,139 @@
+import { DataItem } from '@/service/mongo';
+import { getOpenAIApi } from '@/service/utils/chat';
+import { httpsAgent, getOpenApiKey } from '@/service/utils/tools';
+import type { ChatCompletionRequestMessage } from 'openai';
+import { DataItemSchema } from '@/types/mongoSchema';
+import { ChatModelNameEnum } from '@/constants/model';
+import { pushSplitDataBill } from '@/service/events/pushBill';
+
+export async function generateQA(next = false): Promise<any> {
+  if (global.generatingQA && !next) return;
+  global.generatingQA = true;
+
+  const systemPrompt: ChatCompletionRequestMessage = {
+    role: 'system',
+    content: `总结助手。我会向你发送一段长文本,请从中总结出5至15个问题和答案,答案请尽量详细,请按以下格式返回: "Q1:"\n"A1:"\n"Q2:"\n"A2:"\n`
+  };
+  let dataItem: DataItemSchema | null = null;
+
+  try {
+    // 找出一个需要生成的 dataItem
+    dataItem = await DataItem.findOne({
+      status: { $ne: 0 },
+      times: { $gt: 0 }
+    });
+
+    if (!dataItem) {
+      console.log('没有需要生成 QA 的数据');
+      global.generatingQA = false;
+      return;
+    }
+
+    // 更新状态为生成中
+    await DataItem.findByIdAndUpdate(dataItem._id, {
+      status: 2
+    });
+
+    // 获取 openapi Key
+    let userApiKey, systemKey;
+    try {
+      const key = await getOpenApiKey(dataItem.userId);
+      userApiKey = key.userApiKey;
+      systemKey = key.systemKey;
+    } catch (error) {
+      // 余额不够了, 把用户所有记录改成闲置
+      await DataItem.updateMany({
+        userId: dataItem.userId,
+        status: 0
+      });
+      throw new Error('获取 openai key 失败');
+    }
+
+    console.log('正在生成一个QA, ID:', dataItem._id, 'temperature: ', dataItem.temperature / 100);
+
+    const startTime = Date.now();
+
+    // 获取 openai 请求实例
+    const chatAPI = getOpenAIApi(userApiKey || systemKey);
+    // 请求 chatgpt 获取回答
+    const response = await chatAPI.createChatCompletion(
+      {
+        model: ChatModelNameEnum.GPT35,
+        temperature: dataItem.temperature / 100,
+        n: 1,
+        messages: [
+          systemPrompt,
+          {
+            role: 'user',
+            content: dataItem.text
+          }
+        ]
+      },
+      {
+        timeout: 120000,
+        httpsAgent
+      }
+    );
+    const content = response.data.choices[0].message?.content;
+    // 从 content 中提取 QA
+    const splitResponse = splitText(content || '');
+    // 插入数据库，并修改状态
+    await DataItem.findByIdAndUpdate(dataItem._id, {
+      status: dataItem.temperature >= 90 ? 0 : 1, // 需要生成 4 组内容。0,0.3,0.6,0.9
+      temperature: dataItem.temperature >= 90 ? dataItem.temperature : dataItem.temperature + 30,
+      $push: {
+        rawResponse: content,
+        result: {
+          $each: splitResponse
+        }
+      }
+    });
+    // 计费
+    !userApiKey &&
+      splitResponse.length > 0 &&
+      pushSplitDataBill({
+        userId: dataItem.userId,
+        text: systemPrompt.content + dataItem.text + content
+      });
+    console.log(
+      '生成QA成功，time:',
+      `${(Date.now() - startTime) / 1000}s`,
+      'QA数量：',
+      splitResponse.length
+    );
+  } catch (error: any) {
+    console.log('error: 生成QA错误', dataItem?._id);
+    console.log('response:', error?.response);
+    // 重置状态
+    if (dataItem?._id) {
+      await DataItem.findByIdAndUpdate(dataItem._id, {
+        status: dataItem.times > 0 ? 1 : 0, // 还有重试次数则可以继续进行
+        $inc: {
+          // 剩余尝试次数-1
+          times: -1
+        }
+      });
+    }
+  }
+
+  generateQA(true);
+}
+
+/**
+ * 检查文本是否按格式返回
+ */
+function splitText(text: string) {
+  const regex = /Q\d+:(\s*)(.*)(\s*)A\d+:(\s*)(.*)(\s*)/g; // 匹配Q和A的正则表达式
+  const matches = text.matchAll(regex); // 获取所有匹配到的结果
+
+  const result = []; // 存储最终的结果
+  for (const match of matches) {
+    const q = match[2];
+    const a = match[5];
+    if (q && a) {
+      result.push({ q, a }); // 如果Q和A都存在，就将其添加到结果中
+    }
+  }
+
+  return result;
+}

src/service/events/pushBill.ts

@@ -0,0 +1,105 @@
+import { connectToDatabase, Bill, User } from '../mongo';
+import { modelList, ChatModelNameEnum } from '@/constants/model';
+import { encode } from 'gpt-token-utils';
+import { formatPrice } from '@/utils/user';
+
+export const pushChatBill = async ({
+  modelName,
+  userId,
+  chatId,
+  text
+}: {
+  modelName: string;
+  userId: string;
+  chatId: string;
+  text: string;
+}) => {
+  await connectToDatabase();
+
+  let billId;
+
+  try {
+    // 获取模型单价格
+    const modelItem = modelList.find((item) => item.model === modelName);
+    const unitPrice = modelItem?.price || 5;
+
+    // 计算 token 数量
+    const tokens = encode(text);
+
+    // 计算价格
+    const price = unitPrice * tokens.length;
+    console.log('chat bill');
+    console.log('token len:', tokens.length);
+    console.log('text len: ', text.length);
+    console.log('price: ', `${formatPrice(price)}元`);
+
+    try {
+      // 插入 Bill 记录
+      const res = await Bill.create({
+        userId,
+        type: 'chat',
+        modelName,
+        chatId,
+        textLen: text.length,
+        tokenLen: tokens.length,
+        price
+      });
+      billId = res._id;
+
+      // 账号扣费
+      await User.findByIdAndUpdate(userId, {
+        $inc: { balance: -price }
+      });
+    } catch (error) {
+      console.log('创建账单失败:', error);
+      billId && Bill.findByIdAndDelete(billId);
+    }
+  } catch (error) {
+    console.log(error);
+  }
+};
+
+export const pushSplitDataBill = async ({ userId, text }: { userId: string; text: string }) => {
+  await connectToDatabase();
+
+  let billId;
+
+  try {
+    // 获取模型单价格, 都是用 gpt35 拆分
+    const modelItem = modelList.find((item) => item.model === ChatModelNameEnum.GPT35);
+    const unitPrice = modelItem?.price || 5;
+
+    // 计算 token 数量
+    const tokens = encode(text);
+
+    // 计算价格
+    const price = unitPrice * tokens.length;
+    console.log('splitData bill');
+    console.log('token len:', tokens.length);
+    console.log('text len: ', text.length);
+    console.log('price: ', `${formatPrice(price)}元`);
+
+    try {
+      // 插入 Bill 记录
+      const res = await Bill.create({
+        userId,
+        type: 'splitData',
+        modelName: ChatModelNameEnum.GPT35,
+        textLen: text.length,
+        tokenLen: tokens.length,
+        price
+      });
+      billId = res._id;
+
+      // 账号扣费
+      await User.findByIdAndUpdate(userId, {
+        $inc: { balance: -price }
+      });
+    } catch (error) {
+      console.log('创建账单失败:', error);
+      billId && Bill.findByIdAndDelete(billId);
+    }
+  } catch (error) {
+    console.log(error);
+  }
+};

src/service/models/authCode.ts

@@ -1,4 +1,5 @@
-import { Schema, model, models } from 'mongoose';
+import { Schema, model, models, Model } from 'mongoose';
+import { AuthCodeSchema as AuthCodeType } from '@/types/mongoSchema';
 
 const AuthCodeSchema = new Schema({
   email: {
@@ -21,4 +22,5 @@ const AuthCodeSchema = new Schema({
   }
 });
 
-export const AuthCode = models['auth_code'] || model('auth_code', AuthCodeSchema);
+export const AuthCode: Model<AuthCodeType> =
+  models['auth_code'] || model('auth_code', AuthCodeSchema);

src/service/models/bill.ts

@@ -0,0 +1,45 @@
+import { Schema, model, models, Model } from 'mongoose';
+import { modelList } from '@/constants/model';
+import { BillSchema as BillType } from '@/types/mongoSchema';
+
+const BillSchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  type: {
+    type: String,
+    enum: ['chat', 'splitData', 'return'],
+    required: true
+  },
+  modelName: {
+    type: String,
+    enum: modelList.map((item) => item.model),
+    required: true
+  },
+  chatId: {
+    type: Schema.Types.ObjectId,
+    ref: 'chat'
+  },
+  time: {
+    type: Date,
+    default: () => new Date()
+  },
+  textLen: {
+    // 提示词+响应的总字数
+    type: Number,
+    required: true
+  },
+  tokenLen: {
+    // 折算成 token 的数量
+    type: Number,
+    required: true
+  },
+  price: {
+    type: Number,
+    required: true
+  }
+});
+
+export const Bill: Model<BillType> = models['bill'] || model('bill', BillSchema);

src/service/models/chat.ts

@@ -1,4 +1,5 @@
-import { Schema, model, models } from 'mongoose';
+import { Schema, model, models, Model } from 'mongoose';
+import { ChatSchema as ChatType } from '@/types/mongoSchema';
 
 const ChatSchema = new Schema({
   userId: {
@@ -20,7 +21,31 @@ const ChatSchema = new Schema({
     // 剩余加载次数
     type: Number,
     required: true
+  },
+  updateTime: {
+    type: Number,
+    required: true
+  },
+  isShare: {
+    type: Boolean,
+    default: false
+  },
+  content: {
+    type: [
+      {
+        obj: {
+          type: String,
+          required: true,
+          enum: ['Human', 'AI', 'SYSTEM']
+        },
+        value: {
+          type: String,
+          required: true
+        }
+      }
+    ],
+    default: []
   }
 });
 
-export const Chat = models['chat'] || model('chat', ChatSchema);
+export const Chat: Model<ChatType> = models['chat'] || model('chat', ChatSchema);

src/service/models/chatWindow.ts

@@ -1,28 +0,0 @@
-import { Schema, model, models } from 'mongoose';
-
-const ChatWindowSchema = new Schema({
-  chatId: {
-    type: Schema.Types.ObjectId,
-    ref: 'chat',
-    required: true
-  },
-  updateTime: {
-    type: Number,
-    required: true
-  },
-  content: [
-    {
-      obj: {
-        type: String,
-        required: true,
-        enum: ['Human', 'AI', 'SYSTEM']
-      },
-      value: {
-        type: String,
-        required: true
-      }
-    }
-  ]
-});
-
-export const ChatWindow = models['chatWindow'] || model('chatWindow', ChatWindowSchema);

src/service/models/data.ts

@@ -0,0 +1,24 @@
+import { Schema, model, models, Model } from 'mongoose';
+import { DataItemSchema as Datatype } from '@/types/mongoSchema';
+
+const DataSchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  name: {
+    type: String,
+    required: true
+  },
+  createTime: {
+    type: Date,
+    default: () => new Date()
+  },
+  isDeleted: {
+    type: Boolean,
+    default: false
+  }
+});
+
+export const Data: Model<Datatype> = models['data'] || model('data', DataSchema);