From 0275ab0108e9a411d90d8ef8c8d70e21e498d81a Mon Sep 17 00:00:00 2001 From: Santhosh Kumar Date: Wed, 2 Jul 2025 00:52:32 +0530 Subject: feat: add audio and video support to read_file (#2556) --- packages/core/src/tools/read-many-files.test.ts | 27 +++++++++++++++++++++++++ 1 file changed, 27 insertions(+) (limited to 'packages/core/src/tools') diff --git a/packages/core/src/tools/read-many-files.test.ts b/packages/core/src/tools/read-many-files.test.ts index 666c484c..697b7d1b 100644 --- a/packages/core/src/tools/read-many-files.test.ts +++ b/packages/core/src/tools/read-many-files.test.ts @@ -14,6 +14,33 @@ import fs from 'fs'; // Actual fs for setup import os from 'os'; import { Config } from '../config/config.js'; +vi.mock('mime-types', () => { + const lookup = (filename: string) => { + if (filename.endsWith('.ts') || filename.endsWith('.js')) { + return 'text/plain'; + } + if (filename.endsWith('.png')) { + return 'image/png'; + } + if (filename.endsWith('.pdf')) { + return 'application/pdf'; + } + if (filename.endsWith('.mp3') || filename.endsWith('.wav')) { + return 'audio/mpeg'; + } + if (filename.endsWith('.mp4') || filename.endsWith('.mov')) { + return 'video/mp4'; + } + return false; + }; + return { + default: { + lookup, + }, + lookup, + }; +}); + describe('ReadManyFilesTool', () => { let tool: ReadManyFilesTool; let tempRootDir: string; -- cgit v1.2.3