feat: add .svg support (#3229)

This commit is contained in:
Pugazhendhi 2025-07-07 11:21:32 +05:30 committed by GitHub
parent 97d9386e3f
commit 524ede52d2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 44 additions and 1 deletions

View File

@ -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,

View File

@ -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');