diff --git a/gemini_docs/vertexai/vertexai.rst b/gemini_docs/vertexai/vertexai.rst
index 43c9d5d269..a6633984f1 100644
--- a/gemini_docs/vertexai/vertexai.rst
+++ b/gemini_docs/vertexai/vertexai.rst
@@ -6,6 +6,8 @@ Vertex AI SDK
     :show-inheritance:
     :inherited-members:
 
+    .. autoattribute:: types
+
 .. autoclass:: vertexai.Client
     :members:
     :undoc-members:
@@ -37,6 +39,11 @@ Vertex AI SDK
     :undoc-members:
     :show-inheritance:
 
+.. automodule:: vertexai._genai.types
+    :members:
+    :undoc-members:
+    :show-inheritance:
+
 .. automodule:: vertexai.generative_models
     :members:
     :show-inheritance: