Support File URLs

google-gemini · May 9, 2024 · 74b53e8 · 74b53e8
1 parent 6227ee7
commit 74b53e8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/packages/main/src/files/file-manager.ts b/packages/main/src/files/file-manager.ts
@@ -51,7 +51,11 @@ export class GoogleAIFileManager {
     filePath: string,
     fileMetadata: FileMetadata,
   ): Promise<UploadFileResponse> {
-    const file = readFileSync(filePath);
+    var filePathOrUrl: URL | string = filePath;
+    if (filePath.startsWith("file://")) {
+      filePathOrUrl = new URL(filePath);
+    }
+    const file = readFileSync(filePathOrUrl);
     const url = new FilesRequestUrl(
       FilesTask.UPLOAD,
       this.apiKey,