new framwork

client/src/api/chat.ts

@@ -0,0 +1,79 @@
+import { GET, POST, DELETE, PUT } from './request';
+import type { HistoryItemType } from '@/types/chat';
+import type { InitChatResponse, InitShareChatResponse } from './response/chat';
+import { RequestPaging } from '../types/index';
+import type { ShareChatSchema } from '@/types/mongoSchema';
+import type { ShareChatEditType } from '@/types/model';
+import { Obj2Query } from '@/utils/tools';
+import type { QuoteItemType } from '@/pages/api/openapi/kb/appKbSearch';
+import type { Props as UpdateHistoryProps } from '@/pages/api/chat/history/updateChatHistory';
+
+/**
+ * 获取初始化聊天内容
+ */
+export const getInitChatSiteInfo = (modelId: '' | string, chatId: '' | string) =>
+  GET<InitChatResponse>(`/chat/init?modelId=${modelId}&chatId=${chatId}`);
+
+/**
+ * 获取历史记录
+ */
+export const getChatHistory = (data: RequestPaging) =>
+  POST<HistoryItemType[]>('/chat/history/getHistory', data);
+
+/**
+ * 删除一条历史记录
+ */
+export const delChatHistoryById = (id: string) => GET(`/chat/removeHistory?id=${id}`);
+
+/**
+ * get history quotes
+ */
+export const getHistoryQuote = (params: { chatId: string; historyId: string }) =>
+  GET<(QuoteItemType & { _id: string })[]>(`/chat/history/getHistoryQuote`, params);
+
+/**
+ * update history quote status
+ */
+export const updateHistoryQuote = (params: {
+  chatId: string;
+  historyId: string;
+  quoteId: string;
+}) => GET(`/chat/history/updateHistoryQuote`, params);
+
+/**
+ * 删除一句对话
+ */
+export const delChatRecordByIndex = (chatId: string, contentId: string) =>
+  DELETE(`/chat/delChatRecordByContentId?chatId=${chatId}&contentId=${contentId}`);
+
+/**
+ * 修改历史记录: 标题/置顶
+ */
+export const putChatHistory = (data: UpdateHistoryProps) =>
+  PUT('/chat/history/updateChatHistory', data);
+
+/**
+ * create a shareChat
+ */
+export const createShareChat = (
+  data: ShareChatEditType & {
+    modelId: string;
+  }
+) => POST<string>(`/chat/shareChat/create`, data);
+
+/**
+ * get shareChat
+ */
+export const getShareChatList = (modelId: string) =>
+  GET<ShareChatSchema[]>(`/chat/shareChat/list?modelId=${modelId}`);
+
+/**
+ * delete a  shareChat
+ */
+export const delShareChatById = (id: string) => DELETE(`/chat/shareChat/delete?id=${id}`);
+
+/**
+ * 初始化分享聊天
+ */
+export const initShareChatInfo = (data: { shareId: string; password: string }) =>
+  GET<InitShareChatResponse>(`/chat/shareChat/init?${Obj2Query(data)}`);

client/src/api/common.ts

@@ -0,0 +1 @@
+import { GET, POST, DELETE } from './request';

client/src/api/fetch.ts

@@ -0,0 +1,67 @@
+import { GUIDE_PROMPT_HEADER, NEW_CHATID_HEADER, QUOTE_LEN_HEADER } from '@/constants/chat';
+
+interface StreamFetchProps {
+  url: string;
+  data: any;
+  onMessage: (text: string) => void;
+  abortSignal: AbortController;
+}
+export const streamFetch = ({ url, data, onMessage, abortSignal }: StreamFetchProps) =>
+  new Promise<{
+    responseText: string;
+    newChatId: string;
+    systemPrompt: string;
+    quoteLen: number;
+  }>(async (resolve, reject) => {
+    try {
+      const res = await fetch(url, {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json'
+        },
+        body: JSON.stringify(data),
+        signal: abortSignal.signal
+      });
+      const reader = res.body?.getReader();
+      if (!reader) return;
+
+      const decoder = new TextDecoder();
+
+      const newChatId = decodeURIComponent(res.headers.get(NEW_CHATID_HEADER) || '');
+      const systemPrompt = decodeURIComponent(res.headers.get(GUIDE_PROMPT_HEADER) || '').trim();
+      const quoteLen = res.headers.get(QUOTE_LEN_HEADER)
+        ? Number(res.headers.get(QUOTE_LEN_HEADER))
+        : 0;
+
+      let responseText = '';
+
+      const read = async () => {
+        try {
+          const { done, value } = await reader?.read();
+          if (done) {
+            if (res.status === 200) {
+              resolve({ responseText, newChatId, quoteLen, systemPrompt });
+            } else {
+              const parseError = JSON.parse(responseText);
+              reject(parseError?.message || '请求异常');
+            }
+
+            return;
+          }
+          const text = decoder.decode(value);
+          responseText += text;
+          onMessage(text);
+          read();
+        } catch (err: any) {
+          if (err?.message === 'The user aborted a request.') {
+            return resolve({ responseText, newChatId, quoteLen, systemPrompt });
+          }
+          reject(typeof err === 'string' ? err : err?.message || '请求异常');
+        }
+      };
+      read();
+    } catch (err: any) {
+      console.log(err, '====');
+      reject(typeof err === 'string' ? err : err?.message || '请求异常');
+    }
+  });

