dc1119ca90
* feat: toolNode aiNode readFileNode adapt new version * update docker-compose * update tip * feat: adapt new file version * perf: file input * fix: ts
55 lines
1.6 KiB
TypeScript
55 lines
1.6 KiB
TypeScript
import { detect } from 'jschardet';
|
||
import { documentFileType, imageFileType } from './constants';
|
||
import { ChatFileTypeEnum } from '../../core/chat/constants';
|
||
import { UserChatItemValueItemType } from '../../core/chat/type';
|
||
|
||
export const formatFileSize = (bytes: number): string => {
|
||
if (bytes === 0) return '0 B';
|
||
|
||
const k = 1024;
|
||
const sizes = ['B', 'KB', 'MB', 'GB', 'TB', 'PB', 'EB', 'ZB', 'YB'];
|
||
const i = Math.floor(Math.log(bytes) / Math.log(k));
|
||
|
||
return parseFloat((bytes / Math.pow(k, i)).toFixed(2)) + ' ' + sizes[i];
|
||
};
|
||
|
||
export const detectFileEncoding = (buffer: Buffer) => {
|
||
return detect(buffer.slice(0, 200))?.encoding?.toLocaleLowerCase();
|
||
};
|
||
|
||
// Url => user upload file type
|
||
export const parseUrlToFileType = (url: string): UserChatItemValueItemType['file'] | undefined => {
|
||
const parseUrl = new URL(url, 'https://locaohost:3000');
|
||
|
||
const filename = (() => {
|
||
// Old version file url: https://xxx.com/file/read?filename=xxx.pdf
|
||
const filenameQuery = parseUrl.searchParams.get('filename');
|
||
if (filenameQuery) return filenameQuery;
|
||
|
||
// Common file： https://xxx.com/xxx.pdf?xxxx=xxx
|
||
const pathname = parseUrl.pathname;
|
||
if (pathname) return pathname.split('/').pop();
|
||
})();
|
||
|
||
if (!filename) return;
|
||
|
||
const extension = filename.split('.').pop()?.toLowerCase() || '';
|
||
|
||
if (!extension) return;
|
||
|
||
if (documentFileType.includes(extension)) {
|
||
return {
|
||
type: ChatFileTypeEnum.file,
|
||
name: filename,
|
||
url
|
||
};
|
||
}
|
||
if (imageFileType.includes(extension)) {
|
||
return {
|
||
type: ChatFileTypeEnum.image,
|
||
name: filename,
|
||
url
|
||
};
|
||
}
|
||
};
|