pengjunjie/langflow
1
0
Fork 1
mirror of https://github.com/langflow-ai/langflow.git synced 2026-07-24 11:47:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(models): keep gemini-embedding-001 active

Browse Source 
Only mark text-embedding-004 and embedding-001 as deprecated; the
gemini-embedding-001 model is still served by the Google Generative AI
v1beta endpoint, so leave it visible without the badge.
This commit is contained in:
Keval718
2026-04-28 14:52:51 -04:00
parent 36401bca96
commit 07764ff3db
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/lfx/src/lfx/base/models/google_generative_ai_constants.py
View File

@ -112,9 +112,15 @@ GOOGLE_GENERATIVE_AI_EMBEDDING_MODELS = [
"models/embedding-001",
]
# Embedding models as detailed metadata
# Marked deprecated: not natively supported by the Knowledge Base ingestion
# flow; hidden from the embedding model picker until support is added.
# Embedding models as detailed metadata.
# `text-embedding-004` and `embedding-001` are marked deprecated because they
# are no longer served by the Google Generative AI v1beta endpoint used by KB
# ingestion (404). `gemini-embedding-001` is the current supported model.
_GOOGLE_DEPRECATED_EMBEDDING_MODELS = {
"models/text-embedding-004",
"models/embedding-001",
}
GOOGLE_GENERATIVE_AI_EMBEDDING_MODELS_DETAILED = [
create_model_metadata(
provider="Google Generative AI",
@ -122,7 +128,7 @@ GOOGLE_GENERATIVE_AI_EMBEDDING_MODELS_DETAILED = [
icon="GoogleGenerativeAI",
model_type="embeddings",
default=True,
deprecated=True,
deprecated=name in _GOOGLE_DEPRECATED_EMBEDDING_MODELS,
)
for name in GOOGLE_GENERATIVE_AI_EMBEDDING_MODELS
]