google-adk 0.0.1__py3-none-any.whl → 0.0.2__py3-none-any.whl

google/adk/tests/integration/fixture/home_automation_agent/simple_test.test.json

@@ -0,0 +1,5 @@
+[{
+  "query": "Turn off device_2 in the Bedroom.",
+  "expected_tool_use": [{"tool_name": "set_device_info", "tool_input": {"location": "Bedroom", "device_id": "device_2", "status": "OFF"}}],
+  "reference": "I have set the device_2 status to off."
+}]

google/adk/tests/integration/fixture/home_automation_agent/simple_test2.test.json

@@ -0,0 +1,5 @@
+[{
+  "query": "Turn off device_3 in the Bedroom.",
+  "expected_tool_use": [{"tool_name": "set_device_info", "tool_input": {"location": "Bedroom", "device_id": "device_3", "status": "OFF"}}],
+  "reference": "I have set the device_3 status to off."
+}]

google/adk/tests/integration/fixture/home_automation_agent/test_config.json

@@ -0,0 +1,5 @@
+{
+  "criteria": {
+    "tool_trajectory_avg_score": 1.0
+  }
+}

google/adk/tests/integration/fixture/home_automation_agent/test_files/dependent_tool_calls.test.json

@@ -0,0 +1,18 @@
+[
+  {
+      "query": "Turn off device_2 in the Bedroom.",
+      "expected_tool_use": [{
+          "tool_name": "set_device_info",
+          "tool_input": {"location": "Bedroom", "status": "OFF", "device_id": "device_2"}
+      }],
+      "reference": "I have set the device 2 status to off."
+  },
+  {
+      "query": "What's the status of device_2 in the Bedroom?",
+      "expected_tool_use": [{
+          "tool_name": "get_device_info",
+          "tool_input": {"device_id": "device_2"}
+      }],
+      "reference": "Status of device_2 is off."
+  }
+]

google/adk/tests/integration/fixture/home_automation_agent/test_files/memorizing_past_events/eval_data.test.json

@@ -0,0 +1,17 @@
+[
+  {
+      "query": "Turn off device_2 in the Bedroom.",
+      "expected_tool_use": [
+          {
+              "tool_name": "set_device_info",
+              "tool_input": {"location": "Bedroom", "device_id": "device_2", "status": "OFF"}
+          }
+      ],
+      "reference": "OK. I've turned off device_2 in the Bedroom. Anything else?\n"
+  },
+  {
+      "query": "What's the command I just issued?",
+      "expected_tool_use": [],
+      "reference": "You asked me to turn off device_2 in the Bedroom.\n"
+  }
+]

google/adk/tests/integration/fixture/home_automation_agent/test_files/memorizing_past_events/test_config.json

@@ -0,0 +1,6 @@
+{
+  "criteria": {
+    "tool_trajectory_avg_score": 1.0,
+    "response_match_score": 0.5
+  }
+}

google/adk/tests/integration/fixture/home_automation_agent/test_files/simple_multi_turn_conversation.test.json

@@ -0,0 +1,18 @@
+[
+    {
+        "query": "Turn off device_2 in the Bedroom.",
+        "expected_tool_use": [{
+            "tool_name": "set_device_info",
+            "tool_input": {"location": "Bedroom", "device_id": "device_2", "status": "OFF"}
+        }],
+        "reference": "I have set the device 2 status to off."
+    },
+    {
+        "query": "Turn on device_2 in the Bedroom.",
+        "expected_tool_use": [{
+            "tool_name": "set_device_info",
+            "tool_input": {"location": "Bedroom", "status": "ON", "device_id": "device_2"}
+        }],
+        "reference": "I have set the device 2 status to on."
+    }
+]

google/adk/tests/integration/fixture/home_automation_agent/test_files/simple_test.test.json

