diff --git a/frontend/src/components/EmbeddingSelection/GeminiOptions/index.jsx b/frontend/src/components/EmbeddingSelection/GeminiOptions/index.jsx index 18e1530ac..85183a490 100644 --- a/frontend/src/components/EmbeddingSelection/GeminiOptions/index.jsx +++ b/frontend/src/components/EmbeddingSelection/GeminiOptions/index.jsx @@ -1,15 +1,7 @@ const DEFAULT_MODELS = [ { - id: "embedding-001", - name: "Embedding 001", - }, - { - id: "text-embedding-004", - name: "Text Embedding 004", - }, - { - id: "gemini-embedding-exp-03-07", - name: "Gemini Embedding Exp 03 07", + id: "gemini-embedding-001", + name: "Gemini Embedding 001", }, ]; diff --git a/server/utils/EmbeddingEngines/gemini/index.js b/server/utils/EmbeddingEngines/gemini/index.js index 59eb22cb2..f04a49ffe 100644 --- a/server/utils/EmbeddingEngines/gemini/index.js +++ b/server/utils/EmbeddingEngines/gemini/index.js @@ -1,9 +1,7 @@ const { toChunks } = require("../../helpers"); const MODEL_MAP = { - "embedding-001": 2048, - "text-embedding-004": 2048, - "gemini-embedding-exp-03-07": 8192, + "gemini-embedding-001": 2048, }; class GeminiEmbedder { @@ -13,7 +11,7 @@ class GeminiEmbedder { this.className = "GeminiEmbedder"; const { OpenAI: OpenAIApi } = require("openai"); - this.model = process.env.EMBEDDING_MODEL_PREF || "text-embedding-004"; + this.model = process.env.EMBEDDING_MODEL_PREF || "gemini-embedding-001"; this.openai = new OpenAIApi({ apiKey: process.env.GEMINI_EMBEDDING_API_KEY, // Even models that are v1 in gemini API can be used with v1beta/openai/ endpoint and nobody knows why.