client/src/api/model.ts

@@ -0,0 +1,44 @@
+import { GET, POST, DELETE, PUT } from './request';
+import type { ModelSchema } from '@/types/mongoSchema';
+import type { ModelUpdateParams } from '@/types/model';
+import { RequestPaging } from '../types/index';
+import type { ModelListResponse } from './response/model';
+
+/**
+ * 获取模型列表
+ */
+export const getMyModels = () => GET<ModelListResponse>('/model/list');
+
+/**
+ * 创建一个模型
+ */
+export const postCreateModel = (data: { name: string }) => POST<string>('/model/create', data);
+
+/**
+ * 根据 ID 删除模型
+ */
+export const delModelById = (id: string) => DELETE(`/model/del?modelId=${id}`);
+
+/**
+ * 根据 ID 获取模型
+ */
+export const getModelById = (id: string) => GET<ModelSchema>(`/model/detail?modelId=${id}`);
+
+/**
+ * 根据 ID 更新模型
+ */
+export const putModelById = (id: string, data: ModelUpdateParams) =>
+  PUT(`/model/update?modelId=${id}`, data);
+
+/* 共享市场 */
+/**
+ * 获取共享市场模型
+ */
+export const getShareModelList = (data: { searchText?: string } & RequestPaging) =>
+  POST(`/model/share/getModels`, data);
+
+/**
+ * 收藏/取消收藏模型
+ */
+export const triggerModelCollection = (modelId: string) =>
+  POST<number>(`/model/share/collection?modelId=${modelId}`);

client/src/api/openapi.ts

@@ -0,0 +1,16 @@
+import { GET, POST, DELETE } from './request';
+import { UserOpenApiKey } from '@/types/openapi';
+/**
+ * crete a api key
+ */
+export const createAOpenApiKey = () => POST<string>('/openapi/postKey');
+
+/**
+ * get api keys
+ */
+export const getOpenApiKeys = () => GET<UserOpenApiKey[]>('/openapi/getKeys');
+
+/**
+ * delete api by id
+ */
+export const delOpenApiById = (id: string) => DELETE(`/openapi/delKey?id=${id}`);

client/src/api/plugins/kb.ts

@@ -0,0 +1,85 @@
+import { GET, POST, PUT, DELETE } from '../request';
+import type { KbItemType } from '@/types/plugin';
+import { RequestPaging } from '@/types/index';
+import { TrainingModeEnum } from '@/constants/plugin';
+import {
+  Props as PushDataProps,
+  Response as PushDateResponse
+} from '@/pages/api/openapi/kb/pushData';
+
+export type KbUpdateParams = {
+  id: string;
+  name: string;
+  tags: string;
+  avatar: string;
+};
+
+/* knowledge base */
+export const getKbList = () => GET<KbItemType[]>(`/plugins/kb/list`);
+
+export const getKbById = (id: string) => GET<KbItemType>(`/plugins/kb/detail?id=${id}`);
+
+export const postCreateKb = (data: { name: string }) => POST<string>(`/plugins/kb/create`, data);
+
+export const putKbById = (data: KbUpdateParams) => PUT(`/plugins/kb/update`, data);
+
+export const delKbById = (id: string) => DELETE(`/plugins/kb/delete?id=${id}`);
+
+/* kb data */
+type GetKbDataListProps = RequestPaging & {
+  kbId: string;
+  searchText: string;
+};
+export const getKbDataList = (data: GetKbDataListProps) =>
+  POST(`/plugins/kb/data/getDataList`, data);
+
+/**
+ * 获取导出数据（不分页）
+ */
+export const getExportDataList = (kbId: string) =>
+  GET<[string, string][]>(
+    `/plugins/kb/data/exportModelData`,
+    { kbId },
+    {
+      timeout: 600000
+    }
+  );
+
+/**
+ * 获取模型正在拆分数据的数量
+ */
+export const getTrainingData = (data: { kbId: string; init: boolean }) =>
+  POST<{
+    qaListLen: number;
+    vectorListLen: number;
+  }>(`/plugins/kb/data/getTrainingData`, data);
+
+export const getKbDataItemById = (dataId: string) =>
+  GET(`/plugins/kb/data/getDataById`, { dataId });
+
+/**
+ * 直接push数据
+ */
+export const postKbDataFromList = (data: PushDataProps) =>
+  POST<PushDateResponse>(`/openapi/kb/pushData`, data);
+
+/**
+ * 更新一条数据
+ */
+export const putKbDataById = (data: { dataId: string; a: string; q?: string }) =>
+  PUT('/openapi/kb/updateData', data);
+/**
+ * 删除一条知识库数据
+ */
+export const delOneKbDataByDataId = (dataId: string) =>
+  DELETE(`/openapi/kb/delDataById?dataId=${dataId}`);
+
+/**
+ * 拆分数据
+ */
+export const postSplitData = (data: {
+  kbId: string;
+  chunks: string[];
+  prompt: string;
+  mode: `${TrainingModeEnum}`;
+}) => POST(`/openapi/text/pushData`, data);

