feat(genai): Add Grounding with Enterprise Web Search and Google Maps Samples

Author: holtskinner

genai/tools/requirements.txt

@@ -1,3 +1,3 @@
-google-genai==1.16.1
+google-genai==1.18.0
 # PIl is required for tools_code_execution_with_txt_img.py
 pillow==11.1.0

genai/tools/test_tools_examples.py

@@ -15,13 +15,15 @@
 #
 # Using Google Cloud Vertex AI to test the code samples.
 #
-
 import os
 
+import pytest
 import tools_code_exec_with_txt
 import tools_code_exec_with_txt_local_img
+import tools_enterprise_web_search_with_txt
 import tools_func_def_with_txt
 import tools_func_desc_with_txt
+import tools_google_maps_with_txt
 import tools_google_search_with_txt
 import tools_vais_with_txt
 
@@ -41,6 +43,11 @@ def test_tools_code_exec_with_txt_local_img() -> None:
     assert response
 
 
+def test_tools_enterprise_web_search_with_txt() -> None:
+    response = tools_enterprise_web_search_with_txt.generate_content()
+    assert response
+
+
 def test_tools_func_def_with_txt() -> None:
     response = tools_func_def_with_txt.generate_content()
     assert response
@@ -51,6 +58,14 @@ def test_tools_func_desc_with_txt() -> None:
     assert response
 
 
+@pytest.mark.skip(
+    reason="Google Maps Grounding allowlisting is not set up for the test project."
+)
+def test_tools_google_maps_with_txt() -> None:
+    response = tools_google_maps_with_txt.generate_content()
+    assert response
+
+
 def test_tools_google_search_with_txt() -> None:
     response = tools_google_search_with_txt.generate_content()
     assert response

genai/tools/tools_enterprise_web_search_with_txt.py

@@ -0,0 +1,47 @@
+# 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() -> str:
+    # [START googlegenaisdk_tools_enterprise_web_search_with_txt]
+    from google import genai
+    from google.genai.types import (
+        EnterpriseWebSearch,
+        GenerateContentConfig,
+        HttpOptions,
+        Tool,
+    )
+
+    client = genai.Client(http_options=HttpOptions(api_version="v1"))
+
+    response = client.models.generate_content(
+        model="gemini-2.5-flash-preview-05-20",
+        contents="When is the next total solar eclipse in the United States?",
+        config=GenerateContentConfig(
+            tools=[
+                # Use Enterprise Web Search Tool
+                Tool(enterprise_web_search=EnterpriseWebSearch())
+            ],
+        ),
+    )
+
+    print(response.text)
+    # Example response:
+    # 'The next total solar eclipse in the United States will occur on ...'
+    # [END googlegenaisdk_tools_enterprise_web_search_with_txt]
+    return response.text
+
+
+if __name__ == "__main__":
+    generate_content()

genai/tools/tools_google_maps_with_txt.py

@@ -0,0 +1,60 @@
+# 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() -> str:
+    # [START googlegenaisdk_tools_google_maps_with_txt]
+    from google import genai
+    from google.genai.types import (
+        ApiKeyConfig,
+        AuthConfig,
+        GenerateContentConfig,
+        GoogleMaps,
+        HttpOptions,
+        Tool,
+    )
+
+    # TODO(developer): Update below line with your Google Maps API key
+    GOOGLE_MAPS_API_KEY = "YOUR_GOOGLE_MAPS_API_KEY"
+
+    client = genai.Client(http_options=HttpOptions(api_version="v1"))
+
+    response = client.models.generate_content(
+        model="gemini-2.5-flash-preview-05-20",
+        contents="Recommend a good restaurant in San Francisco",
+        config=GenerateContentConfig(
+            tools=[
+                # Use Google Maps Tool
+                Tool(
+                    google_maps=GoogleMaps(
+                        auth_config=AuthConfig(
+                            api_key_config=ApiKeyConfig(
+                                api_key_string=GOOGLE_MAPS_API_KEY,
+                            )
+                        )
+                    )
+                )
+            ],
+        ),
+    )
+
+    print(response.text)
+    # Example response:
+    # 'The next total solar eclipse in the United States will occur on ...'
+    # [END googlegenaisdk_tools_enterprise_web_search_with_txt]
+    return response.text
+
+
+if __name__ == "__main__":
+    generate_content()