fix: Remove lazy loading of vertexai.types module

PiperOrigin-RevId: 834811652

Author: sararob

vertexai/__init__.py

@@ -18,6 +18,9 @@
 import sys
 
 from google.cloud.aiplatform import version as aiplatform_version
+from vertexai._genai import types
+
+sys.modules[__name__ + ".types"] = types
 
 __version__ = aiplatform_version.__version__
 
@@ -44,14 +47,6 @@ def __getattr__(name):  # type: ignore[no-untyped-def]
             _genai_client = importlib.import_module("._genai.client", __name__)
         return getattr(_genai_client, name)
 
-    if name == "types":
-        global _genai_types
-        if _genai_types is None:
-            _genai_types = importlib.import_module("._genai.types", __name__)
-        if "vertexai.types" not in sys.modules:
-            sys.modules["vertexai.types"] = _genai_types
-        return _genai_types
-
     raise AttributeError(f"module '{__name__}' has no attribute '{name}'")