refactor: update default Gemini and Vertex AI embedder model to gemini-embedding-001 (#4571)
This commit is contained in:
@@ -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")
|
||||
|
||||
@@ -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 = {
|
||||
|
||||
Reference in New Issue
Block a user