Skip to content

Commit bbb9131

Browse files
authored
fix(vertex): update model version check for text embedding (#1184)
1 parent 152daeb commit bbb9131

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

libs/vertexai/langchain_google_vertexai/embeddings.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ def _missing_(cls, value: Any) -> "GoogleEmbeddingModelVersion":
9090
if "textembedding-gecko@003" in value.lower():
9191
return GoogleEmbeddingModelVersion.EMBEDDINGS_DEC_2023
9292
if (
93-
"gemini-embedding-001" in value.lower()
93+
"text-embedding-004" in value.lower()
9494
or "text-multilingual-embedding-002" in value.lower()
9595
or "text-embedding-preview-0409" in value.lower()
9696
or "text-multilingual-embedding-preview-0409" in value.lower()

0 commit comments

Comments
 (0)