perf: 生成对话框时机

c2c73ed2 · archer · 9682c827 · c2c73ed2 · c2c73ed2 · c2c73ed2
Commit c2c73ed2 authored Apr 23, 2023 by archer
25 changed files
--- a/src/api/chat.ts
+++ b/src/api/chat.ts
@@ -3,15 +3,10 @@ import type { ChatItemType, ChatSiteItemType } from '@/types/chat';
 import type { InitChatResponse } from './response/chat';
 /**
- * 获取一个聊天框的ID
- */
-export const getChatSiteId = (modelId: string) => GET<string>(`/chat/generate?modelId=${modelId}`);
-/**
 * 获取初始化聊天内容
 */
-export const getInitChatSiteInfo = (chatId: string) =>
+export const getInitChatSiteInfo = (modelId: string, chatId: '' | string) =>
-  GET<InitChatResponse>(`/chat/init?chatId=${chatId}`);
+  GET<InitChatResponse>(`/chat/init?modelId=${modelId}&chatId=${chatId}`);
 /**
 * 发送 GPT3 prompt
@@ -34,8 +29,11 @@ export const postGPT3SendPrompt = ({
 /**
 * 存储一轮对话
 */
-export const postSaveChat = (data: { chatId: string; prompts: ChatItemType[] }) =>
+export const postSaveChat = (data: {
-  POST('/chat/saveChat', data);
+  modelId: string;
+  chatId: '' | string;
+  prompts: ChatItemType[];
+}) => POST<string>('/chat/saveChat', data);
 /**
 * 删除一句对话

--- a/src/constants/model.ts
+++ b/src/constants/model.ts
-import type { ServiceName, ModelDataType, ModelSchema } from '@/types/mongoSchema';
+import type { ModelDataType, ModelSchema } from '@/types/mongoSchema';
 export enum ModelDataStatusEnum {
  ready = 'ready',
@@ -18,7 +18,6 @@ export const ChatModelNameMap = {
 };
 export type ModelConstantsData = {
-  serviceCompany: `${ServiceName}`;
  name: string;
  model: `${ChatModelNameEnum}`;
  trainName: string; // 空字符串代表不能训练
@@ -30,7 +29,6 @@ export type ModelConstantsData = {
 export const modelList: ModelConstantsData[] = [
  {
-    serviceCompany: 'openai',
    name: 'chatGPT',
    model: ChatModelNameEnum.GPT35,
    trainName: '',
@@ -40,7 +38,6 @@ export const modelList: ModelConstantsData[] = [
    price: 3
  },
  {
-    serviceCompany: 'openai',
    name: '知识库',
    model: ChatModelNameEnum.VECTOR_GPT,
    trainName: 'vector',
@@ -132,7 +129,6 @@ export const defaultModel: ModelSchema = {
    mode: ModelVectorSearchModeEnum.hightSimilarity
  },
  service: {
-    company: 'openai',
    trainId: '',
    chatModel: ChatModelNameEnum.GPT35,
    modelName: ChatModelNameEnum.GPT35

--- a/src/pages/api/chat/chatGpt.ts
+++ b/src/pages/api/chat/chatGpt.ts
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { connectToDatabase } from '@/service/mongo';
-import { getOpenAIApi, authChat } from '@/service/utils/chat';
+import { getOpenAIApi, authChat } from '@/service/utils/auth';
 import { httpsAgent, openaiChatFilter } from '@/service/utils/tools';
 import { ChatCompletionRequestMessage, ChatCompletionRequestMessageRoleEnum } from 'openai';
 import { ChatItemType } from '@/types/chat';
 import { jsonRes } from '@/service/response';
-import type { ModelSchema } from '@/types/mongoSchema';
 import { PassThrough } from 'stream';
 import { modelList } from '@/constants/model';
 import { pushChatBill } from '@/service/events/pushBill';
@@ -28,29 +27,33 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
  });
  try {
-    const { chatId, prompt } = req.body as {
+    const { chatId, prompt, modelId } = req.body as {
      prompt: ChatItemType;
-      chatId: string;
+      modelId: string;
+      chatId: '' | string;
    };
    const { authorization } = req.headers;
-    if (!chatId || !prompt) {
+    if (!modelId || !prompt) {
      throw new Error('缺少参数');
    }
    await connectToDatabase();
    let startTime = Date.now();
-    const { chat, userApiKey, systemKey, userId } = await authChat(chatId, authorization);
+    const { model, content, userApiKey, systemKey, userId } = await authChat({
+      modelId,
+      chatId,
+      authorization
+    });
-    const model: ModelSchema = chat.modelId;
    const modelConstantsData = modelList.find((item) => item.model === model.service.modelName);
    if (!modelConstantsData) {
      throw new Error('模型加载异常');
    }
    // 读取对话内容
-    const prompts = [...chat.content, prompt];
+    const prompts = [...content, prompt];
    // 如果有系统提示词，自动插入
    if (model.systemPrompt) {

--- a/src/pages/api/chat/generate.ts
+++ b/src/pages/api/chat/generate.ts
-import type { NextApiRequest, NextApiResponse } from 'next';
-import { jsonRes } from '@/service/response';
-import { connectToDatabase, Model, Chat } from '@/service/mongo';
-import { authToken } from '@/service/utils/tools';
-import type { ModelSchema } from '@/types/mongoSchema';
-/* 获取我的模型 */
-export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
-  try {
-    const { modelId } = req.query as {
-      modelId: string;
-    };
-    const { authorization } = req.headers;
-    if (!authorization) {
-      throw new Error('无权生成对话');
-    }
-    if (!modelId) {
-      throw new Error('缺少参数');
-    }
-    // 凭证校验
-    const userId = await authToken(authorization);
-    await connectToDatabase();
-    // 校验是否为用户的模型
-    const model = await Model.findOne<ModelSchema>({
-      _id: modelId,
-      userId
-    });
-    if (!model) {
-      throw new Error('无权使用该模型');
-    }
-    // 创建 chat 数据
-    const response = await Chat.create({
-      userId,
-      modelId,
-      content: []
-    });
-    jsonRes(res, {
-      data: response._id // 即聊天框的 ID
-    });
-  } catch (err) {
-    jsonRes(res, {
-      code: 500,
-      error: err
-    });
-  }
-}
--- a/src/pages/api/chat/init.ts
+++ b/src/pages/api/chat/init.ts
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
-import { connectToDatabase, Chat } from '@/service/mongo';
+import { connectToDatabase, Chat, Model } from '@/service/mongo';
-import type { ChatPopulate } from '@/types/mongoSchema';
 import type { InitChatResponse } from '@/api/response/chat';
 import { authToken } from '@/service/utils/tools';
