Add VAIS create corpus test

Author: holtskinner

generative_ai/rag/test_rag_examples.py

@@ -23,6 +23,7 @@
 import create_corpus_feature_store_example
 import create_corpus_pinecone_example
 import create_corpus_vector_search_example
+import create_corpus_vertex_ai_search_example
 import create_corpus_weaviate_example
 import delete_corpus_example
 import delete_file_example
@@ -138,6 +139,16 @@ def test_create_corpus_weaviate() -> None:
     delete_corpus_example.delete_corpus(corpus.name)
 
 
+def test_create_corpus_vertex_ai_search() -> None:
+    VAIS_LOCATION = "us"
+    ENGINE_ID = "test-engine"
+    corpus = create_corpus_vertex_ai_search_example.create_corpus_vertex_ai_search(
+        f"projects/{PROJECT_ID}/locations/{VAIS_LOCATION}/collections/default_collection/engines/{ENGINE_ID}"
+    )
+    assert corpus
+    delete_corpus_example.delete_corpus(corpus.name)
+
+
 def test_get_corpus(test_corpus: pytest.fixture) -> None:
     retrieved_corpus = get_corpus_example.get_corpus(test_corpus.name)
     assert retrieved_corpus.name == test_corpus.name