(bug-fix) : fix VertexAI missing configurations (#1926)

mem0/configs/embeddings/base.py

@@ -25,6 +25,8 @@ class BaseEmbedderConfig(ABC):
         # AzureOpenAI specific
         azure_kwargs: Optional[AzureConfig] = {},
         http_client_proxies: Optional[Union[Dict, str]] = None,
+        # VertexAI specific
+        vertex_credentials_json: Optional[str] = None,
     ):
         """
         Initializes a configuration class instance for the Embeddings.
@@ -63,3 +65,6 @@ class BaseEmbedderConfig(ABC):
 
         # AzureOpenAI specific
         self.azure_kwargs = AzureConfig(**azure_kwargs) or {}
+
+        # VertexAI specific
+        self.vertex_credentials_json = vertex_credentials_json

mem0/embeddings/vertexai.py

@@ -7,7 +7,7 @@ from mem0.configs.embeddings.base import BaseEmbedderConfig
 from mem0.embeddings.base import EmbeddingBase
 
 
-class VertexAI(EmbeddingBase):
+class VertexAIEmbedding(EmbeddingBase):
     def __init__(self, config: Optional[BaseEmbedderConfig] = None):
         super().__init__(config)
 

mem0/utils/factory.py

@@ -42,6 +42,7 @@ class EmbedderFactory:
         "huggingface": "mem0.embeddings.huggingface.HuggingFaceEmbedding",
         "azure_openai": "mem0.embeddings.azure_openai.AzureOpenAIEmbedding",
         "gemini": "mem0.embeddings.gemini.GoogleGenAIEmbedding",
+        "vertexai": "mem0.embeddings.vertexai.VertexAIEmbedding",
     }
 
     @classmethod