docs(genai): Add Grounding Sample for Vertex AI Search

Author: holtskinner

genai/tools/test_tools_examples.py

@@ -23,6 +23,7 @@
 import tools_func_def_with_txt
 import tools_func_desc_with_txt
 import tools_google_search_with_txt
+import tools_vais_with_txt
 
 os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "True"
 os.environ["GOOGLE_CLOUD_LOCATION"] = "us-central1"
@@ -53,3 +54,10 @@ def test_tools_func_desc_with_txt() -> None:
 def test_tools_google_search_with_txt() -> None:
     response = tools_google_search_with_txt.generate_content()
     assert response
+
+
+def test_tools_vais_with_txt() -> None:
+    PROJECT_ID = os.environ.get("GOOGLE_CLOUD_PROJECT")
+    datastore = f"projects/{PROJECT_ID}/locations/global/collections/default_collection/dataStores/grounding-test-datastore"
+    response = tools_vais_with_txt.generate_content(datastore)
+    assert response

genai/tools/tools_vais_with_txt.py

@@ -0,0 +1,58 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+def generate_content(datastore: str) -> str:
+    # [START googlegenaisdk_tools_vais_with_txt]
+    from google import genai
+    from google.genai.types import (
+        GenerateContentConfig,
+        HttpOptions,
+        Retrieval,
+        Tool,
+        VertexAISearch,
+    )
+
+    client = genai.Client(http_options=HttpOptions(api_version="v1"))
+
+    # Load Data Store ID from Vertex AI Search
+    # datastore = "projects/111111111111/locations/global/collections/default_collection/dataStores/data-store-id"
+
+    response = client.models.generate_content(
+        model="gemini-2.0-flash-001",
+        contents="How do I make an appointment to renew my driver's license?",
+        config=GenerateContentConfig(
+            tools=[
+                # Use Vertex AI Search Tool
+                Tool(
+                    retrieval=Retrieval(
+                        vertex_ai_search=VertexAISearch(
+                            datastore=datastore,
+                        )
+                    )
+                )
+            ],
+        ),
+    )
+
+    print(response.text)
+    # Example response:
+    # 'The process for making an appointment to renew your driver's license varies depending on your location. To provide you with the most accurate instructions...'
+    # [END googlegenaisdk_tools_vais_with_txt]
+    return response.text
+
+
+if __name__ == "__main__":
+    datastore = input("Data Store ID: ")
+    generate_content(datastore)