Add vectorstores to golbal namespace (#139)

* Add vectorstores to golbal namespace --------- Co-authored-by: Jorge <[email protected]>
langchain-ai · Apr 11, 2024 · a609de0 · a609de0
1 parent 22c2056
commit a609de0
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git 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
@@ -15,12 +15,16 @@
     "HarmCategory",
     "PydanticFunctionsOutputParser",
     "create_structured_runnable",
-    "VectorSearchVectorStore",
     "VertexAIImageCaptioning",
     "VertexAIImageCaptioningChat",
     "VertexAIImageEditorChat",
     "VertexAIImageGeneratorChat",
     "VertexAIVisualQnAChat",
+    "DataStoreDocumentStorage",
+    "GCSDocumentStorage",
+    "VectorSearchVectorStore",
+    "VectorSearchVectorStoreDatastore",
+    "VectorSearchVectorStoreGCS",
 ]