import { getVectorCountByTeamId } from '../../common/vectorStore/controller'; import { getTeamPlanStatus, getTeamStandPlan } from '../../support/wallet/sub/utils'; import { MongoApp } from '../../core/app/schema'; import { MongoPlugin } from '../../core/plugin/schema'; import { MongoDataset } from '../../core/dataset/schema'; import { DatasetTypeEnum } from '@fastgpt/global/core/dataset/constants'; import { TeamErrEnum } from '@fastgpt/global/common/error/code/team'; export const checkDatasetLimit = async ({ teamId, insertLen = 0 }: { teamId: string; insertLen?: number; }) => { const [{ standardConstants, totalPoints, usedPoints, datasetMaxSize }, usedSize] = await Promise.all([getTeamPlanStatus({ teamId }), getVectorCountByTeamId(teamId)]); if (!standardConstants) return; if (usedSize + insertLen >= datasetMaxSize) { return Promise.reject(TeamErrEnum.datasetSizeNotEnough); } if (usedPoints >= totalPoints) { return Promise.reject(TeamErrEnum.aiPointsNotEnough); } return; }; export const checkTeamAIPoints = async (teamId: string) => { const { standardConstants, totalPoints, usedPoints } = await getTeamPlanStatus({ teamId }); if (!standardConstants) return; if (usedPoints >= totalPoints) { return Promise.reject(TeamErrEnum.aiPointsNotEnough); } return { totalPoints, usedPoints }; }; export const checkTeamDatasetLimit = async (teamId: string) => { const [{ standardConstants }, datasetCount] = await Promise.all([ getTeamStandPlan({ teamId }), MongoDataset.countDocuments({ teamId, type: { $ne: DatasetTypeEnum.folder } }) ]); if (standardConstants && datasetCount >= standardConstants.maxDatasetAmount) { return Promise.reject(TeamErrEnum.datasetAmountNotEnough); } }; export const checkTeamAppLimit = async (teamId: string) => { const [{ standardConstants }, appCount] = await Promise.all([ getTeamStandPlan({ teamId }), MongoApp.count({ teamId }) ]); if (standardConstants && appCount >= standardConstants.maxAppAmount) { return Promise.reject(TeamErrEnum.appAmountNotEnough); } }; export const checkTeamPluginLimit = async (teamId: string) => { const [{ standardConstants }, pluginCount] = await Promise.all([ getTeamStandPlan({ teamId }), MongoPlugin.count({ teamId }) ]); if (standardConstants && pluginCount >= standardConstants.maxAppAmount) { return Promise.reject(TeamErrEnum.pluginAmountNotEnough); } }; export const checkTeamReRankPermission = async (teamId: string) => { const { standardConstants } = await getTeamStandPlan({ teamId }); if (standardConstants && !standardConstants?.permissionReRank) { return false; } return true; };