client/src/api/request.ts

@@ -0,0 +1,124 @@
+import axios, { Method, InternalAxiosRequestConfig, AxiosResponse } from 'axios';
+import { clearCookie } from '@/utils/user';
+import { TOKEN_ERROR_CODE } from '@/service/errorCode';
+
+interface ConfigType {
+  headers?: { [key: string]: string };
+  hold?: boolean;
+  timeout?: number;
+}
+interface ResponseDataType {
+  code: number;
+  message: string;
+  data: any;
+}
+
+/**
+ * 请求开始
+ */
+function requestStart(config: InternalAxiosRequestConfig): InternalAxiosRequestConfig {
+  if (config.headers) {
+    // config.headers.Authorization = getToken();
+  }
+
+  return config;
+}
+
+/**
+ * 请求成功,检查请求头
+ */
+function responseSuccess(response: AxiosResponse<ResponseDataType>) {
+  return response;
+}
+/**
+ * 响应数据检查
+ */
+function checkRes(data: ResponseDataType) {
+  if (data === undefined) {
+    console.log('error->', data, 'data is empty');
+    return Promise.reject('服务器异常');
+  } else if (data.code < 200 || data.code >= 400) {
+    return Promise.reject(data);
+  }
+  return data.data;
+}
+
+/**
+ * 响应错误
+ */
+function responseError(err: any) {
+  console.log('error->', '请求错误', err);
+
+  if (!err) {
+    return Promise.reject({ message: '未知错误' });
+  }
+  if (typeof err === 'string') {
+    return Promise.reject({ message: err });
+  }
+  // 有报错响应
+  if (err?.code in TOKEN_ERROR_CODE) {
+    clearCookie();
+    window.location.replace(
+      `/login?lastRoute=${encodeURIComponent(location.pathname + location.search)}`
+    );
+    return Promise.reject({ message: 'token过期，重新登录' });
+  }
+  return Promise.reject(err);
+}
+
+/* 创建请求实例 */
+const instance = axios.create({
+  timeout: 60000, // 超时时间
+  headers: {
+    'content-type': 'application/json'
+  }
+});
+
+/* 请求拦截 */
+instance.interceptors.request.use(requestStart, (err) => Promise.reject(err));
+/* 响应拦截 */
+instance.interceptors.response.use(responseSuccess, (err) => Promise.reject(err));
+
+function request(url: string, data: any, config: ConfigType, method: Method): any {
+  /* 去空 */
+  for (const key in data) {
+    if (data[key] === null || data[key] === undefined) {
+      delete data[key];
+    }
+  }
+
+  return instance
+    .request({
+      baseURL: '/api',
+      url,
+      method,
+      data: method === 'GET' ? null : data,
+      params: method === 'GET' ? data : null, // get请求不携带data，params放在url上
+      ...config // 用户自定义配置，可以覆盖前面的配置
+    })
+    .then((res) => checkRes(res.data))
+    .catch((err) => responseError(err));
+}
+
+/**
+ * api请求方式
+ * @param {String} url
+ * @param {Any} params
+ * @param {Object} config
+ * @returns
+ */
+export function GET<T>(url: string, params = {}, config: ConfigType = {}): Promise<T> {
+  return request(url, params, config, 'GET');
+}
+
+export function POST<T>(url: string, data = {}, config: ConfigType = {}): Promise<T> {
+  return request(url, data, config, 'POST');
+}
+
+export function PUT<T>(url: string, data = {}, config: ConfigType = {}): Promise<T> {
+  return request(url, data, config, 'PUT');
+}
+
+export function DELETE<T>(url: string, config: ConfigType = {}): Promise<T> {
+  return request(url, {}, config, 'DELETE');
+}

client/src/api/response/chat.d.ts

