perf: chat framwork

Signed-off-by: 晓杰 <2561589453@qq.com>

.env.template

@@ -1,7 +1,9 @@
 # proxy
-AXIOS_PROXY_HOST=127.0.0.1
-AXIOS_PROXY_PORT_FAST=7890
-AXIOS_PROXY_PORT_NORMAL=7890
+# AXIOS_PROXY_HOST=127.0.0.1
+# AXIOS_PROXY_PORT=7890
+# OPENAI_BASE_URL=https://api.openai.com/v1
+# OPENAI_BASE_URL_AUTH=可选的安全凭证
+# 是否开启队列任务。 1-开启，0-关闭（请求parentUrl去执行任务,单机时直接填1）
 queueTask=1
 parentUrl=https://hostname/api/openapi/startEvents
 # email

.github/workflows/release.yml

@@ -5,7 +5,7 @@ on:
 
   push:
     branches:
-      - "main"
+      - 'main'
 
 jobs:
   release:
@@ -14,7 +14,7 @@ jobs:
       - name: Checkout
         uses: actions/checkout@v3
         with:
-          fetch-depth: 0
+          fetch-depth: 1
 
       - name: Install Dependencies
         run: |
@@ -41,6 +41,8 @@ jobs:
         run: |
           docker buildx build \
           --platform linux/amd64,linux/arm64 \
+          --label "org.opencontainers.image.source=https://github.com/${{ github.repository_owner }}/FastGPT" \
+          --label "org.opencontainers.image.description=fast-gpt image" \
           --label "org.opencontainers.image.licenses=MIT" \
           --push \
           -t ${DOCKER_REPO}:latest \

.gitignore

@@ -34,7 +34,6 @@ yarn-error.log*
 # typescript
 *.tsbuildinfo
 next-env.d.ts
-/public/trainData/
 /.vscode/
 platform.json
 testApi/

README.md

@@ -6,30 +6,33 @@ Fast GPT 允许你使用自己的 openai API KEY 来快速的调用 openai 接
 ![KBProcess](docs/imgs/KBProcess.jpg?raw=true "KBProcess")
 
 ## 开发
-复制 .env.template 成 .env.local ，填写核心参数  
-
+**配置环境变量**
 ```bash
-# proxy（不需要代理可忽略）
+# proxy（可选）
 AXIOS_PROXY_HOST=127.0.0.1
-AXIOS_PROXY_PORT_FAST=7890
-AXIOS_PROXY_PORT_NORMAL=7890
+AXIOS_PROXY_PORT=7890
+# openai 中转连接（可选）
+OPENAI_BASE_URL=https://api.openai.com/v1
+OPENAI_BASE_URL_AUTH=可选的安全凭证
+# 是否开启队列任务。 1-开启，0-关闭（请求 parentUrl 去执行任务,单机时直接填1）
 queueTask=1
 parentUrl=https://hostname/api/openapi/startEvents
-# email，参考 nodeMail 获取参数
+# 发送邮箱验证码配置。参考 nodeMail 获取参数，自行百度。
 MY_MAIL=xxx@qq.com
 MAILE_CODE=xxx
-# 阿里短信服务
+# 阿里短信服务（邮箱和短信至少二选一）
 aliAccessKeyId=xxx
 aliAccessKeySecret=xxx
 aliSignName=xxx
 aliTemplateCode=SMS_xxx
-# token（随便填，登录凭证）
+# token（随便填，作为登录凭证）
 TOKEN_KEY=xxx
 # openai key
 OPENAIKEY=sk-xxx
 # mongo连接地址
 MONGODB_URI=mongodb://username:password@0.0.0.0:27017/test?authSource=admin
-MONGODB_NAME=xxx # mongo数据库名称
+# mongo数据库名称
+MONGODB_NAME=xxx 
 # pg 数据库相关内容，和 docker-compose 对上
 PG_HOST=0.0.0.0 
 PG_PORT=8102
@@ -37,13 +40,20 @@ PG_USER=xxx
 PG_PASSWORD=xxx
 PG_DB_NAME=xxx
 ```
-```bash
+**运行**
+```
 pnpm dev
 ```
 
 ## 部署
 
-### 安装 docker 和 docker-compose
+### 代理环境（国外服务器可忽略）
+1. [clash 方案](./docs/proxy/clash.md) - 仅需一台服务器（需要有 clash）
+2. [nginx 方案](./docs/proxy/nginx.md) - 需要一台国外服务器
+3. [cloudflare 方案](./docs/proxy/cloudflare.md) - 需要有域名（每日免费 10w 次代理请求）
+
+### docker 部署
+#### 1. 安装 docker 和 docker-compose
 这个不同系统略有区别，百度安装下。验证安装成功后进行下一步。下面给出一个例子：
 ```bash
 # 安装docker
@@ -57,48 +67,12 @@ docker -v
 docker-compose -v
 ```
 
-### 安装 clash 代理（选）
-```bash
-# 下载包
-curl https://glados.rocks/tools/clash-linux.zip -o clash.zip 
-# 解压
-unzip clash.zip
-# 下载终端配置⽂件（改成自己配置文件路径）
-curl https://update.glados-config.com/clash/98980/8f30944/70870/glados-terminal.yaml > config.yaml
-# 赋予运行权限
-chmod +x ./clash-linux-amd64-v1.10.0 
-# 记得配置端口变量：
-export ALL_PROXY=socks5://127.0.0.1:7891
-export http_proxy=http://127.0.0.1:7890
-export https_proxy=http://127.0.0.1:7890
-export HTTP_PROXY=http://127.0.0.1:7890
-export HTTPS_PROXY=http://127.0.0.1:7890
 
-# 运行脚本: 删除clash - 到 clash 目录 - 删除缓存 - 执行运行. 会生成一个 nohup.out 文件，可以看到 clash 的 logs
-OLD_PROCESS=$(pgrep clash)
-if [ ! -z "$OLD_PROCESS" ]; then
-  echo "Killing old process: $OLD_PROCESS"
-  kill $OLD_PROCESS
-fi
-sleep 2
-cd  **/clash
-rm -f ./nohup.out || true
-rm -f ./cache.db || true
-nohup ./clash-linux-amd64-v1.10.0  -d ./ &
-echo "Restart clash"
-```
+#### 2. 创建3个初始化文件
+手动创建或者直接把 deploy 里内容复制过去
 
-### 本地 docker 打包
-```bash
-docker build -t imageName:tag .
-docker push imageName:tag
-# 或者直接拉镜像，见下方
-```
-
-### 准备初始化文件
 **/root/fast-gpt/pg/init.sql**
 ```sql
-#!/bin/bash
 set -e
 psql -v ON_ERROR_STOP=1 --username "$POSTGRES_USER" --dbname "$POSTGRES_DB" <<-EOSQL
 
@@ -119,6 +93,7 @@ CREATE INDEX modelData_userId_index ON modelData USING HASH (user_id);
 CREATE INDEX modelData_modelId_index ON modelData USING HASH (model_id);
 EOSQL
 ```
+
 **/root/fast-gpt/nginx/nginx.conf**
 ```conf
 user nginx;
@@ -171,6 +146,7 @@ http {
     }
 }
 ```
+
 **/root/fast-gpt/docker-compose.yml**
 ```yml
 version: "3.3"
@@ -181,24 +157,19 @@ services:
     restart: always
     container_name: fast-gpt
     environment:
-      # 代理（不需要代理，可去掉下面三个参数）
-      - AXIOS_PROXY_HOST=127.0.0.1
-      - AXIOS_PROXY_PORT_FAST=7890
-      - AXIOS_PROXY_PORT_NORMAL=7890
-      # 邮箱
+      # - AXIOS_PROXY_HOST=127.0.0.1
+      # - AXIOS_PROXY_PORT=7890
+      # - OPENAI_BASE_URL=https://api.openai.com/v1
+      # - OPENAI_BASE_URL_AUTH=可选的安全凭证
       - MY_MAIL=xxxx@qq.com  
       - MAILE_CODE=xxxx
-      # 阿里云短信
       - aliAccessKeyId=xxxx 
       - aliAccessKeySecret=xxxx
       - aliSignName=xxxxx
       - aliTemplateCode=SMS_xxxx
-      # 登录 key
       - TOKEN_KEY=xxxx 
-      # 是否开启队列任务。 1-开启，0-关闭（请求parentUrl去执行任务,单机时直接填1）
       - queueTask=1
       - parentUrl=https://hostname/api/openapi/startEvents
-      # db
       - MONGODB_URI=mongodb://username:passsword@0.0.0.0:27017/?authSource=admin
       - MONGODB_NAME=xxx
       - PG_HOST=0.0.0.0
@@ -206,7 +177,6 @@ services:
       - PG_USER=xxx
       - PG_PASSWORD=xxx
       - PG_DB_NAME=xxx
-      # openai 账号
       - OPENAIKEY=sk-xxxxx
   nginx: 
     image: nginx:alpine3.17
@@ -246,7 +216,9 @@ services:
       - /root/fast-gpt/mongo/logs:/var/log/mongodb
       - /etc/localtime:/etc/localtime:ro
 ```
-### 辅助运行脚本
+
+#### 3. 运行 docker-compose
+下面是一个辅助脚本，也可以直接 docker-compose up -d
 **run.sh 运行文件**
 ```bash
 #!/bin/bash
@@ -270,5 +242,15 @@ do
 done
 ```
 
-## Mac 可能的问题
-> 因为教程有部分镜像不兼容arm64，所以写个文档指导新手如何快速在mac上面搭建fast-gpt[如何在mac上面部署fastgpt](./docs/mac.md)
+
+## 其他优化点
+### Git Action 自动打包镜像
+.github里拥有一个 git 提交到 main 分支时自动打包 amd64 和 arm64 镜像的 actions。你仅需要提前在 git 配置好 session。  
+
+1. 创建账号 session: 头像 -> settings -> 最底部 Developer settings ->  Personal access tokens -> tokens(classic) -> 创建新 session，把一些看起来需要的权限勾上。
+2. 添加 session 到仓库: 仓库 -> settings -> Secrets and variables -> Actions -> 创建secret
+3. 填写 secret: Name-GH_PAT, Secret-第一步的tokens 
+
+## 其他问题
+### Mac 可能的问题
+> 因为教程有部分镜像不兼容arm64，所以写个文档指导新手如何快速在mac上面搭建fast-gpt[如何在mac上面部署fastgpt](./docs/mac.md)

deploy/docker-compose.yml

@@ -0,0 +1,66 @@
+version: '3.3'
+services:
+  fast-gpt:
+    image: c121914yu/fast-gpt:latest
+    network_mode: host
+    restart: always
+    container_name: fast-gpt
+    environment:
+      # - AXIOS_PROXY_HOST=127.0.0.1
+      # - AXIOS_PROXY_PORT=7890
+      # - OPENAI_BASE_URL=https://api.openai.com/v1
+      # - OPENAI_BASE_URL_AUTH=可选的安全凭证
+      - MY_MAIL=xxxx@qq.com
+      - MAILE_CODE=xxxx
+      - aliAccessKeyId=xxxx
+      - aliAccessKeySecret=xxxx
+      - aliSignName=xxxxx
+      - aliTemplateCode=SMS_xxxx
+      - TOKEN_KEY=xxxx
+      - queueTask=1
+      - parentUrl=https://hostname/api/openapi/startEvents
+      - MONGODB_URI=mongodb://username:passsword@0.0.0.0:27017/?authSource=admin
+      - MONGODB_NAME=xxx
+      - PG_HOST=0.0.0.0
+      - PG_PORT=8100
+      - PG_USER=xxx
+      - PG_PASSWORD=xxx
+      - PG_DB_NAME=xxx
+      - OPENAIKEY=sk-xxxxx
+  nginx:
+    image: nginx:alpine3.17
+    container_name: nginx
+    restart: always
+    network_mode: host
+    volumes:
+      - /root/fast-gpt/nginx/nginx.conf:/etc/nginx/nginx.conf:ro
+      - /root/fast-gpt/nginx/logs:/var/log/nginx
+      - /root/fast-gpt/nginx/ssl/docgpt.key:/ssl/docgpt.key
+      - /root/fast-gpt/nginx/ssl/docgpt.pem:/ssl/docgpt.pem
+  pg:
+    image: ankane/pgvector
+    container_name: pg
+    restart: always
+    ports:
+      - 8100:5432
+    environment:
+      - POSTGRES_USER=xxx
+      - POSTGRES_PASSWORD=xxx
+      - POSTGRES_DB=xxx
+    volumes:
+      - /root/fast-gpt/pg/data:/var/lib/postgresql/data
+      - /root/fast-gpt/pg/init.sql:/docker-entrypoint-initdb.d/init.sh
+      - /etc/localtime:/etc/localtime:ro
+  mongodb:
+    image: mongo:4.0.1
+    container_name: mongo
+    restart: always
+    ports:
+      - 27017:27017
+    environment:
+      - MONGO_INITDB_ROOT_USERNAME=username
+      - MONGO_INITDB_ROOT_PASSWORD=password
+    volumes:
+      - /root/fast-gpt/mongo/data:/data/db
+      - /root/fast-gpt/mongo/logs:/var/log/mongodb
+      - /etc/localtime:/etc/localtime:ro

deploy/nginx/nginx.conf

@@ -0,0 +1,49 @@
+user nginx;
+worker_processes auto;
+worker_rlimit_nofile 51200;
+
+events {
+    worker_connections 1024;
+}
+
+http {
+    access_log off;
+    server_names_hash_bucket_size 512;
+    client_header_buffer_size 32k;
+    large_client_header_buffers 4 32k;
+    client_max_body_size 50M;
+ 
+    gzip  on;
+    gzip_min_length   1k;
+    gzip_buffers  4 8k;
+    gzip_http_version 1.1;
+    gzip_comp_level 6;
+    gzip_vary on;
+    gzip_types  text/plain application/x-javascript text/css application/javascript application/json application/xml;
+    gzip_disable "MSIE [1-6]\.";
+
+    open_file_cache max=1000 inactive=1d;
+    open_file_cache_valid 30s;
+    open_file_cache_min_uses 8;
+    open_file_cache_errors off;
+
+    server {
+        listen 443 ssl;
+        server_name docgpt.ahapocket.cn;
+        ssl_certificate /ssl/docgpt.pem;
+        ssl_certificate_key /ssl/docgpt.key;
+        ssl_session_timeout 5m;
+
+        location / {
+            proxy_pass http://localhost:3000;
+            proxy_set_header Host $host;
+            proxy_set_header X-Real-IP $remote_addr;
+            proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;        
+        }
+    }
+    server {
+        listen 80;
+        server_name docgpt.ahapocket.cn;
+        rewrite ^(.*) https://$server_name$1 permanent;
+    }
+}

deploy/pg/init.sql

@@ -0,0 +1,19 @@
+set -e
+psql -v ON_ERROR_STOP=1 --username "$POSTGRES_USER" --dbname "$POSTGRES_DB" <<-EOSQL
+
+CREATE EXTENSION vector;
+-- init table
+CREATE TABLE modelData (
+    id BIGSERIAL PRIMARY KEY,
+    vector VECTOR(1536),
+    status VARCHAR(50) NOT NULL,
+    user_id VARCHAR(50) NOT NULL,
+    model_id VARCHAR(50) NOT NULL,
+    q TEXT NOT NULL,
+    a TEXT NOT NULL
+);
+-- create index
+CREATE INDEX modelData_status_index ON modelData USING HASH (status);
+CREATE INDEX modelData_userId_index ON modelData USING HASH (user_id);
+CREATE INDEX modelData_modelId_index ON modelData USING HASH (model_id);
+EOSQL

deploy/run.sh

@@ -0,0 +1,19 @@
+#!/bin/bash
+docker-compose pull
+docker-compose up -d
+
+echo "Docker Compose 重新拉取镜像完成！"
+
+# 删除本地旧镜像
+images=$(docker images --format "{{.ID}} {{.Repository}}" | grep fast-gpt)
+
+# 将镜像 ID 和名称放入数组中
+IFS=$'\n' read -rd '' -a image_array <<<"$images"
+
+# 遍历数组并删除所有旧的镜像
+for ((i=1; i<${#image_array[@]}; i++))
+do
+    image=${image_array[$i]}
+    image_id=${image%% *}
+    docker rmi $image_id
+done

docs/proxy/clash.md

@@ -0,0 +1,68 @@
+# 安装 clash 
+
+clash 会在本机启动代理。对应的，你需要配置项目的两个环境变量：
+
+```
+AXIOS_PROXY_HOST=127.0.0.1
+AXIOS_PROXY_PORT=7890
+```
+
+需要注的是，在你的 config.yaml 文件中，最好仅指定 api.openai.com 走代理，其他请求都直连。
+
+**安装clash**
+```bash
+# 下载包
+curl https://glados.rocks/tools/clash-linux.zip -o clash.zip 
+# 解压
+unzip clash.zip
+# 下载终端配置⽂件（改成自己配置文件路径）
+curl https://update.glados-config.com/clash/98980/8f30944/70870/glados-terminal.yaml > config.yaml
+# 赋予运行权限
+chmod +x ./clash-linux-amd64-v1.10.0 
+```
+
+**runClash.sh**
+```sh
+# 记得配置端口变量：
+export ALL_PROXY=socks5://127.0.0.1:7891
+export http_proxy=http://127.0.0.1:7890
+export https_proxy=http://127.0.0.1:7890
+export HTTP_PROXY=http://127.0.0.1:7890
+export HTTPS_PROXY=http://127.0.0.1:7890
+
+# 运行脚本: 删除clash - 到 clash 目录 - 删除缓存 - 执行运行. 会生成一个 nohup.out 文件，可以看到 clash 的 logs
+OLD_PROCESS=$(pgrep clash)
+if [ ! -z "$OLD_PROCESS" ]; then
+  echo "Killing old process: $OLD_PROCESS"
+  kill $OLD_PROCESS
+fi
+sleep 2
+cd  **/clash
+rm -f ./nohup.out || true
+rm -f ./cache.db || true
+nohup ./clash-linux-amd64-v1.10.0  -d ./ &
+echo "Restart clash"
+```
+
+**config.yaml配置例子**
+```yaml
+mixed-port: 7890
+allow-lan: false
+bind-address: '*'
+mode: rule
+log-level: warning
+dns:  
+  enable: true  
+  ipv6: false  
+  nameserver:  
+    - 8.8.8.8
+    - 8.8.4.4 
+  cache-size: 400
+proxies:
+    - 
+proxy-groups:
+  - { name: '♻️ 自动选择', type: url-test,  proxies: [香港V01×1.5], url: 'https://api.openai.com', interval: 3600}
+rules:
+  - 'DOMAIN-SUFFIX,api.openai.com,♻️ 自动选择'
+  - 'MATCH,DIRECT'
+```

docs/proxy/cloudflare.md

@@ -0,0 +1,46 @@
+# cloudflare 代理配置
+
+[来自 "不做了睡觉" 教程](https://gravel-twister-d32.notion.site/FastGPT-API-ba7bb261d5fd4fd9bbb2f0607dacdc9e)
+
+**workers 配置文件**
+
+```js
+const TELEGRAPH_URL = 'https://api.openai.com';
+
+addEventListener('fetch', (event) => {
+  event.respondWith(handleRequest(event.request));
+});
+
+async function handleRequest(request) {
+  // 安全校验
+  if (request.headers.get('auth') !== 'auth_code') {
+    return new Response('UnAuthorization', { status: 403 });
+  }
+
+  const url = new URL(request.url);
+  url.host = TELEGRAPH_URL.replace(/^https?:\/\//, '');
+
+  const modifiedRequest = new Request(url.toString(), {
+    headers: request.headers,
+    method: request.method,
+    body: request.body,
+    redirect: 'follow'
+  });
+
+  const response = await fetch(modifiedRequest);
+  const modifiedResponse = new Response(response.body, response);
+
+  // 添加允许跨域访问的响应头
+  modifiedResponse.headers.set('Access-Control-Allow-Origin', '*');
+
+  return modifiedResponse;
+}
+```
+
+**对应的环境变量**
+务必别忘了填 v1
+
+```
+OPENAI_BASE_URL=https://xxxxxx/v1
+OPENAI_BASE_URL_AUTH=auth_code
+```

docs/proxy/nginx.md

@@ -0,0 +1,72 @@
+# nginx 反向代理 openai 接口
+如果你有国外的服务器，可以通过配置 nginx 反向代理，转发 openai 相关的请求，从而让国内的服务器可以通过访问该 nginx 去访问 openai 接口。
+
+```conf
+user nginx;
+worker_processes auto;
+worker_rlimit_nofile 51200;
+
+events {
+    worker_connections 1024;
+}
+
+http {
+    resolver 8.8.8.8;
+    proxy_ssl_server_name on;
+
+    access_log off;
+    server_names_hash_bucket_size 512;
+    client_header_buffer_size 32k;
+    large_client_header_buffers 4 32k;
+    client_max_body_size 50M;
+ 
+    gzip  on;
+    gzip_min_length   1k;
+    gzip_buffers  4 8k;
+    gzip_http_version 1.1;
+    gzip_comp_level 6;
+    gzip_vary on;
+    gzip_types  text/plain application/x-javascript text/css application/javascript application/json application/xml;
+    gzip_disable "MSIE [1-6]\.";
+
+    open_file_cache max=1000 inactive=1d;
+    open_file_cache_valid 30s;
+    open_file_cache_min_uses 8;
+    open_file_cache_errors off;
+
+    server {
+        listen 443 ssl;
+        server_name your_host;
+        ssl_certificate /ssl/your_host.pem;
+        ssl_certificate_key /ssl/your_host.key;
+        ssl_session_timeout 5m;
+
+        location ~ /openai/(.*) {
+            # auth check
+            if ($http_authkey != "xxxxxx") {
+                return 403;
+            }
+            
+            proxy_pass https://api.openai.com/$1$is_args$args;
+            proxy_set_header Host api.openai.com;
+            proxy_set_header X-Real-IP $remote_addr;
+            proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
+            # 流式响应
+            proxy_set_header Connection '';
+            proxy_http_version 1.1;
+            chunked_transfer_encoding off;
+            proxy_buffering off;
+            proxy_cache off;
+            # 一般响应
+            proxy_buffer_size 128k;
+            proxy_buffers 4 256k;
+            proxy_busy_buffers_size 256k;
+        }
+    }
+    server {
+        listen 80;
+        server_name ai.fastgpt.run;
+        rewrite ^(.*) https://$server_name$1 permanent;
+    }
+}
+```

next.config.js

@@ -2,7 +2,7 @@
 
 const nextConfig = {
   output: 'standalone',
-  reactStrictMode: true,
+  reactStrictMode: false,
   compress: true,
 
   webpack(config) {

public/docs/versionIntro.md

@@ -1,5 +1,5 @@
-### Fast GPT V2.8.1
-* 优化 - 知识库升级，内容条数不上限！
-* 优化 - 导入去重效果，可防止导出后的 csv 重复导入。
-* 优化 - 聊天框，电脑端复制删除图标。
-* 优化 - 聊天框，生成内容时，如果滚动条触底，则会自动向下滚动，不需要手动下滑。
+### Fast GPT V3.1
+
+- 优化 - 模型结构设计，不再区分知识库和对话模型，而是通过开关的形式，手动选择手否需要进行知识库搜索。
+- 新增 - 模型共享市场，可以使用其他用户分享的模型。
+- 新增 - 邀请好友注册功能。

public/js/pdf.js

@@ -406,12 +406,12 @@ function getVerbosityLevel() {
 }
 function info(msg) {
   if (verbosity >= VerbosityLevel.INFOS) {
-    console.log(`Info: ${msg}`);
+    // console.log(`Info: ${msg}`);
   }
 }
 function warn(msg) {
   if (verbosity >= VerbosityLevel.WARNINGS) {
-    console.log(`Warning: ${msg}`);
+    // console.log(`Warning: ${msg}`);
   }
 }
 function unreachable(msg) {
@@ -4206,7 +4206,7 @@ function loadScript(src, removeScriptElement = false) {
   });
 }
 function deprecated(details) {
-  console.log("Deprecated API usage: " + details);
+  // console.log("Deprecated API usage: " + details);
 }
 let pdfDateStringRegex;
 class PDFDateString {

public/js/pdf.worker.js

@@ -1008,12 +1008,12 @@ function getVerbosityLevel() {
 }
 function info(msg) {
   if (verbosity >= VerbosityLevel.INFOS) {
-    console.log(`Info: ${msg}`);
+    // console.log(`Info: ${msg}`);
   }
 }
 function warn(msg) {
   if (verbosity >= VerbosityLevel.WARNINGS) {
-    console.log(`Warning: ${msg}`);
+    // console.log(`Warning: ${msg}`);
   }
 }
 function unreachable(msg) {

src/api/chat.ts

@@ -31,5 +31,5 @@ export const postSaveChat = (data: {
 /**
  * 删除一句对话
  */
-export const delChatRecordByIndex = (chatId: string, index: number) =>
-  DELETE(`/chat/delChatRecordByIndex?chatId=${chatId}&index=${index}`);
+export const delChatRecordByIndex = (chatId: string, contentId: string) =>
+  DELETE(`/chat/delChatRecordByContentId?chatId=${chatId}&contentId=${contentId}`);

src/api/fetch.ts

@@ -1,4 +1,6 @@
 import { getToken } from '../utils/user';
+import { SYSTEM_PROMPT_PREFIX } from '@/constants/chat';
+
 interface StreamFetchProps {
   url: string;
   data: any;
@@ -6,7 +8,7 @@ interface StreamFetchProps {
   abortSignal: AbortController;
 }
 export const streamFetch = ({ url, data, onMessage, abortSignal }: StreamFetchProps) =>
-  new Promise(async (resolve, reject) => {
+  new Promise<{ responseText: string; systemPrompt: string }>(async (resolve, reject) => {
     try {
       const res = await fetch(url, {
         method: 'POST',
@@ -19,30 +21,44 @@ export const streamFetch = ({ url, data, onMessage, abortSignal }: StreamFetchPr
       });
       const reader = res.body?.getReader();
       if (!reader) return;
+
       const decoder = new TextDecoder();
       let responseText = '';
+      let systemPrompt = '';
 
       const read = async () => {
-        const { done, value } = await reader?.read();
-        if (done) {
-          if (res.status === 200) {
-            resolve(responseText);
-          } else {
-            try {
+        try {
+          const { done, value } = await reader?.read();
+          if (done) {
+            if (res.status === 200) {
+              resolve({ responseText, systemPrompt });
+            } else {
               const parseError = JSON.parse(responseText);
               reject(parseError?.message || '请求异常');
-            } catch (err) {
-              reject('请求异常');
             }
-          }
 
-          return;
+            return;
+          }
+          let text = decoder.decode(value).replace(/<br\/>/g, '\n');
+          // check system prompt
+          if (text.includes(SYSTEM_PROMPT_PREFIX)) {
+            const arr = text.split(SYSTEM_PROMPT_PREFIX);
+            systemPrompt = arr.pop() || '';
+
+            text = arr.join('');
+          }
+          responseText += text;
+          onMessage(text);
+
+          read();
+        } catch (err: any) {
+          if (err?.message === 'The user aborted a request.') {
+            return resolve({ responseText, systemPrompt });
+          }
+          reject(typeof err === 'string' ? err : err?.message || '请求异常');
         }
-        const text = decoder.decode(value).replace(/<br\/>/g, '\n');
-        res.status === 200 && onMessage(text);
-        responseText += text;
-        read();
       };
+
       read();
     } catch (err: any) {
       console.log(err, '====');

src/api/model.ts

@@ -1,7 +1,6 @@
 import { GET, POST, DELETE, PUT } from './request';
 import type { ModelSchema, ModelDataSchema } from '@/types/mongoSchema';
-import { ModelUpdateParams } from '@/types/model';
-import { TrainingItemType } from '../types/training';
+import { ModelUpdateParams, ShareModelItem } from '@/types/model';
 import { RequestPaging } from '../types/index';
 import { Obj2Query } from '@/utils/tools';
 
@@ -13,8 +12,7 @@ export const getMyModels = () => GET<ModelSchema[]>('/model/list');
 /**
  * 创建一个模型
  */
-export const postCreateModel = (data: { name: string; serviceModelName: string }) =>
-  POST<ModelSchema>('/model/create', data);
+export const postCreateModel = (data: { name: string }) => POST<string>('/model/create', data);
 
 /**
  * 根据 ID 删除模型
@@ -32,21 +30,7 @@ export const getModelById = (id: string) => GET<ModelSchema>(`/model/detail?mode
 export const putModelById = (id: string, data: ModelUpdateParams) =>
   PUT(`/model/update?modelId=${id}`, data);
 
-export const postTrainModel = (id: string, form: FormData) =>
-  POST(`/model/train/train?modelId=${id}`, form, {
-    headers: {
-      'content-type': 'multipart/form-data'
-    }
-  });
-
-export const putModelTrainingStatus = (id: string) =>
-  PUT(`/model/train/putTrainStatus?modelId=${id}`);
-
-export const getModelTrainings = (id: string) =>
-  GET<TrainingItemType[]>(`/model/train/getTrainings?modelId=${id}`);
-
 /* 模型 data */
-
 type GetModelDataListProps = RequestPaging & {
   modelId: string;
   searchText: string;
@@ -108,3 +92,19 @@ export const putModelDataById = (data: { dataId: string; a: string; q?: string }
  */
 export const delOneModelData = (dataId: string) =>
   DELETE(`/model/data/delModelDataById?dataId=${dataId}`);
+
+/* 共享市场 */
+/**
+ * 获取共享市场模型
+ */
+export const getShareModelList = (data: { searchText?: string } & RequestPaging) =>
+  POST(`/model/share/getModels`, data);
+/**
+ * 获取收藏的模型
+ */
+export const getCollectionModels = () => GET<ShareModelItem[]>(`/model/share/getCollection`);
+/**
+ * 收藏/取消收藏模型
+ */
+export const triggerModelCollection = (modelId: string) =>
+  POST<number>(`/model/share/collection?modelId=${modelId}`);

src/api/request.ts

@@ -1,6 +1,6 @@
 import axios, { Method, InternalAxiosRequestConfig, AxiosResponse } from 'axios';
 import { getToken, clearToken } from '@/utils/user';
-import { TOKEN_ERROR_CODE } from '@/constants/responseCode';
+import { TOKEN_ERROR_CODE } from '@/service/errorCode';
 
 interface ConfigType {
   headers?: { [key: string]: string };

src/api/response/chat.d.ts

@@ -7,7 +7,6 @@ export type InitChatResponse = {
   name: string;
   avatar: string;
   intro: string;
-  chatModel: ModelSchema.service.chatModel; // 对话模型名
-  modelName: ModelSchema.service.modelName; // 底层模型
+  chatModel: ModelSchema['chat']['chatModel']; // 对话模型名
   history: ChatItemType[];
 };

src/components/Icon/icons/chatting.svg

@@ -1,19 +0,0 @@
-<?xml version="1.0" encoding="utf-8"?>
-<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" style="margin: auto; background: none; display: block; shape-rendering: auto;" width="204px" height="204px" viewBox="0 0 100 100" preserveAspectRatio="xMidYMid">
-<circle cx="84" cy="50" r="10" fill="#e15b64">
-    <animate attributeName="r" repeatCount="indefinite" dur="0.5681818181818182s" calcMode="spline" keyTimes="0;1" values="15;0" keySplines="0 0.5 0.5 1" begin="0s"></animate>
-    <animate attributeName="fill" repeatCount="indefinite" dur="2.272727272727273s" calcMode="discrete" keyTimes="0;0.25;0.5;0.75;1" values="#e15b64;#abbd81;#f8b26a;#f47e60;#e15b64" begin="0s"></animate>
-</circle><circle cx="16" cy="50" r="10" fill="#e15b64">
-  <animate attributeName="r" repeatCount="indefinite" dur="2.272727272727273s" calcMode="spline" keyTimes="0;0.25;0.5;0.75;1" values="0;0;15;15;15" keySplines="0 0.5 0.5 1;0 0.5 0.5 1;0 0.5 0.5 1;0 0.5 0.5 1" begin="0s"></animate>
-  <animate attributeName="cx" repeatCount="indefinite" dur="2.272727272727273s" calcMode="spline" keyTimes="0;0.25;0.5;0.75;1" values="16;16;16;50;84" keySplines="0 0.5 0.5 1;0 0.5 0.5 1;0 0.5 0.5 1;0 0.5 0.5 1" begin="0s"></animate>
-</circle><circle cx="50" cy="50" r="10" fill="#f47e60">
-  <animate attributeName="r" repeatCount="indefinite" dur="2.272727272727273s" calcMode="spline" keyTimes="0;0.25;0.5;0.75;1" values="0;0;15;15;15" keySplines="0 0.5 0.5 1;0 0.5 0.5 1;0 0.5 0.5 1;0 0.5 0.5 1" begin="-0.5681818181818182s"></animate>
-  <animate attributeName="cx" repeatCount="indefinite" dur="2.272727272727273s" calcMode="spline" keyTimes="0;0.25;0.5;0.75;1" values="16;16;16;50;84" keySplines="0 0.5 0.5 1;0 0.5 0.5 1;0 0.5 0.5 1;0 0.5 0.5 1" begin="-0.5681818181818182s"></animate>
-</circle><circle cx="84" cy="50" r="10" fill="#f8b26a">
-  <animate attributeName="r" repeatCount="indefinite" dur="2.272727272727273s" calcMode="spline" keyTimes="0;0.25;0.5;0.75;1" values="0;0;15;15;15" keySplines="0 0.5 0.5 1;0 0.5 0.5 1;0 0.5 0.5 1;0 0.5 0.5 1" begin="-1.1363636363636365s"></animate>
-  <animate attributeName="cx" repeatCount="indefinite" dur="2.272727272727273s" calcMode="spline" keyTimes="0;0.25;0.5;0.75;1" values="16;16;16;50;84" keySplines="0 0.5 0.5 1;0 0.5 0.5 1;0 0.5 0.5 1;0 0.5 0.5 1" begin="-1.1363636363636365s"></animate>
-</circle><circle cx="16" cy="50" r="10" fill="#abbd81">
-  <animate attributeName="r" repeatCount="indefinite" dur="2.272727272727273s" calcMode="spline" keyTimes="0;0.25;0.5;0.75;1" values="0;0;15;15;15" keySplines="0 0.5 0.5 1;0 0.5 0.5 1;0 0.5 0.5 1;0 0.5 0.5 1" begin="-1.7045454545454546s"></animate>
-  <animate attributeName="cx" repeatCount="indefinite" dur="2.272727272727273s" calcMode="spline" keyTimes="0;0.25;0.5;0.75;1" values="16;16;16;50;84" keySplines="0 0.5 0.5 1;0 0.5 0.5 1;0 0.5 0.5 1;0 0.5 0.5 1" begin="-1.7045454545454546s"></animate>
-</circle>
-<!-- [ldio] generated by https://loading.io/ --></svg>

src/components/Icon/icons/collectionLight.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="1682602070818" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="2479" xmlns:xlink="http://www.w3.org/1999/xlink" width="128" height="128"><path d="M509.606998 143.114488c9.082866 0 17.327644 4.840238 20.996197 12.331863l97.262184 197.441814c5.613858 11.403724 16.663518 19.358907 29.438473 21.216207l223.738737 32.552393c8.420787 1.215688 15.604396 6.851035 18.23327 14.254655 2.520403 7.18361 0.595564 15.062044-5.084808 20.586874L730.253304 601.611947c-8.949836 8.751315-12.994965 21.171182-10.916631 33.370015l38.011732 222.060515c1.325182 7.737218-2.165316 15.426341-8.905834 19.978007-4.088108 2.741437-8.861832 4.155646-13.812587 4.155646-4.022617 0-7.999185-0.972141-11.425214-2.740414L528.149307 775.671215c-5.768377-3.006474-12.155854-4.552689-18.542308-4.552689-6.364965 0-12.727882 1.547239-18.518772 4.552689L296.254819 878.348736c-3.559059 1.855254-7.602142 2.828418-11.668761 2.828418-4.861728 0-9.723455-1.459235-13.546527-4.022617-6.961552-4.684696-10.475586-12.419867-9.127891-20.155039l38.011732-222.016513c2.078335-12.198833-1.988284-24.619724-10.939143-33.370015L125.02397 441.443038c-5.635347-5.492084-7.55814-13.348006-5.061272-20.453844 2.63092-7.481392 9.812483-13.116739 18.298761-14.332427l223.674269-32.552393c12.839423-1.857301 23.867594-9.813506 29.481452-21.216207l97.194646-197.396789C492.325403 147.965983 500.590648 143.114488 509.606998 143.114488M509.606998 104.904235c-24.043602 0-45.922912 13.226233-56.177464 33.95637L356.189863 336.302419l-223.674269 32.54216c-22.983457 3.304256-42.100864 18.718317-49.481971 39.659255-7.381108 21.048385-1.812275 44.23241 14.431687 60.033281l163.916257 160.125931-38.011732 222.016513c-3.868097 22.408359 6.03239 44.819788 25.458835 57.94676 10.69662 7.116071 23.204491 10.784624 35.757388 10.784624 10.298554 0 20.663622-2.475378 30.055526-7.337105l194.987926-102.7205L704.662463 912.072815c9.369392 4.861728 19.712971 7.337105 29.990035 7.337105 12.57541 0 25.082258-3.668553 35.778878-10.784624 19.426445-13.126972 29.305443-35.538401 25.460882-57.94676l-38.012755-222.016513 163.937746-160.125931c16.22145-15.812127 21.810748-38.984896 14.408151-60.033281-7.402597-20.940938-26.51898-36.353976-49.503461-39.659255L663.04767 336.302419l-97.240695-197.441814C555.619962 118.131491 533.695626 104.904235 509.606998 104.904235L509.606998 104.904235z"  p-id="2480"></path></svg>

src/components/Icon/icons/collectionSolid.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="1682602068431" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="2339" xmlns:xlink="http://www.w3.org/1999/xlink" width="128" height="128"><path d="M335.008 916.629333c-35.914667 22.314667-82.88 10.773333-104.693333-25.557333a77.333333 77.333333 0 0 1-8.96-57.429333l46.485333-198.24a13.141333 13.141333 0 0 0-4.021333-12.864l-152.16-132.586667c-31.605333-27.52-35.253333-75.648-8.234667-107.733333a75.68 75.68 0 0 1 51.733333-26.752L354.848 339.2c4.352-0.362667 8.245333-3.232 10.026667-7.594667l76.938666-188.170666c16.032-39.2 60.618667-57.92 99.52-41.461334a76.309333 76.309333 0 0 1 40.832 41.461334l76.938667 188.16c1.781333 4.373333 5.674667 7.253333 10.026667 7.605333l199.712 16.277333c41.877333 3.413333 72.885333 40.458667 69.568 82.517334a76.938667 76.938667 0 0 1-26.08 51.978666l-152.16 132.586667c-3.541333 3.082667-5.141333 8.074667-4.021334 12.853333l46.485334 198.24c9.621333 41.013333-15.36 82.336-56.138667 92.224a75.285333 75.285333 0 0 1-57.525333-9.237333l-170.976-106.24a11.296 11.296 0 0 0-12.010667 0l-170.986667 106.24z" p-id="2340"></path></svg>

src/components/Icon/icons/export.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="1682957006954" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="2644" xmlns:xlink="http://www.w3.org/1999/xlink" width="48" height="48"><path d="M942.1 593.9c-22.6 0-41 18.3-41 41v204.8c0 22.6-18.4 41-41 41H163.8c-22.6 0-41-18.4-41-41V634.9c0-22.6-18.3-41-41-41s-41 18.3-41 41v204.8c0 67.8 55.1 122.9 122.9 122.9H860c67.8 0 122.9-55.1 122.9-122.9V634.9c0.1-22.6-18.2-41-40.8-41z" p-id="2645"></path><path d="M309.3 363L471 201.3v515.5c0 22.5 18.4 41 41 41 22.5 0 41-18.4 41-41V201.3L714.7 363c15.9 15.9 42 15.9 57.9 0 15.9-15.9 15.9-42 0-57.9L541.5 73.9c-0.2-0.2-0.3-0.4-0.4-0.5-5.7-5.7-12.7-9.3-20.1-10.9-0.2-0.1-0.5-0.2-0.7-0.2-2.7-0.5-5.4-0.8-8.1-0.8-2.7 0-5.5 0.3-8.1 0.8-0.3 0.1-0.5 0.2-0.7 0.2-7.4 1.6-14.4 5.2-20.1 10.9-0.2 0.2-0.3 0.4-0.4 0.5L251.4 305.1c-15.9 15.9-15.9 42 0 57.9 15.9 15.9 42 15.9 57.9 0z" p-id="2646"></path></svg>

src/components/Icon/icons/shareMarket.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="1682599933100" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="5707" xmlns:xlink="http://www.w3.org/1999/xlink" width="128" height="128"><path d="M750.592 668.7232a159.6416 159.6416 0 0 0-128.4608 64.9216l-269.568-138.24a159.3344 159.3344 0 0 0 17.0496-128.6656l261.12-136.704a159.4368 159.4368 0 1 0-31.1296-53.0432L341.1456 412.3648a159.7952 159.7952 0 1 0-32.256 229.7856l286.72 146.9952a159.7952 159.7952 0 1 0 154.88-120.4224z m0-542.72a98.3552 98.3552 0 1 1-98.3552 98.3552 98.4576 98.4576 0 0 1 98.3552-98.304z m-534.2208 484.352A98.3552 98.3552 0 1 1 314.7264 512a98.4576 98.4576 0 0 1-98.3552 98.3552zM750.592 926.72a98.3552 98.3552 0 1 1 98.3552-98.3552A98.4576 98.4576 0 0 1 750.592 926.72z" p-id="5708"></path></svg>

src/components/Icon/icons/stop.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="1682424901088" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="3662" xmlns:xlink="http://www.w3.org/1999/xlink" width="32" height="32"><path d="M885.333333 85.333333H138.666667a53.393333 53.393333 0 0 0-53.333334 53.333334v746.666666a53.393333 53.393333 0 0 0 53.333334 53.333334h746.666666a53.393333 53.393333 0 0 0 53.333334-53.333334V138.666667a53.393333 53.393333 0 0 0-53.333334-53.333334z m-160 602.666667a37.373333 37.373333 0 0 1-37.333333 37.333333H336a37.373333 37.373333 0 0 1-37.333333-37.333333V336a37.373333 37.373333 0 0 1 37.333333-37.333333h352a37.373333 37.373333 0 0 1 37.333333 37.333333z"  p-id="3663"></path></svg>

src/components/Icon/index.tsx

@@ -13,12 +13,16 @@ const map = {
   board: require('./icons/board.svg').default,
   develop: require('./icons/develop.svg').default,
   user: require('./icons/user.svg').default,
-  chatting: require('./icons/chatting.svg').default,
   promotion: require('./icons/promotion.svg').default,
   delete: require('./icons/delete.svg').default,
   withdraw: require('./icons/withdraw.svg').default,
   dbModel: require('./icons/dbModel.svg').default,
-  history: require('./icons/history.svg').default
+  history: require('./icons/history.svg').default,
+  stop: require('./icons/stop.svg').default,
+  shareMarket: require('./icons/shareMarket.svg').default,
+  collectionLight: require('./icons/collectionLight.svg').default,
+  collectionSolid: require('./icons/collectionSolid.svg').default,
+  export: require('./icons/export.svg').default
 };
 
 export type IconName = keyof typeof map;

src/components/Layout/auth.tsx

@@ -7,7 +7,8 @@ import { useQuery } from '@tanstack/react-query';
 
 const unAuthPage: { [key: string]: boolean } = {
   '/': true,
-  '/login': true
+  '/login': true,
+  '/model/share': true
 };
 
 const Auth = ({ children }: { children: JSX.Element }) => {
@@ -33,7 +34,9 @@ const Auth = ({ children }: { children: JSX.Element }) => {
     {
       onError(error) {
         console.log('error->', error);
-        router.replace('/login');
+        router.replace(
+          `/login?lastRoute=${encodeURIComponent(location.pathname + location.search)}`
+        );
         toast();
       },
       onSettled() {

src/components/Layout/index.tsx

@@ -20,12 +20,19 @@ const navbarList = [
     link: '/',
     activeLink: ['/']
   },
+  {
+    label: '共享',
+    icon: 'shareMarket',
+    link: '/model/share',
+    activeLink: ['/model/share']
+  },
   {
     label: '模型',
     icon: 'model',
     link: '/model/list',
     activeLink: ['/model/list', '/model/detail']
   },
+
   {
     label: '账号',
     icon: 'user',

src/components/Markdown/index.tsx

@@ -17,7 +17,7 @@ const Markdown = ({ source, isChatting = false }: { source: string; isChatting?:
 
   return (
     <ReactMarkdown
-      className={`${styles.markdown} ${
+      className={`markdown ${styles.markdown} ${
         isChatting ? (source === '' ? styles.waitingAnimation : styles.animation) : ''
       }`}
       remarkPlugins={[remarkMath]}
@@ -31,6 +31,7 @@ const Markdown = ({ source, isChatting = false }: { source: string; isChatting?:
           return !inline || match ? (
             <Box my={3} borderRadius={'md'} overflow={'hidden'} backgroundColor={'#222'}>
               <Flex
+                className="code-header"
                 py={2}
                 px={5}
                 backgroundColor={useColorModeValue('#323641', 'gray.600')}

src/constants/common.ts

@@ -4,3 +4,823 @@ export enum UserAuthTypeEnum {
 }
 
 export const PRICE_SCALE = 100000;
+
+export const htmlTemplate = `<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <meta http-equiv="X-UA-Compatible" content="IE=edge" />
+    <meta name="viewport" content="width=device-width initial-scale=1.0" />
+    <title>FastGpt</title>
+  </head>
+  <style>
+    .markdown > :first-child {
+      margin-top: 0 !important;
+    }
+    .markdown > :last-child {
+      margin-bottom: 0 !important;
+    }
+    .markdown a.absent {
+      color: #cc0000;
+    }
+    .markdown a.anchor {
+      bottom: 0;
+      cursor: pointer;
+      display: block;
+      left: 0;
+      margin-left: -30px;
+      padding-left: 30px;
+      position: absolute;
+      top: 0;
+    }
+    .markdown h1,
+    .markdown h2,
+    .markdown h3,
+    .markdown h4,
+    .markdown h5,
+    .markdown h6 {
+      cursor: text;
+      font-weight: bold;
+      margin: 10px 0;
+      padding: 0;
+      position: relative;
+    }
+    .markdown h1 .mini-icon-link,
+    .markdown h2 .mini-icon-link,
+    .markdown h3 .mini-icon-link,
+    .markdown h4 .mini-icon-link,
+    .markdown h5 .mini-icon-link,
+    .markdown h6 .mini-icon-link {
+      display: none;
+    }
+    .markdown h1:hover a.anchor,
+    .markdown h2:hover a.anchor,
+    .markdown h3:hover a.anchor,
+    .markdown h4:hover a.anchor,
+    .markdown h5:hover a.anchor,
+    .markdown h6:hover a.anchor {
+      line-height: 1;
+      margin-left: -22px;
+      padding-left: 0;
+      text-decoration: none;
+      top: 15%;
+    }
+    .markdown h1:hover a.anchor .mini-icon-link,
+    .markdown h2:hover a.anchor .mini-icon-link,
+    .markdown h3:hover a.anchor .mini-icon-link,
+    .markdown h4:hover a.anchor .mini-icon-link,
+    .markdown h5:hover a.anchor .mini-icon-link,
+    .markdown h6:hover a.anchor .mini-icon-link {
+      display: inline-block;
+    }
+    .markdown h1 tt,
+    .markdown h1 code,
+    .markdown h2 tt,
+    .markdown h2 code,
+    .markdown h3 tt,
+    .markdown h3 code,
+    .markdown h4 tt,
+    .markdown h4 code,
+    .markdown h5 tt,
+    .markdown h5 code,
+    .markdown h6 tt,
+    .markdown h6 code {
+      font-size: inherit;
+    }
+    .markdown h1 {
+      font-size: 28px;
+    }
+    .markdown h2 {
+      font-size: 24px;
+    }
+    .markdown h3 {
+      font-size: 18px;
+    }
+    .markdown h4 {
+      font-size: 16px;
+    }
+    .markdown h5 {
+      font-size: 14px;
+    }
+    .markdown h6 {
+      font-size: 12px;
+    }
+    .markdown p,
+    .markdown blockquote,
+    .markdown ul,
+    .markdown ol,
+    .markdown dl,
+    .markdown table,
+    .markdown pre {
+      margin: 10px 0;
+    }
+    .markdown > h2:first-child,
+    .markdown > h1:first-child,
+    .markdown > h1:first-child + h2,
+    .markdown > h3:first-child,
+    .markdown > h4:first-child,
+    .markdown > h5:first-child,
+    .markdown > h6:first-child {
+      margin-top: 0;
+      padding-top: 0;
+    }
+    .markdown a:first-child,
+    .markdown > h1,
+    .markdown a:first-child,
+    .markdown > h2,
+    .markdown a:first-child,
+    .markdown > h3,
+    .markdown a:first-child,
+    .markdown > h4,
+    .markdown a:first-child,
+    .markdown > h5,
+    .markdown a:first-child,
+    .markdown > h6 {
+      margin-top: 0;
+      padding-top: 0;
+    }
+    .markdown h1 + p,
+    .markdown h2 + p,
+    .markdown h3 + p,
+    .markdown h4 + p,
+    .markdown h5 + p,
+    .markdown h6 + p {
+      margin-top: 0;
+    }
+    .markdown li p.first {
+      display: inline-block;
+    }
+    .markdown ul,
+    .markdown ol {
+      padding-left: 2em;
+    }
+    .markdown ul.no-list,
+    .markdown ol.no-list {
+      list-style-type: none;
+      padding: 0;
+    }
+    .markdown ul li > :first-child,
+    .markdown ol li > :first-child {
+      margin-top: 0;
+    }
+    .markdown ul ul,
+    .markdown ul ol,
+    .markdown ol ol,
+    .markdown ol ul {
+      margin-bottom: 0;
+    }
+    .markdown dl {
+      padding: 0;
+    }
+    .markdown dl dt {
+      font-size: 14px;
+      font-style: italic;
+      font-weight: bold;
+      margin: 15px 0 5px;
+      padding: 0;
+    }
+    .markdown dl dt:first-child {
+      padding: 0;
+    }
+    .markdown dl dt > :first-child {
+      margin-top: 0;
+    }
+    .markdown dl dt > :last-child {
+      margin-bottom: 0;
+    }
+    .markdown dl dd {
+      margin: 0 0 15px;
+      padding: 0 15px;
+    }
+    .markdown dl dd > *:first-child {
+      margin-top: 0;
+    }
+    .markdown dl dd > *last-child {
+      margin-bottom: none;
+    }
+
+    .markdown blockquote {
+      border-left: solid 4px #dddddd;
+      color: #777777;
+      padding-left: 15px;
+    }
+
+    .markdown blockquote > * :first-child {
+      margin-top: 0;
+    }
+
+    .markdown blockquote > * :last-child {
+      margin-bottom: 0;
+    }
+
+    .markdown table th {
+      font-weight: bold;
+    }
+    .markdown table th,
+    .markdown table td {
+      padding: 6px 13px;
+    }
+    .markdown table tr {
+      background-color: #ffffff;
+    }
+    .markdown table tr:nth-child(2n) {
+      background-color: #f0f0f0;
+    }
+    .markdown img {
+      max-width: 100%;
+    }
+    .markdown span.frame {
+      display: block;
+      overflow: hidden;
+    }
+    .markdown span.frame > span {
+      border: 1px solid #dddddd;
+      display: block;
+      float: left;
+      margin: 13px 0 0;
+      overflow: hidden;
+      padding: 7px;
+      width: auto;
+    }
+    .markdown span.frame span img {
+      display: block;
+      float: left;
+    }
+    .markdown span.frame span span {
+      clear: both;
+      color: #333333;
+      display: block;
+      padding: 5px;
+    }
+
+    .markdown span.align-center {
+      clear: both;
+      display: block;
+      overflow: hidden;
+      text-align: center;
+    }
+
+    .markdown span.align-center > span {
+      display: block;
+      margin: 13px auto;
+      overflow: hidden;
+    }
+
+    .markdown span.align-center span img {
+      margin: 0 auto;
+      text-align: center;
+    }
+
+    .markdown span.align-right {
+      clear: both;
+      display: block;
+      overflow: hidden;
+    }
+
+    .markdown span.align-right > span {
+      display: block;
+      margin: 13px auto;
+      overflow: hidden;
+      text-align: right;
+    }
+
+    .markdown span.align-right img {
+      margin: 0;
+      text-align: right;
+    }
+
+    .markdown span.float-left {
+      display: block;
+      float: left;
+      margin-right: 13px;
+      overflow: hidden;
+    }
+
+    .markdown span.float-left > span {
+      margin: 13px auto;
+    }
+
+    .markdown span.float-right {
+      display: block;
+      float: right;
+      margin-left: 13px;
+      overflow: hidden;
+    }
+
+    .markdown span.float-right > span {
+      display: block;
+      margin: 13px auto;
+      overflow: hidden;
+      text-align: right;
+    }
+
+    .markdown code,
+    .markdown tt {
+      border: 1px solid #eaeaea;
+      border-radius: 3px;
+      margin: 0 2px;
+      padding: 0 5px;
+    }
+    .markdown pre > code {
+      background-color: transparent;
+      border: none;
+      margin: 0;
+      padding: 0;
+      white-space: pre;
+    }
+    .markdown .highlight pre,
+    .markdown pre {
+      border: 1px solid #ccc;
+      border-radius: 3px;
+      font-size: max(0.9em, 14px);
+      line-height: 19px;
+      overflow: auto;
+      padding: 6px 10px;
+    }
+
+    .markdown pre code,
+    .markdown pre tt {
+      background-color: #f8f8f8;
+      border: none;
+    }
+
+    .markdown {
+      text-align: justify;
+      overflow-y: hidden;
+      tab-size: 4;
+      word-spacing: normal;
+      word-break: break-all;
+    }
+    .markdown pre {
+      display: block;
+      width: 100%;
+      padding: 15px;
+      margin: 0;
+      border: none;
+      border-radius: none;
+      background-color: #222 !important;
+      overflow-x: auto;
+      color: #fff;
+    }
+    .markdown pre code {
+      background-color: #222 !important;
+      width: 100%;
+    }
+    .markdown a {
+      text-decoration: underline;
+      color: var(--chakra-colors-blue-600);
+    }
+    .markdown table {
+      border-collapse: separate;
+      border-spacing: 0;
+      color: #718096;
+    }
+    .markdown table thead tr:first-child th {
+      border-bottom-width: 1px;
+      border-left-width: 1px;
+      border-top-width: 1px;
+      border-color: #ccc;
+      background-color: #edf2f7;
+      overflow: hidden;
+    }
+    .markdown table thead tr:first-child th:first-child {
+      border-top-left-radius: 0.375rem;
+    }
+    .markdown table thead tr:first-child th:last-child {
+      border-right-width: 1px;
+      border-top-right-radius: 0.375rem;
+    }
+    .markdown td {
+      border-bottom-width: 1px;
+      border-left-width: 1px;
+      border-color: #ccc;
+    }
+    .markdown td:last-of-type {
+      border-right-width: 1px;
+    }
+    .markdown tbody tr:last-child {
+      overflow: hidden;
+    }
+    .markdown tbody tr:last-child td:first-child {
+      border-bottom-left-radius: 0.375rem;
+    }
+    .markdown tbody tr:last-child td:last-child {
+      border-bottom-right-radius: 0.375rem;
+    }
+    .markdown p {
+      text-align: justify;
+      white-space: pre-wrap;
+    }
+    code[class*='language-'] {
+      color: #d4d4d4;
+      text-shadow: none;
+      direction: ltr;
+      text-align: left;
+      white-space: pre;
+      word-spacing: normal;
+      word-break: normal;
+      line-height: 1.5;
+      -moz-tab-size: 4;
+      -o-tab-size: 4;
+      tab-size: 4;
+      -webkit-hyphens: none;
+      -moz-hyphens: none;
+      hyphens: none;
+    }
+
+    pre[class*='language-'] {
+      color: #d4d4d4;
+      text-shadow: none;
+      direction: ltr;
+      text-align: left;
+      white-space: pre;
+      word-spacing: normal;
+      word-break: normal;
+      line-height: 1.5;
+      -moz-tab-size: 4;
+      -o-tab-size: 4;
+      tab-size: 4;
+      -webkit-hyphens: none;
+      -moz-hyphens: none;
+      hyphens: none;
+      padding: 1em;
+      margin: 0.5em0;
+      overflow: auto;
+      background: #1e1e1e;
+    }
+
+    code[class*='language-'] ::selection,
+    code[class*='language-']::selection,
+    pre[class*='language-'] ::selection,
+    pre[class*='language-']::selection {
+      text-shadow: none;
+      background: #264f78;
+    }
+
+    :not(pre) > code[class*='language-'] {
+      padding: 0.1em 0.3em;
+      border-radius: 0.3em;
+      color: #db4c69;
+      background: #1e1e1e;
+    }
+
+    .namespace {
+      opacity: 0.7;
+    }
+
+    .doctype.doctype-tag {
+      color: #569cd6;
+    }
+
+    .doctype.name {
+      color: #9cdcfe;
+    }
+
+    comment {
+      color: #6a9955;
+    }
+
+    prolog {
+      color: #6a9955;
+    }
+
+    .language-html .language-css .token.punctuation,
+    .language-html .language-javascript .token.punctuation {
+      color: #d4d4d4;
+    }
+
+    punctuation {
+      color: #d4d4d4;
+    }
+
+    boolean {
+      color: #569cd6;
+    }
+
+    constant {
+      color: #9cdcfe;
+    }
+
+    inserted {
+      color: #b5cea8;
+    }
+
+    number {
+      color: #b5cea8;
+    }
+
+    property {
+      color: #9cdcfe;
+    }
+
+    symbol {
+      color: #b5cea8;
+    }
+
+    tag {
+      color: #569cd6;
+    }
+
+    unit {
+      color: #b5cea8;
+    }
+
+    attr-name {
+      color: #9cdcfe;
+    }
+
+    builtin {
+      color: #ce9178;
+    }
+
+    char {
+      color: #ce9178;
+    }
+
+    deleted {
+      color: #ce9178;
+    }
+
+    selector {
+      color: #d7ba7d;
+    }
+
+    string {
+      color: #ce9178;
+    }
+
+    .language-css .token.string.url {
+      text-decoration: underline;
+    }
+
+    entity {
+      color: #569cd6;
+    }
+
+    operator {
+      color: #d4d4d4;
+    }
+
+    operator.arrow {
+      color: #569cd6;
+    }
+
+    atrule {
+      color: #ce9178;
+    }
+
+    atrule.rule {
+      color: #c586c0;
+    }
+
+    atrule.url {
+      color: #9cdcfe;
+    }
+
+    atrule.url.function {
+      color: #dcdcaa;
+    }
+
+    atrule.url.punctuation {
+      color: #d4d4d4;
+    }
+    keyword {
+      color: #569cd6;
+    }
+
+    keyword.control-flow {
+      color: #c586c0;
+    }
+
+    keyword.module {
+      color: #c586c0;
+    }
+
+    function {
+      color: #dcdcaa;
+    }
+
+    function.maybe-class-name {
+      color: #dcdcaa;
+    }
+
+    regex {
+      color: #d16969;
+    }
+
+    important {
+      color: #569cd6;
+    }
+
+    italic {
+      font-style: italic;
+    }
+
+    class-name {
+      color: #4ec9b0;
+    }
+
+    maybe-class-name {
+      color: #4ec9b0;
+    }
+
+    console {
+      color: #9cdcfe;
+    }
+
+    parameter {
+      color: #9cdcfe;
+    }
+
+    interpolation {
+      color: #9cdcfe;
+    }
+    punctuation.interpolation-punctuation {
+      color: #569cd6;
+    }
+    exports.maybe-class-name {
+      color: #9cdcfe;
+    }
+    imports.maybe-class-name {
+      color: #9cdcfe;
+    }
+    variable {
+      color: #9cdcfe;
+    }
+    escape {
+      color: #d7ba7d;
+    }
+    tag.punctuation {
+      color: grey;
+    }
+    cdata {
+      color: grey;
+    }
+    attr-value {
+      color: #ce9178;
+    }
+    attr-value.punctuation {
+      color: #ce9178;
+    }
+    attr-value.punctuation.attr-equals {
+      color: #d4d4d4;
+    }
+    namespace {
+      color: #4ec9b0;
+    }
+    code[class*='language-javascript'],
+    code[class*='language-jsx'],
+    code[class*='language-tsx'],
+    code[class*='language-typescript'] {
+      color: #9cdcfe;
+    }
+
+    pre[class*='language-javascript'],
+    pre[class*='language-jsx'],
+    pre[class*='language-tsx'],
+    pre[class*='language-typescript'] {
+      color: #9cdcfe;
+    }
+
+    code[class*='language-css'] {
+      color: #ce9178;
+    }
+
+    pre[class*='language-css'] {
+      color: #ce9178;
+    }
+
+    code[class*='language-html'] {
+      color: #d4d4d4;
+    }
+
+    pre[class*='language-html'] {
+      color: #d4d4d4;
+    }
+
+    .language-regex .token.anchor {
+      color: #dcdcaa;
+    }
+
+    .language-html .token.punctuation {
+      color: grey;
+    }
+
+    pre[class*='language-'] > code[class*='language-'] {
+      position: relative;
+      z-index: 1;
+    }
+
+    .line-highlight.line-highlight {
+      background: #f7ebc6;
+      box-shadow: inset 5px 0 0 #f7d87c;
+      z-index: 0;
+    }
+    * {
+      box-sizing: border-box;
+    }
+    body,
+    h1,
+    h2,
+    h3,
+    h4,
+    hr,
+    p,
+    blockquote,
+    dl,
+    dt,
+    dd,
+    ul,
+    ol,
+    li,
+    pre,
+    form,
+    fieldset,
+    legend,
+    button,
+    input,
+    textarea,
+    th,
+    td,
+    svg  {
+      margin: 0;
+    }
+    body,
+    html {
+      font-size: 16px;
+      background-color: #fff;
+      color: rgba(0, 0, 0, 0.64);
+      font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Helvetica, Arial, sans-serif,
+        'Apple Color Emoji', 'Segoe UI Emoji', 'Segoe UI Symbol';
+    }
+    pre,
+    code,
+    kbd,
+    samp {
+      font-family: SFMono-Regular, Menlo, Monaco, Consolas, monospace;
+      font-size: 1em;
+    }
+    ::-webkit-scrollbar,
+    ::-webkit-scrollbar {
+      width: 10px;
+      height: 10px;
+    }
+    ::-webkit-scrollbar-track,
+    ::-webkit-scrollbar-track {
+      background: transparent;
+      border-radius: 2px;
+    }
+    ::-webkit-scrollbar-thumb,
+    ::-webkit-scrollbar-thumb {
+      background: #bfbfbf;
+      border-radius: 10px;
+    }
+    ::-webkit-scrollbar-thumb:hover,
+    ::-webkit-scrollbar-thumb:hover {
+      background: #999;
+    }
+  </style>
+  <style>
+    .chat-item {
+      display: flex;
+      align-items: flex-start;
+      padding: 20px 16px;
+      border-bottom: 1px solid rgba(0, 0, 0, 0.1);
+      justify-content: center;
+    }
+    .chat-item img {
+      width: 30px;
+      max-height: 50px;
+      object-fit: contain;
+      margin-right: 10px;
+    }
+    .chat-item:nth-child(even) {
+      background-color: #f6f6f6;
+    }
+    .chat-item:nth-child(odd) {
+      background-color: #ffffff;
+    }
+    .markdown {
+      overflow-x: hidden;
+      max-width: 800px;
+      width: 100%;
+    }
+    @media (max-width: 900px) {
+      html {
+        font-size: 14px;
+      }
+      ::-webkit-scrollbar,
+      ::-webkit-scrollbar {
+        width: 2px;
+        height: 2px;
+      }
+      .chat-item img {
+        width: 20px;
+        max-height: 40px;
+        margin-right: 4px;
+      }
+    }
+  </style>
+  <body>{{CHAT_CONTENT}}</body>
+</html>
+`;

src/constants/data.ts

@@ -1,6 +0,0 @@
-import type { DataType } from '@/types/data';
-
-export const DataTypeTextMap: Record<DataType, string> = {
-  QA: '问答拆分',
-  abstract: '摘要总结'
-};

src/constants/model.ts

@@ -1,64 +1,36 @@
 import type { ModelSchema } from '@/types/mongoSchema';
 
-export enum ModelDataStatusEnum {
-  ready = 'ready',
-  waiting = 'waiting'
-}
-
 export const embeddingModel = 'text-embedding-ada-002';
-export enum ChatModelEnum {
+export type EmbeddingModelType = 'text-embedding-ada-002';
+
+export enum OpenAiChatEnum {
   'GPT35' = 'gpt-3.5-turbo',
   'GPT4' = 'gpt-4',
   'GPT432k' = 'gpt-4-32k'
 }
 
-export enum ModelNameEnum {
-  GPT35 = 'gpt-3.5-turbo',
-  VECTOR_GPT = 'VECTOR_GPT'
-}
+export type ChatModelType = `${OpenAiChatEnum}`;
 
-export const Model2ChatModelMap: Record<`${ModelNameEnum}`, `${ChatModelEnum}`> = {
-  [ModelNameEnum.GPT35]: 'gpt-3.5-turbo',
-  [ModelNameEnum.VECTOR_GPT]: 'gpt-3.5-turbo'
-};
-
-export type ModelConstantsData = {
-  icon: 'model' | 'dbModel';
-  name: string;
-  model: `${ModelNameEnum}`;
-  trainName: string; // 空字符串代表不能训练
-  contextMaxToken: number;
-  maxTemperature: number;
-  price: number; // 多少钱 / 1token，单位: 0.00001元
-};
-
-export const modelList: ModelConstantsData[] = [
-  {
-    icon: 'model',
-    name: 'chatGPT',
-    model: ModelNameEnum.GPT35,
-    trainName: '',
+export const ChatModelMap = {
+  [OpenAiChatEnum.GPT35]: {
+    name: 'ChatGpt',
     contextMaxToken: 4096,
     maxTemperature: 1.5,
     price: 3
   },
-  {
-    icon: 'dbModel',
-    name: '知识库',
-    model: ModelNameEnum.VECTOR_GPT,
-    trainName: 'vector',
-    contextMaxToken: 4096,
-    maxTemperature: 1,
-    price: 3
+  [OpenAiChatEnum.GPT4]: {
+    name: 'Gpt4',
+    contextMaxToken: 8000,
+    maxTemperature: 1.5,
+    price: 30
+  },
+  [OpenAiChatEnum.GPT432k]: {
+    name: 'Gpt4-32k',
+    contextMaxToken: 32000,
+    maxTemperature: 1.5,
+    price: 30
   }
-];
-
-export enum TrainingStatusEnum {
-  pending = 'pending',
-  succeed = 'succeed',
-  errored = 'errored',
-  canceled = 'canceled'
-}
+};
 
 export enum ModelStatusEnum {
   running = 'running',
@@ -86,6 +58,11 @@ export const formatModelStatus = {
   }
 };
 
+export enum ModelDataStatusEnum {
+  ready = 'ready',
+  waiting = 'waiting'
+}
+
 export const ModelDataStatusMap: Record<`${ModelDataStatusEnum}`, string> = {
   ready: '训练完成',
   waiting: '训练中'
@@ -120,23 +97,24 @@ export const ModelVectorSearchModeMap: Record<
 };
 
 export const defaultModel: ModelSchema = {
-  _id: '',
-  userId: '',
-  name: 'modelName',
-  avatar: '',
+  _id: 'modelId',
+  userId: 'userId',
+  name: '模型名称',
+  avatar: '/icon/logo.png',
   status: ModelStatusEnum.pending,
   updateTime: Date.now(),
-  trainingTimes: 0,
-  systemPrompt: '',
-  intro: '',
-  temperature: 5,
-  search: {
-    mode: ModelVectorSearchModeEnum.hightSimilarity
+  chat: {
+    useKb: false,
+    searchMode: ModelVectorSearchModeEnum.hightSimilarity,
+    systemPrompt: '',
+    temperature: 0,
+    chatModel: OpenAiChatEnum.GPT35
   },
-  service: {
-    trainId: '',
-    chatModel: ModelNameEnum.GPT35,
-    modelName: ModelNameEnum.GPT35
+  share: {
+    isShare: false,
+    isShareDetail: false,
+    intro: '',
+    collection: 0
   },
   security: {
     domain: ['*'],

src/constants/redis.ts

@@ -1,6 +0,0 @@
-export const VecModelDataPrefix = 'model:data';
-export const VecModelDataIdx = `idx:${VecModelDataPrefix}:hash`;
-export enum ModelDataStatusEnum {
-  ready = 'ready',
-  waiting = 'waiting'
-}

src/constants/responseCode.ts

@@ -1,20 +0,0 @@
-export const ERROR_CODE: { [key: number]: string } = {
-  400: '请求失败',
-  401: '无权访问',
-  403: '紧张访问',
-  404: '请求不存在',
-  405: '请求方法错误',
-  406: '请求的格式错误',
-  410: '资源已删除',
-  422: '验证错误',
-  500: '服务器发生错误',
-  502: '网关错误',
-  503: '服务器暂时过载或维护',
-  504: '网关超时'
-};
-
-export const TOKEN_ERROR_CODE: { [key: number]: string } = {
-  506: '请先登录',
-  507: '请重新登录',
-  508: '登录已过期'
-};

src/hooks/usePagination.tsx

@@ -41,10 +41,6 @@ export const usePagination = <T = any,>({
     }
   });
 
-  useEffect(() => {
-    mutate(1);
-  }, []);
-
   const Pagination = useCallback(() => {
     return (
       <Flex alignItems={'center'} justifyContent={'end'}>
@@ -93,6 +89,10 @@ export const usePagination = <T = any,>({
     );
   }, [maxPage, mutate, pageNum]);
 
+  useEffect(() => {
+    mutate(1);
+  }, []);
+
   return {
     pageNum,
     pageSize,

src/pages/_app.tsx

@@ -51,8 +51,9 @@ export default function App({ Component, pageProps }: AppProps) {
         />
         <link rel="icon" href="/favicon.ico" />
       </Head>
-      <Script src="/js/qrcode.min.js" strategy="afterInteractive"></Script>
-      <Script src="/js/pdf.js" strategy="afterInteractive"></Script>
+      <Script src="/js/qrcode.min.js" strategy="lazyOnload"></Script>
+      <Script src="/js/pdf.js" strategy="lazyOnload"></Script>
+      <Script src="/js/html2pdf.bundle.min.js" strategy="lazyOnload"></Script>
       <QueryClientProvider client={queryClient}>
         <ChakraProvider theme={theme}>
           <ColorModeScript initialColorMode={theme.config.initialColorMode} />

src/pages/api/chat/chat.ts

@@ -0,0 +1,132 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { connectToDatabase } from '@/service/mongo';
+import { authChat } from '@/service/utils/auth';
+import { modelServiceToolMap } from '@/service/utils/chat';
+import { ChatItemSimpleType } from '@/types/chat';
+import { jsonRes } from '@/service/response';
+import { PassThrough } from 'stream';
+import { ChatModelMap, ModelVectorSearchModeMap } from '@/constants/model';
+import { pushChatBill } from '@/service/events/pushBill';
+import { resStreamResponse } from '@/service/utils/chat';
+import { searchKb } from '@/service/plugins/searchKb';
+import { ChatRoleEnum } from '@/constants/chat';
+
+/* 发送提示词 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  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 {
+    const { chatId, prompt, modelId } = req.body as {
+      prompt: ChatItemSimpleType;
+      modelId: string;
+      chatId: '' | string;
+    };
+
+    const { authorization } = req.headers;
+    if (!modelId || !prompt) {
+      throw new Error('缺少参数');
+    }
+
+    await connectToDatabase();
+    let startTime = Date.now();
+
+    const { model, showModelDetail, content, userApiKey, systemApiKey, userId } = await authChat({
+      modelId,
+      chatId,
+      authorization
+    });
+
+    const modelConstantsData = ChatModelMap[model.chat.chatModel];
+
+    // 读取对话内容
+    const prompts = [...content, prompt];
+
+    // 使用了知识库搜索
+    if (model.chat.useKb) {
+      const { code, searchPrompt } = await searchKb({
+        userApiKey,
+        systemApiKey,
+        text: prompt.value,
+        similarity: ModelVectorSearchModeMap[model.chat.searchMode]?.similarity,
+        model,
+        userId
+      });
+
+      // search result is empty
+      if (code === 201) {
+        return res.send(searchPrompt?.value);
+      }
+
+      searchPrompt && prompts.unshift(searchPrompt);
+    } else {
+      // 没有用知识库搜索，仅用系统提示词
+      model.chat.systemPrompt &&
+        prompts.unshift({
+          obj: ChatRoleEnum.System,
+          value: model.chat.systemPrompt
+        });
+    }
+
+    // 计算温度
+    const temperature = (modelConstantsData.maxTemperature * (model.chat.temperature / 10)).toFixed(
+      2
+    );
+    // console.log(filterPrompts);
+
+    // 发出请求
+    const { streamResponse } = await modelServiceToolMap[model.chat.chatModel].chatCompletion({
+      apiKey: userApiKey || systemApiKey,
+      temperature: +temperature,
+      messages: prompts,
+      stream: true
+    });
+
+    console.log('api response time:', `${(Date.now() - startTime) / 1000}s`);
+
+    step = 1;
+
+    const { totalTokens, finishMessages } = await resStreamResponse({
+      model: model.chat.chatModel,
+      res,
+      stream,
+      chatResponse: streamResponse,
+      prompts,
+      systemPrompt:
+        showModelDetail && prompts[0].obj === ChatRoleEnum.System ? prompts[0].value : ''
+    });
+
+    // 只有使用平台的 key 才计费
+    pushChatBill({
+      isPay: !userApiKey,
+      chatModel: model.chat.chatModel,
+      userId,
+      chatId,
+      textLen: finishMessages.map((item) => item.value).join('').length,
+      tokens: totalTokens
+    });
+  } catch (err: any) {
+    if (step === 1) {
+      // 直接结束流
+      console.log('error，结束');
+      stream.destroy();
+    } else {
+      res.status(500);
+      jsonRes(res, {
+        code: 500,
+        error: err
+      });
+    }
+  }
+}

src/pages/api/chat/chatGpt.ts

@@ -1,126 +0,0 @@
-import type { NextApiRequest, NextApiResponse } from 'next';
-import { connectToDatabase } from '@/service/mongo';
-import { getOpenAIApi, authChat } from '@/service/utils/auth';
-import { httpsAgent, openaiChatFilter } from '@/service/utils/tools';
-import { ChatItemType } from '@/types/chat';
-import { jsonRes } from '@/service/response';
-import { PassThrough } from 'stream';
-import { modelList } from '@/constants/model';
-import { pushChatBill } from '@/service/events/pushBill';
-import { gpt35StreamResponse } from '@/service/utils/openai';
-
-/* 发送提示词 */
-export default async function handler(req: NextApiRequest, res: NextApiResponse) {
-  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 {
-    const { chatId, prompt, modelId } = req.body as {
-      prompt: ChatItemType;
-      modelId: string;
-      chatId: '' | string;
-    };
-
-    const { authorization } = req.headers;
-    if (!modelId || !prompt) {
-      throw new Error('缺少参数');
-    }
-
-    await connectToDatabase();
-    let startTime = Date.now();
-
-    const { model, content, userApiKey, systemKey, userId } = await authChat({
-      modelId,
-      chatId,
-      authorization
-    });
-
-    const modelConstantsData = modelList.find((item) => item.model === model.service.modelName);
-    if (!modelConstantsData) {
-      throw new Error('模型加载异常');
-    }
-
-    // 读取对话内容
-    const prompts = [...content, prompt];
-
-    // 如果有系统提示词，自动插入
-    if (model.systemPrompt) {
-      prompts.unshift({
-        obj: 'SYSTEM',
-        value: model.systemPrompt
-      });
-    }
-
-    // 控制在 tokens 数量，防止超出
-    const filterPrompts = openaiChatFilter({
-      model: model.service.chatModel,
-      prompts,
-      maxTokens: modelConstantsData.contextMaxToken - 500
-    });
-
-    // 计算温度
-    const temperature = modelConstantsData.maxTemperature * (model.temperature / 10);
-    // console.log(filterPrompts);
-    // 获取 chatAPI
-    const chatAPI = getOpenAIApi(userApiKey || systemKey);
-    // 发出请求
-    const chatResponse = await chatAPI.createChatCompletion(
-      {
-        model: model.service.chatModel,
-        temperature,
-        messages: filterPrompts,
-        frequency_penalty: 0.5, // 越大，重复内容越少
-        presence_penalty: -0.5, // 越大，越容易出现新内容
-        stream: true,
-        stop: ['.!?。']
-      },
-      {
-        timeout: 40000,
-        responseType: 'stream',
-        httpsAgent: httpsAgent(!userApiKey)
-      }
-    );
-
-    console.log('api response time:', `${(Date.now() - startTime) / 1000}s`);
-
-    step = 1;
-
-    const { responseContent } = await gpt35StreamResponse({
-      res,
-      stream,
-      chatResponse
-    });
-
-    // 只有使用平台的 key 才计费
-    pushChatBill({
-      isPay: !userApiKey,
-      modelName: model.service.modelName,
-      userId,
-      chatId,
-      messages: filterPrompts.concat({ role: 'assistant', content: responseContent })
-    });
-  } catch (err: any) {
-    if (step === 1) {
-      // 直接结束流
-      console.log('error，结束');
-      stream.destroy();
-    } else {
-      res.status(500);
-      jsonRes(res, {
-        code: 500,
-        error: err
-      });
-    }
-  }
-}

src/pages/api/chat/delChatRecordByContentId.ts

@@ -1,17 +1,17 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, Chat } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    const { chatId, index } = req.query as { chatId: string; index: string };
+    const { chatId, contentId } = req.query as { chatId: string; contentId: string };
     const { authorization } = req.headers;
 
     if (!authorization) {
       throw new Error('无权操作');
     }
-    if (!chatId || !index) {
+    if (!chatId || !contentId) {
       throw new Error('缺少参数');
     }
 
@@ -26,30 +26,13 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       throw new Error('找不到对话');
     }
 
-    // 重新计算 index，跳过已经被删除的内容
-    let unDeleteIndex = +index;
-    let deletedIndex = 0;
-    for (deletedIndex = 0; deletedIndex < chatRecord.content.length; deletedIndex++) {
-      if (!chatRecord.content[deletedIndex].deleted) {
-        unDeleteIndex--;
-        if (unDeleteIndex < 0) {
-          break;
-        }
-      }
-    }
-
-    // 删除最一条数据库记录, 也就是预发送的那一条
+    // 删除一条数据库记录
     await Chat.updateOne(
       {
         _id: chatId,
         userId
       },
-      {
-        $set: {
-          [`content.${deletedIndex}.deleted`]: true,
-          updateTime: Date.now()
-        }
-      }
+      { $pull: { content: { _id: contentId } } }
     );
 
     jsonRes(res);

src/pages/api/chat/getHistory.ts

@@ -1,7 +1,7 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, Chat } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 
 /* 获取历史记录 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {

src/pages/api/chat/init.ts

@@ -2,7 +2,7 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, Chat } from '@/service/mongo';
 import type { InitChatResponse } from '@/api/response/chat';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 import { ChatItemType } from '@/types/chat';
 import { authModel } from '@/service/utils/auth';
 import mongoose from 'mongoose';
@@ -22,7 +22,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     await connectToDatabase();
 
     // 获取 model 数据
-    const { model } = await authModel(modelId, userId);
+    const { model } = await authModel({ modelId, userId, authUser: false, authOwner: false });
 
     // 历史记录
     let history: ChatItemType[] = [];
@@ -30,21 +30,29 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     if (chatId) {
       // 获取 chat.content 数据
       history = await Chat.aggregate([
-        { $match: { _id: new mongoose.Types.ObjectId(chatId) } },
-        { $unwind: '$content' },
-        { $match: { 'content.deleted': false } },
-        { $sort: { 'content._id': -1 } },
-        { $limit: 50 },
+        {
+          $match: {
+            _id: new mongoose.Types.ObjectId(chatId),
+            userId: new mongoose.Types.ObjectId(userId)
+          }
+        },
         {
           $project: {
-            id: '$content._id',
+            content: {
+              $slice: ['$content', -50] // 返回 content 数组的最后50个元素
+            }
+          }
+        },
+        { $unwind: '$content' },
+        {
+          $project: {
+            _id: '$content._id',
             obj: '$content.obj',
-            value: '$content.value'
+            value: '$content.value',
+            systemPrompt: '$content.systemPrompt'
           }
         }
       ]);
-
-      history.reverse();
     }
 
     jsonRes<InitChatResponse>(res, {
@@ -53,9 +61,8 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
         modelId: modelId,
         name: model.name,
         avatar: model.avatar,
-        intro: model.intro,
-        modelName: model.service.modelName,
-        chatModel: model.service.chatModel,
+        intro: model.share.intro,
+        chatModel: model.chat.chatModel,
         history
       }
     });

src/pages/api/chat/removeHistory.ts

@@ -2,7 +2,7 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { ChatItemType } from '@/types/chat';
 import { connectToDatabase, Chat } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 
 /* 获取历史记录 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {

src/pages/api/chat/saveChat.ts

@@ -3,7 +3,8 @@ import { jsonRes } from '@/service/response';
 import { ChatItemType } from '@/types/chat';
 import { connectToDatabase, Chat } from '@/service/mongo';
 import { authModel } from '@/service/utils/auth';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
+import mongoose from 'mongoose';
 
 /* 聊天内容存存储 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
@@ -23,13 +24,16 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     await connectToDatabase();
 
     const content = prompts.map((item) => ({
+      _id: new mongoose.Types.ObjectId(item._id),
       obj: item.obj,
-      value: item.value
+      value: item.value,
+      systemPrompt: item.systemPrompt
     }));
 
+    await authModel({ modelId, userId, authOwner: false });
+
     // 没有 chatId, 创建一个对话
     if (!chatId) {
-      await authModel(modelId, userId);
       const { _id } = await Chat.create({
         userId,
         modelId,

src/pages/api/chat/vectorGpt.ts

@@ -1,188 +0,0 @@
-import type { NextApiRequest, NextApiResponse } from 'next';
-import { connectToDatabase } from '@/service/mongo';
-import { authChat } from '@/service/utils/auth';
-import { httpsAgent, systemPromptFilter, openaiChatFilter } from '@/service/utils/tools';
-import { ChatItemType } from '@/types/chat';
-import { jsonRes } from '@/service/response';
-import { PassThrough } from 'stream';
-import {
-  modelList,
-  ModelVectorSearchModeMap,
-  ModelVectorSearchModeEnum,
-  ModelDataStatusEnum
-} from '@/constants/model';
-import { pushChatBill } from '@/service/events/pushBill';
-import { openaiCreateEmbedding, gpt35StreamResponse } from '@/service/utils/openai';
-import dayjs from 'dayjs';
-import { PgClient } from '@/service/pg';
-
-/* 发送提示词 */
-export default async function handler(req: NextApiRequest, res: NextApiResponse) {
-  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 {
-    const { modelId, chatId, prompt } = req.body as {
-      modelId: string;
-      chatId: '' | string;
-      prompt: ChatItemType;
-    };
-
-    const { authorization } = req.headers;
-    if (!modelId || !prompt) {
-      throw new Error('缺少参数');
-    }
-
-    await connectToDatabase();
-    let startTime = Date.now();
-
-    const { model, content, userApiKey, systemKey, userId } = await authChat({
-      modelId,
-      chatId,
-      authorization
-    });
-
-    const modelConstantsData = modelList.find((item) => item.model === model.service.modelName);
-    if (!modelConstantsData) {
-      throw new Error('模型加载异常');
-    }
-
-    // 读取对话内容
-    const prompts = [...content, prompt];
-
-    // 获取提示词的向量
-    const { vector: promptVector, chatAPI } = await openaiCreateEmbedding({
-      isPay: !userApiKey,
-      apiKey: userApiKey || systemKey,
-      userId,
-      text: prompt.value
-    });
-
-    // 相似度搜素
-    const similarity = ModelVectorSearchModeMap[model.search.mode]?.similarity || 0.22;
-    const vectorSearch = await PgClient.select<{ id: string; q: string; a: string }>('modelData', {
-      fields: ['id', 'q', 'a'],
-      where: [
-        ['status', ModelDataStatusEnum.ready],
-        'AND',
-        ['model_id', model._id],
-        'AND',
-        `vector <=> '[${promptVector}]' < ${similarity}`
-      ],
-      order: [{ field: 'vector', mode: `<=> '[${promptVector}]'` }],
-      limit: 20
-    });
-
-    const formatRedisPrompt: string[] = vectorSearch.rows.map((item) => `${item.q}\n${item.a}`);
-
-    /* 高相似度+退出，无法匹配时直接退出 */
-    if (
-      formatRedisPrompt.length === 0 &&
-      model.search.mode === ModelVectorSearchModeEnum.hightSimilarity
-    ) {
-      return res.send('对不起，你的问题不在知识库中。');
-    }
-    /* 高相似度+无上下文，不添加额外知识 */
-    if (
-      formatRedisPrompt.length === 0 &&
-      model.search.mode === ModelVectorSearchModeEnum.noContext
-    ) {
-      prompts.unshift({
-        obj: 'SYSTEM',
-        value: model.systemPrompt
-      });
-    } else {
-      // 有匹配情况下，system 添加知识库内容。
-      // 系统提示词过滤，最多 2500 tokens
-      const systemPrompt = systemPromptFilter({
-        model: model.service.chatModel,
-        prompts: formatRedisPrompt,
-        maxTokens: 2500
-      });
-
-      prompts.unshift({
-        obj: 'SYSTEM',
-        value: `
-${model.systemPrompt}
-${
-  model.search.mode === ModelVectorSearchModeEnum.hightSimilarity
-    ? `你只能从知识库选择内容回答.不在知识库内容拒绝回复`
-    : ''
-}
-知识库内容为: 当前时间为${dayjs().format('YYYY/MM/DD HH:mm:ss')}\n${systemPrompt}'
-`
-      });
-    }
-
-    // 控制在 tokens 数量，防止超出
-    const filterPrompts = openaiChatFilter({
-      model: model.service.chatModel,
-      prompts,
-      maxTokens: modelConstantsData.contextMaxToken - 500
-    });
-
-    // console.log(filterPrompts);
-    // 计算温度
-    const temperature = modelConstantsData.maxTemperature * (model.temperature / 10);
-
-    // 发出请求
-    const chatResponse = await chatAPI.createChatCompletion(
-      {
-        model: model.service.chatModel,
-        temperature,
-        messages: filterPrompts,
-        frequency_penalty: 0.5, // 越大，重复内容越少
-        presence_penalty: -0.5, // 越大，越容易出现新内容
-        stream: true
-      },
-      {
-        timeout: 40000,
-        responseType: 'stream',
-        httpsAgent: httpsAgent(!userApiKey)
-      }
-    );
-
-    console.log('api response time:', `${(Date.now() - startTime) / 1000}s`);
-
-    step = 1;
-
-    const { responseContent } = await gpt35StreamResponse({
-      res,
-      stream,
-      chatResponse
-    });
-
-    // 只有使用平台的 key 才计费
-    pushChatBill({
-      isPay: !userApiKey,
-      modelName: model.service.modelName,
-      userId,
-      chatId,
-      messages: filterPrompts.concat({ role: 'assistant', content: responseContent })
-    });
-    // jsonRes(res);
-  } catch (err: any) {
-    if (step === 1) {
-      // 直接结束流
-      console.log('error，结束');
-      stream.destroy();
-    } else {
-      res.status(500);
-      jsonRes(res, {
-        code: 500,
-        error: err
-      });
-    }
-  }
-}

src/pages/api/data/delData.ts

@@ -1,47 +0,0 @@
-// 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

@@ -1,48 +0,0 @@
-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

@@ -1,71 +0,0 @@
-// 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 mongoose from 'mongoose';
-
-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 datalist = await Data.aggregate<DataListItem>([
-      {
-        $match: {
-          userId: new mongoose.Types.ObjectId(userId),
-          isDeleted: false
-        }
-      },
-      {
-        $sort: { createTime: -1 } // 按照创建时间倒序排列
-      },
-      {
-        $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(res, {
-      data: datalist
-    });
-  } catch (err) {
-    jsonRes(res, {
-      code: 500,
-      error: err
-    });
-  }
-}

src/pages/api/data/postData.ts

@@ -1,35 +0,0 @@
-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 { DataType } from '@/types/data';
-
-export default async function handler(req: NextApiRequest, res: NextApiResponse) {
-  try {
-    let { name, type } = req.body as { name: string; type: DataType };
-    if (!name || !type) {
-      throw new Error('参数错误');
-    }
-    await connectToDatabase();
-
-    const { authorization } = req.headers;
-
-    const userId = await authToken(authorization);
-
-    // 生成 data 集合
-    const data = await Data.create({
-      userId,
-      name,
-      type
-    });
-
-    jsonRes(res, {
-      data: data._id
-    });
-  } catch (err) {
-    jsonRes(res, {
-      code: 500,
-      error: err
-    });
-  }
-}

src/pages/api/data/putDataName.ts

@@ -1,37 +0,0 @@
-// 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

@@ -1,69 +0,0 @@
-import type { NextApiRequest, NextApiResponse } from 'next';
-import { jsonRes } from '@/service/response';
-import { connectToDatabase, DataItem, Data } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
-import { generateQA } from '@/service/events/generateQA';
-import { generateAbstract } from '@/service/events/generateAbstract';
-import { countChatTokens } from '@/utils/tools';
-
-/* 拆分数据成QA */
-export default async function handler(req: NextApiRequest, res: NextApiResponse) {
-  try {
-    const { text, dataId } = req.body as { text: string; dataId: string };
-    if (!text || !dataId) {
-      throw new Error('参数错误');
-    }
-    await connectToDatabase();
-
-    const { authorization } = req.headers;
-
-    const userId = await authToken(authorization);
-
-    const DataRecord = await Data.findById(dataId);
-
-    if (!DataRecord) {
-      throw new Error('找不到数据集');
-    }
-    const replaceText = text.replace(/[\\n]+/g, ' ');
-
-    // 文本拆分成 chunk
-    let chunks = replaceText.match(/[^!?.。]+[!?.。]/g) || [];
-
-    const dataItems: any[] = [];
-    let splitText = '';
-
-    chunks.forEach((chunk) => {
-      splitText += chunk;
-      const tokens = countChatTokens({ messages: [{ role: 'system', content: splitText }] });
-      if (tokens >= 780) {
-        dataItems.push({
-          userId,
-          dataId,
-          type: DataRecord.type,
-          text: splitText,
-          status: 1
-        });
-        splitText = '';
-      }
-    });
-
-    // 批量插入数据
-    await DataItem.insertMany(dataItems);
-
-    try {
-      generateQA();
-      generateAbstract();
-    } catch (error) {
-      error;
-    }
-
-    jsonRes(res, {
-      data: { chunks, replaceText }
-    });
-  } catch (err) {
-    jsonRes(res, {
-      code: 500,
-      error: err
-    });
-  }
-}

src/pages/api/model/create.ts

@@ -2,15 +2,14 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
-import { ModelStatusEnum, modelList, ModelNameEnum, Model2ChatModelMap } from '@/constants/model';
+import { authToken } from '@/service/utils/auth';
+import { ModelStatusEnum } from '@/constants/model';
 import { Model } from '@/service/models/model';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
-    const { name, serviceModelName } = req.body as {
+    const { name } = req.body as {
       name: string;
-      serviceModelName: `${ModelNameEnum}`;
     };
     const { authorization } = req.headers;
 
@@ -18,46 +17,32 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       throw new Error('无权操作');
     }
 
-    if (!name || !serviceModelName) {
+    if (!name) {
       throw new Error('缺少参数');
     }
 
     // 凭证校验
     const userId = await authToken(authorization);
 
-    const modelItem = modelList.find((item) => item.model === serviceModelName);
-
-    if (!modelItem) {
-      throw new Error('模型不存在');
-    }
-
     await connectToDatabase();
 
     // 上限校验
     const authCount = await Model.countDocuments({
       userId
     });
-    if (authCount >= 20) {
-      throw new Error('上限 20 个模型');
+    if (authCount >= 30) {
+      throw new Error('上限 30 个模型');
     }
 
     // 创建模型
     const response = await Model.create({
       name,
       userId,
-      status: ModelStatusEnum.running,
-      service: {
-        trainId: '',
-        chatModel: Model2ChatModelMap[modelItem.model], // 聊天时用的模型
-        modelName: modelItem.model // 最底层的模型，不会变，用于计费等核心操作
-      }
+      status: ModelStatusEnum.running
     });
 
-    // 根据 id 获取模型信息
-    const model = await Model.findById(response._id);
-
     jsonRes(res, {
-      data: model
+      data: response._id
     });
   } catch (err) {
     jsonRes(res, {

src/pages/api/model/data/delModelDataById.ts

@@ -1,6 +1,6 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 import { PgClient } from '@/service/pg';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {

src/pages/api/model/data/exportModelData.ts

@@ -1,7 +1,7 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 import { PgClient } from '@/service/pg';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {

src/pages/api/model/data/fetchingUrlData.ts

@@ -1,9 +1,9 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 import axios from 'axios';
-import { httpsAgent } from '@/service/utils/tools';
+import { axiosConfig } from '@/service/utils/tools';
 
 /**
  * 读取网站的内容
@@ -22,7 +22,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
     const data = await axios
       .get(url, {
-        httpsAgent: httpsAgent(false)
+        httpsAgent: axiosConfig().httpsAgent
       })
       .then((res) => res.data as string);
 

src/pages/api/model/data/getModelData.ts

@@ -1,9 +1,10 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 import { PgClient } from '@/service/pg';
 import type { PgModelDataItemType } from '@/types/pg';
+import { authModel } from '@/service/utils/auth';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
@@ -36,9 +37,14 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
 
     await connectToDatabase();
 
+    const { model } = await authModel({
+      userId,
+      modelId,
+      authOwner: false
+    });
+
     const where: any = [
-      ['user_id', userId],
-      'AND',
+      ...(model.share.isShareDetail ? [] : [['user_id', userId], 'AND']),
       ['model_id', modelId],
       ...(searchText ? ['AND', `(q LIKE '%${searchText}%' OR a LIKE '%${searchText}%')`] : [])
     ];

src/pages/api/model/data/getSplitData.ts

@@ -1,7 +1,7 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, SplitData, Model } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 
 /* 拆分数据成QA */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {

src/pages/api/model/data/pushModelDataCsv.ts

@@ -1,10 +1,11 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { connectToDatabase, Model } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { connectToDatabase } from '@/service/mongo';
+import { authToken } from '@/service/utils/auth';
 import { generateVector } from '@/service/events/generateVector';
 import { ModelDataStatusEnum } from '@/constants/model';
 import { PgClient } from '@/service/pg';
+import { authModel } from '@/service/utils/auth';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
@@ -28,15 +29,11 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     await connectToDatabase();
 
     // 验证是否是该用户的 model
-    const model = await Model.findOne({
-      _id: modelId,
-      userId
+    await authModel({
+      userId,
+      modelId
     });
 
-    if (!model) {
-      throw new Error('无权操作该模型');
-    }
-
     // 去重
     const searchRes = await Promise.allSettled(
       data.map(async ([q, a]) => {

src/pages/api/model/data/pushModelDataInput.ts

@@ -1,10 +1,11 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { connectToDatabase, Model } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { connectToDatabase } from '@/service/mongo';
+import { authToken } from '@/service/utils/auth';
 import { ModelDataSchema } from '@/types/mongoSchema';
 import { generateVector } from '@/service/events/generateVector';
 import { PgClient } from '@/service/pg';
+import { authModel } from '@/service/utils/auth';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
@@ -28,15 +29,11 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     await connectToDatabase();
 
     // 验证是否是该用户的 model
-    const model = await Model.findOne({
-      _id: modelId,
-      userId
+    await authModel({
+      userId,
+      modelId
     });
 
-    if (!model) {
-      throw new Error('无权操作该模型');
-    }
-
     // 插入记录
     await PgClient.insert('modelData', {
       values: data.map((item) => [

src/pages/api/model/data/putModelData.ts

@@ -1,7 +1,7 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { authToken } from '@/service/utils/tools';
-import { ModelDataStatusEnum } from '@/constants/redis';
+import { authToken } from '@/service/utils/auth';
+import { ModelDataStatusEnum } from '@/constants/model';
 import { generateVector } from '@/service/events/generateVector';
 import { PgClient } from '@/service/pg';
 

src/pages/api/model/data/splitData.ts

@@ -1,7 +1,7 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, SplitData, Model } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 import { generateVector } from '@/service/events/generateVector';
 import { generateQA } from '@/service/events/generateQA';
 import { PgClient } from '@/service/pg';

src/pages/api/model/del.ts

@@ -1,12 +1,9 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { Chat, Model, Training, connectToDatabase } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
-import { getUserApiOpenai } from '@/service/utils/openai';
-import { TrainingStatusEnum } from '@/constants/model';
-import { TrainingItemType } from '@/types/training';
-import { httpsAgent } from '@/service/utils/tools';
+import { Chat, Model, connectToDatabase } from '@/service/mongo';
+import { authToken } from '@/service/utils/auth';
 import { PgClient } from '@/service/pg';
+import { authModel } from '@/service/utils/auth';
 
 /* 获取我的模型 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
@@ -25,18 +22,14 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     // 凭证校验
     const userId = await authToken(authorization);
 
+    await connectToDatabase();
+
     // 验证是否是该用户的 model
-    const model = await Model.findOne({
-      _id: modelId,
+    await authModel({
+      modelId,
       userId
     });
 
-    if (!model) {
-      throw new Error('无权操作该模型');
-    }
-
-    await connectToDatabase();
-
     // 删除 pg 中所有该模型的数据
     await PgClient.delete('modelData', {
       where: [['user_id', userId], 'AND', ['model_id', modelId]]
@@ -47,31 +40,6 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       modelId
     });
 
-    // 查看是否正在训练
-    const training: TrainingItemType | null = await Training.findOne({
-      modelId,
-      status: TrainingStatusEnum.pending
-    });
-
-    // 如果正在训练，需要删除openai上的相关信息
-    if (training) {
-      const { openai } = await getUserApiOpenai(userId);
-      // 获取训练记录
-      const tuneRecord = await openai.retrieveFineTune(training.tuneId, {
-        httpsAgent: httpsAgent(false)
-      });
-
-      // 删除训练文件
-      openai.deleteFile(tuneRecord.data.training_files[0].id, { httpsAgent: httpsAgent(false) });
-      // 取消训练
-      openai.cancelFineTune(training.tuneId, { httpsAgent: httpsAgent(false) });
-    }
-
-    // 删除对应训练记录
-    await Training.deleteMany({
-      modelId
-    });
-
     // 删除模型
     await Model.deleteOne({
       _id: modelId,

src/pages/api/model/detail.tsx

@@ -1,9 +1,8 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
-import { Model } from '@/service/models/model';
-import type { ModelSchema } from '@/types/mongoSchema';
+import { authToken } from '@/service/utils/auth';
+import { authModel } from '@/service/utils/auth';
 
 /* 获取我的模型 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
@@ -14,7 +13,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       throw new Error('无权操作');
     }
 
-    const { modelId } = req.query;
+    const { modelId } = req.query as { modelId: string };
 
     if (!modelId) {
       throw new Error('参数错误');
@@ -25,16 +24,12 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
 
     await connectToDatabase();
 
-    // 根据 userId 获取模型信息
-    const model = await Model.findOne<ModelSchema>({
+    const { model } = await authModel({
+      modelId,
       userId,
-      _id: modelId
+      authOwner: false
     });
 
-    if (!model) {
-      throw new Error('模型不存在');
-    }
-
     jsonRes(res, {
       data: model
     });

src/pages/api/model/list.ts

@@ -1,10 +1,10 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 import { Model } from '@/service/models/model';
 
-/* 获取我的模型 */
+/* 获取模型列表 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
     const { authorization } = req.headers;

src/pages/api/model/share/collection.ts

@@ -0,0 +1,44 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Collection, Model } from '@/service/mongo';
+import { authToken } from '@/service/utils/auth';
+
+/* 模型收藏切换 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
+  try {
+    const { modelId } = req.query as { modelId: string };
+
+    if (!modelId) {
+      throw new Error('缺少参数');
+    }
+    // 凭证校验
+    const userId = await authToken(req.headers.authorization);
+
+    await connectToDatabase();
+
+    const collectionRecord = await Collection.findOne({
+      userId,
+      modelId
+    });
+
+    if (collectionRecord) {
+      await Collection.findByIdAndRemove(collectionRecord._id);
+    } else {
+      await Collection.create({
+        userId,
+        modelId
+      });
+    }
+
+    await Model.findByIdAndUpdate(modelId, {
+      'share.collection': await Collection.countDocuments({ modelId })
+    });
+
+    jsonRes(res);
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

src/pages/api/model/share/getCollection.ts

@@ -0,0 +1,37 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Collection } from '@/service/mongo';
+import { authToken } from '@/service/utils/auth';
+import type { ShareModelItem } from '@/types/model';
+
+/* 获取模型列表 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
+  try {
+    // 凭证校验
+    const userId = await authToken(req.headers.authorization);
+
+    await connectToDatabase();
+
+    // get my collections
+    const collections = await Collection.find({
+      userId
+    }).populate('modelId', '_id avatar name userId share');
+
+    jsonRes<ShareModelItem[]>(res, {
+      data: collections
+        .map((item: any) => ({
+          _id: item.modelId?._id,
+          avatar: item.modelId?.avatar || '/icon/logo.png',
+          name: item.modelId?.name || '',
+          userId: item.modelId?.userId || '',
+          share: item.modelId?.share || {},
+          isCollection: true
+        }))
+        .filter((item) => item.share.isShare)
+    });
+  } catch (err) {
+    jsonRes(res, {
+      data: []
+    });
+  }
+}

src/pages/api/model/share/getModels.ts

@@ -0,0 +1,60 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { connectToDatabase, Collection, Model } from '@/service/mongo';
+import { authToken } from '@/service/utils/auth';
+import type { PagingData } from '@/types';
+import type { ShareModelItem } from '@/types/model';
+
+/* 获取模型列表 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
+  try {
+    const {
+      searchText = '',
+      pageNum = 1,
+      pageSize = 20
+    } = req.body as { searchText: string; pageNum: number; pageSize: number };
+
+    await connectToDatabase();
+
+    const regex = new RegExp(searchText, 'i');
+
+    const where = {
+      $and: [
+        { 'share.isShare': true },
+        { $or: [{ name: { $regex: regex } }, { 'share.intro': { $regex: regex } }] }
+      ]
+    };
+
+    // 获取被分享的模型
+    const [models, total] = await Promise.all([
+      Model.find(where, '_id avatar name userId share')
+        .sort({
+          'share.collection': -1
+        })
+        .limit(pageSize)
+        .skip((pageNum - 1) * pageSize),
+      Model.countDocuments(where)
+    ]);
+
+    jsonRes<PagingData<ShareModelItem>>(res, {
+      data: {
+        pageNum,
+        pageSize,
+        data: models.map((item) => ({
+          _id: item._id,
+          avatar: item.avatar || '/icon/logo.png',
+          name: item.name,
+          userId: item.userId,
+          share: item.share,
+          isCollection: false
+        })),
+        total
+      }
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}

src/pages/api/model/train/getTrainings.ts

@@ -1,52 +0,0 @@
-import type { NextApiRequest, NextApiResponse } from 'next';
-import { jsonRes } from '@/service/response';
-import { connectToDatabase, Training } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
-
-// 关闭next默认的bodyParser处理方式
-export const config = {
-  api: {
-    bodyParser: false
-  }
-};
-
-/* 获取模型训练记录 */
-export default async function handler(req: NextApiRequest, res: NextApiResponse) {
-  try {
-    const { authorization } = req.headers;
-
-    if (!authorization) {
-      throw new Error('无权操作');
-    }
-    const { modelId } = req.query;
-    if (!modelId) {
-      throw new Error('参数错误');
-    }
-    await authToken(authorization);
-
-    await connectToDatabase();
-
-    /* 获取 modelId 下的 training 记录 */
-    const records = await Training.find({
-      modelId
-    });
-
-    jsonRes(res, {
-      data: records
-    });
-  } catch (err: any) {
-    /* 清除上传的文件，关闭训练记录 */
-    // @ts-ignore
-    if (openai) {
-      // @ts-ignore
-      uploadFileId && openai.deleteFile(uploadFileId);
-      // @ts-ignore
-      trainId && openai.cancelFineTune(trainId);
-    }
-
-    jsonRes(res, {
-      code: 500,
-      error: err
-    });
-  }
-}

src/pages/api/model/train/putTrainStatus.ts

@@ -1,106 +0,0 @@
-import type { NextApiRequest, NextApiResponse } from 'next';
-import { jsonRes } from '@/service/response';
-import { connectToDatabase, Model, Training } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
-import { getUserApiOpenai } from '@/service/utils/openai';
-import type { ModelSchema } from '@/types/mongoSchema';
-import { TrainingItemType } from '@/types/training';
-import { ModelStatusEnum, TrainingStatusEnum } from '@/constants/model';
-import { OpenAiTuneStatusEnum } from '@/service/constants/training';
-import { httpsAgent } from '@/service/utils/tools';
-
-/* 更新训练状态 */
-export default async function handler(req: NextApiRequest, res: NextApiResponse) {
-  try {
-    const { authorization } = req.headers;
-
-    if (!authorization) {
-      throw new Error('无权操作');
-    }
-    const { modelId } = req.query as { modelId: string };
-    if (!modelId) {
-      throw new Error('参数错误');
-    }
-    const userId = await authToken(authorization);
-
-    await connectToDatabase();
-
-    // 获取模型
-    const model = await Model.findById<ModelSchema>(modelId);
-
-    if (!model || model.status !== 'training') {
-      throw new Error('模型不在训练中');
-    }
-
-    // 查询正在训练中的训练记录
-    const training: TrainingItemType | null = await Training.findOne({
-      modelId,
-      status: 'pending'
-    });
-
-    if (!training) {
-      throw new Error('找不到训练记录');
-    }
-
-    // 用户的 openai 实例
-    const { openai } = await getUserApiOpenai(userId);
-
-    // 获取 openai 的训练情况
-    const { data } = await openai.retrieveFineTune(training.tuneId, {
-      httpsAgent: httpsAgent(false)
-    });
-    // console.log(data);
-    if (data.status === OpenAiTuneStatusEnum.succeeded) {
-      // 删除训练文件
-      openai.deleteFile(data.training_files[0].id, { httpsAgent: httpsAgent(false) });
-
-      // 更新模型状态和模型内容
-      await Model.findByIdAndUpdate(modelId, {
-        status: ModelStatusEnum.running,
-        updateTime: new Date(),
-        service: {
-          ...model.service,
-          trainId: data.fine_tuned_model, // 训练完后，再次训练和对话使用的 model 是一样的
-          chatModel: data.fine_tuned_model
-        }
-      });
-      // 更新训练数据
-      await Training.findByIdAndUpdate(training._id, {
-        status: TrainingStatusEnum.succeed
-      });
-
-      return jsonRes(res, {
-        data: '模型微调完成'
-      });
-    }
-
-    /* 取消微调 */
-    if (data.status === OpenAiTuneStatusEnum.cancelled) {
-      // 删除训练文件
-      openai.deleteFile(data.training_files[0].id, { httpsAgent: httpsAgent(false) });
-
-      // 更新模型
-      await Model.findByIdAndUpdate(modelId, {
-        status: ModelStatusEnum.running,
-        updateTime: new Date()
-      });
-      // 更新训练数据
-      await Training.findByIdAndUpdate(training._id, {
-        status: TrainingStatusEnum.canceled
-      });
-
-      return jsonRes(res, {
-        data: '模型微调已取消'
-      });
-    }
-
-    jsonRes(res, {
-      data: '模型还在训练中'
-    });
-  } catch (err: any) {
-    jsonRes(res, {
-      code: 500,
-      error: err
-    });
-  }
-}

src/pages/api/model/train/train.ts

@@ -1,130 +0,0 @@
-// 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, Model, Training } from '@/service/mongo';
-import formidable from 'formidable';
-import { authToken } from '@/service/utils/tools';
-import { getUserApiOpenai } from '@/service/utils/openai';
-import { join } from 'path';
-import fs from 'fs';
-import type { ModelSchema } from '@/types/mongoSchema';
-import type { OpenAIApi } from 'openai';
-import { ModelStatusEnum, TrainingStatusEnum } from '@/constants/model';
-import { httpsAgent } from '@/service/utils/tools';
-
-// 关闭next默认的bodyParser处理方式
-export const config = {
-  api: {
-    bodyParser: false
-  }
-};
-
-/* 上传文件，开始微调 */
-export default async function handler(req: NextApiRequest, res: NextApiResponse) {
-  let openai: OpenAIApi, trainId: string, uploadFileId: string;
-
-  try {
-    const { authorization } = req.headers;
-
-    if (!authorization) {
-      throw new Error('无权操作');
-    }
-    const { modelId } = req.query;
-
-    if (!modelId) {
-      throw new Error('参数错误');
-    }
-    const userId = await authToken(authorization);
-
-    await connectToDatabase();
-
-    // 获取模型的状态
-    const model = await Model.findById<ModelSchema>(modelId);
-
-    if (!model || model.status !== 'running') {
-      throw new Error('模型正忙');
-    }
-
-    // const trainingType = model.service.modelType
-    const trainingType = model.service.trainId; // 目前都默认是 openai text-davinci-03
-
-    // 获取用户的 API Key 实例化后的对象
-    const user = await getUserApiOpenai(userId);
-    openai = user.openai;
-
-    // 接收文件并保存
-    const form = formidable({
-      uploadDir: join(process.cwd(), 'public/trainData'),
-      keepExtensions: true
-    });
-
-    const { files } = await new Promise<{
-      fields: formidable.Fields;
-      files: formidable.Files;
-    }>((resolve, reject) => {
-      form.parse(req, (err, fields, files) => {
-        if (err) return reject(err);
-        resolve({ fields, files });
-      });
-    });
-    const file = files.file;
-
-    // 上传文件到 openai
-    // @ts-ignore
-    const uploadRes = await openai.createFile(
-      // @ts-ignore
-      fs.createReadStream(file.filepath),
-      'fine-tune',
-      { httpsAgent: httpsAgent(false) }
-    );
-    uploadFileId = uploadRes.data.id; // 记录上传文件的 ID
-
-    // 开始训练
-    const trainRes = await openai.createFineTune(
-      {
-        training_file: uploadFileId,
-        model: trainingType,
-        suffix: model.name,
-        n_epochs: 4
-      },
-      { httpsAgent: httpsAgent(false) }
-    );
-
-    trainId = trainRes.data.id; // 记录训练 ID
-
-    // 创建训练记录
-    await Training.create({
-      serviceName: 'openai',
-      tuneId: trainId,
-      status: TrainingStatusEnum.pending,
-      modelId
-    });
-
-    // 修改模型状态
-    await Model.findByIdAndUpdate(modelId, {
-      $inc: {
-        trainingTimes: +1
-      },
-      updateTime: new Date(),
-      status: ModelStatusEnum.training
-    });
-
-    jsonRes(res, {
-      data: 'start training'
-    });
-  } catch (err: any) {
-    /* 清除上传的文件，关闭训练记录 */
-    // @ts-ignore
-    if (openai) {
-      // @ts-ignore
-      uploadFileId && openai.deleteFile(uploadFileId, { httpsAgent: httpsAgent(false) });
-      // @ts-ignore
-      trainId && openai.cancelFineTune(trainId, { httpsAgent: httpsAgent(false) });
-    }
-
-    jsonRes(res, {
-      code: 500,
-      error: err
-    });
-  }
-}

src/pages/api/model/update.ts

@@ -1,15 +1,15 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 import { Model } from '@/service/models/model';
 import type { ModelUpdateParams } from '@/types/model';
+import { authModel } from '@/service/utils/auth';
 
 /* 获取我的模型 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
-    const { name, search, service, security, systemPrompt, intro, temperature } =
-      req.body as ModelUpdateParams;
+    const { name, avatar, chat, share, security } = req.body as ModelUpdateParams;
     const { modelId } = req.query as { modelId: string };
     const { authorization } = req.headers;
 
@@ -17,7 +17,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       throw new Error('无权操作');
     }
 
-    if (!name || !service || !security || !modelId) {
+    if (!name || !chat || !security || !modelId) {
       throw new Error('参数错误');
     }
 
@@ -26,6 +26,11 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
 
     await connectToDatabase();
 
+    await authModel({
+      modelId,
+      userId
+    });
+
     // 更新模型
     await Model.updateOne(
       {
@@ -34,11 +39,11 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       },
       {
         name,
-        systemPrompt,
-        intro,
-        temperature,
-        search,
-        // service,
+        avatar,
+        chat,
+        'share.isShare': share.isShare,
+        'share.isShareDetail': share.isShareDetail,
+        'share.intro': share.intro,
         security
       }
     );

src/pages/api/openapi/chat/chat.ts

@@ -0,0 +1,148 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { connectToDatabase } from '@/service/mongo';
+import { authOpenApiKey, authModel } from '@/service/utils/auth';
+import { modelServiceToolMap, resStreamResponse } from '@/service/utils/chat';
+import { ChatItemSimpleType } from '@/types/chat';
+import { jsonRes } from '@/service/response';
+import { PassThrough } from 'stream';
+import { ChatModelMap, ModelVectorSearchModeMap } from '@/constants/model';
+import { pushChatBill } from '@/service/events/pushBill';
+import { searchKb } from '@/service/plugins/searchKb';
+import { ChatRoleEnum } from '@/constants/chat';
+
+/* 发送提示词 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  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 {
+    const {
+      prompts,
+      modelId,
+      isStream = true
+    } = req.body as {
+      prompts: ChatItemSimpleType[];
+      modelId: string;
+      isStream: boolean;
+    };
+
+    if (!prompts || !modelId) {
+      throw new Error('缺少参数');
+    }
+    if (!Array.isArray(prompts)) {
+      throw new Error('prompts is not array');
+    }
+    if (prompts.length > 30 || prompts.length === 0) {
+      throw new Error('prompts length range 1-30');
+    }
+
+    await connectToDatabase();
+    let startTime = Date.now();
+
+    /* 凭证校验 */
+    const { apiKey, userId } = await authOpenApiKey(req);
+
+    const { model } = await authModel({
+      userId,
+      modelId
+    });
+
+    const modelConstantsData = ChatModelMap[model.chat.chatModel];
+
+    // 使用了知识库搜索
+    if (model.chat.useKb) {
+      const similarity = ModelVectorSearchModeMap[model.chat.searchMode]?.similarity || 0.22;
+
+      const { code, searchPrompt } = await searchKb({
+        systemApiKey: apiKey,
+        text: prompts[prompts.length - 1].value,
+        similarity,
+        model,
+        userId
+      });
+
+      // search result is empty
+      if (code === 201) {
+        return res.send(searchPrompt?.value);
+      }
+
+      searchPrompt && prompts.unshift(searchPrompt);
+    } else {
+      // 没有用知识库搜索，仅用系统提示词
+      if (model.chat.systemPrompt) {
+        prompts.unshift({
+          obj: ChatRoleEnum.System,
+          value: model.chat.systemPrompt
+        });
+      }
+    }
+
+    // 计算温度
+    const temperature = (modelConstantsData.maxTemperature * (model.chat.temperature / 10)).toFixed(
+      2
+    );
+
+    // 发出请求
+    const { streamResponse, responseMessages, responseText, totalTokens } =
+      await modelServiceToolMap[model.chat.chatModel].chatCompletion({
+        apiKey,
+        temperature: +temperature,
+        messages: prompts,
+        stream: isStream
+      });
+
+    console.log('api response time:', `${(Date.now() - startTime) / 1000}s`);
+
+    let textLen = 0;
+    let tokens = totalTokens;
+
+    if (isStream) {
+      step = 1;
+      const { finishMessages, totalTokens } = await resStreamResponse({
+        model: model.chat.chatModel,
+        res,
+        stream,
+        chatResponse: streamResponse,
+        prompts
+      });
+      textLen = finishMessages.map((item) => item.value).join('').length;
+      tokens = totalTokens;
+    } else {
+      textLen = responseMessages.map((item) => item.value).join('').length;
+      jsonRes(res, {
+        data: responseText
+      });
+    }
+
+    pushChatBill({
+      isPay: true,
+      chatModel: model.chat.chatModel,
+      userId,
+      textLen,
+      tokens
+    });
+  } catch (err: any) {
+    if (step === 1) {
+      // 直接结束流
+      console.log('error，结束');
+      stream.destroy();
+    } else {
+      res.status(500);
+      jsonRes(res, {
+        code: 500,
+        error: err
+      });
+    }
+  }
+}

src/pages/api/openapi/chat/chatGpt.ts

@@ -1,147 +0,0 @@
-import type { NextApiRequest, NextApiResponse } from 'next';
-import { connectToDatabase, Model } from '@/service/mongo';
-import { getOpenAIApi } from '@/service/utils/auth';
-import { httpsAgent, openaiChatFilter, authOpenApiKey } from '@/service/utils/tools';
-import { ChatCompletionRequestMessage, ChatCompletionRequestMessageRoleEnum } from 'openai';
-import { ChatItemType } from '@/types/chat';
-import { jsonRes } from '@/service/response';
-import { PassThrough } from 'stream';
-import { modelList } from '@/constants/model';
-import { pushChatBill } from '@/service/events/pushBill';
-import { gpt35StreamResponse } from '@/service/utils/openai';
-
-/* 发送提示词 */
-export default async function handler(req: NextApiRequest, res: NextApiResponse) {
-  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 {
-    const {
-      prompts,
-      modelId,
-      isStream = true
-    } = req.body as {
-      prompts: ChatItemType[];
-      modelId: string;
-      isStream: boolean;
-    };
-
-    if (!prompts || !modelId) {
-      throw new Error('缺少参数');
-    }
-    if (!Array.isArray(prompts)) {
-      throw new Error('prompts is not array');
-    }
-    if (prompts.length > 30 || prompts.length === 0) {
-      throw new Error('prompts length range 1-30');
-    }
-
-    await connectToDatabase();
-    let startTime = Date.now();
-
-    const { apiKey, userId } = await authOpenApiKey(req);
-
-    const model = await Model.findOne({
-      _id: modelId,
-      userId
-    });
-
-    if (!model) {
-      throw new Error('无权使用该模型');
-    }
-
-    const modelConstantsData = modelList.find((item) => item.model === model.service.modelName);
-    if (!modelConstantsData) {
-      throw new Error('模型加载异常');
-    }
-
-    // 如果有系统提示词，自动插入
-    if (model.systemPrompt) {
-      prompts.unshift({
-        obj: 'SYSTEM',
-        value: model.systemPrompt
-      });
-    }
-
-    // 控制在 tokens 数量，防止超出
-    const filterPrompts = openaiChatFilter({
-      model: model.service.chatModel,
-      prompts,
-      maxTokens: modelConstantsData.contextMaxToken - 500
-    });
-
-    // console.log(filterPrompts);
-    // 计算温度
-    const temperature = modelConstantsData.maxTemperature * (model.temperature / 10);
-
-    // 获取 chatAPI
-    const chatAPI = getOpenAIApi(apiKey);
-    // 发出请求
-    const chatResponse = await chatAPI.createChatCompletion(
-      {
-        model: model.service.chatModel,
-        temperature,
-        messages: filterPrompts,
-        frequency_penalty: 0.5, // 越大，重复内容越少
-        presence_penalty: -0.5, // 越大，越容易出现新内容
-        stream: isStream,
-        stop: ['.!?。']
-      },
-      {
-        timeout: 40000,
-        responseType: isStream ? 'stream' : 'json',
-        httpsAgent: httpsAgent(true)
-      }
-    );
-
-    console.log('api response time:', `${(Date.now() - startTime) / 1000}s`);
-
-    step = 1;
-    let responseContent = '';
-
-    if (isStream) {
-      const streamResponse = await gpt35StreamResponse({
-        res,
-        stream,
-        chatResponse
-      });
-      responseContent = streamResponse.responseContent;
-    } else {
-      responseContent = chatResponse.data.choices?.[0]?.message?.content || '';
-      jsonRes(res, {
-        data: responseContent
-      });
-    }
-
-    // 只有使用平台的 key 才计费
-    pushChatBill({
-      isPay: true,
-      modelName: model.service.modelName,
-      userId,
-      messages: filterPrompts.concat({ role: 'assistant', content: responseContent })
-    });
-  } catch (err: any) {
-    if (step === 1) {
-      // 直接结束流
-      console.log('error，结束');
-      stream.destroy();
-    } else {
-      res.status(500);
-      jsonRes(res, {
-        code: 500,
-        error: err
-      });
-    }
-  }
-}

src/pages/api/openapi/chat/lafGpt.ts

@@ -1,20 +1,14 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { connectToDatabase, Model } from '@/service/mongo';
-import { getOpenAIApi } from '@/service/utils/auth';
-import { authOpenApiKey } from '@/service/utils/tools';
-import { httpsAgent, openaiChatFilter, systemPromptFilter } from '@/service/utils/tools';
-import { ChatItemType } from '@/types/chat';
+import { authOpenApiKey } from '@/service/utils/auth';
+import { resStreamResponse, modelServiceToolMap } from '@/service/utils/chat';
+import { ChatItemSimpleType } from '@/types/chat';
 import { jsonRes } from '@/service/response';
 import { PassThrough } from 'stream';
-import {
-  ModelNameEnum,
-  modelList,
-  ModelVectorSearchModeMap,
-  ChatModelEnum
-} from '@/constants/model';
+import { ChatModelMap, ModelVectorSearchModeMap } from '@/constants/model';
 import { pushChatBill } from '@/service/events/pushBill';
-import { openaiCreateEmbedding, gpt35StreamResponse } from '@/service/utils/openai';
-import { PgClient } from '@/service/pg';
+import { searchKb } from '@/service/plugins/searchKb';
+import { ChatRoleEnum } from '@/constants/chat';
 
 /* 发送提示词 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
@@ -38,7 +32,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       modelId,
       isStream = true
     } = req.body as {
-      prompt: ChatItemType;
+      prompt: ChatItemSimpleType;
       modelId: string;
       isStream: boolean;
     };
@@ -59,26 +53,20 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       throw new Error('找不到模型');
     }
 
-    const modelConstantsData = modelList.find((item) => item.model === ModelNameEnum.VECTOR_GPT);
-    if (!modelConstantsData) {
-      throw new Error('模型已下架');
-    }
+    const modelConstantsData = ChatModelMap[model.chat.chatModel];
+
     console.log('laf gpt start');
 
-    // 获取 chatAPI
-    const chatAPI = getOpenAIApi(apiKey);
-
     // 请求一次 chatgpt 拆解需求
-    const promptResponse = await chatAPI.createChatCompletion(
-      {
-        model: ChatModelEnum.GPT35,
-        temperature: 0,
-        frequency_penalty: 0.5, // 越大，重复内容越少
-        presence_penalty: -0.5, // 越大，越容易出现新内容
-        messages: [
-          {
-            role: 'system',
-            content: `服务端逻辑生成器.根据用户输入的需求,拆解成 laf 云函数实现的步骤,只返回步骤,按格式返回步骤: 1.\n2.\n3.\n ......
+    const { responseText: resolveText, totalTokens: resolveTokens } = await modelServiceToolMap[
+      model.chat.chatModel
+    ].chatCompletion({
+      apiKey,
+      temperature: 0,
+      messages: [
+        {
+          obj: ChatRoleEnum.System,
+          value: `服务端逻辑生成器.根据用户输入的需求,拆解成 laf 云函数实现的步骤,只返回步骤,按格式返回步骤: 1.\n2.\n3.\n ......
 下面是一些例子:
 一个 hello world 例子
 1. 返回字符串: "hello world"
@@ -111,111 +99,67 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 5. 获取当前时间,记录为 updateTime.
 6. 更新数据库数据,表为"blogs",更新符合 blogId 的记录的内容为{blogText, tags, updateTime}.
 7. 返回结果 "更新博客记录成功"`
-          },
-          {
-            role: 'user',
-            content: prompt.value
-          }
-        ]
-      },
-      {
-        timeout: 180000,
-        httpsAgent: httpsAgent(true)
-      }
-    );
-
-    const promptResolve = promptResponse.data.choices?.[0]?.message?.content || '';
-    if (!promptResolve) {
-      throw new Error('gpt 异常');
-    }
-
-    prompt.value += ` ${promptResolve}`;
-    console.log('prompt resolve success, time:', `${(Date.now() - startTime) / 1000}s`);
-
-    // 获取提示词的向量
-    const { vector: promptVector } = await openaiCreateEmbedding({
-      isPay: true,
-      apiKey,
-      userId,
-      text: prompt.value
+        },
+        {
+          obj: ChatRoleEnum.Human,
+          value: prompt.value
+        }
+      ],
+      stream: false
     });
 
+    prompt.value += ` ${resolveText}`;
+    console.log('prompt resolve success, time:', `${(Date.now() - startTime) / 1000}s`);
+
     // 读取对话内容
     const prompts = [prompt];
 
-    // 相似度搜索
-    const similarity = ModelVectorSearchModeMap[model.search.mode]?.similarity || 0.22;
-    const vectorSearch = await PgClient.select<{ id: string; q: string; a: string }>('modelData', {
-      fields: ['id', 'q', 'a'],
-      order: [{ field: 'vector', mode: `<=> '[${promptVector}]'` }],
-      where: [
-        ['model_id', model._id],
-        'AND',
-        ['user_id', userId],
-        'AND',
-        `vector <=> '[${promptVector}]' < ${similarity}`
-      ],
-      limit: 30
+    // 获取向量匹配到的提示词
+    const { searchPrompt } = await searchKb({
+      systemApiKey: apiKey,
+      similarity: ModelVectorSearchModeMap[model.chat.searchMode]?.similarity,
+      text: prompt.value,
+      model,
+      userId
     });
 
-    const formatRedisPrompt: string[] = vectorSearch.rows.map((item) => `${item.q}\n${item.a}`);
+    searchPrompt && prompts.unshift(searchPrompt);
 
-    // system 筛选，最多 2500 tokens
-    const systemPrompt = systemPromptFilter({
-      model: model.service.chatModel,
-      prompts: formatRedisPrompt,
-      maxTokens: 2500
-    });
-
-    prompts.unshift({
-      obj: 'SYSTEM',
-      value: `${model.systemPrompt} 知识库是最新的,下面是知识库内容:${systemPrompt}`
-    });
-
-    // 控制上下文 tokens 数量，防止超出
-    const filterPrompts = openaiChatFilter({
-      model: model.service.chatModel,
-      prompts,
-      maxTokens: modelConstantsData.contextMaxToken - 500
-    });
-
-    // console.log(filterPrompts);
     // 计算温度
-    const temperature = modelConstantsData.maxTemperature * (model.temperature / 10);
-
-    // 发出请求
-    const chatResponse = await chatAPI.createChatCompletion(
-      {
-        model: model.service.chatModel,
-        temperature,
-        messages: filterPrompts,
-        frequency_penalty: 0.5, // 越大，重复内容越少
-        presence_penalty: -0.5, // 越大，越容易出现新内容
-        stream: isStream
-      },
-      {
-        timeout: 180000,
-        responseType: isStream ? 'stream' : 'json',
-        httpsAgent: httpsAgent(true)
-      }
+    const temperature = (modelConstantsData.maxTemperature * (model.chat.temperature / 10)).toFixed(
+      2
     );
 
-    console.log('code response. time:', `${(Date.now() - startTime) / 1000}s`);
+    // 发出请求
+    const { streamResponse, responseMessages, responseText, totalTokens } =
+      await modelServiceToolMap[model.chat.chatModel].chatCompletion({
+        apiKey,
+        temperature: +temperature,
+        messages: prompts,
+        stream: isStream
+      });
 
-    step = 1;
-    let responseContent = '';
+    console.log('api response time:', `${(Date.now() - startTime) / 1000}s`);
+
+    let textLen = resolveText.length;
+    let tokens = resolveTokens;
 
     if (isStream) {
-      const streamResponse = await gpt35StreamResponse({
+      step = 1;
+      const { finishMessages, totalTokens } = await resStreamResponse({
+        model: model.chat.chatModel,
         res,
         stream,
-        chatResponse
+        chatResponse: streamResponse,
+        prompts
       });
-      responseContent = streamResponse.responseContent;
+      textLen += finishMessages.map((item) => item.value).join('').length;
+      tokens += totalTokens;
     } else {
-      responseContent = chatResponse.data.choices?.[0]?.message?.content || '';
+      textLen += responseMessages.map((item) => item.value).join('').length;
+      tokens += totalTokens;
       jsonRes(res, {
-        data: responseContent
+        data: responseText
       });
     }
 
@@ -223,9 +167,10 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
     pushChatBill({
       isPay: true,
-      modelName: model.service.modelName,
+      chatModel: model.chat.chatModel,
       userId,
-      messages: filterPrompts.concat({ role: 'assistant', content: responseContent })
+      textLen,
+      tokens
     });
   } catch (err: any) {
     if (step === 1) {

src/pages/api/openapi/chat/vectorGpt.ts

@@ -1,217 +0,0 @@
-import type { NextApiRequest, NextApiResponse } from 'next';
-import { connectToDatabase, Model } from '@/service/mongo';
-import {
-  httpsAgent,
-  systemPromptFilter,
-  authOpenApiKey,
-  openaiChatFilter
-} from '@/service/utils/tools';
-import { ChatCompletionRequestMessage, ChatCompletionRequestMessageRoleEnum } from 'openai';
-import { ChatItemType } from '@/types/chat';
-import { jsonRes } from '@/service/response';
-import { PassThrough } from 'stream';
-import {
-  modelList,
-  ModelVectorSearchModeMap,
-  ModelVectorSearchModeEnum,
-  ModelDataStatusEnum
-} from '@/constants/model';
-import { pushChatBill } from '@/service/events/pushBill';
-import { openaiCreateEmbedding, gpt35StreamResponse } from '@/service/utils/openai';
-import dayjs from 'dayjs';
-import { PgClient } from '@/service/pg';
-
-/* 发送提示词 */
-export default async function handler(req: NextApiRequest, res: NextApiResponse) {
-  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 {
-    const {
-      prompts,
-      modelId,
-      isStream = true
-    } = req.body as {
-      prompts: ChatItemType[];
-      modelId: string;
-      isStream: boolean;
-    };
-
-    if (!prompts || !modelId) {
-      throw new Error('缺少参数');
-    }
-    if (!Array.isArray(prompts)) {
-      throw new Error('prompts is not array');
-    }
-    if (prompts.length > 30 || prompts.length === 0) {
-      throw new Error('prompts length range 1-30');
-    }
-
-    await connectToDatabase();
-    let startTime = Date.now();
-
-    /* 凭证校验 */
-    const { apiKey, userId } = await authOpenApiKey(req);
-
-    const model = await Model.findOne({
-      _id: modelId,
-      userId
-    });
-
-    if (!model) {
-      throw new Error('无权使用该模型');
-    }
-
-    const modelConstantsData = modelList.find((item) => item.model === model?.service?.modelName);
-    if (!modelConstantsData) {
-      throw new Error('模型初始化异常');
-    }
-
-    // 获取提示词的向量
-    const { vector: promptVector, chatAPI } = await openaiCreateEmbedding({
-      isPay: true,
-      apiKey,
-      userId,
-      text: prompts[prompts.length - 1].value // 取最后一个
-    });
-
-    // 相似度搜素
-    const similarity = ModelVectorSearchModeMap[model.search.mode]?.similarity || 0.22;
-    const vectorSearch = await PgClient.select<{ id: string; q: string; a: string }>('modelData', {
-      fields: ['id', 'q', 'a'],
-      where: [
-        ['status', ModelDataStatusEnum.ready],
-        'AND',
-        ['model_id', model._id],
-        'AND',
-        `vector <=> '[${promptVector}]' < ${similarity}`
-      ],
-      order: [{ field: 'vector', mode: `<=> '[${promptVector}]'` }],
-      limit: 20
-    });
-
-    const formatRedisPrompt: string[] = vectorSearch.rows.map((item) => `${item.q}\n${item.a}`);
-
-    // system 合并
-    if (prompts[0].obj === 'SYSTEM') {
-      formatRedisPrompt.unshift(prompts.shift()?.value || '');
-    }
-
-    /* 高相似度+退出，无法匹配时直接退出 */
-    if (
-      formatRedisPrompt.length === 0 &&
-      model.search.mode === ModelVectorSearchModeEnum.hightSimilarity
-    ) {
-      return res.send('对不起，你的问题不在知识库中。');
-    }
-    /* 高相似度+无上下文，不添加额外知识 */
-    if (
-      formatRedisPrompt.length === 0 &&
-      model.search.mode === ModelVectorSearchModeEnum.noContext
-    ) {
-      prompts.unshift({
-        obj: 'SYSTEM',
-        value: model.systemPrompt
-      });
-    } else {
-      // 有匹配或者低匹配度模式情况下，添加知识库内容。
-      // 系统提示词过滤，最多 2500 tokens
-      const systemPrompt = systemPromptFilter({
-        model: model.service.chatModel,
-        prompts: formatRedisPrompt,
-        maxTokens: 2500
-      });
-
-      prompts.unshift({
-        obj: 'SYSTEM',
-        value: `
-${model.systemPrompt}
-${
-  model.search.mode === ModelVectorSearchModeEnum.hightSimilarity
-    ? `你只能从知识库选择内容回答.不在知识库内容拒绝回复`
-    : ''
-}
-知识库内容为: 当前时间为${dayjs().format('YYYY/MM/DD HH:mm:ss')}\n${systemPrompt}'
-`
-      });
-    }
-
-    // 控制在 tokens 数量，防止超出
-    const filterPrompts = openaiChatFilter({
-      model: model.service.chatModel,
-      prompts,
-      maxTokens: modelConstantsData.contextMaxToken - 500
-    });
-
-    // console.log(filterPrompts);
-    // 计算温度
-    const temperature = modelConstantsData.maxTemperature * (model.temperature / 10);
-
-    // 发出请求
-    const chatResponse = await chatAPI.createChatCompletion(
-      {
-        model: model.service.chatModel,
-        temperature,
-        messages: filterPrompts,
-        frequency_penalty: 0.5, // 越大，重复内容越少
-        presence_penalty: -0.5, // 越大，越容易出现新内容
-        stream: isStream
-      },
-      {
-        timeout: 180000,
-        responseType: isStream ? 'stream' : 'json',
-        httpsAgent: httpsAgent(true)
-      }
-    );
-
-    console.log('api response time:', `${(Date.now() - startTime) / 1000}s`);
-
-    step = 1;
-    let responseContent = '';
-
-    if (isStream) {
-      const streamResponse = await gpt35StreamResponse({
-        res,
-        stream,
-        chatResponse
-      });
-      responseContent = streamResponse.responseContent;
-    } else {
-      responseContent = chatResponse.data.choices?.[0]?.message?.content || '';
-      jsonRes(res, {
-        data: responseContent
-      });
-    }
-
-    pushChatBill({
-      isPay: true,
-      modelName: model.service.modelName,
-      userId,
-      messages: filterPrompts.concat({ role: 'assistant', content: responseContent })
-    });
-    // jsonRes(res);
-  } catch (err: any) {
-    if (step === 1) {
-      // 直接结束流
-      console.log('error，结束');
-      stream.destroy();
-    } else {
-      res.status(500);
-      jsonRes(res, {
-        code: 500,
-        error: err
-      });
-    }
-  }
-}

src/pages/api/openapi/delKey.ts

@@ -2,7 +2,7 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, OpenApi } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {

src/pages/api/openapi/getKeys.ts

@@ -2,7 +2,7 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, OpenApi } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 import { UserOpenApiKey } from '@/types/openapi';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {

src/pages/api/openapi/postKey.ts

@@ -2,7 +2,7 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, OpenApi } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 import { customAlphabet } from 'nanoid';
 const nanoid = customAlphabet('abcdefghijklmnopqrstuvwxyz1234567890');
 

src/pages/api/user/checkPayResult.ts

@@ -1,7 +1,7 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, User, Pay } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 import { PaySchema, UserModelSchema } from '@/types/mongoSchema';
 import dayjs from 'dayjs';
 import { getPayResult } from '@/service/utils/wxpay';

src/pages/api/user/getBill.ts

@@ -2,7 +2,7 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, Bill } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 import type { BillSchema } from '@/types/mongoSchema';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {

src/pages/api/user/getPayCode.ts

@@ -1,6 +1,6 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 import { customAlphabet } from 'nanoid';
 import { connectToDatabase, Pay } from '@/service/mongo';
 import { PRICE_SCALE } from '@/constants/common';

src/pages/api/user/getPayOrders.ts

@@ -1,6 +1,6 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 import { connectToDatabase, Pay } from '@/service/mongo';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {

src/pages/api/user/promotion/getPromotionData.ts

@@ -2,7 +2,7 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, User, promotionRecord } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 import mongoose from 'mongoose';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {

src/pages/api/user/promotion/getPromotions.ts

@@ -2,7 +2,7 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, promotionRecord } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {

src/pages/api/user/tokenLogin.ts

@@ -3,7 +3,7 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase } from '@/service/mongo';
 import { User } from '@/service/models/user';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {

src/pages/api/user/update.ts

@@ -3,7 +3,7 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { User } from '@/service/models/user';
 import { connectToDatabase } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
+import { authToken } from '@/service/utils/auth';
 import { UserUpdateParams } from '@/types/user';
 
 /* 更新一些基本信息 */

src/pages/chat/components/Empty.tsx

@@ -3,12 +3,7 @@ import { Card, Box } from '@chakra-ui/react';
 import { useMarkdown } from '@/hooks/useMarkdown';
 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>
-  );
+const Empty = ({ modelName, intro }: { modelName: string; intro: string }) => {
   const { data: chatProblem } = useMarkdown({ url: '/chatProblem.md' });
   const { data: versionIntro } = useMarkdown({ url: '/versionIntro.md' });
 
@@ -24,7 +19,9 @@ const Empty = ({ intro }: { intro: string }) => {
     >
       {!!intro && (
         <Card p={4} mb={10}>
-          <Header>模型介绍</Header>
+          <Box fontSize={'xl'} fontWeight={'600'} textAlign={'center'} pb={2}>
+            {modelName} 介绍
+          </Box>
           <Box whiteSpace={'pre-line'}>{intro}</Box>
         </Card>
       )}

src/pages/chat/components/SlideBar.tsx

@@ -1,4 +1,4 @@
-import React, { useRef, useEffect } from 'react';
+import React, { useRef, useEffect, useMemo, useCallback } from 'react';
 import { AddIcon, ChatIcon, DeleteIcon, MoonIcon, SunIcon } from '@chakra-ui/icons';
 import {
   Box,
@@ -13,7 +13,11 @@ import {
   IconButton,
   useDisclosure,
   useColorMode,
-  useColorModeValue
+  useColorModeValue,
+  Menu,
+  MenuButton,
+  MenuList,
+  MenuItem
 } from '@chakra-ui/react';
 import { useUserStore } from '@/store/user';
 import { useMutation, useQuery } from '@tanstack/react-query';
@@ -22,16 +26,21 @@ import { getToken } from '@/utils/user';
 import MyIcon from '@/components/Icon';
 import WxConcat from '@/components/WxConcat';
 import { getChatHistory, delChatHistoryById } from '@/api/chat';
-import { modelList } from '@/constants/model';
+import { getCollectionModels } from '@/api/model';
+import type { ChatSiteItemType } from '../index';
+import { fileDownload } from '@/utils/file';
+import { htmlTemplate } from '@/constants/common';
 
 const SlideBar = ({
   chatId,
   modelId,
+  history,
   resetChat,
   onClose
 }: {
   chatId: string;
   modelId: string;
+  history: ChatSiteItemType[];
   resetChat: (modelId?: string, chatId?: string) => void;
   onClose: () => void;
 }) => {
@@ -41,14 +50,42 @@ const SlideBar = ({
   const { isOpen: isOpenWx, onOpen: onOpenWx, onClose: onCloseWx } = useDisclosure();
   const preChatId = useRef('chatId'); // 用于校验上一次chatId的情况,判断是否需要刷新历史记录
 
-  const { isSuccess } = useQuery(['getMyModels'], getMyModels, {
-    cacheTime: 5 * 60 * 1000
+  const { isSuccess, refetch: fetchMyModels } = useQuery(['getMyModels'], getMyModels, {
+    cacheTime: 5 * 60 * 1000,
+    enabled: false
   });
 
+  const { data: collectionModels = [], refetch: fetchCollectionModels } = useQuery(
+    [getCollectionModels],
+    getCollectionModels,
+    {
+      cacheTime: 5 * 60 * 1000,
+      enabled: false
+    }
+  );
+
+  const models = useMemo(() => {
+    const myModelList = myModels.map((item) => ({
+      id: item._id,
+      name: item.name,
+      icon: 'model' as any
+    }));
+    const collectionList = collectionModels
+      .map((item) => ({
+        id: item._id,
+        name: item.name,
+        icon: 'collectionSolid' as any
+      }))
+      .filter((model) => !myModelList.find((item) => item.id === model.id));
+
+    return myModelList.concat(collectionList);
+  }, [collectionModels, myModels]);
+
   const { data: chatHistory = [], mutate: loadChatHistory } = useMutation({
     mutationFn: getChatHistory
   });
 
+  // update history
   useEffect(() => {
     if (chatId && preChatId.current === '') {
       loadChatHistory();
@@ -56,9 +93,76 @@ const SlideBar = ({
     preChatId.current = chatId;
   }, [chatId, loadChatHistory]);
 
+  // init history
   useEffect(() => {
-    loadChatHistory();
-  }, [loadChatHistory]);
+    setTimeout(() => {
+      fetchMyModels();
+      fetchCollectionModels();
+      loadChatHistory();
+    }, 1500);
+  }, [fetchCollectionModels, fetchMyModels, loadChatHistory]);
+
+  /**
+   * export md
+   */
+  const onclickExportMd = useCallback(() => {
+    fileDownload({
+      text: history.map((item) => item.value).join('\n'),
+      type: 'text/markdown',
+      filename: 'chat.md'
+    });
+  }, [history]);
+
+  const getHistoryHtml = useCallback(() => {
+    const historyDom = document.getElementById('history');
+    if (!historyDom) return;
+    const dom = Array.from(historyDom.children).map((child, i) => {
+      const avatar = `<img src="${
+        child.querySelector<HTMLImageElement>('.avatar')?.src
+      }" alt="" />`;
+
+      const chatContent = child.querySelector<HTMLDivElement>('.markdown');
+
+      if (!chatContent) {
+        return '';
+      }
+
+      const chatContentClone = chatContent.cloneNode(true) as HTMLDivElement;
+
+      const codeHeader = chatContentClone.querySelectorAll('.code-header');
+      codeHeader.forEach((childElement: any) => {
+        childElement.remove();
+      });
+
+      return `<div class="chat-item">
+        ${avatar}
+        ${chatContentClone.outerHTML}
+      </div>`;
+    });
+    const html = htmlTemplate.replace('{{CHAT_CONTENT}}', dom.join('\n'));
+    return html;
+  }, []);
+
+  const onclickExportHtml = useCallback(() => {
+    const html = getHistoryHtml();
+    html &&
+      fileDownload({
+        text: html,
+        type: 'text/html',
+        filename: '聊天记录.html'
+      });
+  }, [getHistoryHtml]);
+
+  const onclickExportPdf = useCallback(() => {
+    const html = getHistoryHtml();
+
+    html &&
+      // @ts-ignore
+      html2pdf(html, {
+        margin: 0,
+        filename: `聊天记录.pdf`
+      });
+  }, [getHistoryHtml]);
 
   const RenderHistory = () => (
     <>
@@ -121,7 +225,7 @@ const SlideBar = ({
     onClick: () => void;
     children: JSX.Element | string;
   }) => (
-    <Box px={3} mb={3}>
+    <Box px={3} mb={2}>
       <Flex
         alignItems={'center'}
         p={2}
@@ -152,7 +256,7 @@ const SlideBar = ({
           w={'90%'}
           variant={'white'}
           h={'40px'}
-          mb={4}
+          mb={2}
           mx={'auto'}
           leftIcon={<AddIcon />}
           onClick={() => resetChat()}
@@ -165,9 +269,9 @@ const SlideBar = ({
         {isSuccess && (
           <>
             <Box>
-              {myModels.map((item) => (
+              {models.map((item) => (
                 <Flex
-                  key={item._id}
+                  key={item.id}
                   alignItems={'center'}
                   p={3}
                   borderRadius={'md'}
@@ -178,28 +282,19 @@ const SlideBar = ({
                   }}
                   fontSize={'xs'}
                   border={'1px solid transparent'}
-                  {...(item._id === modelId
+                  {...(item.id === modelId
                     ? {
                         borderColor: 'rgba(255,255,255,0.5)',
                         backgroundColor: 'rgba(255,255,255,0.1)'
                       }
                     : {})}
                   onClick={async () => {
-                    if (item._id === modelId) return;
-                    resetChat(item._id);
+                    if (item.id === modelId) return;
+                    resetChat(item.id);
                     onClose();
                   }}
                 >
-                  <MyIcon
-                    name={
-                      modelList.find((model) => model.model === item.service.modelName)?.icon ||
-                      'model'
-                    }
-                    mr={2}
-                    color={'white'}
-                    w={'16px'}
-                    h={'16px'}
-                  />
+                  <MyIcon name={item.icon} mr={2} color={'white'} w={'16px'} h={'16px'} />
                   <Box className={'textEllipsis'} flex={'1 0 0'} w={0}>
                     {item.name}
                   </Box>
@@ -223,7 +318,33 @@ const SlideBar = ({
         </Accordion>
       </Box>
 
-      <Divider my={4} colorScheme={useColorModeValue('gray', 'white')} />
+      <Divider my={3} colorScheme={useColorModeValue('gray', 'white')} />
+
+      {history.length > 0 && (
+        <Menu autoSelect={false}>
+          <MenuButton
+            mx={3}
+            mb={2}
+            p={2}
+            display={'flex'}
+            alignItems={'center'}
+            cursor={'pointer'}
+            borderRadius={'md'}
+            textAlign={'left'}
+            _hover={{
+              backgroundColor: 'rgba(255,255,255,0.2)'
+            }}
+          >
+            <MyIcon name="export" fill={'white'} w={'18px'} h={'18px'} mr={4} />
+            导出聊天
+          </MenuButton>
+          <MenuList fontSize={'sm'} color={'blackAlpha.800'}>
+            <MenuItem onClick={onclickExportHtml}>HTML格式</MenuItem>
+            <MenuItem onClick={onclickExportPdf}>PDF格式</MenuItem>
+            <MenuItem onClick={onclickExportMd}>Markdown格式</MenuItem>
+          </MenuList>
+        </Menu>
+      )}
 
       <RenderButton onClick={() => router.push('/')}>
         <>

src/pages/chat/index.module.scss

@@ -0,0 +1,11 @@
+.stopIcon {
+  animation: zoomStopIcon 0.4s infinite alternate;
+}
+@keyframes zoomStopIcon {
+  0% {
+    transform: scale(0.8);
+  }
+  100% {
+    transform: scale(1.2);
+  }
+}

src/pages/chat/index.tsx

@@ -1,6 +1,5 @@
 import React, { useCallback, useState, useRef, useMemo, useEffect } from 'react';
 import { useRouter } from 'next/router';
-import Image from 'next/image';
 import { getInitChatSiteInfo, delChatRecordByIndex, postSaveChat } from '@/api/chat';
 import type { InitChatResponse } from '@/api/response/chat';
 import type { ChatItemType } from '@/types/chat';
@@ -16,30 +15,37 @@ import {
   Menu,
   MenuButton,
   MenuList,
-  MenuItem
+  MenuItem,
+  Image,
+  Button,
+  Modal,
+  ModalOverlay,
+  ModalContent,
+  ModalBody,
+  ModalCloseButton
 } from '@chakra-ui/react';
 import { useToast } from '@/hooks/useToast';
 import { useScreen } from '@/hooks/useScreen';
 import { useQuery } from '@tanstack/react-query';
-import { ModelNameEnum } from '@/constants/model';
+import { OpenAiChatEnum } from '@/constants/model';
 import dynamic from 'next/dynamic';
 import { useGlobalStore } from '@/store/global';
 import { useCopyData } from '@/utils/tools';
 import { streamFetch } from '@/api/fetch';
-import Icon from '@/components/Icon';
 import MyIcon from '@/components/Icon';
 import { throttle } from 'lodash';
-import { customAlphabet } from 'nanoid';
-const nanoid = customAlphabet('abcdefghijklmnopqrstuvwxyz1234567890', 5);
+import { Types } from 'mongoose';
+import Markdown from '@/components/Markdown';
+import { HUMAN_ICON, LOGO_ICON } from '@/constants/chat';
 
 const SlideBar = dynamic(() => import('./components/SlideBar'));
 const Empty = dynamic(() => import('./components/Empty'));
-const Markdown = dynamic(() => import('@/components/Markdown'));
+
+import styles from './index.module.scss';
 
 const textareaMinH = '22px';
 
 export type ChatSiteItemType = {
-  id: string;
   status: 'loading' | 'finish';
 } & ChatItemType;
 
@@ -55,18 +61,20 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
 
   // 中断请求
   const controller = useRef(new AbortController());
+  const isResetPage = useRef(false);
+
   const [chatData, setChatData] = useState<ChatType>({
     chatId,
     modelId,
     name: '',
-    avatar: '',
+    avatar: '/icon/logo.png',
     intro: '',
-    chatModel: '',
-    modelName: '',
+    chatModel: OpenAiChatEnum.GPT35,
     history: []
   }); // 聊天框整体数据
 
-  const [inputVal, setInputVal] = useState(''); // 输入的内容
+  const [inputVal, setInputVal] = useState(''); // user input prompt
+  const [showSystemPrompt, setShowSystemPrompt] = useState('');
 
   const isChatting = useMemo(
     () => chatData.history[chatData.history.length - 1]?.status === 'loading',
@@ -133,17 +141,15 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
 
         setChatData({
           ...res,
-          history: res.history.map((item: any, i) => ({
-            obj: item.obj,
-            value: item.value,
-            id: item.id || `${nanoid()}-${i}`,
+          history: res.history.map((item) => ({
+            ...item,
             status: 'finish'
           }))
         });
         if (isScroll && res.history.length > 0) {
           setTimeout(() => {
             scrollToBottom('auto');
-          }, 1200);
+          }, 1000);
         }
       } catch (e: any) {
         toast({
@@ -152,7 +158,7 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
           isClosable: true,
           duration: 5000
         });
-        router.replace('/model/list');
+        router.back();
       }
       setLoading(false);
       return null;
@@ -164,7 +170,9 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
   const resetChat = useCallback(
     async (modelId = chatData.modelId, chatId = '') => {
       // 强制中断流
+      isResetPage.current = true;
       controller.current?.abort();
+
       try {
         router.replace(`/chat?modelId=${modelId}&chatId=${chatId}`);
         loadChatInfo({
@@ -186,25 +194,20 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
 
   // gpt 对话
   const gptChatPrompt = useCallback(
-    async (prompts: ChatSiteItemType) => {
-      const urlMap: Record<string, string> = {
-        [ModelNameEnum.GPT35]: '/api/chat/chatGpt',
-        [ModelNameEnum.VECTOR_GPT]: '/api/chat/vectorGpt'
-      };
-
-      if (!urlMap[chatData.modelName]) return Promise.reject('找不到模型');
-
+    async (prompts: ChatSiteItemType[]) => {
       // create abort obj
       const abortSignal = new AbortController();
       controller.current = abortSignal;
+      isResetPage.current = false;
 
       const prompt = {
-        obj: prompts.obj,
-        value: prompts.value
+        obj: prompts[0].obj,
+        value: prompts[0].value
       };
+
       // 流请求，获取数据
-      const res = await streamFetch({
-        url: urlMap[chatData.modelName],
+      const { responseText, systemPrompt } = await streamFetch({
+        url: '/api/chat/chat',
         data: {
           prompt,
           chatId,
@@ -226,22 +229,33 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
         abortSignal
       });
 
-      let id = '';
-      // 保存对话信息
+      // 重置了页面，说明退出了当前聊天, 不缓存任何内容
+      if (isResetPage.current) {
+        return;
+      }
+
+      let newChatId = '';
+      // save chat record
       try {
-        id = await postSaveChat({
+        newChatId = await postSaveChat({
           modelId,
           chatId,
           prompts: [
-            prompt,
             {
+              _id: prompts[0]._id,
+              obj: 'Human',
+              value: prompt.value
+            },
+            {
+              _id: prompts[1]._id,
               obj: 'AI',
-              value: res as string
+              value: responseText,
+              systemPrompt
             }
           ]
         });
-        if (id) {
-          router.replace(`/chat?modelId=${modelId}&chatId=${id}`);
+        if (newChatId) {
+          router.replace(`/chat?modelId=${modelId}&chatId=${newChatId}`);
         }
       } catch (err) {
         toast({
@@ -252,20 +266,21 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
         });
       }
 
-      // 设置完成状态
+      // 设置聊天内容为完成状态
       setChatData((state) => ({
         ...state,
-        chatId: id || state.chatId, // 如果有 Id，说明是新创建的对话
+        chatId: newChatId || state.chatId, // 如果有 Id，说明是新创建的对话
         history: state.history.map((item, index) => {
           if (index !== state.history.length - 1) return item;
           return {
             ...item,
-            status: 'finish'
+            status: 'finish',
+            systemPrompt
           };
         })
       }));
     },
-    [chatData.modelName, chatId, generatingMessage, modelId, router, toast]
+    [chatId, generatingMessage, modelId, router, toast]
   );
 
   /**
@@ -294,13 +309,13 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
     const newChatList: ChatSiteItemType[] = [
       ...chatData.history,
       {
-        id: nanoid(),
+        _id: String(new Types.ObjectId()),
         obj: 'Human',
         value: val,
         status: 'finish'
       },
       {
-        id: nanoid(),
+        _id: String(new Types.ObjectId()),
         obj: 'AI',
         value: '',
         status: 'loading'
@@ -320,7 +335,7 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
     }, 100);
 
     try {
-      await gptChatPrompt(newChatList[newChatList.length - 2]);
+      await gptChatPrompt(newChatList.slice(-2));
     } catch (err: any) {
       toast({
         title: typeof err === 'string' ? err : err?.message || '聊天出错了~',
@@ -340,11 +355,11 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
 
   // 删除一句话
   const delChatRecord = useCallback(
-    async (index: number) => {
+    async (index: number, id: string) => {
       setLoading(true);
       try {
         // 删除数据库最后一句
-        await delChatRecordByIndex(chatId, index);
+        await delChatRecordByIndex(chatId, id);
 
         setChatData((state) => ({
           ...state,
@@ -380,7 +395,7 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
   // 更新流中断对象
   useEffect(() => {
     return () => {
-      // eslint-disable-next-line react-hooks/exhaustive-deps
+      isResetPage.current = true;
       controller.current?.abort();
     };
   }, []);
@@ -397,6 +412,7 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
             resetChat={resetChat}
             chatId={chatId}
             modelId={modelId}
+            history={chatData.history}
             onClose={onCloseSlider}
           />
         </Box>
@@ -412,7 +428,7 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
             px={7}
           >
             <Box onClick={onOpenSlider}>
-              <Icon
+              <MyIcon
                 name={'menu'}
                 w={'20px'}
                 h={'20px'}
@@ -428,6 +444,7 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
                 resetChat={resetChat}
                 chatId={chatId}
                 modelId={modelId}
+                history={chatData.history}
                 onClose={onCloseSlider}
               />
             </DrawerContent>
@@ -441,10 +458,18 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
         flexDirection={'column'}
       >
         {/* 聊天内容 */}
-        <Box ref={ChatBox} pb={[4, 0]} flex={'1 0 0'} h={0} w={'100%'} overflowY={'auto'}>
+        <Box
+          id={'history'}
+          ref={ChatBox}
+          pb={[4, 0]}
+          flex={'1 0 0'}
+          h={0}
+          w={'100%'}
+          overflowY={'auto'}
+        >
           {chatData.history.map((item, index) => (
             <Box
-              key={item.id}
+              key={item._id}
               py={media(9, 6)}
               px={media(4, 2)}
               backgroundColor={
@@ -454,28 +479,46 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
               borderBottom={'1px solid rgba(0,0,0,0.1)'}
             >
               <Flex maxW={'750px'} m={'auto'} alignItems={'flex-start'}>
-                <Menu>
+                <Menu autoSelect={false}>
                   <MenuButton as={Box} mr={media(4, 1)} cursor={'pointer'}>
                     <Image
-                      src={item.obj === 'Human' ? '/icon/human.png' : '/icon/logo.png'}
-                      alt="/icon/logo.png"
-                      width={media(30, 20)}
-                      height={media(30, 20)}
+                      className="avatar"
+                      src={item.obj === 'Human' ? HUMAN_ICON : chatData.avatar || LOGO_ICON}
+                      alt="avatar"
+                      w={['20px', '30px']}
+                      maxH={'50px'}
+                      objectFit={'contain'}
                     />
                   </MenuButton>
                   <MenuList fontSize={'sm'}>
                     <MenuItem onClick={() => onclickCopy(item.value)}>复制</MenuItem>
-                    <MenuItem onClick={() => delChatRecord(index)}>删除该行</MenuItem>
+                    <MenuItem onClick={() => delChatRecord(index, item._id)}>删除该行</MenuItem>
                   </MenuList>
                 </Menu>
-                <Box flex={'1 0 0'} w={0} overflow={'hidden'} id={`chat${index}`}>
+                <Box flex={'1 0 0'} w={0} overflow={'hidden'}>
                   {item.obj === 'AI' ? (
-                    <Markdown
-                      source={item.value}
-                      isChatting={isChatting && index === chatData.history.length - 1}
-                    />
+                    <>
+                      <Markdown
+                        source={item.value}
+                        isChatting={isChatting && index === chatData.history.length - 1}
+                      />
+                      {item.systemPrompt && (
+                        <Button
+                          size={'xs'}
+                          mt={2}
+                          fontWeight={'normal'}
+                          colorScheme={'gray'}
+                          variant={'outline'}
+                          onClick={() => setShowSystemPrompt(item.systemPrompt || '')}
+                        >
+                          查看提示词
+                        </Button>
+                      )}
+                    </>
                   ) : (
-                    <Box whiteSpace={'pre-wrap'}>{item.value}</Box>
+                    <Box className="markdown" whiteSpace={'pre-wrap'}>
+                      <Box as={'p'}>{item.value}</Box>
+                    </Box>
                   )}
                 </Box>
                 {isPc && (
@@ -497,14 +540,16 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
                       _hover={{
                         color: 'red.600'
                       }}
-                      onClick={() => delChatRecord(index)}
+                      onClick={() => delChatRecord(index, item._id)}
                     />
                   </Flex>
                 )}
               </Flex>
             </Box>
           ))}
-          {chatData.history.length === 0 && <Empty intro={chatData.intro} />}
+          {chatData.history.length === 0 && (
+            <Empty modelName={chatData.name} intro={chatData.intro} />
+          )}
         </Box>
         {/* 发送区 */}
         <Box m={media('20px auto', '0 auto')} w={'100%'} maxW={media('min(750px, 100%)', 'auto')}>
@@ -559,33 +604,51 @@ const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
             <Flex
               alignItems={'center'}
               justifyContent={'center'}
-              h={'30px'}
-              w={'30px'}
+              h={'25px'}
+              w={'25px'}
               position={'absolute'}
               right={['12px', '20px']}
               bottom={'15px'}
-              onClick={sendPrompt}
             >
               {isChatting ? (
-                <Icon
-                  style={{ transform: 'translateY(4px)' }}
-                  h={'30px'}
-                  w={'30px'}
-                  name={'chatting'}
+                <MyIcon
+                  className={styles.stopIcon}
+                  width={['22px', '25px']}
+                  height={['22px', '25px']}
+                  cursor={'pointer'}
+                  name={'stop'}
+                  color={useColorModeValue('gray.500', 'white')}
+                  onClick={() => {
+                    controller.current?.abort();
+                  }}
                 />
               ) : (
-                <Icon
+                <MyIcon
                   name={'chatSend'}
                   width={['18px', '20px']}
                   height={['18px', '20px']}
                   cursor={'pointer'}
-                  fill={useColorModeValue('#718096', 'white')}
-                ></Icon>
+                  color={useColorModeValue('gray.500', 'white')}
+                  onClick={sendPrompt}
+                />
               )}
             </Flex>
           </Box>
         </Box>
       </Flex>
+
+      {/* system prompt show modal */}
+      {
+        <Modal isOpen={!!showSystemPrompt} onClose={() => setShowSystemPrompt('')}>
+          <ModalOverlay />
+          <ModalContent maxW={'min(90vw, 600px)'} pr={2} maxH={'80vh'} overflowY={'auto'}>
+            <ModalCloseButton />
+            <ModalBody pt={10} fontSize={'sm'} whiteSpace={'pre-wrap'} textAlign={'justify'}>
+              {showSystemPrompt}
+            </ModalBody>
+          </ModalContent>
+        </Modal>
+      }
     </Flex>
   );
 };

src/pages/index.tsx

@@ -22,9 +22,9 @@ const Home = () => {
 
       <Card p={5} mt={4} textAlign={'center'}>
         <Box>
-          {/* <Link href="https://beian.miit.gov.cn/" target="_blank">
-            浙B2-20080101
-          </Link> */}
+          <Link href="https://beian.miit.gov.cn/" target="_blank">
+            浙ICP备2023011255号-1
+          </Link>
         </Box>
         <Box>Made by FastGpt Team.</Box>
       </Card>

src/pages/login/components/ForgetPasswordForm.tsx

@@ -78,7 +78,7 @@ const RegisterForm = ({ setPageType, loginSuccess }: Props) => {
         找回 FastGPT 账号
       </Box>
       <form onSubmit={handleSubmit(onclickFindPassword)}>
-        <FormControl mt={8} isInvalid={!!errors.username}>
+        <FormControl mt={5} isInvalid={!!errors.username}>
           <Input
             placeholder="邮箱/手机号"
             size={mediaLgMd}
@@ -95,7 +95,7 @@ const RegisterForm = ({ setPageType, loginSuccess }: Props) => {
             {!!errors.username && errors.username.message}
           </FormErrorMessage>
         </FormControl>
-        <FormControl mt={8} isInvalid={!!errors.username}>
+        <FormControl mt={5} isInvalid={!!errors.username}>
           <Flex>
             <Input
               flex={1}
@@ -121,7 +121,7 @@ const RegisterForm = ({ setPageType, loginSuccess }: Props) => {
             {!!errors.code && errors.code.message}
           </FormErrorMessage>
         </FormControl>
-        <FormControl mt={8} isInvalid={!!errors.password}>
+        <FormControl mt={5} isInvalid={!!errors.password}>
           <Input
             type={'password'}
             placeholder="新密码"
@@ -142,7 +142,7 @@ const RegisterForm = ({ setPageType, loginSuccess }: Props) => {
             {!!errors.password && errors.password.message}
           </FormErrorMessage>
         </FormControl>
-        <FormControl mt={8} isInvalid={!!errors.password2}>
+        <FormControl mt={5} isInvalid={!!errors.password2}>
           <Input
             type={'password'}
             placeholder="确认密码"
@@ -168,7 +168,7 @@ const RegisterForm = ({ setPageType, loginSuccess }: Props) => {
         </Box>
         <Button
           type="submit"
-          mt={8}
+          mt={5}
           w={'100%'}
           size={mediaLgMd}
           colorScheme="blue"

src/pages/login/components/RegisterForm.tsx

@@ -8,6 +8,7 @@ import type { ResLogin } from '@/api/response/user';
 import { useScreen } from '@/hooks/useScreen';
 import { useToast } from '@/hooks/useToast';
 import { useRouter } from 'next/router';
+import { postCreateModel } from '@/api/model';
 
 interface Props {
   loginSuccess: (e: ResLogin) => void;
@@ -64,6 +65,10 @@ const RegisterForm = ({ setPageType, loginSuccess }: Props) => {
           title: `注册成功`,
           status: 'success'
         });
+        // aut register a model
+        postCreateModel({
+          name: '模型1'
+        });
       } catch (error: any) {
         toast({
           title: error.message || '注册异常',
@@ -81,7 +86,7 @@ const RegisterForm = ({ setPageType, loginSuccess }: Props) => {
         注册 FastGPT 账号
       </Box>
       <form onSubmit={handleSubmit(onclickRegister)}>
-        <FormControl mt={8} isInvalid={!!errors.username}>
+        <FormControl mt={5} isInvalid={!!errors.username}>
           <Input
             placeholder="邮箱/手机号"
             size={mediaLgMd}
@@ -98,7 +103,7 @@ const RegisterForm = ({ setPageType, loginSuccess }: Props) => {
             {!!errors.username && errors.username.message}
           </FormErrorMessage>
         </FormControl>
-        <FormControl mt={8} isInvalid={!!errors.username}>
+        <FormControl mt={5} isInvalid={!!errors.username}>
           <Flex>
             <Input
               flex={1}
@@ -124,7 +129,7 @@ const RegisterForm = ({ setPageType, loginSuccess }: Props) => {
             {!!errors.code && errors.code.message}
           </FormErrorMessage>
         </FormControl>
-        <FormControl mt={8} isInvalid={!!errors.password}>
+        <FormControl mt={5} isInvalid={!!errors.password}>
           <Input
             type={'password'}
             placeholder="密码"
@@ -145,7 +150,7 @@ const RegisterForm = ({ setPageType, loginSuccess }: Props) => {
             {!!errors.password && errors.password.message}
           </FormErrorMessage>
         </FormControl>
-        <FormControl mt={8} isInvalid={!!errors.password2}>
+        <FormControl mt={5} isInvalid={!!errors.password2}>
           <Input
             type={'password'}
             placeholder="确认密码"
@@ -171,7 +176,7 @@ const RegisterForm = ({ setPageType, loginSuccess }: Props) => {
         </Box>
         <Button
           type="submit"
-          mt={8}
+          mt={5}
           w={'100%'}
           size={mediaLgMd}
           colorScheme="blue"

src/pages/login/index.tsx

@@ -13,6 +13,7 @@ const ForgetPasswordForm = dynamic(() => import('./components/ForgetPasswordForm
 
 const Login = () => {
   const router = useRouter();
+  const { lastRoute = '' } = router.query as { lastRoute: string };
   const { isPc } = useScreen();
   const [pageType, setPageType] = useState<`${PageTypeEnum}`>(PageTypeEnum.login);
   const { setUserInfo } = useUserStore();
@@ -20,9 +21,11 @@ const Login = () => {
   const loginSuccess = useCallback(
     (res: ResLogin) => {
       setUserInfo(res.user, res.token);
-      router.push('/model/list');
+      setTimeout(() => {
+        router.push(lastRoute ? decodeURIComponent(lastRoute) : '/model/list');
+      }, 100);
     },
-    [router, setUserInfo]
+    [lastRoute, router, setUserInfo]
   );
 
   function DynamicComponent({ type }: { type: `${PageTypeEnum}` }) {
@@ -42,12 +45,19 @@ const Login = () => {
   }, [router]);
 
   return (
-    <Box className={styles.loginPage} h={'100%'} p={isPc ? '10vh 10vw' : 0}>
+    <Flex
+      alignItems={'center'}
+      justifyContent={'center'}
+      className={styles.loginPage}
+      h={'100%'}
+      px={[0, '10vw']}
+    >
       <Flex
-        maxW={'1240px'}
-        m={'auto'}
-        backgroundColor={'#fff'}
         height="100%"
+        w={'100%'}
+        maxW={'1240px'}
+        maxH={['auto', '660px']}
+        backgroundColor={'#fff'}
         alignItems={'center'}
         justifyContent={'center'}
         p={10}
@@ -80,7 +90,7 @@ const Login = () => {
           <DynamicComponent type={pageType} />
         </Box>
       </Flex>
-    </Box>
+    </Flex>
   );
 };
 

src/pages/model/detail/components/InputDataModal.tsx

@@ -118,32 +118,28 @@ const InputDataModal = ({
           px={6}
           pb={2}
         >
-          <Box flex={2} mr={[0, 4]} mb={[4, 0]} h={['230px', '100%']}>
-            <Box h={'30px'}>问题</Box>
+          <Box flex={1} mr={[0, 4]} mb={[4, 0]} h={['230px', '100%']}>
+            <Box h={'30px'}>{'匹配的知识点'}</Box>
             <Textarea
-              placeholder={
-                '相关问题，可以输入多个问法, 最多 1000 字。例如：\n1. laf 是什么？\n2. laf 可以做什么？\n3. laf怎么用'
-              }
-              maxLength={1000}
+              placeholder={'匹配的知识点。这部分内容会被搜索，请把控内容的质量。最多 1500 字。'}
+              maxLength={1500}
               resize={'none'}
               h={'calc(100% - 30px)'}
               {...register(`q`, {
-                required: '相关问题，可以回车输入多个问法'
+                required: true
               })}
             />
           </Box>
-          <Box flex={3} h={['330px', '100%']}>
-            <Box h={'30px'}>知识点</Box>
+          <Box flex={1} h={['330px', '100%']}>
+            <Box h={'30px'}>补充知识</Box>
             <Textarea
               placeholder={
-                '知识点，最多 2000 字。例如：\n1. laf是一个云函数开发平台。\n2. laf 什么都能做。\n3. 下面是使用 laf 的例子: ……'
+                '补充知识。这部分内容不会被搜索，但会作为"匹配的知识点"的内容补充，你可以讲一些细节的内容填写在这里。最多 1500 字。'
               }
-              maxLength={2000}
+              maxLength={1500}
               resize={'none'}
               h={'calc(100% - 30px)'}
-              {...register(`a`, {
-                required: '知识点'
-              })}
+              {...register('a')}
             />
           </Box>
         </Box>

src/pages/model/detail/components/ModelDataCard.tsx

@@ -16,8 +16,10 @@ import {
   MenuButton,
   MenuList,
   MenuItem,
-  Input
+  Input,
+  Tooltip
 } from '@chakra-ui/react';
+import { QuestionOutlineIcon } from '@chakra-ui/icons';
 import type { BoxProps } from '@chakra-ui/react';
 import type { ModelDataItemType } from '@/types/model';
 import { ModelDataStatusMap } from '@/constants/model';
@@ -39,15 +41,16 @@ import InputModal, { FormData as InputDataType } from './InputDataModal';
 const SelectFileModal = dynamic(() => import('./SelectFileModal'));
 const SelectCsvModal = dynamic(() => import('./SelectCsvModal'));
 
-const ModelDataCard = ({ modelId }: { modelId: string }) => {
+const ModelDataCard = ({ modelId, isOwner }: { modelId: string; isOwner: boolean }) => {
   const { Loading, setIsLoading } = useLoading();
   const lastSearch = useRef('');
   const [searchText, setSearchText] = useState('');
   const tdStyles = useRef<BoxProps>({
     fontSize: 'xs',
+    minW: '150px',
     maxW: '500px',
-    whiteSpace: 'pre-wrap',
     maxH: '250px',
+    whiteSpace: 'pre-wrap',
     overflowY: 'auto'
   });
   const {
@@ -93,10 +96,16 @@ const ModelDataCard = ({ modelId }: { modelId: string }) => {
     (num = 1) => {
       getData(num);
       refetch();
+      return null;
     },
     [getData, refetch]
   );
 
+  useQuery(['refetchData'], () => refetchData(pageNum), {
+    refetchInterval: 5000,
+    enabled: splitDataLen > 0
+  });
+
   // 获取所有的数据，并导出 json
   const { mutate: onclickExport, isLoading: isLoadingExport = false } = useMutation({
     mutationFn: () => getExportDataList(modelId),
@@ -126,51 +135,54 @@ const ModelDataCard = ({ modelId }: { modelId: string }) => {
     <>
       <Flex>
         <Box fontWeight={'bold'} fontSize={'lg'} flex={1} mr={2}>
-          模型数据: {total}组
-          <Box as={'span'} fontSize={'sm'}>
-            （测试版本）
-          </Box>
+          知识库数据: {total}组
         </Box>
-        <IconButton
-          icon={<RepeatIcon />}
-          aria-label={'refresh'}
-          variant={'outline'}
-          mr={4}
-          size={'sm'}
-          onClick={() => refetchData(pageNum)}
-        />
-        <Button
-          variant={'outline'}
-          mr={2}
-          size={'sm'}
-          isLoading={isLoadingExport}
-          title={'换行数据导出时，会进行格式转换'}
-          onClick={() => onclickExport()}
-        >
-          导出
-        </Button>
-        <Menu autoSelect={false}>
-          <MenuButton as={Button} size={'sm'}>
-            导入
-          </MenuButton>
-          <MenuList>
-            <MenuItem
-              onClick={() =>
-                setEditInputData({
-                  a: '',
-                  q: ''
-                })
-              }
+        {isOwner && (
+          <>
+            <IconButton
+              icon={<RepeatIcon />}
+              aria-label={'refresh'}
+              variant={'outline'}
+              mr={4}
+              size={'sm'}
+              onClick={() => refetchData(pageNum)}
+            />
+            <Button
+              variant={'outline'}
+              mr={2}
+              size={'sm'}
+              isLoading={isLoadingExport}
+              title={'换行数据导出时，会进行格式转换'}
+              onClick={() => onclickExport()}
             >
-              手动输入
-            </MenuItem>
-            <MenuItem onClick={onOpenSelectFileModal}>文本/文件拆分</MenuItem>
-            <MenuItem onClick={onOpenSelectCsvModal}>csv 问答对导入</MenuItem>
-          </MenuList>
-        </Menu>
+              导出
+            </Button>
+            <Menu autoSelect={false}>
+              <MenuButton as={Button} size={'sm'}>
+                导入
+              </MenuButton>
+              <MenuList>
+                <MenuItem
+                  onClick={() =>
+                    setEditInputData({
+                      a: '',
+                      q: ''
+                    })
+                  }
+                >
+                  手动输入
+                </MenuItem>
+                <MenuItem onClick={onOpenSelectFileModal}>文本/文件拆分</MenuItem>
+                <MenuItem onClick={onOpenSelectCsvModal}>csv 问答对导入</MenuItem>
+              </MenuList>
+            </Menu>
+          </>
+        )}
       </Flex>
       <Flex mt={4}>
-        {splitDataLen > 0 && <Box fontSize={'xs'}>{splitDataLen}条数据正在拆分...</Box>}
+        {isOwner && splitDataLen > 0 && (
+          <Box fontSize={'xs'}>{splitDataLen}条数据正在拆分，请耐心等待...</Box>
+        )}
         <Box flex={1} />
         <Input
           maxW={'240px'}
@@ -198,10 +210,19 @@ const ModelDataCard = ({ modelId }: { modelId: string }) => {
           <Table variant={'simple'} w={'100%'}>
             <Thead>
               <Tr>
-                <Th>{'匹配内容(问题)'}</Th>
-                <Th>对应答案</Th>
+                <Th>
+                  匹配的知识点
+                  <Tooltip
+                    label={
+                      '对话时，会将用户的问题和知识库的 "匹配知识点" 进行比较，找到最相似的前 n 条记录，将这些记录的 "匹配知识点"+"补充知识点" 作为 chatgpt 的系统提示词。'
+                    }
+                  >
+                    <QuestionOutlineIcon ml={1} />
+                  </Tooltip>
+                </Th>
+                <Th>补充知识</Th>
                 <Th>状态</Th>
-                <Th>操作</Th>
+                {isOwner && <Th>操作</Th>}
               </Tr>
             </Thead>
             <Tbody>
@@ -214,33 +235,35 @@ const ModelDataCard = ({ modelId }: { modelId: string }) => {
                     <Box {...tdStyles.current}>{item.a || '-'}</Box>
                   </Td>
                   <Td>{ModelDataStatusMap[item.status]}</Td>
-                  <Td>
-                    <IconButton
-                      mr={5}
-                      icon={<EditIcon />}
-                      variant={'outline'}
-                      aria-label={'delete'}
-                      size={'sm'}
-                      onClick={() =>
-                        setEditInputData({
-                          dataId: item.id,
-                          q: item.q,
-                          a: item.a
-                        })
-                      }
-                    />
-                    <IconButton
-                      icon={<DeleteIcon />}
-                      variant={'outline'}
-                      colorScheme={'gray'}
-                      aria-label={'delete'}
-                      size={'sm'}
-                      onClick={async () => {
-                        await delOneModelData(item.id);
-                        refetchData(pageNum);
-                      }}
-                    />
-                  </Td>
+                  {isOwner && (
+                    <Td>
+                      <IconButton
+                        mr={5}
+                        icon={<EditIcon />}
+                        variant={'outline'}
+                        aria-label={'delete'}
+                        size={'sm'}
+                        onClick={() =>
+                          setEditInputData({
+                            dataId: item.id,
+                            q: item.q,
+                            a: item.a
+                          })
+                        }
+                      />
+                      <IconButton
+                        icon={<DeleteIcon />}
+                        variant={'outline'}
+                        colorScheme={'gray'}
+                        aria-label={'delete'}
+                        size={'sm'}
+                        onClick={async () => {
+                          await delOneModelData(item.id);
+                          refetchData(pageNum);
+                        }}
+                      />
+                    </Td>
+                  )}
                 </Tr>
               ))}
             </Tbody>

src/pages/model/detail/components/ModelEditForm.tsx

@@ -1,4 +1,4 @@
-import React, { useState } from 'react';
+import React, { useState, useCallback } from 'react';
 import {
   Box,
   Card,
@@ -13,22 +13,28 @@ import {
   SliderMark,
   Tooltip,
   Button,
-  Select
+  Select,
+  Grid,
+  Switch,
+  Image
 } from '@chakra-ui/react';
 import { QuestionOutlineIcon } from '@chakra-ui/icons';
 import type { ModelSchema } from '@/types/mongoSchema';
 import { UseFormReturn } from 'react-hook-form';
-import { modelList, ModelVectorSearchModeMap } from '@/constants/model';
+import { ChatModelMap, ModelVectorSearchModeMap } from '@/constants/model';
 import { formatPrice } from '@/utils/user';
 import { useConfirm } from '@/hooks/useConfirm';
+import { useSelectFile } from '@/hooks/useSelectFile';
+import { useToast } from '@/hooks/useToast';
+import { fileToBase64 } from '@/utils/file';
 
 const ModelEditForm = ({
   formHooks,
-  canTrain,
+  isOwner,
   handleDelModel
 }: {
   formHooks: UseFormReturn<ModelSchema>;
-  canTrain: boolean;
+  isOwner: boolean;
   handleDelModel: () => void;
 }) => {
   const { openConfirm, ConfirmChild } = useConfirm({
@@ -36,12 +42,55 @@ const ModelEditForm = ({
   });
   const { register, setValue, getValues } = formHooks;
   const [refresh, setRefresh] = useState(false);
+  const { File, onOpen: onOpenSelectFile } = useSelectFile({
+    fileType: '.jpg,.png',
+    multiple: false
+  });
+  const { toast } = useToast();
+
+  const onSelectFile = useCallback(
+    async (e: File[]) => {
+      const file = e[0];
+      if (!file) return;
+
+      if (file.size > 100 * 1024) {
+        return toast({
+          title: '头像需小于 100kb',
+          status: 'warning'
+        });
+      }
+
+      const base64 = (await fileToBase64(file)) as string;
+      setValue('avatar', base64);
+      setRefresh((state) => !state);
+    },
+    [setValue, toast]
+  );
 
   return (
     <>
       <Card p={4}>
-        <Flex justifyContent={'space-between'} alignItems={'center'}>
-          <Box fontWeight={'bold'}>基本信息</Box>
+        <Box fontWeight={'bold'}>基本信息</Box>
+        <Flex alignItems={'center'} mt={4}>
+          <Box flex={'0 0 80px'} w={0}>
+            modelId:
+          </Box>
+          <Box>{getValues('_id')}</Box>
+        </Flex>
+        <Flex mt={4} alignItems={'center'}>
+          <Box flex={'0 0 80px'} w={0}>
+            头像:
+          </Box>
+          <Image
+            src={getValues('avatar') || '/icon/logo.png'}
+            alt={'avatar'}
+            w={['28px', '36px']}
+            h={['28px', '36px']}
+            objectFit={'cover'}
+            cursor={isOwner ? 'pointer' : 'default'}
+            title={'点击切换头像'}
+            onClick={() => isOwner && onOpenSelectFile()}
+          />
         </Flex>
         <FormControl mt={4}>
           <Flex alignItems={'center'}>
@@ -49,47 +98,48 @@ const ModelEditForm = ({
               名称:
             </Box>
             <Input
+              isDisabled={!isOwner}
               {...register('name', {
                 required: '展示名称不能为空'
               })}
             ></Input>
           </Flex>
-          <Flex alignItems={'center'} mt={5}>
-            <Box flex={'0 0 80px'} w={0}>
-              modelId:
-            </Box>
-            <Box>{getValues('_id')}</Box>
-          </Flex>
         </FormControl>
+
         <Flex alignItems={'center'} mt={5}>
           <Box flex={'0 0 80px'} w={0}>
-            模型类型:
+            对话模型:
           </Box>
-          <Box>{modelList.find((item) => item.model === getValues('service.modelName'))?.name}</Box>
+          <Box>{ChatModelMap[getValues('chat.chatModel')].name}</Box>
         </Flex>
         <Flex alignItems={'center'} mt={5}>
           <Box flex={'0 0 80px'} w={0}>
             价格:
           </Box>
           <Box>
-            {formatPrice(
-              modelList.find((item) => item.model === getValues('service.modelName'))?.price || 0,
-              1000
-            )}
+            {formatPrice(ChatModelMap[getValues('chat.chatModel')].price, 1000)}
             元/1K tokens(包括上下文和回答)
           </Box>
         </Flex>
-        <Flex mt={5} alignItems={'center'}>
-          <Box flex={'0 0 150px'}>删除模型和数据集</Box>
-          <Button
-            colorScheme={'gray'}
-            variant={'outline'}
-            size={'sm'}
-            onClick={openConfirm(handleDelModel)}
-          >
-            删除模型
-          </Button>
+        <Flex alignItems={'center'} mt={5}>
+          <Box flex={'0 0 80px'} w={0}>
+            收藏人数:
+          </Box>
+          <Box>{getValues('share.collection')}人</Box>
         </Flex>
+        {isOwner && (
+          <Flex mt={5} alignItems={'center'}>
+            <Box flex={'0 0 150px'}>删除模型和知识库</Box>
+            <Button
+              colorScheme={'gray'}
+              variant={'outline'}
+              size={'sm'}
+              onClick={openConfirm(handleDelModel)}
+            >
+              删除模型
+            </Button>
+          </Flex>
+        )}
       </Card>
       <Card p={4}>
         <Box fontWeight={'bold'}>模型效果</Box>
@@ -109,14 +159,15 @@ const ModelEditForm = ({
               min={0}
               max={10}
               step={1}
-              value={getValues('temperature')}
+              value={getValues('chat.temperature')}
+              isDisabled={!isOwner}
               onChange={(e) => {
-                setValue('temperature', e);
+                setValue('chat.temperature', e);
                 setRefresh(!refresh);
               }}
             >
               <SliderMark
-                value={getValues('temperature')}
+                value={getValues('chat.temperature')}
                 textAlign="center"
                 bg="blue.500"
                 color="white"
@@ -126,7 +177,7 @@ const ModelEditForm = ({
                 fontSize={'xs'}
                 transform={'translate(-50%, -200%)'}
               >
-                {getValues('temperature')}
+                {getValues('chat.temperature')}
               </SliderMark>
               <SliderTrack>
                 <SliderFilledTrack />
@@ -135,34 +186,103 @@ const ModelEditForm = ({
             </Slider>
           </Flex>
         </FormControl>
-        {canTrain && (
-          <FormControl mt={4}>
-            <Flex alignItems={'center'}>
-              <Box flex={'0 0 70px'}>搜索模式</Box>
-              <Select {...register('search.mode', { required: '搜索模式不能为空' })}>
-                {Object.entries(ModelVectorSearchModeMap).map(([key, { text }]) => (
-                  <option key={key} value={key}>
-                    {text}
-                  </option>
-                ))}
-              </Select>
-            </Flex>
-          </FormControl>
+        <Flex mt={4} alignItems={'center'}>
+          <Box mr={4}>知识库搜索</Box>
+          <Switch
+            isChecked={getValues('chat.useKb')}
+            onChange={() => {
+              setValue('chat.useKb', !getValues('chat.useKb'));
+              setRefresh(!refresh);
+            }}
+          />
+        </Flex>
+        {getValues('chat.useKb') && (
+          <Flex mt={4} alignItems={'center'}>
+            <Box mr={4} whiteSpace={'nowrap'}>
+              搜索模式&emsp;
+            </Box>
+            <Select
+              isDisabled={!isOwner}
+              {...register('chat.searchMode', { required: '搜索模式不能为空' })}
+            >
+              {Object.entries(ModelVectorSearchModeMap).map(([key, { text }]) => (
+                <option key={key} value={key}>
+                  {text}
+                </option>
+              ))}
+            </Select>
+          </Flex>
         )}
+
         <Box mt={4}>
           <Box mb={1}>系统提示词</Box>
           <Textarea
-            rows={6}
+            rows={8}
             maxLength={-1}
-            {...register('systemPrompt')}
-            placeholder={
-              canTrain
-                ? '训练的模型会根据知识库内容，生成一部分系统提示词，因此在对话时需要消耗更多的 tokens。你可以增加提示词，让效果更符合预期。例如: \n1. 请根据知识库内容回答用户问题。\n2. 知识库是电影《铃芽之旅》的内容，根据知识库内容回答。无关问题，拒绝回复！'
-                : '模型默认的 prompt 词，通过调整该内容，可以生成一个限定范围的模型。\n注意，改功能会影响对话的整体朝向！'
-            }
+            isDisabled={!isOwner}
+            placeholder={'模型默认的 prompt 词，通过调整该内容，可以引导模型聊天方向。'}
+            {...register('chat.systemPrompt')}
           />
         </Box>
       </Card>
+      {isOwner && (
+        <Card p={4} gridColumnStart={[1, 1]} gridColumnEnd={[2, 3]}>
+          <Box fontWeight={'bold'}>分享设置</Box>
+
+          <Grid gridTemplateColumns={['1fr', '1fr 410px']} gridGap={5}>
+            <Box>
+              <Flex mt={5} alignItems={'center'}>
+                <Box mr={3}>模型分享:</Box>
+                <Switch
+                  isChecked={getValues('share.isShare')}
+                  onChange={() => {
+                    setValue('share.isShare', !getValues('share.isShare'));
+                    setRefresh(!refresh);
+                  }}
+                />
+                <Box ml={12} mr={3}>
+                  分享模型细节:
+                </Box>
+                <Switch
+                  isChecked={getValues('share.isShareDetail')}
+                  onChange={() => {
+                    setValue('share.isShareDetail', !getValues('share.isShareDetail'));
+                    setRefresh(!refresh);
+                  }}
+                />
+              </Flex>
+              <Box mt={5}>
+                <Box>模型介绍</Box>
+                <Textarea
+                  mt={1}
+                  rows={6}
+                  maxLength={150}
+                  {...register('share.intro')}
+                  placeholder={'介绍模型的功能、场景等，吸引更多人来使用！最多150字。'}
+                />
+              </Box>
+            </Box>
+            <Box
+              textAlign={'justify'}
+              fontSize={'sm'}
+              border={'1px solid #f4f4f4'}
+              borderRadius={'sm'}
+              p={3}
+            >
+              <Box fontWeight={'bold'}>Tips</Box>
+              <Box mt={1} as={'ul'} pl={4}>
+                <li>
+                  开启模型分享后，你的模型将会出现在共享市场，可供 FastGpt
+                  所有用户使用。用户使用时不会消耗你的 tokens，而是消耗使用者的 tokens。
+                </li>
+                <li>开启分享详情后，其他用户可以查看该模型的特有数据：温度、提示词和数据集。</li>
+              </Box>
+            </Box>
+          </Grid>
+        </Card>
+      )}
+      <File onSelect={onSelectFile} />
+
       {/* <Card p={4}>
         <Box fontWeight={'bold'}>安全策略</Box>
         <FormControl mt={2}>