remove deprecated value for Google embedding

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

View File

@@ -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",
},
];

View File

@@ -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.