refactor: update default Gemini and Vertex AI embedder model to gemini-embedding-001 (#4571)

This commit is contained in:
Kartik
2026-03-28 20:39:27 +05:30
committed by GitHub
parent ece654811a
commit cd2dd7cc54
5 changed files with 11 additions and 11 deletions
+1 -1
View File
@@ -12,7 +12,7 @@ class GoogleGenAIEmbedding(EmbeddingBase):
def __init__(self, config: Optional[BaseEmbedderConfig] = None):
super().__init__(config)
self.config.model = self.config.model or "models/text-embedding-004"
self.config.model = self.config.model or "models/gemini-embedding-001"
self.config.embedding_dims = self.config.embedding_dims or self.config.output_dimensionality or 768
api_key = self.config.api_key or os.getenv("GOOGLE_API_KEY")
+1 -1
View File
@@ -12,7 +12,7 @@ class VertexAIEmbedding(EmbeddingBase):
def __init__(self, config: Optional[BaseEmbedderConfig] = None):
super().__init__(config)
self.config.model = self.config.model or "text-embedding-004"
self.config.model = self.config.model or "gemini-embedding-001"
self.config.embedding_dims = self.config.embedding_dims or 256
self.embedding_types = {