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

2024-10-03 21:34:14 +05:30
parent c09c4926a7
commit d107b639b3
5 changed files with 15 additions and 8 deletions
--- a/mem0/configs/embeddings/base.py
+++ b/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