export const PgDatasetTableName = 'modeldata';

/* ------------ dataset -------------- */
export enum DatasetTypeEnum {
  folder = 'folder',
  dataset = 'dataset'
}

export const DatasetTypeMap = {
  [DatasetTypeEnum.folder]: {
    name: 'folder'
  },
  [DatasetTypeEnum.dataset]: {
    name: 'dataset'
  }
};

/* ------------ collection -------------- */
export enum DatasetCollectionTypeEnum {
  folder = 'folder',
  file = 'file',
  link = 'link',
  virtual = 'virtual'
}
export const DatasetCollectionTypeMap = {
  [DatasetCollectionTypeEnum.folder]: {
    name: 'core.dataset.folder'
  },
  [DatasetCollectionTypeEnum.file]: {
    name: 'core.dataset.file'
  },
  [DatasetCollectionTypeEnum.link]: {
    name: 'core.dataset.link'
  },
  [DatasetCollectionTypeEnum.virtual]: {
    name: 'core.dataset.Virtual File'
  }
};
export enum DatasetCollectionTrainingModeEnum {
  manual = 'manual',
  chunk = 'chunk',
  qa = 'qa'
}
export const DatasetCollectionTrainingTypeMap = {
  [DatasetCollectionTrainingModeEnum.manual]: {
    label: 'core.dataset.collection.training.type manual'
  },
  [DatasetCollectionTrainingModeEnum.chunk]: {
    label: 'core.dataset.collection.training.type chunk'
  },
  [DatasetCollectionTrainingModeEnum.qa]: {
    label: 'core.dataset.collection.training.type qa'
  }
};

/* ------------ data -------------- */
export enum DatasetDataIndexTypeEnum {
  chunk = 'chunk',
  qa = 'qa',
  summary = 'summary',
  hypothetical = 'hypothetical',
  custom = 'custom'
}
export const DatasetDataIndexTypeMap = {
  [DatasetDataIndexTypeEnum.chunk]: {
    name: 'dataset.data.indexes.chunk'
  },
  [DatasetDataIndexTypeEnum.summary]: {
    name: 'dataset.data.indexes.summary'
  },
  [DatasetDataIndexTypeEnum.hypothetical]: {
    name: 'dataset.data.indexes.hypothetical'
  },
  [DatasetDataIndexTypeEnum.qa]: {
    name: 'dataset.data.indexes.qa'
  },
  [DatasetDataIndexTypeEnum.custom]: {
    name: 'dataset.data.indexes.custom'
  }
};

/* ------------ training -------------- */
export enum TrainingModeEnum {
  chunk = 'chunk',
  qa = 'qa'
}

export const TrainingTypeMap = {
  [TrainingModeEnum.chunk]: {
    label: 'core.dataset.training.type chunk'
  },
  [TrainingModeEnum.qa]: {
    label: 'core.dataset.training.type qa'
  }
};

/* ------------ search -------------- */
export enum DatasetSearchModeEnum {
  embedding = 'embedding',
  embeddingReRank = 'embeddingReRank',
  embFullTextReRank = 'embFullTextReRank'
}

export const DatasetSearchModeMap = {
  [DatasetSearchModeEnum.embedding]: {
    icon: 'core/dataset/modeEmbedding',
    title: 'core.dataset.search.mode.embedding',
    desc: 'core.dataset.search.mode.embedding desc',
    value: DatasetSearchModeEnum.embedding
  },
  [DatasetSearchModeEnum.embeddingReRank]: {
    icon: 'core/dataset/modeEmbeddingRerank',
    title: 'core.dataset.search.mode.embeddingReRank',
    desc: 'core.dataset.search.mode.embeddingReRank desc',
    value: DatasetSearchModeEnum.embeddingReRank
  },
  [DatasetSearchModeEnum.embFullTextReRank]: {
    icon: 'core/dataset/modeEmbFTRerank',
    title: 'core.dataset.search.mode.embFullTextReRank',
    desc: 'core.dataset.search.mode.embFullTextReRank desc',
    value: DatasetSearchModeEnum.embFullTextReRank
  }
};

export const FolderAvatarSrc = '/imgs/files/folder.svg';