feat(api): add google ai file manager

RajaRakoto · Nov 24, 2024 · 8a5b59d · 8a5b59d
1 parent cfe6da5
commit 8a5b59d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/api/index.ts b/src/api/index.ts
@@ -1,12 +1,14 @@
 /* libs */
+import { GoogleAIFileManager } from "@google/generative-ai/server";
 import { GoogleGenerativeAI } from "@google/generative-ai";
 
 /* orm */
 import { rGemini } from "../../drizzle/query";
 
 // ==============================
 
-const genAI = new GoogleGenerativeAI(rGemini.apiKey || "");
+export const genAI = new GoogleGenerativeAI(rGemini.apiKey || "");
+export const fileManager = new GoogleAIFileManager(rGemini.apiKey || "");
 
 export const geminiModel = genAI.getGenerativeModel({
   model: rGemini.model,