mirror of
https://github.com/labring/FastGPT.git
synced 2025-10-18 09:24:03 +00:00
23 lines
628 B
TypeScript
23 lines
628 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 }, usedSize] = await Promise.all([
|
|
getTeamDatasetValidSub({ teamId, freeSize }),
|
|
getVectorCountByTeamId(teamId)
|
|
]);
|
|
|
|
if (usedSize + insertLen >= maxSize) {
|
|
return Promise.reject(`数据库容量不足，无法继续添加。可以在账号页面进行扩容。`);
|
|
}
|
|
return;
|
|
};
|