Function handler

projects/app/src/pages/api/core/dataset/createWithFiles.ts:45–187

(req: ApiRequestProps)

import { parseApiInput } from '@fastgpt/service/common/zod/requestParseError';

async function handler(req: ApiRequestProps): Promise<CreateDatasetWithFilesResponse> {
  const { datasetParams, files } = parseApiInput({
    req,
    bodySchema: CreateDatasetWithFilesBodySchema
  }).body;
  const {
    parentId,
    name,
    avatar,
    vectorModel = getDefaultEmbeddingModel()?.model,
    agentModel = getDefaultLLMModel()?.model,
    vlmModel = getDefaultVLMModel()?.model
  } = datasetParams;

  const { teamId, tmbId, userId } = parentId
    ? await authDataset({
        req,
        datasetId: parentId,
        authToken: true,
        authApiKey: true,
        per: WritePermissionVal
      })
    : await authUserPer({
        req,
        authToken: true,
        authApiKey: true,
        per: TeamDatasetCreatePermissionVal
      });

  // check limit
  await checkTeamDatasetLimit(teamId);

  try {
    const result = await mongoSessionRun(async (session) => {
      // 1. Create dataset
      const [dataset] = await MongoDataset.create(
        [
          {
            ...parseParentIdInMongo(parentId),
            name,
            teamId,
            tmbId,
            vectorModel,
            agentModel,
            vlmModel,
            avatar,
            intro: '',
            type: DatasetTypeEnum.dataset
          }
        ],
        { session, ordered: true }
      );

      // 2. Create permission
      await MongoResourcePermission.insertOne({
        teamId,
        tmbId,
        resourceId: dataset._id,