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

google/adk/tests/integration/fixture/flow_complex_spark/sample.session.json

@@ -1,190 +0,0 @@
-{
-  "id": "683bcc98-7359-4b65-9cdb-307f85ae62c0",
-  "context": {
-    "_time": "2024-12-15 11:52:21.958542"
-  },
-  "events": [
-    {
-      "invocation_id": "UfYAgiCT",
-      "author": "user",
-      "content": {
-        "parts": [
-          {
-            "text": "Research on the factors that affect floods in Florida. "
-          }
-        ],
-        "role": "user"
-      },
-      "options": {},
-      "is_greeting": false,
-      "id": "qKEz3Vss",
-      "timestamp": 1734292270.899412
-    },
-    {
-      "invocation_id": "UfYAgiCT",
-      "author": "spark_assistant",
-      "content": {
-        "parts": [
-          {
-            "function_call": {
-              "args": {
-                "agent_name": "research_assistant"
-              },
-              "name": "transfer_to_other_agent"
-            }
-          }
-        ],
-        "role": "model"
-      },
-      "options": {},
-      "is_greeting": false,
-      "id": "MEQQWBGe",
-      "timestamp": 1734292272.522898
-    },
-    {
-      "invocation_id": "UfYAgiCT",
-      "author": "spark_assistant",
-      "content": {
-        "parts": [
-          {
-            "function_response": {
-              "name": "transfer_to_other_agent",
-              "response": {
-                "status": "ok",
-                "target_agent_name": "research_assistant",
-                "message": "Transfered to research_assistant"
-              }
-            }
-          }
-        ],
-        "role": "user"
-      },
-      "options": {
-        "skip_summarization": true,
-        "function_call_event_id": "MEQQWBGe"
-      },
-      "is_greeting": false,
-      "id": "QjLptHaG",
-      "timestamp": 1734292272.524573
-    },
-    {
-      "invocation_id": "UfYAgiCT",
-      "author": "research_plan_agent",
-      "content": {
-        "parts": [
-          {
-            "text": "[research_plan_agent]:  I can help you with that! Here is a research plan for the factors that affect floods in Florida:\n\n**Factors that affect floods in Florida**\nI will research the factors that affect floods in Florida, such as rainfall, sea level rise, and urbanization.\n\n**Rainfall in Florida**\nI will research the average rainfall in Florida, as well as the frequency and intensity of rainfall events.\n\n**Sea Level Rise in Florida**\nI will research the rate of sea level rise in Florida, as well as the impact of sea level rise on flooding.\n\n**Urbanization in Florida**\nI will research the impact of urbanization on flooding in Florida, such as the effects of impervious surfaces and drainage systems.\n\n**Flood History in Florida**\nI will research the history of floods in Florida, including the frequency, severity, and causes of past floods.\n\n**Flood Mitigation Efforts in Florida**\nI will research the flood mitigation efforts that are currently in place in Florida, as well as the effectiveness of these efforts.\n\nI will start researching now. \n"
-          }
-        ],
-        "role": "model"
-      },
-      "options": {},
-      "is_greeting": false,
-      "id": "NcWJjpm9",
-      "timestamp": 1734292274.707415
-    },
-    {
-      "invocation_id": "UfYAgiCT",
-      "author": "question_generation_agent",
-      "content": {
-        "parts": [
-          {
-            "text": "1. [question_generation_agent]: What are the specific types of rainfall events that contribute to flooding in Florida?\n2. [question_generation_agent]: How does the rate of sea level rise in Florida compare to the global average?\n3. [question_generation_agent]: What are the specific ways that urbanization in Florida contributes to increased flooding?\n4. [question_generation_agent]: Are there any specific regions in Florida that are particularly vulnerable to flooding?\n5. [question_generation_agent]: What are the most significant flood events in Florida's history, and what were their causes?\n6. [question_generation_agent]: What are the different types of flood mitigation efforts currently in place in Florida?\n7. [question_generation_agent]: How effective have these flood mitigation efforts been in reducing flood risk in Florida?\n8. [question_generation_agent]: What are the potential future impacts of climate change on flooding in Florida?\n9. [question_generation_agent]: What are the economic and social consequences of flooding in Florida?\n10. [question_generation_agent]: What are the recommendations for improving flood mitigation efforts in Florida? \n"
-          }
-        ],
-        "role": "model"
-      },
-      "options": {},
-      "is_greeting": false,
-      "id": "9DZFuhg7",
-      "timestamp": 1734292277.283785
-    },
-    {
-      "invocation_id": "UfYAgiCT",
-      "author": "information_retrieval_agent",
-      "content": {
-        "parts": [
-          {
-            "text": "[information_retrieval_agent]: **No. 1: Types of Rainfall Events Contributing to Flooding in Florida**\n\nFlorida experiences various types of rainfall events that contribute to flooding. These include:\n\n* **Tropical Storms and Hurricanes:** These storms bring heavy rainfall over extended periods, often exceeding the capacity of drainage systems and leading to widespread flooding.\n* **Thunderstorms:** While less intense than hurricanes, thunderstorms can produce heavy downpours in a short time, causing localized flooding, especially in urban areas with limited drainage.\n* **Frontal Systems:** Cold fronts moving across Florida can bring significant rainfall, particularly during the winter months, leading to flooding in low-lying areas.\n* **El Niño-Southern Oscillation (ENSO):** This climate pattern can influence rainfall patterns in Florida, with El Niño years often associated with increased rainfall and flooding.\n\n[information_retrieval_agent]: **No. 2: Sea Level Rise Rate in Florida Compared to Global Average**\n\nFlorida's sea level rise rate is significantly higher than the global average. While the global average sea level rise is about 3.4 millimeters per year, Florida's rate is estimated to be around 9 millimeters per year. This accelerated rate is attributed to factors such as land subsidence and the Gulf Stream's influence.\n\n[information_retrieval_agent]: **No. 3: Urbanization's Impact on Flooding in Florida**\n\nUrbanization in Florida contributes to increased flooding through several mechanisms:\n\n* **Impervious Surfaces:** Concrete and asphalt surfaces in urban areas reduce infiltration, forcing rainwater to run off into drainage systems, which can become overwhelmed during heavy rainfall.\n* **Modified Drainage Patterns:** Urban development often alters natural drainage patterns, leading to concentrated runoff and increased flood risk in certain areas.\n* **Increased Runoff Velocity:** Urban surfaces accelerate runoff, increasing the volume and speed of water flowing into drainage systems, potentially exceeding their capacity.\n\n[information_retrieval_agent]: **No. 4: Regions in Florida Particularly Vulnerable to Flooding**\n\nSeveral regions in Florida are particularly vulnerable to flooding due to their geographic characteristics and exposure to various flood risks:\n\n* **Coastal Areas:** Coastal areas are susceptible to storm surge, sea level rise, and coastal erosion, increasing their vulnerability to flooding.\n* **Low-Lying Areas:** Areas with low elevation are prone to flooding from heavy rainfall and rising sea levels.\n* **Urban Areas:** Urban areas with limited drainage capacity and high concentrations of impervious surfaces are at risk of flooding during heavy rainfall events.\n\n[information_retrieval_agent]: **No. 5: Significant Flood Events in Florida's History**\n\nFlorida has experienced numerous significant flood events throughout its history, with some notable examples:\n\n* **Hurricane Andrew (1992):** This Category 5 hurricane caused widespread flooding in South Florida, primarily due to storm surge and heavy rainfall.\n* **Hurricane Irma (2017):** This Category 4 hurricane brought significant rainfall and storm surge, leading to extensive flooding across the state.\n* **Hurricane Michael (2018):** This Category 5 hurricane caused significant flooding in the Panhandle region, primarily due to heavy rainfall and storm surge.\n\n[information_retrieval_agent]: **No. 6: Flood Mitigation Efforts in Florida**\n\nFlorida has implemented various flood mitigation efforts to reduce flood risk and protect communities:\n\n* **Levees and Seawalls:** These structures are designed to protect coastal areas from storm surge and rising sea levels.\n* **Drainage Systems:** Drainage systems are designed to collect and remove excess rainwater, reducing the risk of flooding in urban areas.\n* **Floodplain Management:** Regulations are in place to restrict development in floodplains, minimizing the risk of damage and loss of life.\n* **Early Warning Systems:** Advanced weather forecasting and warning systems provide timely alerts to residents, allowing them to prepare for potential flooding.\n\n[information_retrieval_agent]: **No. 7: Effectiveness of Flood Mitigation Efforts in Florida**\n\nThe effectiveness of flood mitigation efforts in Florida varies depending on the specific measures implemented and the severity of the flood event. While some efforts have been successful in reducing flood risk, others have proven less effective, particularly in the face of extreme weather events.\n\n[information_retrieval_agent]: **No. 8: Potential Future Impacts of Climate Change on Flooding in Florida**\n\nClimate change is expected to exacerbate flooding in Florida through several mechanisms:\n\n* **Increased Rainfall Intensity:** Climate change is projected to increase the frequency and intensity of rainfall events, leading to more frequent and severe flooding.\n* **Sea Level Rise:** Continued sea level rise will increase the risk of coastal flooding, particularly during storm surge events.\n* **Ocean Warming:** Warmer ocean temperatures can lead to more intense hurricanes, further increasing the risk of flooding.\n\n[information_retrieval_agent]: **No. 9: Economic and Social Consequences of Flooding in Florida**\n\nFlooding in Florida has significant economic and social consequences:\n\n* **Property Damage:** Flooding can cause extensive damage to homes, businesses, and infrastructure, resulting in substantial financial losses.\n* **Disruption of Transportation:** Flooding can disrupt transportation systems, including roads, bridges, and airports, leading to economic losses and social disruption.\n* **Health Risks:** Flooding can create health risks, such as waterborne diseases and injuries from debris.\n* **Displacement of Residents:** Flooding can force residents to evacuate their homes, leading to displacement and social disruption.\n\n[information_retrieval_agent]: **No. 10: Recommendations for Improving Flood Mitigation Efforts in Florida**\n\nTo improve flood mitigation efforts in Florida, several recommendations can be considered:\n\n* **Invest in Infrastructure:** Increased investment in drainage systems, levees, and seawalls is crucial to enhance flood protection.\n* **Promote Sustainable Development:** Encourage development practices that minimize impervious surfaces and preserve natural drainage patterns.\n* **Strengthen Early Warning Systems:** Enhance weather forecasting and warning systems to provide timely and accurate alerts to residents.\n* **Implement Comprehensive Floodplain Management:** Enforce strict regulations to restrict development in floodplains and minimize flood risk.\n* **Promote Public Awareness:** Educate residents about flood risks and encourage them to take steps to prepare for potential flooding.\n\n[exit] \n"
-          }
-        ],
-        "role": "model"
-      },
-      "options": {},
-      "is_greeting": false,
-      "id": "o11CKE8Q",
-      "timestamp": 1734292284.446259
-    },
-    {
-      "invocation_id": "UfYAgiCT",
-      "author": "summary_agent",
-      "content": {
-        "parts": [
-          {
-            "text": "## Flooding in Florida: A Comprehensive Analysis of Causes, Impacts, and Mitigation Strategies\n\n**Abstract:** This essay explores the complex issue of flooding in Florida, a state highly vulnerable to various flood threats. It examines the key factors contributing to flooding, including rainfall patterns, sea level rise, and urbanization, and analyzes their historical impact.  Furthermore, it evaluates the effectiveness of existing flood mitigation efforts and explores recommendations for future improvements. \n\n**Rainfall Patterns and Flood Risk:**\n\nFlorida's climate is characterized by frequent and intense rainfall events, significantly contributing to flood risk.  Tropical storms and hurricanes, particularly during the hurricane season, bring heavy rainfall over prolonged periods, overwhelming drainage systems and causing widespread flooding.  Thunderstorms, though less intense, can generate heavy downpours in short bursts, leading to localized flooding, especially in urban areas.  Furthermore, frontal systems, particularly during winter months, can bring substantial rainfall, exacerbating flooding in low-lying areas. El Niño-Southern Oscillation (ENSO) events can also influence rainfall patterns, with El Niño years often associated with increased rainfall and subsequent flooding.\n\n**Sea Level Rise and Coastal Flooding:**\n\nFlorida's sea level rise rate surpasses the global average, posing a significant threat to coastal areas.  The rate of sea level rise in Florida is estimated to be around 9 millimeters per year, compared to the global average of 3.4 millimeters per year.  This accelerated rise is attributed to factors such as land subsidence and the Gulf Stream's influence.  As sea levels continue to rise, coastal areas face an increased risk of flooding, particularly during storm surge events, leading to erosion, saltwater intrusion, and potential displacement of coastal communities. \n\n**Urbanization and Flood Risk:**\n\nUrbanization in Florida, while contributing to economic growth, also exacerbates flood risk.  The increasing development of impervious surfaces, such as concrete and asphalt, reduces rainwater infiltration, forcing excess water to flow into drainage systems. This can overwhelm the capacity of drainage systems, leading to flooding in urban areas.  Furthermore, urban development often alters natural drainage patterns, diverting water flow and increasing flooding risks in specific locations.  Finally, urban surfaces accelerate runoff velocity, increasing the volume and speed of water entering drainage systems, further enhancing the risk of flooding.\n\n**Historical Flood Events and Their Impact:**\n\nFlorida has experienced numerous significant flood events throughout its history, highlighting the state's vulnerability.  Hurricane Andrew in 1992, a Category 5 hurricane, caused widespread flooding in South Florida, primarily due to storm surge and heavy rainfall.  Hurricane Irma in 2017, a Category 4 hurricane, brought significant rainfall and storm surge, leading to extensive flooding across the state.  Hurricane Michael in 2018, a Category 5 hurricane, caused substantial flooding in the Panhandle region, primarily due to heavy rainfall and storm surge.  These events demonstrate the devastating impact of flooding on infrastructure, property, and human lives. \n\n**Flood Mitigation Efforts and Their Effectiveness:**\n\nFlorida has implemented various flood mitigation efforts to reduce flood risk and protect communities.  These efforts include construction of levees and seawalls to protect coastal areas from storm surge and rising sea levels, construction of drainage systems to collect and remove excess rainwater, implementation of floodplain management regulations to restrict development in floodplains, and development of early warning systems to provide timely alerts to residents.  While these efforts have contributed to reducing flood risk in some areas, their effectiveness varies depending on the severity of the flood event.  Extreme weather events, such as hurricanes, can overwhelm existing infrastructure and mitigation measures, highlighting the need for continuous improvement and adaptation.\n\n**Future Impacts of Climate Change on Flooding:**\n\nClimate change is expected to exacerbate flooding in Florida, posing significant challenges for the state's future.  Increased rainfall intensity, driven by climate change, is projected to lead to more frequent and severe flooding events.  Continued sea level rise will further increase the risk of coastal flooding, especially during storm surge events.  Ocean warming, a consequence of climate change, can contribute to more intense hurricanes, further amplifying flood risk.  These compounding factors necessitate proactive measures to adapt to changing climate conditions and mitigate future flood risks. \n\n**Economic and Social Consequences of Flooding:**\n\nFlooding in Florida carries significant economic and social consequences.  Property damage from flooding can be extensive, resulting in substantial financial losses for individuals, businesses, and the state.  Flooding disrupts transportation systems, including roads, bridges, and airports, causing economic losses and social disruption.  Flooding can also create health risks, such as waterborne diseases and injuries from debris.  Furthermore, displacement of residents due to flooding can lead to social disruption and strain on existing resources.  The economic and social costs of flooding underscore the importance of effective flood mitigation and adaptation strategies.\n\n**Recommendations for Improving Flood Mitigation Efforts:**\n\nTo improve flood mitigation efforts in Florida and address the challenges posed by climate change, several recommendations are crucial.  Increased investment in infrastructure, including drainage systems, levees, and seawalls, is essential to enhance flood protection.  Promoting sustainable development practices, such as minimizing impervious surfaces and preserving natural drainage patterns, can reduce runoff and mitigate flood risk.  Strengthening early warning systems to provide timely and accurate alerts to residents is crucial for preparedness and response.  Implementing comprehensive floodplain management regulations to restrict development in floodplains and minimize flood risk is essential.  Finally, promoting public awareness about flood risks and encouraging residents to take steps to prepare for potential flooding is vital for community resilience.\n\n**Conclusion:**\n\nFlooding in Florida is a complex issue with multifaceted causes, impacts, and mitigation strategies.  Rainfall patterns, sea level rise, and urbanization all contribute to flood risk, while historical events highlight the state's vulnerability.  Existing flood mitigation efforts, while valuable, need continuous improvement and adaptation to address the challenges posed by climate change.  Investing in infrastructure, promoting sustainable development, strengthening early warning systems, implementing comprehensive floodplain management, and fostering public awareness are key recommendations for enhancing flood resilience and safeguarding Florida's communities from the devastating impacts of flooding. \n"
-          }
-        ],
-        "role": "model"
-      },
-      "options": {},
-      "is_greeting": false,
-      "id": "om1YdTV1",
-      "timestamp": 1734292291.449593
-    },
-    {
-      "invocation_id": "UfYAgiCT",
-      "author": "spark_assistant",
-      "content": {
-        "parts": [
-          {
-            "function_call": {
-              "args": {
-                "agent_name": "spark_assistant"
-              },
-              "name": "transfer_to_other_agent"
-            }
-          }
-        ],
-        "role": "model"
-      },
-      "options": {
-        "skip_summarization": true
-      },
-      "is_greeting": false,
-      "id": "oRxzw1Wp",
-      "timestamp": 1734292291.451956
-    },
-    {
-      "invocation_id": "UfYAgiCT",
-      "author": "spark_assistant",
-      "content": {
-        "parts": [
-          {
-            "function_response": {
-              "name": "transfer_to_other_agent",
-              "response": {
-                "status": "ok",
-                "target_agent_name": "spark_assistant",
-                "message": "Transfered to spark_assistant"
-              }
-            }
-          }
-        ],
-        "role": "user"
-      },
-      "options": {
-        "skip_summarization": true,
-        "function_call_event_id": "oRxzw1Wp"
-      },
-      "is_greeting": false,
-      "id": "s2OiE3PP",
-      "timestamp": 1734292291.453157
-    }
-  ],
-  "past_events": [],
-  "current_agent_name": "spark_assistant",
-  "pending_events": {},
-  "artifacts": {},
-  "last_update_time": 1734292291.453157,
-  "event_logs": []
-}

google/adk/tests/integration/fixture/hello_world_agent/__init__.py

@@ -1,15 +0,0 @@
-# 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/hello_world_agent/agent.py

@@ -1,95 +0,0 @@
-# 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.
-
-# Hello world agent from agent 1.0 - https://colab.sandbox.google.com/drive/1Zq-nqmgK0nCERCv8jKIaoeTTgbNn6oSo?resourcekey=0-GYaz9pFT4wY8CI8Cvjy5GA#scrollTo=u3X3XwDOaCv9
-import random
-
-from google.adk import Agent
-from google.genai import types
-
-
-def roll_die(sides: int) -> int:
-  """Roll a die and return the rolled result.
-
-  Args:
-    sides: The integer number of sides the die has.
-
-  Returns:
-    An integer of the result of rolling the die.
-  """
-  return random.randint(1, sides)
-
-
-def check_prime(nums: list[int]) -> list[str]:
-  """Check if a given list of numbers are prime.
-
-  Args:
-    nums: The list of numbers to check.
-
-  Returns:
-    A str indicating which number is prime.
-  """
-  primes = set()
-  for number in nums:
-    number = int(number)
-    if number <= 1:
-      continue
-    is_prime = True
-    for i in range(2, int(number**0.5) + 1):
-      if number % i == 0:
-        is_prime = False
-        break
-    if is_prime:
-      primes.add(number)
-  return (
-      'No prime numbers found.'
-      if not primes
-      else f"{', '.join(str(num) for num in primes)} are prime numbers."
-  )
-
-
-root_agent = Agent(
-    model='gemini-2.0-flash-001',
-    name='data_processing_agent',
-    instruction="""
-      You roll dice and answer questions about the outcome of the dice rolls.
-      You can roll dice of different sizes.
-      You can use multiple tools in parallel by calling functions in parallel(in one request and in one round).
-      The only things you do are roll dice for the user and discuss the outcomes.
-      It is ok to discuss previous dice roles, and comment on the dice rolls.
-      When you are asked to roll a die, you must call the roll_die tool with the number of sides. Be sure to pass in an integer. Do not pass in a string.
-      You should never roll a die on your own.
-      When checking prime numbers, call the check_prime tool with a list of integers. Be sure to pass in a list of integers. You should never pass in a string.
-      You should not check prime numbers before calling the tool.
-      When you are asked to roll a die and check prime numbers, you should always make the following two function calls:
-      1. You should first call the roll_die tool to get a roll. Wait for the function response before calling the check_prime tool.
-      2. After you get the function response from roll_die tool, you should call the check_prime tool with the roll_die result.
-        2.1 If user asks you to check primes based on previous rolls, make sure you include the previous rolls in the list.
-      3. When you respond, you must include the roll_die result from step 1.
-      You should always perform the previous 3 steps when asking for a roll and checking prime numbers.
-      You should not rely on the previous history on prime results.
-    """,
-    tools=[
-        roll_die,
-        check_prime,
-    ],
-    generate_content_config=types.GenerateContentConfig(
-        safety_settings=[
-            types.SafetySetting(  # avoid false alarm about rolling dice.
-                category=types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
-                threshold=types.HarmBlockThreshold.OFF,
-            ),
-        ]
-    ),
-)

google/adk/tests/integration/fixture/hello_world_agent/roll_die.test.json

@@ -1,24 +0,0 @@
-[
-  {
-    "query": "Hi who are you?",
-    "expected_tool_use": [],
-    "reference": "I am a data processing agent. I can roll dice and check if the results are prime numbers. What would you like me to do? \n"
-  },
-  {
-    "query": "What can you do?",
-    "expected_tool_use": [],
-    "reference": "I can roll dice for you of different sizes, and I can check if the results are prime numbers.  I can also remember previous rolls if you'd like to check those for primes as well.  What would you like me to do? \n"
-  },
-  {
-    "query": "Can you roll a die with 6 sides",
-    "expected_tool_use": [
-      {
-        "tool_name": "roll_die",
-        "tool_input": {
-          "sides": 6
-        }
-      }
-    ],
-    "reference": null
-  }
-]

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

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

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

@@ -1,15 +0,0 @@
-# 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/home_automation_agent/agent.py

@@ -1,304 +0,0 @@
-# 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,
-    ],
-)