mirror of
https://github.com/labring/FastGPT.git
synced 2025-07-23 05:12:39 +00:00
126 lines
3.2 KiB
TypeScript
126 lines
3.2 KiB
TypeScript
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';
|