docs(samples): Fixed Vertex AI Vector Search Create Index `index_upda…

…te_method` PiperOrigin-RevId: 691943549
googleapis · Oct 31, 2024 · 7dff586 · 7dff586
1 parent 6c7f439
commit 7dff586
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/samples/model-builder/vector_search/vector_search_create_index_sample.py b/samples/model-builder/vector_search/vector_search_create_index_sample.py
@@ -40,7 +40,7 @@ def vector_search_create_index(
         approximate_neighbors_count=150,
         leaf_node_embedding_count=500,
         leaf_nodes_to_search_percent=7,
-        index_update_method="batch_update",  # Options: stream_update, batch_update
+        index_update_method="BATCH_UPDATE",  # Options: STREAM_UPDATE, BATCH_UPDATE
         distance_measure_type=aiplatform.matching_engine.matching_engine_index_config.DistanceMeasureType.DOT_PRODUCT_DISTANCE,
     )
 
@@ -73,7 +73,7 @@ def vector_search_create_streaming_index(
         approximate_neighbors_count=150,
         leaf_node_embedding_count=500,
         leaf_nodes_to_search_percent=7,
-        index_update_method="stream_update",  # Options: stream_update, batch_update
+        index_update_method="STREAM_UPDATE",  # Options: STREAM_UPDATE, BATCH_UPDATE
         distance_measure_type=aiplatform.matching_engine.matching_engine_index_config.DistanceMeasureType.DOT_PRODUCT_DISTANCE,
     )
 

diff --git a/samples/model-builder/vector_search/vector_search_create_index_sample_test.py b/samples/model-builder/vector_search/vector_search_create_index_sample_test.py
@@ -44,7 +44,7 @@ def test_vector_search_create_index_sample(
         approximate_neighbors_count=ANY,
         leaf_node_embedding_count=ANY,
         leaf_nodes_to_search_percent=ANY,
-        index_update_method="batch_update",
+        index_update_method="BATCH_UPDATE",
         distance_measure_type=ANY,
     )
 
@@ -75,6 +75,6 @@ def test_vector_search_create_streaming_index_sample(
         approximate_neighbors_count=ANY,
         leaf_node_embedding_count=ANY,
         leaf_nodes_to_search_percent=ANY,
-        index_update_method="stream_update",
+        index_update_method="STREAM_UPDATE",
         distance_measure_type=ANY,
     )