@@ -0,0 +1,17 @@
+[
+  {
+      "query": "Turn off device_2 in the Bedroom.",
+      "expected_tool_use": [
+          {
+              "tool_name": "set_device_info",
+              "tool_input": {"location": "Bedroom", "device_id": "device_2", "status": "OFF"}
+          }
+      ],
+      "reference": "OK. I've turned off device_2 in the Bedroom. Anything else?\n"
+  },
+  {
+      "query": "What's the command I just issued?",
+      "expected_tool_use": [],
+      "reference": "You asked me to turn off device_2 in the Bedroom.\n"
+  }
+]

google/adk/tests/integration/fixture/home_automation_agent/test_files/simple_test2.test.json

@@ -0,0 +1,5 @@
+[{
+  "query": "Turn off device_3 in the Bedroom.",
+  "expected_tool_use": [{"tool_name": "set_device_info", "tool_input": {"location": "Bedroom", "device_id": "device_3", "status": "OFF"}}],
+  "reference": "I have set the device_3 status to off."
+}]

google/adk/tests/integration/fixture/home_automation_agent/test_files/test_config.json

@@ -0,0 +1,5 @@
+{
+  "criteria": {
+    "tool_trajectory_avg_score": 1.0
+  }
+}

google/adk/tests/integration/fixture/tool_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

google/adk/tests/integration/fixture/tool_agent/agent.py

