mirror of
https://github.com/Mintplex-Labs/anything-llm
synced 2026-04-25 17:15:37 +02:00
remove deprecated value for Google embedding
This commit is contained in:
@@ -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",
|
||||
},
|
||||
];
|
||||
|
||||
|
||||
@@ -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.
|
||||
|
||||
Reference in New Issue
Block a user