sij/anything-llm
1
0
Fork 0
mirror of https://github.com/Mintplex-Labs/anything-llm.git synced 2025-05-02 17:07:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

patch gemini embedding key preference

Browse source 
resolves
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

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