feat(api): add gemini api wrapper

RajaRakoto · Nov 23, 2024 · f05f8a6 · f05f8a6
1 parent e8ffeeb
commit f05f8a6
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/api/index.ts b/src/api/index.ts
@@ -0,0 +1,13 @@
+/* libs */
+import { GoogleGenerativeAI } from "@google/generative-ai";
+
+/* orm */
+import { rGemini } from "../../drizzle/query";
+
+// ==============================
+
+const genAI = new GoogleGenerativeAI(rGemini.apiKey || "");
+
+export const geminiModel = genAI.getGenerativeModel({
+  model: rGemini.model,
+});