patch gemini embedding key preference

resolves #2972
Mintplex-Labs · Jan 14, 2025 · 4d5d8d4 · 4d5d8d4
1 parent 21af810
commit 4d5d8d4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/server/utils/EmbeddingEngines/gemini/index.js b/server/utils/EmbeddingEngines/gemini/index.js
@@ -1,9 +1,9 @@
 class GeminiEmbedder {
   constructor() {
-    if (!process.env.GEMINI_API_KEY)
+    if (!process.env.GEMINI_EMBEDDING_API_KEY)
       throw new Error("No Gemini API key was set.");
     const { GoogleGenerativeAI } = require("@google/generative-ai");
-    const genAI = new GoogleGenerativeAI(process.env.GEMINI_API_KEY);
+    const genAI = new GoogleGenerativeAI(process.env.GEMINI_EMBEDDING_API_KEY);
     this.model = process.env.EMBEDDING_MODEL_PREF || "text-embedding-004";
     this.gemini = genAI.getGenerativeModel({ model: this.model });