@@ -0,0 +1,26 @@
+import type { ChatPopulate, ModelSchema } from '@/types/mongoSchema';
+import type { ChatItemType } from '@/types/chat';
+
+export interface InitChatResponse {
+  chatId: string;
+  modelId: string;
+  model: {
+    name: string;
+    avatar: string;
+    intro: string;
+    canUse: boolean;
+  };
+  chatModel: ModelSchema['chat']['chatModel']; // 对话模型名
+  history: ChatItemType[];
+}
+
+export interface InitShareChatResponse {
+  maxContext: number;
+  userAvatar: string;
+  model: {
+    name: string;
+    avatar: string;
+    intro: string;
+  };
+  chatModel: ModelSchema['chat']['chatModel']; // 对话模型名
+}

client/src/api/response/model.d.ts

@@ -0,0 +1,6 @@
+import { ModelListItemType } from '@/types/model';
+
+export type ModelListResponse = {
+  myModels: ModelListItemType[];
+  myCollectionModels: ModelListItemType[];
+};

client/src/api/response/user.d.ts

@@ -0,0 +1,12 @@
+import type { UserType } from '@/types/user';
+import type { PromotionRecordSchema } from '@/types/mongoSchema';
+export interface ResLogin {
+  user: UserType;
+}
+
+export interface PromotionRecordType {
+  _id: PromotionRecordSchema['_id'];
+  type: PromotionRecordSchema['type'];
+  createTime: PromotionRecordSchema['createTime'];
+  amount: PromotionRecordSchema['amount'];
+}

client/src/api/system.ts

@@ -0,0 +1,7 @@
+import { GET, POST, PUT } from './request';
+import type { ChatModelItemType } from '@/constants/model';
+import type { InitDateResponse } from '@/pages/api/system/getInitData';
+
+export const getInitData = () => GET<InitDateResponse>('/system/getInitData');
+
+export const getSystemModelList = () => GET<ChatModelItemType[]>('/system/getModels');

client/src/api/user.ts

@@ -0,0 +1,89 @@
+import { GET, POST, PUT } from './request';
+import { createHashPassword, Obj2Query } from '@/utils/tools';
+import { ResLogin, PromotionRecordType } from './response/user';
+import { UserAuthTypeEnum } from '@/constants/common';
+import { UserBillType, UserType, UserUpdateParams } from '@/types/user';
+import type { PagingData, RequestPaging } from '@/types';
+import { informSchema, PaySchema } from '@/types/mongoSchema';
+
+export const sendAuthCode = (data: {
+  username: string;
+  type: `${UserAuthTypeEnum}`;
+  googleToken: string;
+}) => POST('/user/sendAuthCode', data);
+
+export const getTokenLogin = () => GET<UserType>('/user/tokenLogin');
+
+/* get promotion init data */
+export const getPromotionInitData = () =>
+  GET<{
+    invitedAmount: number;
+    historyAmount: number;
+    residueAmount: number;
+  }>('/user/promotion/getPromotionData');
+
+export const postRegister = ({
+  username,
+  password,
+  code,
+  inviterId
+}: {
+  username: string;
+  code: string;
+  password: string;
+  inviterId: string;
+}) =>
+  POST<ResLogin>('/user/register', {
+    username,
+    code,
+    inviterId,
+    password: createHashPassword(password)
+  });
+
+export const postFindPassword = ({
+  username,
+  code,
+  password
+}: {
+  username: string;
+  code: string;
+  password: string;
+}) =>
+  POST<ResLogin>('/user/updatePasswordByCode', {
+    username,
+    code,
+    password: createHashPassword(password)
+  });
+
+export const postLogin = ({ username, password }: { username: string; password: string }) =>
+  POST<ResLogin>('/user/loginByPassword', {
+    username,
+    password: createHashPassword(password)
+  });
+
+export const loginOut = () => GET('/user/loginout');
+
+export const putUserInfo = (data: UserUpdateParams) => PUT('/user/update', data);
+
+export const getUserBills = (data: RequestPaging) =>
+  GET<PagingData<UserBillType>>(`/user/getBill?${Obj2Query(data)}`);
+
+export const getPayOrders = () => GET<PaySchema[]>(`/user/getPayOrders`);
+
+export const getPayCode = (amount: number) =>
+  GET<{
+    codeUrl: string;
+    payId: string;
+  }>(`/user/getPayCode?amount=${amount}`);
+
+export const checkPayResult = (payId: string) => GET<number>(`/user/checkPayResult?payId=${payId}`);
+
+/* promotion records */
+export const getPromotionRecords = (data: RequestPaging) =>
+  GET<PromotionRecordType>(`/user/promotion/getPromotions?${Obj2Query(data)}`);
+
+export const getInforms = (data: RequestPaging) =>
+  POST<PagingData<informSchema>>(`/user/inform/list`, data);
+
+export const getUnreadCount = () => GET<number>(`/user/inform/countUnread`);
+export const readInform = (id: string) => GET(`/user/inform/read`, { id });