michael/merlyn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

patch gemini embedding key preference

Browse Source 
resolves #2972
This commit is contained in:
timothycarambat 2025-01-14 08:20:57 -08:00
parent 21af81085a
commit 4d5d8d4dec
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server/utils/EmbeddingEngines/gemini/index.js
View File

@ -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 });