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;
};