@@ -0,0 +1,218 @@
+# 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.
+
+import os
+from typing import Any
+
+from crewai_tools import DirectoryReadTool
+from google.adk import Agent
+from google.adk.tools.agent_tool import AgentTool
+from google.adk.tools.crewai_tool import CrewaiTool
+from google.adk.tools.langchain_tool import LangchainTool
+from google.adk.tools.retrieval.files_retrieval import FilesRetrieval
+from google.adk.tools.retrieval.vertex_ai_rag_retrieval import VertexAiRagRetrieval
+from langchain_community.tools import ShellTool
+from pydantic import BaseModel
+
+
+class TestCase(BaseModel):
+  case: str
+
+
+class Test(BaseModel):
+  test_title: list[str]
+
+
+def simple_function(param: str) -> str:
+  if isinstance(param, str):
+    return "Called simple function successfully"
+  return "Called simple function with wrong param type"
+
+
+def no_param_function() -> str:
+  return "Called no param function successfully"
+
+
+def no_output_function(param: str):
+  return
+
+
+def multiple_param_types_function(
+    param1: str, param2: int, param3: float, param4: bool
+) -> str:
+  if (
+      isinstance(param1, str)
+      and isinstance(param2, int)
+      and isinstance(param3, float)
+      and isinstance(param4, bool)
+  ):
+    return "Called multiple param types function successfully"
+  return "Called multiple param types function with wrong param types"
+
+
+def throw_error_function(param: str) -> str:
+  raise ValueError("Error thrown by throw_error_function")
+
+
+def list_str_param_function(param: list[str]) -> str:
+  if isinstance(param, list) and all(isinstance(item, str) for item in param):
+    return "Called list str param function successfully"
+  return "Called list str param function with wrong param type"
+
+
+def return_list_str_function(param: str) -> list[str]:
+  return ["Called return list str function successfully"]
+
+
+def complex_function_list_dict(
+    param1: dict[str, Any], param2: list[dict[str, Any]]
+) -> list[Test]:
+  if (
+      isinstance(param1, dict)
+      and isinstance(param2, list)
+      and all(isinstance(item, dict) for item in param2)
+  ):
+    return [
+        Test(test_title=["function test 1", "function test 2"]),
+        Test(test_title=["retrieval test"]),
+    ]
+  raise ValueError("Wrong param")
+
+
+def repetive_call_1(param: str):
+  return f"Call repetive_call_2 tool with param {param + '_repetive'}"
+
+
+def repetive_call_2(param: str):
+  return param
+
+
+test_case_retrieval = FilesRetrieval(
+    name="test_case_retrieval",
+    description="General guidence for agent test cases",
+    input_dir=os.path.join(os.path.dirname(__file__), "files"),
+)
+
+valid_rag_retrieval = VertexAiRagRetrieval(
+    name="valid_rag_retrieval",
+    rag_corpora=[
+        "projects/1096655024998/locations/us-central1/ragCorpora/4985766262475849728"
+    ],
+    description="General guidence for agent test cases",
+)
+
+invalid_rag_retrieval = VertexAiRagRetrieval(
+    name="invalid_rag_retrieval",
+    rag_corpora=[
+        "projects/1096655024998/locations/us-central1/InValidRagCorporas/4985766262475849728"
+    ],
+    description="Invalid rag retrieval resource name",
+)
+
+non_exist_rag_retrieval = VertexAiRagRetrieval(
+    name="non_exist_rag_retrieval",
+    rag_corpora=[
+        "projects/1096655024998/locations/us-central1/RagCorpora/1234567"
+    ],
+    description="Non exist rag retrieval resource name",
+)
+
+shell_tool = LangchainTool(ShellTool())
+
+docs_tool = CrewaiTool(
+    name="direcotry_read_tool",
+    description="use this to find files for you.",
+    tool=DirectoryReadTool(directory="."),
+)
+
+no_schema_agent = Agent(
+    model="gemini-1.5-flash",
+    name="no_schema_agent",
+    instruction="""Just say 'Hi'
+""",
+)
+
+schema_agent = Agent(
+    model="gemini-1.5-flash",
+    name="schema_agent",
+    instruction="""
+    You will be given a test case.
+    Return a list of the received test case appended with '_success' and '_failure' as test_titles
+""",
+    input_schema=TestCase,
+    output_schema=Test,
+)
+
+no_input_schema_agent = Agent(
+    model="gemini-1.5-flash",
+    name="no_input_schema_agent",
+    instruction="""
+    Just return ['Tools_success, Tools_failure']
+""",
+    output_schema=Test,
+)
+
+no_output_schema_agent = Agent(
+    model="gemini-1.5-flash",
+    name="no_output_schema_agent",
+    instruction="""
+    Just say 'Hi'
+""",
+    input_schema=TestCase,
+)
+
+single_function_agent = Agent(
+    model="gemini-1.5-flash",
+    name="single_function_agent",
+    description="An agent that calls a single function",
+    instruction="When calling tools, just return what the tool returns.",
+    tools=[simple_function],
+)
+
+root_agent = Agent(
+    model="gemini-1.5-flash",
+    name="tool_agent",
+    description="An agent that can call other tools",
+    instruction="When calling tools, just return what the tool returns.",
+    tools=[
+        simple_function,
+        no_param_function,
+        no_output_function,
+        multiple_param_types_function,
+        throw_error_function,
+        list_str_param_function,
+        return_list_str_function,
+        # complex_function_list_dict,
+        repetive_call_1,
+        repetive_call_2,
+        test_case_retrieval,
+        valid_rag_retrieval,
+        invalid_rag_retrieval,
+        non_exist_rag_retrieval,
+        shell_tool,
+        docs_tool,
+        AgentTool(
+            agent=no_schema_agent,
+        ),
+        AgentTool(
+            agent=schema_agent,
+        ),
+        AgentTool(
+            agent=no_input_schema_agent,
+        ),
+        AgentTool(
+            agent=no_output_schema_agent,
+        ),
+    ],
+)

google/adk/tests/integration/fixture/trip_planner_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

google/adk/tests/integration/fixture/trip_planner_agent/agent.py

@@ -0,0 +1,110 @@
+# 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.
+
+# https://github.com/crewAIInc/crewAI-examples/tree/main/trip_planner
+
+from google.adk import Agent
+
+# Agent that selects the best city for the trip.
+identify_agent = Agent(
+    name='identify_agent',
+    description='Select the best city based on weather, season, and prices.',
+    instruction="""
+  Analyze and select the best city for the trip based
+  on specific criteria such as weather patterns, seasonal
+  events, and travel costs. This task involves comparing
+  multiple cities, considering factors like current weather
+  conditions, upcoming cultural or seasonal events, and
+  overall travel expenses.
+
+  Your final answer must be a detailed
+  report on the chosen city, and everything you found out
+  about it, including the actual flight costs, weather
+  forecast and attractions.
+
+  Traveling from: {origin}
+  City Options: {cities}
+  Trip Date: {range}
+  Traveler Interests: {interests}
+""",
+)
+
+# Agent that gathers information about the city.
+gather_agent = Agent(
+    name='gather_agent',
+    description='Provide the BEST insights about the selected city',
+    instruction="""
+  As a local expert on this city you must compile an
+  in-depth guide for someone traveling there and wanting
+  to have THE BEST trip ever!
+  Gather information about key attractions, local customs,
+  special events, and daily activity recommendations.
+  Find the best spots to go to, the kind of place only a
+  local would know.
+  This guide should provide a thorough overview of what
+  the city has to offer, including hidden gems, cultural
+  hotspots, must-visit landmarks, weather forecasts, and
+  high level costs.
+
+  The final answer must be a comprehensive city guide,
+  rich in cultural insights and practical tips,
+  tailored to enhance the travel experience.
+
+  Trip Date: {range}
+  Traveling from: {origin}
+  Traveler Interests: {interests}
+""",
+)
+
+# Agent that plans the trip.
+plan_agent = Agent(
+    name='plan_agent',
+    description="""Create the most amazing travel itineraries with budget and
+    packing suggestions for the city""",
+    instruction="""
+  Expand this guide into a full 7-day travel
+  itinerary with detailed per-day plans, including
+  weather forecasts, places to eat, packing suggestions,
+  and a budget breakdown.
+
+  You MUST suggest actual places to visit, actual hotels
+  to stay and actual restaurants to go to.
+
+  This itinerary should cover all aspects of the trip,
+  from arrival to departure, integrating the city guide
+  information with practical travel logistics.
+
+  Your final answer MUST be a complete expanded travel plan,
+  formatted as markdown, encompassing a daily schedule,
+  anticipated weather conditions, recommended clothing and
+  items to pack, and a detailed budget, ensuring THE BEST
+  TRIP EVER. Be specific and give it a reason why you picked
+  each place, what makes them special!
+
+  Trip Date: {range}
+  Traveling from: {origin}
+  Traveler Interests: {interests}
+""",
+)
+
+root_agent = Agent(
+    model='gemini-2.0-flash-001',
+    name='trip_planner',
+    description='Plan the best trip ever',
+    instruction="""
+  Your goal is to plan the best trip according to information listed above.
+  You describe why did you choose the city, list top 3
+  attactions and provide a detailed itinerary for each day.""",
+    sub_agents=[identify_agent, gather_agent, plan_agent],
+)

google/adk/tests/integration/fixture/trip_planner_agent/initial.session.json

@@ -0,0 +1,13 @@
+{
+  "id": "test_id",
+  "app_name": "trip_planner_agent",
+  "user_id": "test_user",
+  "state": {
+    "origin": "San Francisco",
+    "interests": "Food, Shopping, Museums",
+    "range": "1000 miles",
+    "cities": ""
+  },
+  "events": [],
+  "last_update_time": 1741218714.258285
+}

google/adk/tests/integration/fixture/trip_planner_agent/test_config.json

@@ -0,0 +1,5 @@
+{
+  "criteria": {
+    "response_match_score": 0.5
+  }
+}

google/adk/tests/integration/fixture/trip_planner_agent/test_files/initial.session.json

