docs(samples): Update Vertex AI Gemini Grounding with new syntax from v1.64.0 (GoogleCloudPlatform#12537)

* docs(samples): Update Vertex AI Gemini Grounding with new syntax from v1.64.0

- https://github.com/googleapis/python-aiplatform/releases/tag/v1.64.0

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* Updated testing

---------

Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>

Author: holtskinner

generative_ai/gemini_grounding_example.py

@@ -53,23 +53,34 @@ def generate_text_with_grounding_web() -> GenerationResponse:
 
 
 def generate_text_with_grounding_vertex_ai_search(
-    data_store_path: str,
+    data_store_id: str,
 ) -> GenerationResponse:
     # [START generativeaionvertexai_gemini_grounding_with_vais]
     import vertexai
 
-    from vertexai.preview.generative_models import grounding
-    from vertexai.generative_models import GenerationConfig, GenerativeModel, Tool
+    from vertexai.preview.generative_models import (
+        GenerationConfig,
+        GenerativeModel,
+        Tool,
+        grounding,
+    )
 
     # TODO (developer): update project_id
     vertexai.init(project=PROJECT_ID, location="us-central1")
 
     model = GenerativeModel("gemini-1.5-flash-001")
 
-    # TODO(developer): Update and un-comment below line with the correct data store path
-    # data_store_path = "projects/{project_id}/locations/{location}/collections/default_collection/dataStores/{data_store_id}"
+    # TODO(developer): Update project id, location, and data store id for your Vertex AI Search data store.
+    # data_store_id = "DATA_STORE_ID"
+
     tool = Tool.from_retrieval(
-        grounding.Retrieval(grounding.VertexAISearch(datastore=data_store_path))
+        grounding.Retrieval(
+            grounding.VertexAISearch(
+                datastore=data_store_id,
+                project=PROJECT_ID,
+                location="global",
+            )
+        )
     )
 
     prompt = "How do I make an appointment to renew my driver's license?"

generative_ai/requirements.txt

@@ -3,13 +3,12 @@ pandas==2.0.3; python_version == '3.8'
 pandas==2.1.4; python_version > '3.8'
 pillow==9.5.0; python_version < '3.8'
 pillow==10.3.0; python_version >= '3.8'
-# google-cloud-aiplatform[pipelines,rapid_evaluation,reasoningengine,tokenization]==1.57.0
-google-cloud-aiplatform[all]==1.63.0
+google-cloud-aiplatform[all]==1.64.0
 sentencepiece==0.2.0
 google-auth==2.29.0
 anthropic[vertex]==0.28.0
 langchain-core==0.2.11
 langchain-google-vertexai==1.0.6
 numpy<2
 openai==1.30.5
-immutabledict==4.2.0
+immutabledict==4.2.0

generative_ai/test_gemini_examples.py

@@ -136,9 +136,8 @@ def test_gemini_grounding_web_example() -> None:
 
 
 def test_gemini_grounding_vais_example() -> None:
-    data_store_path = f"projects/{PROJECT_ID}/locations/global/collections/default_collection/dataStores/grounding-test-datastore"
     response = gemini_grounding_example.generate_text_with_grounding_vertex_ai_search(
-        data_store_path=data_store_path,
+        "grounding-test-datastore"
     )
     assert response