FastGPT/packages/global/core/dataset/read.ts

import { DatasetSourceReadTypeEnum, ImportDataSourceEnum } from './constants';

export const rawTextBackupPrefix = 'index,content';

export const importType2ReadType = (type: ImportDataSourceEnum) => {
  if (type === ImportDataSourceEnum.csvTable || type === ImportDataSourceEnum.fileLocal) {
    return DatasetSourceReadTypeEnum.fileLocal;
  }
  if (type === ImportDataSourceEnum.fileLink) {
    return DatasetSourceReadTypeEnum.link;
  }
  if (type === ImportDataSourceEnum.externalFile) {
    return DatasetSourceReadTypeEnum.externalFile;
  }
  return DatasetSourceReadTypeEnum.link;
};