import type { PushDatasetDataChunkProps } from '../api';
import type { TrainingModeEnum } from '../constants';

export type PushDataToTrainingQueueProps = {
  teamId: string;
  tmbId: string;
  datasetId: string;
  collectionId: string;

  data: PushDatasetDataChunkProps[];
  mode?: TrainingModeEnum;

  agentModel: string;
  vectorModel: string;
  vlmModel?: string;

  indexSize?: number;

  billId?: string;
  session?: ClientSession;
};
