diff --git a/gemini_docs/vertexai/vertexai.rst b/gemini_docs/vertexai/vertexai.rst
index a6633984f1..085054b801 100644
--- a/gemini_docs/vertexai/vertexai.rst
+++ b/gemini_docs/vertexai/vertexai.rst
@@ -24,21 +24,70 @@ Vertex AI SDK
     :undoc-members:
     :show-inheritance:
 
-.. automodule:: vertexai._genai.agent_engines
+.. automodule:: vertexai._genai.prompt_optimizer
     :members:
     :undoc-members:
     :show-inheritance:
 
-.. automodule:: vertexai._genai.prompt_optimizer
+.. automodule:: vertexai._genai.prompts
     :members:
     :undoc-members:
     :show-inheritance:
 
-.. automodule:: vertexai._genai.prompts
+.. automodule:: vertexai._genai.agent_engines
     :members:
     :undoc-members:
     :show-inheritance:
 
+    .. autoclass:: vertexai._genai.agent_engines.AgentEngines
+        :members:
+        .. autoattribute:: memories
+        .. autoattribute:: sessions
+        .. autoattribute:: sandboxes
+
+    Sessions
+    --------
+    .. automodule:: vertexai._genai.sessions
+        :members:
+        :undoc-members:
+        :show-inheritance:
+
+        .. autoclass:: vertexai._genai.sessions.Sessions
+            :members:
+            .. autoattribute:: events
+
+        Events
+        ~~~~~~
+        .. automodule:: vertexai._genai.session_events
+            :members:
+            :undoc-members:
+            :show-inheritance:
+
+    Memories
+    --------
+    .. automodule:: vertexai._genai.memories
+        :members:
+        :undoc-members:
+        :show-inheritance:
+
+        .. autoclass:: vertexai._genai.memories.Memories
+            :members:
+            .. autoattribute:: revisions
+
+        Revisions
+        ~~~~~~~~~
+        .. automodule:: vertexai._genai.memory_revisions
+            :members:
+            :undoc-members:
+            :show-inheritance:
+
+    Sandboxes
+    --------
+    .. automodule:: vertexai._genai.sandboxes
+        :members:
+        :undoc-members:
+        :show-inheritance:
+
 .. automodule:: vertexai._genai.types
     :members:
     :undoc-members: