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

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

@@ -0,0 +1,304 @@
+# 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
+import sys
+
+from google.adk import Agent
+
+DEVICE_DB = {
+    "device_1": {"status": "ON", "location": "Living Room"},
+    "device_2": {"status": "OFF", "location": "Bedroom"},
+    "device_3": {"status": "OFF", "location": "Kitchen"},
+}
+
+TEMPERATURE_DB = {
+    "Living Room": 22,
+    "Bedroom": 20,
+    "Kitchen": 24,
+}
+
+SCHEDULE_DB = {
+    "device_1": {"time": "18:00", "status": "ON"},
+    "device_2": {"time": "22:00", "status": "OFF"},
+}
+
+USER_PREFERENCES_DB = {
+    "user_x": {"preferred_temp": 21, "location": "Bedroom"},
+    "user_x": {"preferred_temp": 21, "location": "Living Room"},
+    "user_y": {"preferred_temp": 23, "location": "Living Room"},
+}
+
+
+def reset_data():
+  global DEVICE_DB
+  global TEMPERATURE_DB
+  global SCHEDULE_DB
+  global USER_PREFERENCES_DB
+  DEVICE_DB = {
+      "device_1": {"status": "ON", "location": "Living Room"},
+      "device_2": {"status": "OFF", "location": "Bedroom"},
+      "device_3": {"status": "OFF", "location": "Kitchen"},
+  }
+
+  TEMPERATURE_DB = {
+      "Living Room": 22,
+      "Bedroom": 20,
+      "Kitchen": 24,
+  }
+
+  SCHEDULE_DB = {
+      "device_1": {"time": "18:00", "status": "ON"},
+      "device_2": {"time": "22:00", "status": "OFF"},
+  }
+
+  USER_PREFERENCES_DB = {
+      "user_x": {"preferred_temp": 21, "location": "Bedroom"},
+      "user_x": {"preferred_temp": 21, "location": "Living Room"},
+      "user_y": {"preferred_temp": 23, "location": "Living Room"},
+  }
+
+
+def get_device_info(device_id: str) -> dict:
+  """Get the current status and location of a AC device.
+
+  Args:
+      device_id (str): The unique identifier of the device.
+
+  Returns:
+      dict: A dictionary containing the following fields, or 'Device not found'
+      if the device_id does not exist:
+        - status: The current status of the device (e.g., 'ON', 'OFF')
+        - location: The location where the device is installed (e.g., 'Living
+        Room', 'Bedroom', ''Kitchen')
+  """
+  return DEVICE_DB.get(device_id, "Device not found")
+
+
+# def set_device_info(device_id: str, updates: dict) -> str:
+# """Update the information of a AC device, specifically its status and/or location.
+
+# Args:
+#     device_id (str): Required. The unique identifier of the device.
+#     updates (dict): Required. A dictionary containing the fields to be
+#       updated. Supported keys: - "status" (str): The new status to set for the
+#       device. Accepted values: 'ON', 'OFF'. **Only these values are allowed.**
+#       - "location" (str): The new location to set for the device. Accepted
+#       values: 'Living Room', 'Bedroom', 'Kitchen'. **Only these values are
+#         allowed.**
+
+
+# Returns:
+#     str: A message indicating whether the device information was successfully
+#     updated.
+# """
+# if device_id in DEVICE_DB:
+#   if "status" in updates:
+#     DEVICE_DB[device_id]["status"] = updates["status"]
+#   if "location" in updates:
+#     DEVICE_DB[device_id]["location"] = updates["location"]
+#   return f"Device {device_id} information updated: {updates}."
+# return "Device not found"
+def set_device_info(
+    device_id: str, status: str = "", location: str = ""
+) -> str:
+  """Update the information of a AC device, specifically its status and/or location.
+
+  Args:
+      device_id (str): Required. The unique identifier of the device.
+      status (str): The new status to set for the
+        device. Accepted values: 'ON', 'OFF'. **Only these values are allowed.**
+      location (str): The new location to set for the device. Accepted
+        values: 'Living Room', 'Bedroom', 'Kitchen'. **Only these values are
+          allowed.**
+
+  Returns:
+      str: A message indicating whether the device information was successfully
+      updated.
+  """
+  if device_id in DEVICE_DB:
+    if status:
+      DEVICE_DB[device_id]["status"] = status
+      return f"Device {device_id} information updated: status -> {status}."
+    if location:
+      DEVICE_DB[device_id]["location"] = location
+      return f"Device {device_id} information updated: location -> {location}."
+  return "Device not found"
+
+
+def get_temperature(location: str) -> int:
+  """Get the current temperature in celsius of a location (e.g., 'Living Room', 'Bedroom', 'Kitchen').
+
+  Args:
+      location (str): The location for which to retrieve the temperature (e.g.,
+        'Living Room', 'Bedroom', 'Kitchen').
+
+  Returns:
+      int: The current temperature in celsius in the specified location, or
+      'Location not found' if the location does not exist.
+  """
+  return TEMPERATURE_DB.get(location, "Location not found")
+
+
+def set_temperature(location: str, temperature: int) -> str:
+  """Set the desired temperature in celsius for a location.
+
+  Acceptable range of temperature: 18-30 celsius. If it's out of the range, do
+  not call this tool.
+
+  Args:
+      location (str): The location where the temperature should be set.
+      temperature (int): The desired temperature as integer to set in celsius.
+        Acceptable range: 18-30 celsius.
+
+  Returns:
+      str: A message indicating whether the temperature was successfully set.
+  """
+  if location in TEMPERATURE_DB:
+    TEMPERATURE_DB[location] = temperature
+    return f"Temperature in {location} set to {temperature}°C."
+  return "Location not found"
+
+
+def get_user_preferences(user_id: str) -> dict:
+  """Get the temperature preferences and preferred location of a user_id.
+
+  user_id must be provided.
+
+  Args:
+      user_id (str): The unique identifier of the user.
+
+  Returns:
+      dict: A dictionary containing the following fields, or 'User not found' if
+      the user_id does not exist:
+        - preferred_temp: The user's preferred temperature.
+        - location: The location where the user prefers to be.
+  """
+  return USER_PREFERENCES_DB.get(user_id, "User not found")
+
+
+def set_device_schedule(device_id: str, time: str, status: str) -> str:
+  """Schedule a device to change its status at a specific time.
+
+  Args:
+      device_id (str): The unique identifier of the device.
+      time (str): The time at which the device should change its status (format:
+        'HH:MM').
+      status (str): The status to set for the device at the specified time
+        (e.g., 'ON', 'OFF').
+
+  Returns:
+      str: A message indicating whether the schedule was successfully set.
+  """
+  if device_id in DEVICE_DB:
+    SCHEDULE_DB[device_id] = {"time": time, "status": status}
+    return f"Device {device_id} scheduled to turn {status} at {time}."
+  return "Device not found"
+
+
+def get_device_schedule(device_id: str) -> dict:
+  """Retrieve the schedule of a device.
+
+  Args:
+      device_id (str): The unique identifier of the device.
+
+  Returns:
+      dict: A dictionary containing the following fields, or 'Schedule not
+      found' if the device_id does not exist:
+        - time: The scheduled time for the device to change its status (format:
+        'HH:MM').
+        - status: The status that will be set at the scheduled time (e.g., 'ON',
+        'OFF').
+  """
+  return SCHEDULE_DB.get(device_id, "Schedule not found")
+
+
+def celsius_to_fahrenheit(celsius: int) -> float:
+  """Convert Celsius to Fahrenheit.
+
+  You must call this to do the conversion of temperature, so you can get the
+  precise number in required format.
+
+  Args:
+      celsius (int): Temperature in Celsius.
+
+  Returns:
+      float: Temperature in Fahrenheit.
+  """
+  return (celsius * 9 / 5) + 32
+
+
+def fahrenheit_to_celsius(fahrenheit: float) -> int:
+  """Convert Fahrenheit to Celsius.
+
+  You must call this to do the conversion of temperature, so you can get the
+  precise number in required format.
+
+  Args:
+      fahrenheit (float): Temperature in Fahrenheit.
+
+  Returns:
+      int: Temperature in Celsius.
+  """
+  return int((fahrenheit - 32) * 5 / 9)
+
+
+def list_devices(status: str = "", location: str = "") -> list:
+  """Retrieve a list of AC devices, filtered by status and/or location when provided.
+
+  For cost efficiency, always apply as many filters (status and location) as
+  available in the input arguments.
+
+  Args:
+      status (str, optional): The status to filter devices by (e.g., 'ON',
+        'OFF'). Defaults to None.
+      location (str, optional): The location to filter devices by (e.g., 'Living
+        Room', 'Bedroom', ''Kitchen'). Defaults to None.
+
+  Returns:
+      list: A list of dictionaries, each containing the device ID, status, and
+      location, or an empty list if no devices match the criteria.
+  """
+  devices = []
+  for device_id, info in DEVICE_DB.items():
+    if ((not status) or info["status"] == status) and (
+        (not location) or info["location"] == location
+    ):
+      devices.append({
+          "device_id": device_id,
+          "status": info["status"],
+          "location": info["location"],
+      })
+  return devices if devices else "No devices found matching the criteria."
+
+
+root_agent = Agent(
+    model="gemini-2.0-flash-001",
+    name="Home_automation_agent",
+    instruction="""
+    You are Home Automation Agent. You are responsible for controlling the devices in the home.
+    """,
+    tools=[
+        get_device_info,
+        set_device_info,
+        get_temperature,
+        set_temperature,
+        get_user_preferences,
+        set_device_schedule,
+        get_device_schedule,
+        celsius_to_fahrenheit,
+        fahrenheit_to_celsius,
+        list_devices,
+    ],
+)

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