Revert "sub plan page (#885)" (#886)

This reverts commit 443ad37b6a.
2024-02-23 17:48:15 +08:00
parent 443ad37b6a
commit fd9b6291af
246 changed files with 4281 additions and 6286 deletions
--- a/projects/app/src/pages/api/core/ai/agent/createQuestionGuide.ts
+++ b/projects/app/src/pages/api/core/ai/agent/createQuestionGuide.ts
@@ -2,7 +2,7 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@fastgpt/service/common/response';
 import { connectToDatabase } from '@/service/mongo';
 import type { CreateQuestionGuideParams } from '@/global/core/ai/api.d';
-import { pushQuestionGuideUsage } from '@/service/support/wallet/usage/push';
+import { pushQuestionGuideBill } from '@/service/support/wallet/bill/push';
 import { createQuestionGuide } from '@fastgpt/service/core/ai/functions/createQuestionGuide';
 import { authCertOrShareId } from '@fastgpt/service/support/permission/auth/common';

@@ -19,7 +19,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<

    const qgModel = global.llmModels[0];

-    const { result, charsLength } = await createQuestionGuide({
+    const { result, inputTokens, outputTokens } = await createQuestionGuide({
      messages,
      model: qgModel.model
    });
@@ -28,8 +28,9 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
      data: result
    });

-    pushQuestionGuideUsage({
-      charsLength,
+    pushQuestionGuideBill({
+      inputTokens,
+      outputTokens,
      teamId,
      tmbId
    });
--- a/projects/app/src/pages/api/core/app/create.ts
+++ b/projects/app/src/pages/api/core/app/create.ts
@@ -6,7 +6,6 @@ import { AppTypeEnum } from '@fastgpt/global/core/app/constants';
 import { MongoApp } from '@fastgpt/service/core/app/schema';
 import { authUserNotVisitor } from '@fastgpt/service/support/permission/auth/user';
 import { SimpleModeTemplate_FastGPT_Universal } from '@/global/core/app/constants';
-import { checkTeamAppLimit } from '@/service/support/permission/teamLimit';

 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
  try {
@@ -26,7 +25,12 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
    const { teamId, tmbId } = await authUserNotVisitor({ req, authToken: true });

    // 上限校验
-    await checkTeamAppLimit(teamId);
+    const authCount = await MongoApp.countDocuments({
+      teamId
+    });
+    if (authCount >= 50) {
+      throw new Error('每个团队上限 50 个应用');
+    }

    // 创建模型
    const response = await MongoApp.create({
--- a/projects/app/src/pages/api/core/app/data/totalUsage.ts
+++ b/projects/app/src/pages/api/core/app/data/totalUsage.ts
@@ -0,0 +1,51 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@fastgpt/service/common/response';
+import { connectToDatabase } from '@/service/mongo';
+import { authCert } from '@fastgpt/service/support/permission/auth/common';
+import { Types } from '@fastgpt/service/common/mongo';
+import { MongoBill } from '@fastgpt/service/support/wallet/bill/schema';
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    await connectToDatabase();
+    const { appId, start, end } = req.body as { appId: string; start: number; end: number };
+    const { userId } = await authCert({ req, authToken: true });
+
+    const result = await MongoBill.aggregate([
+      {
+        $match: {
+          appId: new Types.ObjectId(appId),
+          userId: new Types.ObjectId(userId),
+          time: { $gte: new Date(start) }
+        }
+      },
+      {
+        $group: {
+          _id: {
+            year: { $year: '$time' },
+            month: { $month: '$time' },
+            day: { $dayOfMonth: '$time' }
+          },
+          total: { $sum: '$total' }
+        }
+      },
+      {
+        $project: {
+          _id: 0,
+          date: { $dateFromParts: { year: '$_id.year', month: '$_id.month', day: '$_id.day' } },
+          total: 1
+        }
+      },
+      { $sort: { date: 1 } }
+    ]);
+
+    jsonRes(res, {
+      data: result
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}
--- a/projects/app/src/pages/api/core/app/form2Modules/fastgpt-simple.ts
+++ b/projects/app/src/pages/api/core/app/form2Modules/fastgpt-simple.ts
@@ -92,7 +92,7 @@ function simpleChatTemplate({ formData, maxToken }: Props): ModuleItemType[] {
        },
        {
          key: 'model',
-          type: 'selectLLMModel',
+          type: 'selectChatModel',
          label: 'core.module.input.label.aiModel',
          required: true,
          valueType: 'string',
@@ -471,7 +471,7 @@ function datasetTemplate({ formData, maxToken }: Props): ModuleItemType[] {
        },
        {
          key: 'model',
-          type: 'selectLLMModel',
+          type: 'selectChatModel',
          label: 'core.module.input.label.aiModel',
          required: true,
          valueType: 'string',
--- a/projects/app/src/pages/api/core/app/form2Modules/fastgpt-universal.ts
+++ b/projects/app/src/pages/api/core/app/form2Modules/fastgpt-universal.ts
@@ -88,7 +88,7 @@ function simpleChatTemplate(formData: AppSimpleEditFormType): ModuleItemType[] {
        },
        {
          key: 'model',
-          type: 'selectLLMModel',
+          type: 'selectChatModel',
          label: 'core.module.input.label.aiModel',
          required: true,
          valueType: 'string',
@@ -498,7 +498,7 @@ function datasetTemplate(formData: AppSimpleEditFormType): ModuleItemType[] {
        },
        {
          key: 'model',
-          type: 'selectLLMModel',
+          type: 'selectChatModel',
          label: 'core.module.input.label.aiModel',
          required: true,
          valueType: 'string',
--- a/projects/app/src/pages/api/core/app/update.ts
+++ b/projects/app/src/pages/api/core/app/update.ts
@@ -12,7 +12,7 @@ import { getLLMModel } from '@/service/core/ai/model';
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
  try {
    await connectToDatabase();
-    const { name, avatar, type, simpleTemplateId, intro, modules, permission, teamTags } =
+    const { name, avatar, type, simpleTemplateId, intro, modules, permission } =
      req.body as AppUpdateParams;
    const { appId } = req.query as { appId: string };

@@ -65,7 +65,6 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
        avatar,
        intro,
        permission,
-        teamTags: teamTags,
        ...(modules && {
          modules
        })
--- a/projects/app/src/pages/api/core/app/updateTeamTasg.ts
+++ b/projects/app/src/pages/api/core/app/updateTeamTasg.ts
@@ -1,82 +0,0 @@
-import type { NextApiRequest, NextApiResponse } from 'next';
-import { jsonRes } from '@fastgpt/service/common/response';
-import { connectToDatabase } from '@/service/mongo';
-import { MongoApp } from '@fastgpt/service/core/app/schema';
-import type { AppUpdateParams } from '@fastgpt/global/core/app/api';
-import { authApp } from '@fastgpt/service/support/permission/auth/app';
-import { FlowNodeTypeEnum } from '@fastgpt/global/core/module/node/constant';
-import { ModuleInputKeyEnum } from '@fastgpt/global/core/module/constants';
-import { getLLMModel } from '@/service/core/ai/model';
-
-/* 获取我的模型 */
-export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
-  try {
-    await connectToDatabase();
-    const { name, avatar, type, simpleTemplateId, intro, modules, permission, teamTags } =
-      req.body as AppUpdateParams;
-    const { appId } = req.query as { appId: string };
-
-    if (!appId) {
-      throw new Error('appId is empty');
-    }
-
-    // 凭证校验
-    await authApp({ req, authToken: true, appId, per: permission ? 'owner' : 'w' });
-
-    // check modules
-    // 1. dataset search limit, less than model quoteMaxToken
-    if (modules) {
-      let maxTokens = 3000;
-
-      modules.forEach((item) => {
-        if (item.flowType === FlowNodeTypeEnum.chatNode) {
-          const model =
-            item.inputs.find((item) => item.key === ModuleInputKeyEnum.aiModel)?.value || '';
-          const chatModel = getLLMModel(model);
-          const quoteMaxToken = chatModel.quoteMaxToken || 3000;
-
-          maxTokens = Math.max(maxTokens, quoteMaxToken);
-        }
-      });
-
-      modules.forEach((item) => {
-        if (item.flowType === FlowNodeTypeEnum.datasetSearchNode) {
-          item.inputs.forEach((input) => {
-            if (input.key === ModuleInputKeyEnum.datasetMaxTokens) {
-              const val = input.value as number;
-              if (val > maxTokens) {
-                input.value = maxTokens;
-              }
-            }
-          });
-        }
-      });
-    }
-
-    // 更新模型
-    await MongoApp.findOneAndUpdate(
-      {
-        _id: appId
-      },
-      {
-        name,
-        type,
-        simpleTemplateId,
-        avatar,
-        intro,
-        permission,
-        teamTags: teamTags,
-        ...(modules && {
-          modules
-        })
-      }
-    );
-
-    jsonRes(res);
-  } catch (err) {
-    jsonRes(res, {
-      code: 500,
-      error: err
-    });
-  }
-}
--- a/projects/app/src/pages/api/core/chat/chatTest.ts
+++ b/projects/app/src/pages/api/core/chat/chatTest.ts
@@ -4,13 +4,13 @@ import { sseErrRes } from '@fastgpt/service/common/response';
 import { sseResponseEventEnum } from '@fastgpt/service/common/response/constant';
 import { responseWrite } from '@fastgpt/service/common/response';
 import type { ModuleItemType } from '@fastgpt/global/core/module/type.d';
-import { pushChatUsage } from '@/service/support/wallet/usage/push';
-import { UsageSourceEnum } from '@fastgpt/global/support/wallet/usage/constants';
+import { pushChatBill } from '@/service/support/wallet/bill/push';
+import { BillSourceEnum } from '@fastgpt/global/support/wallet/bill/constants';
 import type { ChatItemType } from '@fastgpt/global/core/chat/type';
 import { authApp } from '@fastgpt/service/support/permission/auth/app';
 import { dispatchModules } from '@/service/moduleDispatch';
 import { authCert } from '@fastgpt/service/support/permission/auth/common';
-import { getUserChatInfoAndAuthTeamPoints } from '@/service/support/permission/auth/team';
+import { getUserAndAuthBalance } from '@fastgpt/service/support/user/controller';

 export type Props = {
  history: ChatItemType[];
@@ -50,10 +50,13 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
    ]);

    // auth balance
-    const { user } = await getUserChatInfoAndAuthTeamPoints(tmbId);
+    const user = await getUserAndAuthBalance({
+      tmbId,
+      minBalance: 0
+    });

    /* start process */
-    const { responseData, moduleDispatchBills } = await dispatchModules({
+    const { responseData } = await dispatchModules({
      res,
      mode: 'test',
      teamId,
@@ -82,13 +85,13 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
    });
    res.end();

-    pushChatUsage({
+    pushChatBill({
      appName,
      appId,
      teamId,
      tmbId,
-      source: UsageSourceEnum.fastgpt,
-      moduleDispatchBills
+      source: BillSourceEnum.fastgpt,
+      response: responseData
    });
  } catch (err: any) {
    res.status(500);
--- a/projects/app/src/pages/api/core/chat/clearHistories.ts
+++ b/projects/app/src/pages/api/core/chat/clearHistories.ts
@@ -14,13 +14,10 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
    await connectToDatabase();
    const { appId, shareId, outLinkUid } = req.query as ClearHistoriesProps;

-    let chatAppId = appId;
-
    const match = await (async () => {
      if (shareId && outLinkUid) {
-        const { appId, uid } = await authOutLink({ shareId, outLinkUid });
+        const { uid } = await authOutLink({ shareId, outLinkUid });

-        chatAppId = appId;
        return {
          shareId,
          outLinkUid: uid
@@ -44,11 +41,11 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
    const idList = list.map((item) => item.chatId);

    await MongoChatItem.deleteMany({
-      appId: chatAppId,
+      appId,
      chatId: { $in: idList }
    });
    await MongoChat.deleteMany({
-      appId: chatAppId,
+      appId,
      chatId: { $in: idList }
    });

--- a/projects/app/src/pages/api/core/chat/getHistories.ts
+++ b/projects/app/src/pages/api/core/chat/getHistories.ts
@@ -28,13 +28,6 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
          }
        };
      }
-      if (appId && outLinkUid) {
-        return {
-          shareId,
-          outLinkUid: outLinkUid,
-          source: ChatSourceEnum.team
-        };
-      }
      if (appId) {
        const { tmbId } = await authCert({ req, authToken: true });
        return {
@@ -43,7 +36,6 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
          source: ChatSourceEnum.online
        };
      }
-
      return Promise.reject('Params are error');
    })();

--- a/projects/app/src/pages/api/core/chat/item/getSpeech.ts
+++ b/projects/app/src/pages/api/core/chat/item/getSpeech.ts
@@ -3,9 +3,9 @@ import { jsonRes } from '@fastgpt/service/common/response';
 import { connectToDatabase } from '@/service/mongo';
 import { GetChatSpeechProps } from '@/global/core/chat/api.d';
 import { text2Speech } from '@fastgpt/service/core/ai/audio/speech';
-import { pushAudioSpeechUsage } from '@/service/support/wallet/usage/push';
+import { pushAudioSpeechBill } from '@/service/support/wallet/bill/push';
 import { authCertOrShareId } from '@fastgpt/service/support/permission/auth/common';
-import { authType2UsageSource } from '@/service/support/wallet/usage/utils';
+import { authType2BillSource } from '@/service/support/wallet/bill/utils';
 import { getAudioSpeechModel } from '@/service/core/ai/model';
 import { MongoTTSBuffer } from '@fastgpt/service/common/buffer/tts/schema';

@@ -54,12 +54,12 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
      speed: ttsConfig.speed,
      onSuccess: async ({ model, buffer }) => {
        try {
-          pushAudioSpeechUsage({
+          pushAudioSpeechBill({
            model: model,
            charsLength: input.length,
            tmbId,
            teamId,
-            source: authType2UsageSource({ authType })
+            source: authType2BillSource({ authType })
          });

          await MongoTTSBuffer.create({
--- a/projects/app/src/pages/api/core/chat/outLink/getInforByTeamId.ts
+++ b/projects/app/src/pages/api/core/chat/outLink/getInforByTeamId.ts
@@ -1,37 +0,0 @@
-import type { NextApiRequest, NextApiResponse } from 'next';
-import { jsonRes } from '@fastgpt/service/common/response';
-import { connectToDatabase } from '@/service/mongo';
-import type { chatByTeamProps } from '@/global/core/chat/api.d';
-import axios from 'axios';
-import { MongoChatItem } from '@fastgpt/service/core/chat/chatItemSchema';
-import { getChatItems } from '@fastgpt/service/core/chat/controller';
-import { selectShareResponse } from '@/utils/service/core/chat';
-export default async function handler(req: NextApiRequest, res: NextApiResponse) {
-  try {
-    await connectToDatabase();
-
-    let { teamId, appId, outLinkUid } = req.query as chatByTeamProps;
-
-    const history = await MongoChatItem.find({
-      appId: appId,
-      outLinkUid: outLinkUid,
-      teamId: teamId
-    });
-
-    jsonRes(res, {
-      data: history
-    });
-  } catch (err) {
-    jsonRes(res, {
-      code: 500,
-      data: req.query,
-      error: err
-    });
-  }
-}
-
-export const config = {
-  api: {
-    responseLimit: '10mb'
-  }
-};
--- a/projects/app/src/pages/api/core/chat/team/init.ts
+++ b/projects/app/src/pages/api/core/chat/team/init.ts
@@ -1,91 +0,0 @@
-import type { NextApiRequest, NextApiResponse } from 'next';
-import { jsonRes } from '@fastgpt/service/common/response';
-import { connectToDatabase } from '@/service/mongo';
-import { getGuideModule } from '@fastgpt/global/core/module/utils';
-import { getChatModelNameListByModules } from '@/service/core/app/module';
-import { ModuleOutputKeyEnum } from '@fastgpt/global/core/module/constants';
-import type { InitChatProps, InitChatResponse } from '@/global/core/chat/api.d';
-import { MongoChat } from '@fastgpt/service/core/chat/chatSchema';
-import { MongoApp } from '@fastgpt/service/core/app/schema';
-import { getChatItems } from '@fastgpt/service/core/chat/controller';
-import { AppErrEnum } from '@fastgpt/global/common/error/code/app';
-
-export default async function handler(req: NextApiRequest, res: NextApiResponse) {
-  try {
-    await connectToDatabase();
-
-    let { appId, chatId, outLinkUid } = req.query as {
-      chatId?: string;
-      appId?: string;
-      outLinkUid?: string;
-    };
-
-    if (!appId) {
-      return jsonRes(res, {
-        code: 501,
-        message: "You don't have an app yet"
-      });
-    }
-
-    // auth app permission
-    const [chat, app] = await Promise.all([
-      //   authApp({
-      //     req,
-      //     authToken: false,
-      //     appId,
-      //     per: 'r'
-      //   }),
-      chatId ? MongoChat.findOne({ appId, chatId }) : undefined,
-      MongoApp.findById(appId).lean()
-    ]);
-    if (!app) {
-      throw new Error(AppErrEnum.unExist);
-    }
-
-    // auth chat permission
-    // if (chat && chat.outLinkUid !== outLinkUid) {
-    //   throw new Error(ChatErrEnum.unAuthChat);
-    // }
-    // // auth chat permission
-    // if (chat && !app.canWrite && String(tmbId) !== String(chat?.tmbId)) {
-    //   throw new Error(ChatErrEnum.unAuthChat);
-    // }
-
-    // get app and history
-    const { history } = await getChatItems({
-      appId,
-      chatId,
-      limit: 30,
-      field: `dataId obj value adminFeedback userBadFeedback userGoodFeedback ${ModuleOutputKeyEnum.responseData}`
-    });
-
-    jsonRes<InitChatResponse>(res, {
-      data: {
-        chatId,
-        appId,
-        title: chat?.title || '新对话',
-        userAvatar: undefined,
-        variables: chat?.variables || {},
-        history,
-        app: {
-          userGuideModule: getGuideModule(app.modules),
-          chatModels: getChatModelNameListByModules(app.modules),
-          name: app.name,
-          avatar: app.avatar,
-          intro: app.intro
-        }
-      }
-    });
-  } catch (err) {
-    jsonRes(res, {
-      code: 500,
-      error: err
-    });
-  }
-}
-
-export const config = {
-  api: {
-    responseLimit: '10mb'
-  }
-};
--- a/projects/app/src/pages/api/core/chat/teamInit.ts
+++ b/projects/app/src/pages/api/core/chat/teamInit.ts
@@ -1,81 +0,0 @@
-import type { NextApiRequest, NextApiResponse } from 'next';
-import { jsonRes } from '@fastgpt/service/common/response';
-import { connectToDatabase } from '@/service/mongo';
-import { authApp } from '@fastgpt/service/support/permission/auth/app';
-import { getGuideModule } from '@fastgpt/global/core/module/utils';
-import { getChatModelNameListByModules } from '@/service/core/app/module';
-import { ModuleOutputKeyEnum } from '@fastgpt/global/core/module/constants';
-import type { InitChatProps, InitChatResponse } from '@/global/core/chat/api.d';
-import { MongoChat } from '@fastgpt/service/core/chat/chatSchema';
-import { getChatItems } from '@fastgpt/service/core/chat/controller';
-import { ChatErrEnum } from '@fastgpt/global/common/error/code/chat';
-
-export default async function handler(req: NextApiRequest, res: NextApiResponse) {
-  try {
-    await connectToDatabase();
-
-    let { appId, chatId, loadCustomFeedbacks } = req.query as InitChatProps;
-
-    if (!appId) {
-      return jsonRes(res, {
-        code: 501,
-        message: "You don't have an app yet"
-      });
-    }
-
-    // auth app permission
-    const [{ app, tmbId }, chat] = await Promise.all([
-      authApp({
-        req,
-        authToken: true,
-        appId,
-        per: 'r'
-      }),
-      chatId ? MongoChat.findOne({ appId, chatId }) : undefined
-    ]);
-
-    // // auth chat permission
-    // if (chat && !app.canWrite && String(tmbId) !== String(chat?.tmbId)) {
-    //   throw new Error(ChatErrEnum.unAuthChat);
-    // }
-
-    // get app and history
-    const { history } = await getChatItems({
-      appId,
-      chatId,
-      limit: 30,
-      field: `dataId obj value adminFeedback userBadFeedback userGoodFeedback ${
-        ModuleOutputKeyEnum.responseData
-      } ${loadCustomFeedbacks ? 'customFeedbacks' : ''}`
-    });
-
-    jsonRes<InitChatResponse>(res, {
-      data: {
-        chatId,
-        appId,
-        title: chat?.title || '新对话',
-        userAvatar: undefined,
-        variables: chat?.variables || {},
-        history,
-        app: {
-          userGuideModule: getGuideModule(app.modules),
-          chatModels: getChatModelNameListByModules(app.modules),
-          name: app.name,
-          avatar: app.avatar,
-          intro: app.intro
-        }
-      }
-    });
-  } catch (err) {
-    jsonRes(res, {
-      code: 500,
-      error: err
-    });
-  }
-}
-
-export const config = {
-  api: {
-    responseLimit: '10mb'
-  }
-};
--- a/projects/app/src/pages/api/core/chat/updateHistory.ts
+++ b/projects/app/src/pages/api/core/chat/updateHistory.ts
@@ -24,7 +24,6 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
    await MongoChat.findOneAndUpdate(
      { appId, chatId },
      {
-        updateTime: new Date(),
        ...(customTitle !== undefined && { customTitle }),
        ...(top !== undefined && { top })
      }
--- a/projects/app/src/pages/api/core/dataset/collection/create/link.ts
+++ b/projects/app/src/pages/api/core/dataset/collection/create/link.ts
@@ -11,12 +11,13 @@ import {
  TrainingModeEnum,
  DatasetCollectionTypeEnum
 } from '@fastgpt/global/core/dataset/constants';
-import { checkDatasetLimit } from '@/service/support/permission/teamLimit';
+import { checkDatasetLimit } from '@fastgpt/service/support/permission/limit/dataset';
 import { predictDataLimitLength } from '@fastgpt/global/core/dataset/utils';
-import { createTrainingUsage } from '@fastgpt/service/support/wallet/usage/controller';
-import { UsageSourceEnum } from '@fastgpt/global/support/wallet/usage/constants';
+import { createTrainingBill } from '@fastgpt/service/support/wallet/bill/controller';
+import { BillSourceEnum } from '@fastgpt/global/support/wallet/bill/constants';
 import { getLLMModel, getVectorModel } from '@/service/core/ai/model';
 import { reloadCollectionChunks } from '@fastgpt/service/core/dataset/collection/utils';
+import { getStandardSubPlan } from '@/service/support/wallet/sub/utils';
 import { mongoSessionRun } from '@fastgpt/service/common/mongo/sessionRun';

 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
@@ -42,7 +43,8 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
    // 1. check dataset limit
    await checkDatasetLimit({
      teamId,
-      insertLen: predictDataLimitLength(trainingType, new Array(10))
+      insertLen: predictDataLimitLength(trainingType, new Array(10)),
+      standardPlans: getStandardSubPlan()
    });

    const { _id: collectionId } = await mongoSessionRun(async (session) => {
@@ -64,11 +66,11 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
      });

      // 3. create bill and start sync
-      const { billId } = await createTrainingUsage({
+      const { billId } = await createTrainingBill({
        teamId,
        tmbId,
        appName: 'core.dataset.collection.Sync Collection',
-        billSource: UsageSourceEnum.training,
+        billSource: BillSourceEnum.training,
        vectorModel: getVectorModel(dataset.vectorModel).name,
        agentModel: getLLMModel(dataset.agentModel).name,
        session
--- a/projects/app/src/pages/api/core/dataset/collection/create/text.ts
+++ b/projects/app/src/pages/api/core/dataset/collection/create/text.ts
@@ -12,13 +12,14 @@ import {
  DatasetCollectionTypeEnum
 } from '@fastgpt/global/core/dataset/constants';
 import { splitText2Chunks } from '@fastgpt/global/common/string/textSplitter';
-import { checkDatasetLimit } from '@/service/support/permission/teamLimit';
+import { checkDatasetLimit } from '@fastgpt/service/support/permission/limit/dataset';
 import { predictDataLimitLength } from '@fastgpt/global/core/dataset/utils';
 import { pushDataToTrainingQueue } from '@/service/core/dataset/data/controller';
 import { hashStr } from '@fastgpt/global/common/string/tools';
-import { createTrainingUsage } from '@fastgpt/service/support/wallet/usage/controller';
-import { UsageSourceEnum } from '@fastgpt/global/support/wallet/usage/constants';
+import { createTrainingBill } from '@fastgpt/service/support/wallet/bill/controller';
+import { BillSourceEnum } from '@fastgpt/global/support/wallet/bill/constants';
 import { getLLMModel, getVectorModel } from '@/service/core/ai/model';
+import { getStandardSubPlan } from '@/service/support/wallet/sub/utils';

 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
  try {
@@ -52,7 +53,8 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
    // 2. check dataset limit
    await checkDatasetLimit({
      teamId,
-      insertLen: predictDataLimitLength(trainingType, chunks)
+      insertLen: predictDataLimitLength(trainingType, chunks),
+      standardPlans: getStandardSubPlan()
    });

    // 3. create collection and training bill
@@ -72,11 +74,11 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
        hashRawText: hashStr(text),
        rawTextLength: text.length
      }),
-      createTrainingUsage({
+      createTrainingBill({
        teamId,
        tmbId,
        appName: name,
-        billSource: UsageSourceEnum.training,
+        billSource: BillSourceEnum.training,
        vectorModel: getVectorModel(dataset.vectorModel)?.name,
        agentModel: getLLMModel(dataset.agentModel)?.name
      })
--- a/projects/app/src/pages/api/core/dataset/collection/sync/link.ts
+++ b/projects/app/src/pages/api/core/dataset/collection/sync/link.ts
@@ -12,8 +12,8 @@ import {
  DatasetCollectionTypeEnum
 } from '@fastgpt/global/core/dataset/constants';
 import { DatasetErrEnum } from '@fastgpt/global/common/error/code/dataset';
-import { createTrainingUsage } from '@fastgpt/service/support/wallet/usage/controller';
-import { UsageSourceEnum } from '@fastgpt/global/support/wallet/usage/constants';
+import { createTrainingBill } from '@fastgpt/service/support/wallet/bill/controller';
+import { BillSourceEnum } from '@fastgpt/global/support/wallet/bill/constants';
 import { getLLMModel, getVectorModel } from '@/service/core/ai/model';
 import { createOneCollection } from '@fastgpt/service/core/dataset/collection/controller';
 import { mongoSessionRun } from '@fastgpt/service/common/mongo/sessionRun';
@@ -56,11 +56,11 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<

    await mongoSessionRun(async (session) => {
      // create training bill
-      const { billId } = await createTrainingUsage({
+      const { billId } = await createTrainingBill({
        teamId: collection.teamId,
        tmbId,
        appName: 'core.dataset.collection.Sync Collection',
-        billSource: UsageSourceEnum.training,
+        billSource: BillSourceEnum.training,
        vectorModel: vectorModelData.name,
        agentModel: agentModelData.name,
        session
--- a/projects/app/src/pages/api/core/dataset/create.ts
+++ b/projects/app/src/pages/api/core/dataset/create.ts
@@ -7,7 +7,6 @@ import { createDefaultCollection } from '@fastgpt/service/core/dataset/collectio
 import { authUserNotVisitor } from '@fastgpt/service/support/permission/auth/user';
 import { DatasetTypeEnum } from '@fastgpt/global/core/dataset/constants';
 import { getLLMModel, getVectorModel, getDatasetModel } from '@/service/core/ai/model';
-import { checkTeamDatasetLimit } from '@/service/support/permission/teamLimit';

 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
  try {
@@ -32,7 +31,13 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
    }

    // check limit
-    await checkTeamDatasetLimit(teamId);
+    const authCount = await MongoDataset.countDocuments({
+      teamId,
+      type: DatasetTypeEnum.dataset
+    });
+    if (authCount >= 50) {
+      throw new Error('每个团队上限 50 个知识库');
+    }

    const { _id } = await MongoDataset.create({
      name,
--- a/projects/app/src/pages/api/core/dataset/data/delete.ts
+++ b/projects/app/src/pages/api/core/dataset/data/delete.ts
@@ -3,7 +3,8 @@ import { jsonRes } from '@fastgpt/service/common/response';
 import { withNextCors } from '@fastgpt/service/common/middle/cors';
 import { connectToDatabase } from '@/service/mongo';
 import { authDatasetData } from '@/service/support/permission/auth/dataset';
-import { deleteDatasetData } from '@/service/core/dataset/data/controller';
+import { MongoDatasetData } from '@fastgpt/service/core/dataset/data/schema';
+import { deleteDatasetDataVector } from '@fastgpt/service/common/vectorStore/controller';

 export default withNextCors(async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
  try {
@@ -25,7 +26,19 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
      per: 'w'
    });

-    await deleteDatasetData(datasetData);
+    // update mongo data update time
+    await MongoDatasetData.findByIdAndUpdate(dataId, {
+      updateTime: new Date()
+    });
+
+    // delete vector data
+    await deleteDatasetDataVector({
+      teamId,
+      idList: datasetData.indexes.map((item) => item.dataId)
+    });
+
+    // delete mongo data
+    await MongoDatasetData.findByIdAndDelete(dataId);

    jsonRes(res, {
      data: 'success'
--- a/projects/app/src/pages/api/core/dataset/data/insertData.ts
+++ b/projects/app/src/pages/api/core/dataset/data/insertData.ts
@@ -12,10 +12,12 @@ import { hasSameValue } from '@/service/core/dataset/data/utils';
 import { insertData2Dataset } from '@/service/core/dataset/data/controller';
 import { authDatasetCollection } from '@fastgpt/service/support/permission/auth/dataset';
 import { getCollectionWithDataset } from '@fastgpt/service/core/dataset/controller';
-import { pushGenerateVectorUsage } from '@/service/support/wallet/usage/push';
+import { authTeamBalance } from '@/service/support/permission/auth/bill';
+import { pushGenerateVectorBill } from '@/service/support/wallet/bill/push';
 import { InsertOneDatasetDataProps } from '@/global/core/dataset/api';
 import { simpleText } from '@fastgpt/global/common/string/tools';
-import { checkDatasetLimit } from '@/service/support/permission/teamLimit';
+import { checkDatasetLimit } from '@fastgpt/service/support/permission/limit/dataset';
+import { getStandardSubPlan } from '@/service/support/wallet/sub/utils';

 export default withNextCors(async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
  try {
@@ -41,7 +43,8 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex

    await checkDatasetLimit({
      teamId,
-      insertLen: 1
+      insertLen: 1,
+      standardPlans: getStandardSubPlan()
    });

    // auth collection and get dataset
@@ -49,7 +52,7 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
      {
        datasetId: { _id: datasetId, vectorModel }
      }
-    ] = await Promise.all([getCollectionWithDataset(collectionId)]);
+    ] = await Promise.all([getCollectionWithDataset(collectionId), authTeamBalance(teamId)]);

    // format data
    const formatQ = simpleText(q);
@@ -87,7 +90,7 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
      indexes: formatIndexes
    });

-    pushGenerateVectorUsage({
+    pushGenerateVectorBill({
      teamId,
      tmbId,
      charsLength,
--- a/projects/app/src/pages/api/core/dataset/data/pushData.ts
+++ b/projects/app/src/pages/api/core/dataset/data/pushData.ts
@@ -8,9 +8,10 @@ import type {
  PushDatasetDataResponse
 } from '@fastgpt/global/core/dataset/api.d';
 import { authDatasetCollection } from '@fastgpt/service/support/permission/auth/dataset';
-import { checkDatasetLimit } from '@/service/support/permission/teamLimit';
+import { checkDatasetLimit } from '@fastgpt/service/support/permission/limit/dataset';
 import { predictDataLimitLength } from '@fastgpt/global/core/dataset/utils';
 import { pushDataToTrainingQueue } from '@/service/core/dataset/data/controller';
+import { getStandardSubPlan } from '@/service/support/wallet/sub/utils';

 export default withNextCors(async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
  try {
@@ -37,7 +38,8 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
    // auth dataset limit
    await checkDatasetLimit({
      teamId,
-      insertLen: predictDataLimitLength(collection.trainingType, data)
+      insertLen: predictDataLimitLength(collection.trainingType, data),
+      standardPlans: getStandardSubPlan()
    });

    jsonRes<PushDatasetDataResponse>(res, {
--- a/projects/app/src/pages/api/core/dataset/data/update.ts
+++ b/projects/app/src/pages/api/core/dataset/data/update.ts
@@ -4,9 +4,9 @@ import { withNextCors } from '@fastgpt/service/common/middle/cors';
 import { connectToDatabase } from '@/service/mongo';
 import { updateData2Dataset } from '@/service/core/dataset/data/controller';
 import { authDatasetData } from '@/service/support/permission/auth/dataset';
-import { pushGenerateVectorUsage } from '@/service/support/wallet/usage/push';
+import { authTeamBalance } from '@/service/support/permission/auth/bill';
+import { pushGenerateVectorBill } from '@/service/support/wallet/bill/push';
 import { UpdateDatasetDataProps } from '@/global/core/dataset/api';
-import { checkDatasetLimit } from '@/service/support/permission/teamLimit';

 export default withNextCors(async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
  try {
@@ -29,10 +29,7 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
    });

    // auth team balance
-    await checkDatasetLimit({
-      teamId,
-      insertLen: 1
-    });
+    await authTeamBalance(teamId);

    const { charsLength } = await updateData2Dataset({
      dataId: id,
@@ -42,7 +39,7 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
      model: vectorModel
    });

-    pushGenerateVectorUsage({
+    pushGenerateVectorBill({
      teamId,
      tmbId,
      charsLength,
--- a/projects/app/src/pages/api/core/dataset/searchTest.ts
+++ b/projects/app/src/pages/api/core/dataset/searchTest.ts
@@ -4,16 +4,14 @@ import { withNextCors } from '@fastgpt/service/common/middle/cors';
 import type { SearchTestProps, SearchTestResponse } from '@/global/core/dataset/api.d';
 import { connectToDatabase } from '@/service/mongo';
 import { authDataset } from '@fastgpt/service/support/permission/auth/dataset';
-import { pushGenerateVectorUsage } from '@/service/support/wallet/usage/push';
+import { authTeamBalance } from '@/service/support/permission/auth/bill';
+import { pushGenerateVectorBill } from '@/service/support/wallet/bill/push';
 import { searchDatasetData } from '@/service/core/dataset/data/controller';
 import { updateApiKeyUsage } from '@fastgpt/service/support/openapi/tools';
-import { UsageSourceEnum } from '@fastgpt/global/support/wallet/usage/constants';
+import { BillSourceEnum } from '@fastgpt/global/support/wallet/bill/constants';
 import { getLLMModel } from '@/service/core/ai/model';
+import { queryExtension } from '@fastgpt/service/core/ai/functions/queryExtension';
 import { datasetSearchQueryExtension } from '@fastgpt/service/core/dataset/search/utils';
-import {
-  checkTeamAIPoints,
-  checkTeamReRankPermission
-} from '@/service/support/permission/teamLimit';

 export default withNextCors(async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
  try {
@@ -45,7 +43,7 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
      per: 'r'
    });
    // auth balance
-    await checkTeamAIPoints(teamId);
+    await authTeamBalance(teamId);

    // query extension
    const extensionModel =
@@ -67,27 +65,28 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
      similarity,
      datasetIds: [datasetId],
      searchMode,
-      usingReRank: usingReRank && (await checkTeamReRankPermission(teamId))
+      usingReRank
    });

    // push bill
-    const { totalPoints } = pushGenerateVectorUsage({
+    const { total } = pushGenerateVectorBill({
      teamId,
      tmbId,
      charsLength,
      model: dataset.vectorModel,
-      source: apikey ? UsageSourceEnum.api : UsageSourceEnum.fastgpt,
+      source: apikey ? BillSourceEnum.api : BillSourceEnum.fastgpt,

      ...(aiExtensionResult &&
        extensionModel && {
          extensionModel: extensionModel.name,
-          extensionCharsLength: aiExtensionResult.charsLength
+          extensionInputTokens: aiExtensionResult.inputTokens,
+          extensionOutputTokens: aiExtensionResult.outputTokens
        })
    });
    if (apikey) {
      updateApiKeyUsage({
        apikey,
-        totalPoints: totalPoints
+        usage: total
      });
    }

--- a/projects/app/src/pages/api/core/plugin/create.ts
+++ b/projects/app/src/pages/api/core/plugin/create.ts
@@ -4,7 +4,6 @@ import { connectToDatabase } from '@/service/mongo';
 import type { CreateOnePluginParams } from '@fastgpt/global/core/plugin/controller';
 import { authUserNotVisitor } from '@fastgpt/service/support/permission/auth/user';
 import { MongoPlugin } from '@fastgpt/service/core/plugin/schema';
-import { checkTeamPluginLimit } from '@/service/support/permission/teamLimit';

 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
  try {
@@ -12,8 +11,6 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
    const { teamId, tmbId } = await authUserNotVisitor({ req, authToken: true });
    const body = req.body as CreateOnePluginParams;

-    await checkTeamPluginLimit(teamId);
-
    const { _id } = await MongoPlugin.create({
      ...body,
      teamId,