117 lines
2.8 KiB
TypeScript
117 lines
2.8 KiB
TypeScript
export const PgDatasetTableName = 'modeldata';
|
|
|
|
export enum DatasetTypeEnum {
|
|
folder = 'folder',
|
|
dataset = 'dataset'
|
|
}
|
|
|
|
export const DatasetTypeMap = {
|
|
[DatasetTypeEnum.folder]: {
|
|
name: 'folder'
|
|
},
|
|
[DatasetTypeEnum.dataset]: {
|
|
name: 'dataset'
|
|
}
|
|
};
|
|
|
|
export enum DatasetCollectionTypeEnum {
|
|
file = 'file',
|
|
folder = 'folder',
|
|
link = 'link',
|
|
virtual = 'virtual'
|
|
}
|
|
|
|
export const DatasetCollectionTypeMap = {
|
|
[DatasetCollectionTypeEnum.file]: {
|
|
name: 'dataset.file'
|
|
},
|
|
[DatasetCollectionTypeEnum.folder]: {
|
|
name: 'dataset.folder'
|
|
},
|
|
[DatasetCollectionTypeEnum.link]: {
|
|
name: 'dataset.link'
|
|
},
|
|
[DatasetCollectionTypeEnum.virtual]: {
|
|
name: 'dataset.Virtual File'
|
|
}
|
|
};
|
|
|
|
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'
|
|
}
|
|
};
|
|
|
|
export enum TrainingModeEnum {
|
|
'chunk' = 'chunk',
|
|
'qa' = 'qa'
|
|
// 'hypothetical' = 'hypothetical',
|
|
// 'summary' = 'summary',
|
|
// 'multipleIndex' = 'multipleIndex'
|
|
}
|
|
export const TrainingTypeMap = {
|
|
[TrainingModeEnum.chunk]: {
|
|
name: 'chunk'
|
|
},
|
|
[TrainingModeEnum.qa]: {
|
|
name: 'qa'
|
|
}
|
|
// [TrainingModeEnum.hypothetical]: {
|
|
// name: 'hypothetical'
|
|
// },
|
|
// [TrainingModeEnum.summary]: {
|
|
// name: 'summary'
|
|
// },
|
|
// [TrainingModeEnum.multipleIndex]: {
|
|
// name: 'multipleIndex'
|
|
// }
|
|
};
|
|
|
|
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';
|