From f1a92e967469cc3fc25d6f122445eb358fe7aa93 Mon Sep 17 00:00:00 2001
From: Ariz Chang <arizchang@google.com>
Date: Fri, 23 May 2025 14:35:41 -0700
Subject: [PATCH] chore: add google search agent for testing

PiperOrigin-RevId: 762571269
---
 .../samples/google_search_agent/__init__.py   | 15 ++++++++++
 .../samples/google_search_agent/agent.py      | 30 +++++++++++++++++++
 2 files changed, 45 insertions(+)
 create mode 100644 contributing/samples/google_search_agent/__init__.py
 create mode 100644 contributing/samples/google_search_agent/agent.py

diff --git a/contributing/samples/google_search_agent/__init__.py b/contributing/samples/google_search_agent/__init__.py
new file mode 100644
index 0000000..c48963c
--- /dev/null
+++ b/contributing/samples/google_search_agent/__init__.py
@@ -0,0 +1,15 @@
+# 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
+#
+#     http://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.
+
+from . import agent
diff --git a/contributing/samples/google_search_agent/agent.py b/contributing/samples/google_search_agent/agent.py
new file mode 100644
index 0000000..9ec1c88
--- /dev/null
+++ b/contributing/samples/google_search_agent/agent.py
@@ -0,0 +1,30 @@
+# 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
+#
+#     http://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.
+
+from google.genai import Client
+
+from google.adk import Agent
+from google.adk.tools import google_search
+
+# Only Vertex AI supports image generation for now.
+client = Client()
+
+root_agent = Agent(
+    model='gemini-2.0-flash-001',
+    name='root_agent',
+    description="""an agent whose job it is to perform Google search queries and answer questions about the results.""",
+    instruction="""You are an agent whose job is to perform Google search queries and answer questions about the results.
+""",
+    tools=[google_search],
+)