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

google/adk/tests/integration/fixture/ecommerce_customer_service_agent/order_query.test.json

@@ -0,0 +1,69 @@
+[
+  {
+    "query": "Send an email to user user_a whose email address is alice@example.com",
+    "expected_tool_use": [
+      {
+        "tool_name": "send_email",
+        "tool_input": {
+          "email": "alice@example.com",
+          "user_id": "user_a"
+        }
+      }
+    ],
+    "reference": "Email sent to alice@example.com for user id user_a."
+  },
+  {
+    "query": "Can you tell me the status of my order with ID 1?",
+    "expected_tool_use": [
+      {
+        "tool_name": "get_order_status",
+        "tool_input": {
+          "order_id": "1"
+        }
+      }
+    ],
+    "reference": "Your order with ID 1 is FINISHED."
+  },
+  {
+    "query": "Cancel all pending order for the user with user id user_a",
+    "expected_tool_use": [
+      {
+        "tool_name": "get_order_ids_for_user",
+        "tool_input": {
+          "user_id": "user_a"
+        }
+      },
+      {
+        "tool_name": "get_order_status",
+        "tool_input": {
+          "order_id": "1"
+        }
+      },
+      {
+        "tool_name": "get_order_status",
+        "tool_input": {
+          "order_id": "4"
+        }
+      },
+      {
+        "tool_name": "cancel_order",
+        "tool_input": {
+          "order_id": "4"
+        }
+      }
+    ],
+    "reference": "I have checked your orders and order 4 was in pending status, so I have cancelled it. Order 1 was already finished and couldn't be cancelled.\n"
+  },
+  {
+    "query": "What orders have I placed under the username user_b?",
+    "expected_tool_use": [
+      {
+        "tool_name": "get_order_ids_for_user",
+        "tool_input": {
+          "user_id": "user_b"
+        }
+      }
+    ],
+    "reference": "User user_b has placed one order with order ID 2.\n"
+  }
+]

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

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

google/adk/tests/integration/fixture/flow_complex_spark/__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/flow_complex_spark/agent.py

@@ -0,0 +1,182 @@
+# 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 import Agent
+from google.genai import types
+
+research_plan_agent = Agent(
+    model="gemini-1.5-flash",
+    name="research_plan_agent",
+    description="I can help generate research plan.",
+    instruction="""\
+Your task is to create a research plan according to the user's query.
+
+# Here are the instructions for creating the research plan:
+
++ Focus on finding specific things, e.g. products, data, etc.
++ Have the personality of a work colleague that is very helpful and explains things very nicely.
++ Don't mention your name unless you are asked.
++ Think about the most common things that you would need to research.
++ Think about possible answers when creating the plan.
++ Your task is to create the sections that should be researched. You will output high level headers, preceded by ##
++ Underneath each header, write a short sentence on what we want to find there.
++ The headers will follow the logical analysis pattern, as well as logical exploration pattern.
++ The headers should be a statement, not be in the form of questions.
++ The header will not include roman numerals or anything of the sort, e.g. ":", etc
++ Do not include things that you cannot possibly know about from using Google Search: e.g. sales forecasting, competitors, profitability analysis, etc.
++ Do not have an executive summary
++ In each section describe specifically what will be researched.
++ Never use "we will", but rather "I will".
++ Don't ask for clarifications from the user.
++ Do not ask the user for clarifications or if they have any other questions.
++ All headers should be bolded.
++ If you have steps in the plan that depend on other information, make sure they are 2 diferent sections in the plan.
++ At the end mention that you will start researching.
+
+# Instruction on replying format
+
++ Start with your name as "[research_plan_agent]: ".
++ Output the content you want to say.
+
+Output summary:
+""",
+    flow="single",
+    sub_agents=[],
+    generate_content_config=types.GenerateContentConfig(
+        temperature=0.1,
+    ),
+)
+
+
+question_generation_agent = Agent(
+    model="gemini-1.5-flash",
+    name="question_generation_agent",
+    description="I can help generate questions related to user's question.",
+    instruction="""\
+Generate questions related to the research plan generated by research_plan_agent.
+
+# Instruction on replying format
+
+Your reply should be a numbered lsit.
+
+For each question, reply in the following format: "[question_generation_agent]: [generated questions]"
+
+Here is an example of the generated question list:
+
+1. [question_generation_agent]: which state is San Jose in?
+2. [question_generation_agent]: how google website is designed?
+""",
+    flow="single",
+    sub_agents=[],
+    generate_content_config=types.GenerateContentConfig(
+        temperature=0.1,
+    ),
+)
+
+information_retrieval_agent = Agent(
+    model="gemini-1.5-flash",
+    name="information_retrieval_agent",
+    description=(
+        "I can help retrieve information related to question_generation_agent's"
+        " question."
+    ),
+    instruction="""\
+Inspect all the questions after "[question_generation_agent]: " and asnwer them.
+
+# Instruction on replying format
+
+Always start with "[information_retrieval_agent]: "
+
+For the answer of one question:
+
+- Start with a title with one line summary of the reply.
+- The title line should be bolded and starts with No.x of the corresponding question.
+- Have a paragraph of detailed explain.
+
+# Instruction on exiting the loop
+
+- If you see there are less than 20 questions by "question_generation_agent", do not say "[exit]".
+- If you see there are already great or equal to 20 questions asked by "question_generation_agent", say "[exit]" at last to exit the loop.
+""",
+    flow="single",
+    sub_agents=[],
+    generate_content_config=types.GenerateContentConfig(
+        temperature=0.1,
+    ),
+)
+
+question_sources_generation_agent = Agent(
+    model="gemini-1.5-flash",
+    name="question_sources_generation_agent",
+    description=(
+        "I can help generate questions and retrieve related information."
+    ),
+    instruction="Generate questions and retrieve information.",
+    flow="loop",
+    sub_agents=[
+        question_generation_agent,
+        information_retrieval_agent,
+    ],
+    generate_content_config=types.GenerateContentConfig(
+        temperature=0.1,
+    ),
+)
+
+summary_agent = Agent(
+    model="gemini-1.5-flash",
+    name="summary_agent",
+    description="I can help summarize information of previous content.",
+    instruction="""\
+Summarize information in all historical messages that were replied by "question_generation_agent" and "information_retrieval_agent".
+
+# Instruction on replying format
+
+- The output should be like an essay that has a title, an abstract, multiple paragraphs for each topic and a conclusion.
+- Each paragraph should maps to one or more question in historical content.
+""",
+    flow="single",
+    generate_content_config=types.GenerateContentConfig(
+        temperature=0.8,
+    ),
+)
+
+research_assistant = Agent(
+    model="gemini-1.5-flash",
+    name="research_assistant",
+    description="I can help with research question.",
+    instruction="Help customers with their need.",
+    flow="sequential",
+    sub_agents=[
+        research_plan_agent,
+        question_sources_generation_agent,
+        summary_agent,
+    ],
+    generate_content_config=types.GenerateContentConfig(
+        temperature=0.1,
+    ),
+)
+
+spark_agent = Agent(
+    model="gemini-1.5-flash",
+    name="spark_assistant",
+    description="I can help with non-research question.",
+    instruction="Help customers with their need.",
+    flow="auto",
+    sub_agents=[research_assistant],
+    generate_content_config=types.GenerateContentConfig(
+        temperature=0.1,
+    ),
+)
+
+root_agent = spark_agent

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

@@ -0,0 +1,190 @@
+{
+  "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

@@ -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/hello_world_agent/agent.py

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

@@ -0,0 +1,24 @@
+[
+  {
+    "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

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

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