feat: add .svg support (#3229)
This commit is contained in:
parent
97d9386e3f
commit
524ede52d2
|
@ -211,6 +211,11 @@ describe('fileUtils', () => {
|
|||
expect(detectFileType('file.jpg')).toBe('image');
|
||||
});
|
||||
|
||||
it('should detect svg type by extension', () => {
|
||||
expect(detectFileType('image.svg')).toBe('svg');
|
||||
expect(detectFileType('image.icon.svg')).toBe('svg');
|
||||
});
|
||||
|
||||
it('should detect pdf type by extension', () => {
|
||||
mockMimeLookup.mockReturnValueOnce('application/pdf');
|
||||
expect(detectFileType('file.pdf')).toBe('pdf');
|
||||
|
@ -355,6 +360,26 @@ describe('fileUtils', () => {
|
|||
expect(result.returnDisplay).toContain('Read pdf file: document.pdf');
|
||||
});
|
||||
|
||||
it('should read an SVG file as text when under 1MB', async () => {
|
||||
const svgContent = `
|
||||
<svg xmlns="http://www.w3.org/2000/svg" width="100" height="100">
|
||||
<rect width="100" height="100" fill="blue" />
|
||||
</svg>
|
||||
`;
|
||||
const testSvgFilePath = path.join(tempRootDir, 'test.svg');
|
||||
actualNodeFs.writeFileSync(testSvgFilePath, svgContent, 'utf-8');
|
||||
|
||||
mockMimeLookup.mockReturnValue('image/svg+xml');
|
||||
|
||||
const result = await processSingleFileContent(
|
||||
testSvgFilePath,
|
||||
tempRootDir,
|
||||
);
|
||||
|
||||
expect(result.llmContent).toBe(svgContent);
|
||||
expect(result.returnDisplay).toContain('Read SVG as text');
|
||||
});
|
||||
|
||||
it('should skip binary files', async () => {
|
||||
actualNodeFs.writeFileSync(
|
||||
testBinaryFilePath,
|
||||
|
|
|
@ -98,7 +98,7 @@ export function isBinaryFile(filePath: string): boolean {
|
|||
*/
|
||||
export function detectFileType(
|
||||
filePath: string,
|
||||
): 'text' | 'image' | 'pdf' | 'audio' | 'video' | 'binary' {
|
||||
): 'text' | 'image' | 'pdf' | 'audio' | 'video' | 'binary' | 'svg' {
|
||||
const ext = path.extname(filePath).toLowerCase();
|
||||
|
||||
// The mimetype for "ts" is MPEG transport stream (a video format) but we want
|
||||
|
@ -107,6 +107,10 @@ export function detectFileType(
|
|||
return 'text';
|
||||
}
|
||||
|
||||
if (ext === '.svg') {
|
||||
return 'svg';
|
||||
}
|
||||
|
||||
const lookedUpMimeType = mime.lookup(filePath); // Returns false if not found, or the mime type string
|
||||
if (lookedUpMimeType) {
|
||||
if (lookedUpMimeType.startsWith('image/')) {
|
||||
|
@ -235,6 +239,20 @@ export async function processSingleFileContent(
|
|||
returnDisplay: `Skipped binary file: ${relativePathForDisplay}`,
|
||||
};
|
||||
}
|
||||
case 'svg': {
|
||||
const SVG_MAX_SIZE_BYTES = 1 * 1024 * 1024;
|
||||
if (stats.size > SVG_MAX_SIZE_BYTES) {
|
||||
return {
|
||||
llmContent: `Cannot display content of SVG file larger than 1MB: ${relativePathForDisplay}`,
|
||||
returnDisplay: `Skipped large SVG file (>1MB): ${relativePathForDisplay}`,
|
||||
};
|
||||
}
|
||||
const content = await fs.promises.readFile(filePath, 'utf8');
|
||||
return {
|
||||
llmContent: content,
|
||||
returnDisplay: `Read SVG as text: ${relativePathForDisplay}`,
|
||||
};
|
||||
}
|
||||
case 'text': {
|
||||
const content = await fs.promises.readFile(filePath, 'utf8');
|
||||
const lines = content.split('\n');
|
||||
|
|
Loading…
Reference in New Issue