mirror of
https://github.com/labring/FastGPT.git
synced 2025-07-27 00:17:31 +00:00

Co-authored-by: Archer <545436317@qq.com> Co-authored-by: heheer <71265218+newfish-cmyk@users.noreply.github.com>
21 lines
614 B
TypeScript
21 lines
614 B
TypeScript
import { getVectorCountByTeamId } from '../../../common/vectorStore/controller';
|
|
import { getTeamDatasetValidSub } from '../../wallet/sub/utils';
|
|
|
|
export const checkDatasetLimit = async ({
|
|
teamId,
|
|
freeSize = Infinity,
|
|
insertLen = 0
|
|
}: {
|
|
teamId: string;
|
|
freeSize?: number;
|
|
insertLen?: number;
|
|
}) => {
|
|
const { maxSize } = await getTeamDatasetValidSub({ teamId, freeSize });
|
|
const usedSize = await getVectorCountByTeamId(teamId);
|
|
|
|
if (usedSize + insertLen >= maxSize) {
|
|
return Promise.reject(`数据库容量不足,无法继续添加。可以在账号页面进行扩容。`);
|
|
}
|
|
return;
|
|
};
|