diff --git a/libs/vertexai/langchain_google_vertexai/__init__.py b/libs/vertexai/langchain_google_vertexai/__init__.py
index 4768a057..5d08c169 100644
--- a/libs/vertexai/langchain_google_vertexai/__init__.py
+++ b/libs/vertexai/langchain_google_vertexai/__init__.py
@@ -13,7 +13,13 @@
 )
 from langchain_google_vertexai.llms import VertexAI
 from langchain_google_vertexai.model_garden import VertexAIModelGarden
-from langchain_google_vertexai.vectorstores.vectorstores import VectorSearchVectorStore
+from langchain_google_vertexai.vectorstores import (
+    DataStoreDocumentStorage,
+    GCSDocumentStorage,
+    VectorSearchVectorStore,
+    VectorSearchVectorStoreDatastore,
+    VectorSearchVectorStoreGCS,
+)
 from langchain_google_vertexai.vision_models import (
     VertexAIImageCaptioning,
     VertexAIImageCaptioningChat,
@@ -37,10 +43,14 @@
     "HarmCategory",
     "PydanticFunctionsOutputParser",
     "create_structured_runnable",
-    "VectorSearchVectorStore",
     "VertexAIImageCaptioning",
     "VertexAIImageCaptioningChat",
     "VertexAIImageEditorChat",
     "VertexAIImageGeneratorChat",
     "VertexAIVisualQnAChat",
+    "DataStoreDocumentStorage",
+    "GCSDocumentStorage",
+    "VectorSearchVectorStore",
+    "VectorSearchVectorStoreDatastore",
+    "VectorSearchVectorStoreGCS",
 ]
diff --git a/libs/vertexai/tests/unit_tests/test_imports.py b/libs/vertexai/tests/unit_tests/test_imports.py
index 02f721fe..fc578c88 100644
--- a/libs/vertexai/tests/unit_tests/test_imports.py
+++ b/libs/vertexai/tests/unit_tests/test_imports.py
@@ -15,12 +15,16 @@
     "HarmCategory",
     "PydanticFunctionsOutputParser",
     "create_structured_runnable",
-    "VectorSearchVectorStore",
     "VertexAIImageCaptioning",
     "VertexAIImageCaptioningChat",
     "VertexAIImageEditorChat",
     "VertexAIImageGeneratorChat",
     "VertexAIVisualQnAChat",
+    "DataStoreDocumentStorage",
+    "GCSDocumentStorage",
+    "VectorSearchVectorStore",
+    "VectorSearchVectorStoreDatastore",
+    "VectorSearchVectorStoreGCS",
 ]