+import { ChatItemType } from '@/types/chat';
+import { authModel } from '@/service/utils/auth';
 /* 初始化我的聊天框，需要身份验证 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
@@ -11,43 +12,46 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
    const { authorization } = req.headers;
    const userId = await authToken(authorization);
-    const { chatId } = req.query as { chatId: string };
+    const { modelId, chatId } = req.query as { modelId: string; chatId: '' | string };
-    if (!chatId) {
+    if (!modelId) {
      throw new Error('缺少参数');
    }
    await connectToDatabase();
-    // 获取 chat 数据
+    // 获取 model 数据
-    const chat = await Chat.findOne<ChatPopulate>({
+    const { model } = await authModel(modelId, userId);
-      _id: chatId,
-      userId
+    // 历史记录
-    }).populate({
+    let history: ChatItemType[] = [];
-      path: 'modelId',
-      options: {
+    if (chatId) {
-        strictPopulate: false
+      // 获取 chat 数据
+      const chat = await Chat.findOne({
+        _id: chatId,
+        userId,
+        modelId
+      });
+      if (!chat) {
+        throw new Error('聊天框不存在');
      }
-    });
-    if (!chat) {
+      // filter 被 deleted 的内容
-      throw new Error('聊天框不存在');
+      history = chat.content.filter((item) => item.deleted !== true);
    }
-    // filter 掉被 deleted 的内容
-    chat.content = chat.content.filter((item) => item.deleted !== true);
-    const model = chat.modelId;
    jsonRes<InitChatResponse>(res, {
      data: {
-        chatId: chat._id,
+        chatId: chatId || '',
-        modelId: model._id,
+        modelId: modelId,
        name: model.name,
        avatar: model.avatar,
        intro: model.intro,
        modelName: model.service.modelName,
        chatModel: model.service.chatModel,
-        history: chat.content
+        history
      }
    });
  } catch (err) {

--- a/src/pages/api/chat/saveChat.ts
+++ b/src/pages/api/chat/saveChat.ts
@@ -2,34 +2,53 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 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';
 /* 聊天内容存存储 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
  try {
-    const { chatId, prompts } = req.body as {
+    const { chatId, modelId, prompts } = req.body as {
-      chatId: string;
+      chatId: '' | string;
+      modelId: string;
      prompts: ChatItemType[];
    };
-    if (!chatId || !prompts) {
+    if (!prompts) {
      throw new Error('缺少参数');
    }
+    const userId = await authToken(req.headers.authorization);
    await connectToDatabase();
-    // 存入库
+    const content = prompts.map((item) => ({
-    await Chat.findByIdAndUpdate(chatId, {
+      obj: item.obj,
-      $push: {
+      value: item.value
-        content: {
+    }));
-          $each: prompts.map((item) => ({
-            obj: item.obj,
-            value: item.value
-          }))
-        }
-      },
-      updateTime: new Date()
-    });
+    // 没有 chatId, 创建一个对话
+    if (!chatId) {
+      await authModel(modelId, userId);
+      const { _id } = await Chat.create({
+        userId,
+        modelId,
+        content
+      });
+      return jsonRes(res, {
+        data: _id
+      });
+    } else {
+      // 已经有记录，追加入库
+      await Chat.findByIdAndUpdate(chatId, {
+        $push: {
+          content: {
+            $each: content
+          }
+        },
+        updateTime: new Date()
+      });
+    }
    jsonRes(res);
  } catch (err) {
    jsonRes(res, {

--- a/src/pages/api/chat/vectorGpt.ts
+++ b/src/pages/api/chat/vectorGpt.ts
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { connectToDatabase } from '@/service/mongo';
-import { authChat } from '@/service/utils/chat';
+import { authChat } from '@/service/utils/auth';
 import { httpsAgent, systemPromptFilter, openaiChatFilter } from '@/service/utils/tools';
 import { ChatCompletionRequestMessage, ChatCompletionRequestMessageRoleEnum } from 'openai';
 import { ChatItemType } from '@/types/chat';
@@ -35,29 +35,33 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
  });
  try {
-    const { chatId, prompt } = req.body as {
+    const { modelId, chatId, prompt } = req.body as {
+      modelId: string;
+      chatId: '' | string;
      prompt: ChatItemType;
-      chatId: string;
    };
    const { authorization } = req.headers;
-    if (!chatId || !prompt) {
+    if (!modelId || !prompt) {
      throw new Error('缺少参数');
    }
    await connectToDatabase();
    let startTime = Date.now();
-    const { chat, userApiKey, systemKey, userId } = await authChat(chatId, authorization);
+    const { model, content, userApiKey, systemKey, userId } = await authChat({
+      modelId,
+      chatId,
+      authorization
+    });
-    const model: ModelSchema = chat.modelId;
    const modelConstantsData = modelList.find((item) => item.model === model.service.modelName);
    if (!modelConstantsData) {
      throw new Error('模型加载异常');
    }
    // 读取对话内容
-    const prompts = [...chat.content, prompt];
+    const prompts = [...content, prompt];
    // 获取提示词的向量
    const { vector: promptVector, chatAPI } = await openaiCreateEmbedding({

--- a/src/pages/api/model/create.ts
+++ b/src/pages/api/model/create.ts
@@ -47,7 +47,6 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
      userId,
      status: ModelStatusEnum.running,
      service: {
-        company: modelItem.serviceCompany,
        trainId: '',
        chatModel: ChatModelNameMap[modelItem.model], // 聊天时用的模型
        modelName: modelItem.model // 最底层的模型，不会变，用于计费等核心操作

--- a/src/pages/api/model/data/splitData.ts
+++ b/src/pages/api/model/data/splitData.ts
@@ -36,14 +36,14 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
    const textList: string[] = [];
    let splitText = '';
-    /* 取 3k ~ 4K tokens 内容 */
+    /* 取 2.5k ~ 3.5K tokens 内容 */
    chunks.forEach((chunk) => {
      const tokens = encode(splitText + chunk).length;
-      if (tokens >= 4000) {
+      if (tokens >= 3500) {
-        // 超过 4000，不要这块内容
+        // 超过 3500，不要这块内容
        splitText && textList.push(splitText);
        splitText = chunk;
-      } else if (tokens >= 3000) {
+      } else if (tokens >= 2500) {
        // 超过 3000，取内容
        splitText && textList.push(splitText + chunk);
        splitText = '';

--- a/src/pages/api/openapi/chat/chatGpt.ts
+++ b/src/pages/api/openapi/chat/chatGpt.ts
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { connectToDatabase, Model } from '@/service/mongo';
-import { getOpenAIApi } from '@/service/utils/chat';
+import { getOpenAIApi } from '@/service/utils/auth';
 import { httpsAgent, openaiChatFilter, authOpenApiKey } from '@/service/utils/tools';
 import { ChatCompletionRequestMessage, ChatCompletionRequestMessageRoleEnum } from 'openai';
 import { ChatItemType } from '@/types/chat';

--- a/src/pages/api/openapi/chat/lafGpt.ts
+++ b/src/pages/api/openapi/chat/lafGpt.ts
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { connectToDatabase, Model } from '@/service/mongo';
-import { getOpenAIApi } from '@/service/utils/chat';
+import { getOpenAIApi } from '@/service/utils/auth';
 import { authOpenApiKey } from '@/service/utils/tools';
 import { httpsAgent, openaiChatFilter, systemPromptFilter } from '@/service/utils/tools';
 import { ChatCompletionRequestMessage, ChatCompletionRequestMessageRoleEnum } from 'openai';

--- a/src/pages/chat/components/SlideBar.tsx
+++ b/src/pages/chat/components/SlideBar.tsx
@@ -11,13 +11,6 @@ import {
  Flex,
  Divider,
  IconButton,
-  Modal,
-  ModalOverlay,
-  ModalContent,
-  ModalHeader,
-  ModalFooter,
-  ModalBody,
-  ModalCloseButton,
  useDisclosure,
  useColorMode,
  useColorModeValue
@@ -29,8 +22,6 @@ import { useRouter } from 'next/router';
 import { getToken } from '@/utils/user';
 import MyIcon from '@/components/Icon';
 import { useCopyData } from '@/utils/tools';
-import Markdown from '@/components/Markdown';
-import { getChatSiteId } from '@/api/chat';
 import WxConcat from '@/components/WxConcat';
 import { useMarkdown } from '@/hooks/useMarkdown';
@@ -42,7 +33,7 @@ const SlideBar = ({
 }: {
  chatId: string;
  modelId: string;
-  resetChat: () => void;
+  resetChat: (modelId?: string, chatId?: string) => void;
  onClose: () => void;
 }) => {
  const router = useRouter();
@@ -86,7 +77,7 @@ const SlideBar = ({
            : {})}
          onClick={() => {
            if (item.chatId === chatId) return;
-            router.replace(`/chat?chatId=${item.chatId}`);
+            resetChat(modelId, item.chatId);
            onClose();
          }}
        >
@@ -155,7 +146,7 @@ const SlideBar = ({
          mb={4}
          mx={'auto'}
          leftIcon={<AddIcon />}
-          onClick={resetChat}
+          onClick={() => resetChat()}
        >
          新对话
        </Button>
@@ -194,7 +185,7 @@ const SlideBar = ({
                      : {})}
                    onClick={async () => {
                      if (item._id === modelId) return;
-                      router.replace(`/chat?chatId=${await getChatSiteId(item._id)}`);
+                      resetChat(item._id);
                      onClose();
                    }}
                  >
@@ -260,49 +251,6 @@ const SlideBar = ({
        />
      </Flex>
-      {/* 分享提示modal */}
-      <Modal isOpen={isOpenShare} onClose={onCloseShare}>
-        <ModalOverlay />
-        <ModalContent color={useColorModeValue('blackAlpha.700', 'white')}>
-          <ModalHeader>分享对话</ModalHeader>
-          <ModalCloseButton />
-          <ModalBody>
-            <Markdown source={shareHint} />
-          </ModalBody>
-          <ModalFooter>
-            <Button colorScheme="gray" variant={'outline'} mr={3} onClick={onCloseShare}>
-              取消
-            </Button>
-            {getToken() && (
-              <Button
-                variant="outline"
-                mr={3}
-                onClick={async () => {
-                  copyData(
-                    `${location.origin}/chat?chatId=${await getChatSiteId(modelId)}`,
-                    '已复制分享链接'
-                  );
-                  onCloseShare();
-                  onClose();
-                }}
-              >
-                分享空白对话
-              </Button>
-            )}
-            <Button
-              onClick={() => {
-                copyData(`${location.origin}/chat?chatId=${chatId}`, '已复制分享链接');
-                onCloseShare();
-                onClose();
-              }}
-            >
-              分享聊天记录
-            </Button>
-          </ModalFooter>
-        </ModalContent>
-      </Modal>
      {/* wx 联系 */}
      {isOpenWx && <WxConcat onClose={onCloseWx} />}
    </Flex>

--- a/src/pages/chat/index.tsx
+++ b/src/pages/chat/index.tsx
 import React, { useCallback, useState, useRef, useMemo, useEffect } from 'react';
 import { useRouter } from 'next/router';
 import Image from 'next/image';
-import { getInitChatSiteInfo, getChatSiteId, delChatRecordByIndex, postSaveChat } from '@/api/chat';
+import { getInitChatSiteInfo, delChatRecordByIndex, postSaveChat } from '@/api/chat';
 import type { InitChatResponse } from '@/api/response/chat';
 import { ChatSiteItemType } from '@/types/chat';
 import {
@@ -41,18 +41,17 @@ interface ChatType extends InitChatResponse {
  history: ChatSiteItemType[];
 }
-const Chat = ({ chatId }: { chatId: string }) => {
+const Chat = ({ modelId, chatId }: { modelId: string; chatId: string }) => {
+  const router = useRouter();
  const ChatBox = useRef<HTMLDivElement>(null);
  const TextareaDom = useRef<HTMLTextAreaElement>(null);
-  const { toast } = useToast();
-  const router = useRouter();
  // 中断请求
  const controller = useRef(new AbortController());
  const [chatData, setChatData] = useState<ChatType>({
-    chatId: '',
+    chatId,
-    modelId: '',
+    modelId,
    name: '',
    avatar: '',
    intro: '',
@@ -60,6 +59,7 @@ const Chat = ({ chatId }: { chatId: string }) => {
    modelName: '',
    history: []
  }); // 聊天框整体数据
  const [inputVal, setInputVal] = useState(''); // 输入的内容
  const isChatting = useMemo(
@@ -68,6 +68,7 @@ const Chat = ({ chatId }: { chatId: string }) => {
  );
  const { isOpen: isOpenSlider, onClose: onCloseSlider, onOpen: onOpenSlider } = useDisclosure();
+  const { toast } = useToast();
  const { copyData } = useCopyData();
  const { isPc, media } = useScreen();
  const { setLoading } = useGlobalStore();
@@ -108,19 +109,72 @@ const Chat = ({ chatId }: { chatId: string }) => {
    }, 100);
  }, []);
-  // 重载对话
+  // 获取对话信息
-  const resetChat = useCallback(async () => {
+  const loadChatInfo = useCallback(
-    if (!chatData) return;
+    async ({
-    try {
+      modelId,
-      router.replace(`/chat?chatId=${await getChatSiteId(chatData.modelId)}`);
+      chatId,
-    } catch (error: any) {
+      isLoading = false,
-      toast({
+      isScroll = false
-        title: error?.message || '生成新对话失败',
+    }: {
-        status: 'warning'
+      modelId: string;
-      });
+      chatId: string;
-    }
+      isLoading?: boolean;
-    onCloseSlider();
+      isScroll?: boolean;
-  }, [chatData, onCloseSlider, router, toast]);
+    }) => {
+      isLoading && setLoading(true);
+      try {
+        const res = await getInitChatSiteInfo(modelId, chatId);
+        setChatData({
+          ...res,
+          history: res.history.map((item) => ({
+            ...item,
+            status: 'finish'
+          }))
+        });
+        if (isScroll && res.history.length > 0) {
+          setTimeout(() => {
+            scrollToBottom('auto');
+          }, 2000);
+        }
+      } catch (e: any) {
+        toast({
+          title: e?.message || '获取对话信息异常,请检查地址',
+          status: 'error',
+          isClosable: true,
+          duration: 5000
+        });
+        router.replace('/model/list');
+      }
+      setLoading(false);
+      return null;
+    },
+    [router, scrollToBottom, setLoading, toast]
+  );
+  // 重载新的对话
+  const resetChat = useCallback(
+    async (modelId = chatData.modelId, chatId = '') => {
+      // 强制中断流
+      controller.current?.abort();
+      try {
+        router.replace(`/chat?modelId=${modelId}&chatId=${chatId}`);
+        loadChatInfo({
+          modelId,
+          chatId,
+          isLoading: true,
+          isScroll: true
+        });
+      } catch (error: any) {
+        toast({
+          title: error?.message || '生成新对话失败',
+          status: 'warning'
+        });
+      }
+      onCloseSlider();
+    },
+    [chatData.modelId, loadChatInfo, onCloseSlider, router, toast]
+  );
  // gpt 对话
  const gptChatPrompt = useCallback(
@@ -132,6 +186,10 @@ const Chat = ({ chatId }: { chatId: string }) => {
      if (!urlMap[chatData.modelName]) return Promise.reject('找不到模型');
+      // create abort obj
+      const abortSignal = new AbortController();
+      controller.current = abortSignal;
      const prompt = {
        obj: prompts.obj,
        value: prompts.value
@@ -141,7 +199,8 @@ const Chat = ({ chatId }: { chatId: string }) => {
        url: urlMap[chatData.modelName],
        data: {
          prompt,
-          chatId
+          chatId,
+          modelId
        },
        onMessage: (text: string) => {
          setChatData((state) => ({
@@ -156,12 +215,14 @@ const Chat = ({ chatId }: { chatId: string }) => {
          }));
          generatingMessage();
        },
-        abortSignal: controller.current
+        abortSignal
      });
+      let id = '';
      // 保存对话信息
      try {
-        await postSaveChat({
+        id = await postSaveChat({
+          modelId,
          chatId,
          prompts: [
            prompt,
@@ -171,6 +232,9 @@ const Chat = ({ chatId }: { chatId: string }) => {
            }
          ]
        });
+        if (id) {
+          router.replace(`/chat?modelId=${modelId}&chatId=${id}`);
+        }
      } catch (err) {
        toast({
          title: '对话出现异常, 继续对话会导致上下文丢失，请刷新页面',
@@ -183,6 +247,7 @@ const Chat = ({ chatId }: { chatId: string }) => {
      // 设置完成状态
      setChatData((state) => ({
        ...state,
+        chatId: id || state.chatId, // 如果有 Id，说明是新创建的对话
        history: state.history.map((item, index) => {
          if (index !== state.history.length - 1) return item;
          return {
@@ -192,7 +257,7 @@ const Chat = ({ chatId }: { chatId: string }) => {
        })
      }));
    },
-    [chatData.modelName, chatId, generatingMessage, toast]
+    [chatData.modelName, chatId, generatingMessage, modelId, router, toast]
  );
  /**
@@ -210,7 +275,7 @@ const Chat = ({ chatId }: { chatId: string }) => {
    // 去除空行
    const val = inputVal.trim().replace(/\n\s*/g, '\n');
-    if (!chatData?.modelId || !val) {
+    if (!val) {
      toast({
        title: '内容为空',
        status: 'warning'
@@ -271,12 +336,12 @@ const Chat = ({ chatId }: { chatId: string }) => {
      }));
    }
  }, [
-    inputVal,
-    chatData,
    isChatting,
+    inputVal,
+    chatData.history,
    resetInputVal,
-    scrollToBottom,
    toast,
+    scrollToBottom,
    gptChatPrompt,
    pushChatHistory,
    chatId
@@ -312,50 +377,22 @@ const Chat = ({ chatId }: { chatId: string }) => {
  );
  // 初始化聊天框
-  useQuery(
+  useQuery(['init'], () =>
-    ['init', chatId],
+    loadChatInfo({
-    () => {
+      modelId,
-      setLoading(true);
+      chatId,
-      return getInitChatSiteInfo(chatId);
+      isLoading: true,
-    },
+      isScroll: true
-    {
+    })
-      onSuccess(res) {
-        setChatData({
-          ...res,
-          history: res.history.map((item) => ({
-            ...item,
-            status: 'finish'
-          }))
-        });
-        if (res.history.length > 0) {
-          setTimeout(() => {
-            scrollToBottom('auto');
-          }, 2000);
-        }
-      },
-      onError(e: any) {
-        toast({
-          title: e?.message || '初始化异常,请检查地址',
-          status: 'error',
-          isClosable: true,
-          duration: 5000
-        });
-        router.push('/model/list');
-      },
-      onSettled() {
-        setLoading(false);
-      }
-    }
  );
  // 更新流中断对象
  useEffect(() => {
-    controller.current = new AbortController();
    return () => {
      // eslint-disable-next-line react-hooks/exhaustive-deps
      controller.current?.abort();
    };
-  }, [chatId]);
+  }, []);
  return (
    <Flex
@@ -368,7 +405,7 @@ const Chat = ({ chatId }: { chatId: string }) => {
          <SlideBar
            resetChat={resetChat}
            chatId={chatId}
-            modelId={chatData.modelId}
+            modelId={modelId}
            onClose={onCloseSlider}
          />
        </Box>
@@ -399,7 +436,7 @@ const Chat = ({ chatId }: { chatId: string }) => {
              <SlideBar
                resetChat={resetChat}
                chatId={chatId}
-                modelId={chatData.modelId}
+                modelId={modelId}
                onClose={onCloseSlider}
              />
            </DrawerContent>
@@ -565,9 +602,10 @@ const Chat = ({ chatId }: { chatId: string }) => {
 export default Chat;
 export async function getServerSideProps(context: any) {
-  const chatId = context?.query?.chatId || 'noid';
+  const modelId = context?.query?.modelId || '';
+  const chatId = context?.query?.chatId || '';
  return {
-    props: { chatId }
+    props: { modelId, chatId }
  };
 }
--- a/src/pages/index.tsx
+++ b/src/pages/index.tsx
 import React, { useEffect } from 'react';
-import { Card } from '@chakra-ui/react';
+import { Card, Box, Link } from '@chakra-ui/react';
 import Markdown from '@/components/Markdown';
 import { useMarkdown } from '@/hooks/useMarkdown';
 import { useRouter } from 'next/router';
@@ -15,9 +15,20 @@ const Home = () => {
  }, [inviterId]);
  return (
-    <Card p={5} lineHeight={2}>
+    <>
-      <Markdown source={data} isChatting={false} />
+      <Card p={5} lineHeight={2}>
-    </Card>
+        <Markdown source={data} isChatting={false} />
+      </Card>
+      <Card p={5} mt={4} textAlign={'center'}>
+        <Box>
+          {/* <Link href="https://beian.miit.gov.cn/" target="_blank">
+            浙B2-20080101
+          </Link> */}
+        </Box>
+        <Box>Made by FastGpt Team.</Box>
+      </Card>
+    </>
  );
 };

--- a/src/pages/model/detail/index.tsx
+++ b/src/pages/model/detail/index.tsx
 import React, { useCallback, useState, useRef, useMemo, useEffect } from 'react';
 import { useRouter } from 'next/router';
 import { getModelById, delModelById, putModelTrainingStatus, putModelById } from '@/api/model';
-import { getChatSiteId } from '@/api/chat';
 import type { ModelSchema } from '@/types/mongoSchema';
 import { Card, Box, Flex, Button, Tag, Grid } from '@chakra-ui/react';
 import { useToast } from '@/hooks/useToast';
@@ -70,14 +69,12 @@ const ModelDetail = ({ modelId }: { modelId: string }) => {
  const handlePreviewChat = useCallback(async () => {
    setLoading(true);
    try {
-      const chatId = await getChatSiteId(model._id);
+      router.push(`/chat?modelId=${modelId}`);
-      router.push(`/chat?chatId=${chatId}`);
    } catch (err) {
      console.log('error->', err);
    }
    setLoading(false);
-  }, [setLoading, model, router]);
+  }, [setLoading, router, modelId]);
  /* 上传数据集,触发微调 */
  // const startTraining = useCallback(

--- a/src/pages/model/list/index.tsx
+++ b/src/pages/model/list/index.tsx
 import React, { useState, useCallback } from 'react';
 import { Box, Button, Flex, Card } from '@chakra-ui/react';
-import { getChatSiteId } from '@/api/chat';
 import type { ModelSchema } from '@/types/mongoSchema';
 import { useRouter } from 'next/router';
 import ModelTable from './components/ModelTable';

--- a/src/service/events/generateAbstract.ts
+++ b/src/service/events/generateAbstract.ts
 import { DataItem } from '@/service/mongo';
-import { getOpenAIApi } from '@/service/utils/chat';
+import { getOpenAIApi } from '@/service/utils/auth';
 import { httpsAgent } from '@/service/utils/tools';
 import { getOpenApiKey } from '../utils/openai';
 import type { ChatCompletionRequestMessage } from 'openai';

--- a/src/service/events/generateQA.ts
+++ b/src/service/events/generateQA.ts
 import { SplitData } from '@/service/mongo';
-import { getOpenAIApi } from '@/service/utils/chat';
+import { getOpenAIApi } from '@/service/utils/auth';
 import { httpsAgent } from '@/service/utils/tools';
 import { getOpenApiKey } from '../utils/openai';
 import type { ChatCompletionRequestMessage } from 'openai';

--- a/src/service/events/pushBill.ts
+++ b/src/service/events/pushBill.ts
@@ -14,7 +14,7 @@ export const pushChatBill = async ({
  isPay: boolean;
  modelName: string;
  userId: string;
-  chatId?: string;
+  chatId?: '' | string;
  text: string;
 }) => {
  let billId;
@@ -42,7 +42,7 @@ export const pushChatBill = async ({
          userId,
          type: 'chat',
          modelName,
-          chatId,
+          chatId: chatId ? chatId : undefined,
          textLen: text.length,
          tokenLen: tokens,
          price

--- a/src/service/models/model.ts
+++ b/src/service/models/model.ts
@@ -53,11 +53,6 @@ const ModelSchema = new Schema({
    }
  },
  service: {
-    company: {
-      type: String,
-      required: true,
-      enum: ['openai']
-    },
    trainId: {
      // 训练时需要的 ID， 不能训练的模型没有这个值。
      type: String,

--- a/src/service/utils/auth.ts
+++ b/src/service/utils/auth.ts
+import { Configuration, OpenAIApi } from 'openai';
+import { Chat, Model } from '../mongo';
+import type { ModelSchema } from '@/types/mongoSchema';
+import { authToken } from './tools';
+import { getOpenApiKey } from './openai';
+import type { ChatItemType } from '@/types/chat';
+export const getOpenAIApi = (apiKey: string) => {
+  const configuration = new Configuration({
+    apiKey
+  });
+  return new OpenAIApi(configuration, undefined);
+};
+// 模型使用权校验
+export const authModel = async (modelId: string, userId: string) => {
+  // 获取 model 数据
+  const model = await Model.findById<ModelSchema>(modelId);
+  if (!model) {
+    return Promise.reject('模型不存在');
+  }
+  // 凭证校验
+  if (userId !== String(model.userId)) {
+    return Promise.reject('无权使用该模型');
+  }
+  return { model };
+};
+// 获取对话校验
+export const authChat = async ({
+  modelId,
+  chatId,
+  authorization
+}: {
+  modelId: string;
+  chatId: '' | string;
+  authorization?: string;
+}) => {
+  const userId = await authToken(authorization);
+  // 获取 model 数据
+  const { model } = await authModel(modelId, userId);
+  // 聊天内容
+  let content: ChatItemType[] = [];
+  if (chatId) {
+    // 获取 chat 数据
+    const chat = await Chat.findById(chatId);
+    if (!chat) {
+      return Promise.reject('对话不存在');
+    }
+    // filter 掉被 deleted 的内容
+    content = chat.content.filter((item) => item.deleted !== true);
+  }
+  // 获取 user 的 apiKey
+  const { userApiKey, systemKey } = await getOpenApiKey(userId);
+  return {
+    userApiKey,
+    systemKey,
+    content,
+    userId,
+    model
+  };
+};
--- a/src/service/utils/chat.ts
+++ b/src/service/utils/chat.ts
-import { Configuration, OpenAIApi } from 'openai';
-import { Chat } from '../mongo';
-import type { ChatPopulate } from '@/types/mongoSchema';
-import { authToken } from './tools';
-import { getOpenApiKey } from './openai';
-export const getOpenAIApi = (apiKey: string) => {
-  const configuration = new Configuration({
-    apiKey
-  });
-  return new OpenAIApi(configuration, undefined);
-};
-export const authChat = async (chatId: string, authorization?: string) => {
-  // 获取 chat 数据
-  const chat = await Chat.findById<ChatPopulate>(chatId).populate({
-    path: 'modelId',
-    options: {
-      strictPopulate: false
-    }
-  });
-  if (!chat || !chat.modelId || !chat.userId) {
-    return Promise.reject('模型不存在');
-  }
-  // 凭证校验
-  const userId = await authToken(authorization);
-  if (userId !== String(chat.userId._id)) {
-    return Promise.reject('无权使用该对话');
-  }
-  // 获取 user 的 apiKey
-  const { user, userApiKey, systemKey } = await getOpenApiKey(chat.userId as unknown as string);
-  // filter 掉被 deleted 的内容
-  chat.content = chat.content.filter((item) => item.deleted !== true);
-  return {
-    userApiKey,
-    systemKey,
-    chat,
-    userId: user._id
-  };
-};
--- a/src/service/utils/openai.ts
+++ b/src/service/utils/openai.ts
 import type { NextApiResponse } from 'next';
 import type { PassThrough } from 'stream';
 import { createParser, ParsedEvent, ReconnectInterval } from 'eventsource-parser';
-import { getOpenAIApi } from '@/service/utils/chat';
+import { getOpenAIApi } from '@/service/utils/auth';
 import { httpsAgent } from './tools';
 import { User } from '../models/user';
 import { formatPrice } from '@/utils/user';

--- a/src/store/chat.ts
+++ b/src/store/chat.ts
@@ -2,7 +2,6 @@ import { create } from 'zustand';
 import { devtools, persist } from 'zustand/middleware';
 import { immer } from 'zustand/middleware/immer';
 import type { HistoryItem } from '@/types/chat';
-import { getChatSiteId } from '@/api/chat';
 type Props = {
  chatHistory: HistoryItem[];
@@ -10,7 +9,6 @@ type Props = {
  updateChatHistory: (chatId: string, title: string) => void;
  removeChatHistoryByWindowId: (chatId: string) => void;
  clearHistory: () => void;
-  generateChatWindow: (modelId: string) => Promise<string>;
 };
 export const useChatStore = create<Props>()(
  devtools(
@@ -40,9 +38,6 @@ export const useChatStore = create<Props>()(
          set((state) => {
            state.chatHistory = [];
          });
-        },
-        generateChatWindow(modelId: string) {
-          return getChatSiteId(modelId);
        }
      })),
      {

--- a/src/types/mongoSchema.d.ts
+++ b/src/types/mongoSchema.d.ts
@@ -7,8 +7,6 @@ import {
 } from '@/constants/model';
 import type { DataType } from './data';
-export type ServiceName = 'openai';
 export interface UserModelSchema {
  _id: string;
  username: string;
@@ -46,7 +44,6 @@ export interface ModelSchema {
    mode: `${ModelVectorSearchModeEnum}`;
  };
  service: {
-    company: ServiceName;
    trainId: string; // 训练的模型，训练后就是训练的模型id
    chatModel: string; // 聊天时用的模型，训练后就是训练的模型
    modelName: `${ChatModelNameEnum}`; // 底层模型名称，不会变
@@ -86,7 +83,6 @@ export interface ModelSplitDataSchema {
 export interface TrainingSchema {
  _id: string;
-  serviceName: ServiceName;
  tuneId: string;
  modelId: string;
  status: `${TrainingStatusEnum}`;