mirror of
https://github.com/labring/FastGPT.git
synced 2025-07-22 20:37:48 +00:00

* perf: local file create collection * rename middleware * perf: remove code * feat: next14 * feat: external file dataset * collection tags field * external file dataset doc * fix: ts
56 lines
1.6 KiB
TypeScript
56 lines
1.6 KiB
TypeScript
import { parentPort } from 'worker_threads';
|
|
import { readFileRawText } from './extension/rawText';
|
|
import { ReadRawTextByBuffer, ReadRawTextProps } from './type';
|
|
import { readHtmlRawText } from './extension/html';
|
|
import { readPdfFile } from './extension/pdf';
|
|
import { readDocsFile } from './extension/docx';
|
|
import { readPptxRawText } from './extension/pptx';
|
|
import { readXlsxRawText } from './extension/xlsx';
|
|
import { readCsvRawText } from './extension/csv';
|
|
|
|
parentPort?.on('message', async (props: ReadRawTextProps<Uint8Array>) => {
|
|
const readRawContentByFileBuffer = async (params: ReadRawTextByBuffer) => {
|
|
switch (params.extension) {
|
|
case 'txt':
|
|
case 'md':
|
|
return readFileRawText(params);
|
|
case 'html':
|
|
return readHtmlRawText(params);
|
|
case 'pdf':
|
|
return readPdfFile(params);
|
|
case 'docx':
|
|
return readDocsFile(params);
|
|
case 'pptx':
|
|
return readPptxRawText(params);
|
|
case 'xlsx':
|
|
return readXlsxRawText(params);
|
|
case 'csv':
|
|
return readCsvRawText(params);
|
|
default:
|
|
return Promise.reject('Only support .txt, .md, .html, .pdf, .docx, pptx, .csv, .xlsx');
|
|
}
|
|
};
|
|
|
|
// params.buffer: Uint8Array -> buffer
|
|
const buffer = Buffer.from(props.buffer);
|
|
const newProps: ReadRawTextByBuffer = {
|
|
...props,
|
|
buffer
|
|
};
|
|
|
|
try {
|
|
parentPort?.postMessage({
|
|
type: 'success',
|
|
data: await readRawContentByFileBuffer(newProps)
|
|
});
|
|
} catch (error) {
|
|
console.log(error);
|
|
parentPort?.postMessage({
|
|
type: 'error',
|
|
data: error
|
|
});
|
|
}
|
|
|
|
process.exit();
|
|
});
|