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

remove deprecated value for Google embedding

Browse Source
This commit is contained in:
Timothy Carambat 2026-02-08 14:05:34 -08:00
parent 8f7e0fb1f8
commit 152ac93e27
2 changed files with 4 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
frontend/src/components/EmbeddingSelection/GeminiOptions/index.jsx
View File

@ -1,15 +1,7 @@
const DEFAULT_MODELS = [ const DEFAULT_MODELS = [
{ {
id: "embedding-001", id: "gemini-embedding-001",
name: "Embedding 001", name: "Gemini Embedding 001",
},
{
id: "text-embedding-004",
name: "Text Embedding 004",
},
{
id: "gemini-embedding-exp-03-07",
name: "Gemini Embedding Exp 03 07",
}, },
]; ];

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

@ -1,9 +1,7 @@
const { toChunks } = require("../../helpers"); const { toChunks } = require("../../helpers");
const MODEL_MAP = { const MODEL_MAP = {
"embedding-001": 2048, "gemini-embedding-001": 2048,
"text-embedding-004": 2048,
"gemini-embedding-exp-03-07": 8192,
}; };
class GeminiEmbedder { class GeminiEmbedder {
@ -13,7 +11,7 @@ class GeminiEmbedder {
this.className = "GeminiEmbedder"; this.className = "GeminiEmbedder";
const { OpenAI: OpenAIApi } = require("openai"); 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({ this.openai = new OpenAIApi({
apiKey: process.env.GEMINI_EMBEDDING_API_KEY, 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. // Even models that are v1 in gemini API can be used with v1beta/openai/ endpoint and nobody knows why.