diff --git a/libs/vertexai/langchain_google_vertexai/__init__.py b/libs/vertexai/langchain_google_vertexai/__init__.py index 81aec35a..858d0f43 100644 --- a/libs/vertexai/langchain_google_vertexai/__init__.py +++ b/libs/vertexai/langchain_google_vertexai/__init__.py @@ -11,6 +11,7 @@ ) from langchain_google_vertexai.llms import VertexAI from langchain_google_vertexai.model_garden import VertexAIModelGarden +from langchain_google_vertexai.vectorstores.vectorstores import VectorSearchVectorStore __all__ = [ "ChatVertexAI", @@ -27,4 +28,5 @@ "HarmCategory", "PydanticFunctionsOutputParser", "create_structured_runnable", + "VectorSearchVectorStore", ] diff --git a/libs/vertexai/tests/integration_tests/test_vectorstores.py b/libs/vertexai/tests/integration_tests/test_vectorstores.py index 57c1d92f..39face5e 100644 --- a/libs/vertexai/tests/integration_tests/test_vectorstores.py +++ b/libs/vertexai/tests/integration_tests/test_vectorstores.py @@ -41,6 +41,7 @@ def sdk_manager() -> VectorSearchSDKManager: return sdk_manager +@pytest.mark.skip("CI testing not set up") def test_vector_search_sdk_manager(sdk_manager: VectorSearchSDKManager): gcs_client = sdk_manager.get_gcs_client() assert isinstance(gcs_client, storage.Client) @@ -55,6 +56,7 @@ def test_vector_search_sdk_manager(sdk_manager: VectorSearchSDKManager): assert isinstance(endpoint, MatchingEngineIndexEndpoint) +@pytest.mark.skip("CI testing not set up") def test_gcs_document_storage(sdk_manager: VectorSearchSDKManager): bucket = sdk_manager.get_gcs_bucket(os.environ["GCS_BUCKET_NAME"]) prefix = "integration-test" @@ -78,6 +80,7 @@ def test_gcs_document_storage(sdk_manager: VectorSearchSDKManager): assert original_text == retrieved_text +@pytest.mark.skip("CI testing not set up") def test_datastore_document_storage(sdk_manager: VectorSearchSDKManager): ds_client = sdk_manager.get_datastore_client(namespace="Foo") @@ -100,6 +103,7 @@ def test_datastore_document_storage(sdk_manager: VectorSearchSDKManager): assert original_text == retrieved_text +@pytest.mark.skip("CI testing not set up") def test_public_endpoint_vector_searcher(sdk_manager: VectorSearchSDKManager): index = sdk_manager.get_index(os.environ["INDEX_ID"]) endpoint = sdk_manager.get_endpoint(os.environ["ENDPOINT_ID"]) @@ -116,6 +120,7 @@ def test_public_endpoint_vector_searcher(sdk_manager: VectorSearchSDKManager): assert len(matching_neighbors_list) == 2 +@pytest.mark.skip("CI testing not set up") def test_vector_store(): embeddings = VertexAIEmbeddings(model_name="textembedding-gecko-default") @@ -143,6 +148,7 @@ def test_vector_store(): assert isinstance(doc, Document) +@pytest.mark.skip("CI testing not set up") def test_vector_store_update_index(): embeddings = VertexAIEmbeddings(model_name="textembedding-gecko-default") diff --git a/libs/vertexai/tests/unit_tests/test_imports.py b/libs/vertexai/tests/unit_tests/test_imports.py index e1b1ab1e..ab722808 100644 --- a/libs/vertexai/tests/unit_tests/test_imports.py +++ b/libs/vertexai/tests/unit_tests/test_imports.py @@ -15,6 +15,7 @@ "HarmCategory", "PydanticFunctionsOutputParser", "create_structured_runnable", + "VectorSearchVectorStore", ]