fix: avoid init vertexai during unittests

libs/vertexai/.gitignore

@@ -1 +1,2 @@
 __pycache__
+.mypy_cache_test

libs/vertexai/tests/integration_tests/conftest.py

@@ -2,7 +2,7 @@
 
 import pytest
 from _pytest.tmpdir import TempPathFactory
-from vertexai.vision_models import Image
+from vertexai.vision_models import Image  # type: ignore
 
 
 @pytest.fixture

libs/vertexai/tests/integration_tests/test_embeddings.py

@@ -4,12 +4,12 @@
 `gcloud auth login` first).
 """
 import pytest
-from vertexai.language_models import TextEmbeddingModel
-from vertexai.vision_models import MultiModalEmbeddingModel
+from vertexai.language_models import TextEmbeddingModel  # type: ignore
+from vertexai.vision_models import MultiModalEmbeddingModel  # type: ignore
 
 from langchain_google_vertexai.embeddings import (
-    VertexAIEmbeddings,
     GoogleEmbeddingModelType,
+    VertexAIEmbeddings,
 )
 
 

libs/vertexai/tests/unit_tests/test_embeddings.py

@@ -1,7 +1,8 @@
-from typing import Any
+from typing import Any, Dict
 from unittest.mock import MagicMock
 
 import pytest
+from pydantic.v1 import root_validator
 
 from langchain_google_vertexai import VertexAIEmbeddings
 from langchain_google_vertexai.embeddings import GoogleEmbeddingModelType
@@ -39,4 +40,12 @@ class MockVertexAIEmbeddings(VertexAIEmbeddings):
 
     def __init__(self, model_name, **kwargs: Any) -> None:
         super().__init__(model_name, **kwargs)
-        self.client = MagicMock()
+
+    @classmethod
+    def _init_vertexai(cls, values: Dict) -> None:
+        pass
+
+    @root_validator()
+    def validate_environment(cls, values: Dict) -> Dict:
+        values["client"] = MagicMock()
+        return values