@@ -0,0 +1,13 @@
+{
+  "id": "test_id",
+  "app_name": "trip_planner_agent",
+  "user_id": "test_user",
+  "state": {
+    "origin": "San Francisco",
+    "interests": "Food, Shopping, Museums",
+    "range": "1000 miles",
+    "cities": ""
+  },
+  "events": [],
+  "last_update_time": 1741218714.258285
+}

google/adk/tests/integration/fixture/trip_planner_agent/test_files/test_config.json

@@ -0,0 +1,5 @@
+{
+  "criteria": {
+    "response_match_score": 0.5
+  }
+}

google/adk/tests/integration/fixture/trip_planner_agent/test_files/trip_inquiry_sub_agent.test.json

@@ -0,0 +1,7 @@
+[
+  {
+    "query": "Based on my interests, where should I go, Yosemite national park or Los Angeles?",
+    "expected_tool_use": [],
+    "reference": "Given your interests in food, shopping, and museums, Los Angeles would be a better choice than Yosemite National Park. Yosemite is primarily focused on outdoor activities and natural landscapes, while Los Angeles offers a diverse range of culinary experiences, shopping districts, and world-class museums. I will now gather information to create an in-depth guide for your trip to Los Angeles.\n"
+  }
+]

google/adk/tests/integration/fixture/trip_planner_agent/trip_inquiry.test.json

@@ -0,0 +1,19 @@
+[
+  {
+    "query": "Hi, who are you? What can you do?",
+    "expected_tool_use": [],
+    "reference": "I am trip_planner, and my goal is to plan the best trip ever. I can describe why a city was chosen, list its top attractions, and provide a detailed itinerary for each day of the trip.\n"
+  },
+  {
+    "query": "I want to travel from San Francisco to an European country in fall next year. I am considering London and Paris. What is your advice?",
+    "expected_tool_use": [
+      {
+        "tool_name": "transfer_to_agent",
+        "tool_input": {
+          "agent_name": "indentify_agent"
+        }
+      }
+    ],
+    "reference": "Okay, I can help you analyze London and Paris to determine which city is better for your trip next fall. I will consider weather patterns, seasonal events, travel costs (including flights from San Francisco), and your interests (food, shopping, and museums). After gathering this information, I'll provide a detailed report on my chosen city.\n"
+  }
+]

google/adk/tests/integration/models/__init__.py

@@ -0,0 +1,14 @@
+# 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.
+

google/adk/tests/integration/models/test_google_llm.py

@@ -0,0 +1,65 @@
+# 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.adk.models import LlmRequest
+from google.adk.models import LlmResponse
+from google.adk.models.google_llm import Gemini
+from google.genai import types
+from google.genai.types import Content
+from google.genai.types import Part
+import pytest
+
+
+@pytest.fixture
+def gemini_llm():
+  return Gemini(model="gemini-1.5-flash")
+
+
+@pytest.fixture
+def llm_request():
+  return LlmRequest(
+      model="gemini-1.5-flash",
+      contents=[Content(role="user", parts=[Part.from_text(text="Hello")])],
+      config=types.GenerateContentConfig(
+          temperature=0.1,
+          response_modalities=[types.Modality.TEXT],
+          system_instruction="You are a helpful assistant",
+      ),
+  )
+
+
+@pytest.mark.asyncio
+async def test_generate_content_async(gemini_llm, llm_request):
+  async for response in gemini_llm.generate_content_async(llm_request):
+    assert isinstance(response, LlmResponse)
+    assert response.content.parts[0].text
+
+
+@pytest.mark.asyncio
+async def test_generate_content_async_stream(gemini_llm, llm_request):
+  responses = [
+      resp
+      async for resp in gemini_llm.generate_content_async(
+          llm_request, stream=True
+      )
+  ]
+  text = ""
+  for i in range(len(responses) - 1):
+    assert responses[i].partial is True
+    assert responses[i].content.parts[0].text
+    text += responses[i].content.parts[0].text
+
+  # Last message should be accumulated text
+  assert responses[-1].content.parts[0].text == text
+  assert not responses[-1].partial