Feat/agent tool resilience sample

contributing/samples/agent_tool_resilience/README.md

@@ -0,0 +1,39 @@
+# AgentTool Resilience: Timeout, Retry, and Redirect Patterns
+
+This sample demonstrates how to handle failures, timeouts, and partial results from downstream agents in multi-agent workflows using ADK.
+
+## Running the Demo
+
+```bash
+adk web contributing/samples/agent_tool_resilience
+```
+
+Then in the web UI, select `agent_tool_resilience` from the dropdown and try:
+1. Simple query: "What is quantum computing?"
+2. Complex query: (very detailed research request) as "Research quantum computing applications in healthcare, finance, cryptography, 
+          logistics, weather prediction, drug discovery, and machine learning. 
+          For each domain, provide: historical context, current state-of-the-art, 
+          technical challenges, recent breakthroughs in 2024, comparison with classical 
+          approaches, economic impact, and future roadmap for the next 10 years."
+3. Timeout scenario: Set timeout to 5 seconds in `agent.py` and use a complex query
+
+## Features Demonstrated
+
+- **Timeout Protection**: Custom `TimeoutAgentTool` wrapper adds timeout handling to sub-agents
+- **Automatic Retry**: `ReflectAndRetryToolPlugin` handles retries with structured guidance
+- **Dynamic Fallback**: Coordinator agent routes to alternative agents when primary fails
+- **Error Recovery**: Specialized agent provides user-friendly error analysis
+
+## Expected Behavior
+
+1. **Normal Operation**: Primary agent handles the query successfully
+2. **Timeout Scenario**: Primary times out → Fallback agent is automatically tried
+3. **Failure Scenario**: Primary fails → Retry → Fallback → Error recovery agent provides guidance
+
+## Architecture
+
+The sample includes:
+- `coordinator_agent` - Routes requests and handles errors
+- `research_agent_primary` - Primary agent with timeout protection (5s)
+- `research_agent_fallback` - Fallback agent with longer timeout (60s)
+- `error_recovery_agent` - Analyzes failures and provides recommendations

contributing/samples/agent_tool_resilience/__init__.py

@@ -0,0 +1,18 @@
+# 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
+
+__all__ = ['agent']
+

contributing/samples/agent_tool_resilience/agent.py

@@ -0,0 +1,253 @@
+# 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.
+
+"""Sample demonstrating AgentTool resilience: timeout, retry, and redirect patterns.
+
+This sample shows how to handle failures, timeouts, and partial results
+from downstream agents in multi-agent workflows, including:
+- Timeout protection for sub-agents
+- Automatic retry with ReflectAndRetryToolPlugin
+- Dynamic rerouting to alternative agents
+- Error handling without leaking complexity to users
+"""
+
+import asyncio
+from typing import Any
+
+from google.adk import Agent
+from google.adk.apps import App
+from google.adk.plugins import ReflectAndRetryToolPlugin
+from google.adk.tools import AgentTool
+from google.adk.tools.google_search_tool import google_search
+from google.adk.tools.tool_context import ToolContext
+import time
+from google.genai import types
+from google.adk.events.event import Event
+
+
+# ============================================================================
+# Custom TimeoutAgentTool Wrapper
+# ============================================================================
+
+class TimeoutAgentTool(AgentTool):
+  """AgentTool with timeout protection.
+
+  This wrapper adds timeout handling to AgentTool, catching TimeoutError
+  and returning a structured error response that ReflectAndRetryToolPlugin
+  can process.
+  """
+
+  def __init__(
+      self,
+      agent,
+      timeout: float = 30.0,
+      timeout_error_message: str = "Sub-agent execution timed out",
+      **kwargs
+  ):
+    """Initialize TimeoutAgentTool.
+
+    Args:
+      agent: The agent to wrap.
+      timeout: Timeout in seconds for sub-agent execution.
+      timeout_error_message: Custom error message for timeout.
+      **kwargs: Additional arguments passed to AgentTool.
+    """
+    super().__init__(agent, **kwargs)
+    self.timeout = timeout
+    self.timeout_error_message = timeout_error_message
+
+  async def run_async(
+      self,
+      *,
+      args: dict[str, Any],
+      tool_context: ToolContext,
+  ) -> Any:
+    """Run with timeout protection."""
+    try:
+      return await asyncio.wait_for(
+          super().run_async(args=args, tool_context=tool_context),
+          timeout=self.timeout
+      )
+    except asyncio.TimeoutError:
+      # Return structured error that ReflectAndRetryToolPlugin can handle
+      return {
+          "error": "TimeoutError",
+          "message": self.timeout_error_message,
+          "timeout_seconds": self.timeout,
+          "agent_name": self.agent.name,
+      }
+
+
+
+# ============================================================================
+# Sub-Agents with Different Characteristics
+# ============================================================================
+
+# Primary agent - may be slow or fail
+research_agent_primary = Agent(
+    name='research_agent_primary',
+    model='gemini-2.5-flash',
+    description='Primary research agent for complex queries (may be slow)',
+    instruction="""
+    You are a thorough research assistant. When given a research task:
+    1. Acknowledge the task
+    2. ALWAYS use the google_search tool to find current information
+    3. Break down the information into detailed steps
+    4. Provide a comprehensive summary based on the search results
+
+    IMPORTANT: You MUST use google_search for every research query. Do not
+    respond without searching first. Be thorough and detailed in your responses.
+    """,
+    tools=[google_search],
+)
+
+# Fallback agent - faster, simpler
+research_agent_fallback = Agent(
+    name='research_agent_fallback',
+    model='gemini-2.5-flash',
+    description='Fallback research agent for simpler queries or when primary fails',
+    instruction="""
+    You are a research assistant focused on quick, concise answers.
+    When given a research task:
+    1. ALWAYS use the google_search tool first to find information
+    2. Provide a direct, well-structured response based on the search results
+    3. Keep your response concise without excessive detail
+
+    IMPORTANT: You MUST use google_search for every research query. Do not
+    respond without searching first.
+    """,
+    tools=[google_search],
+)
+
+# Specialized agent for error recovery
+error_recovery_agent = Agent(
+    name='error_recovery_agent',
+    model='gemini-2.5-flash',
+    description='Agent that handles error scenarios and provides alternative approaches',
+    instruction="""
+    You are an error recovery specialist. When you receive an error message
+    or failure report, analyze what went wrong and suggest:
+    1. What the error means
+    2. Why it might have occurred
+    3. Alternative approaches to achieve the goal
+    4. Recommendations for the user
+
+    Be helpful and constructive in your analysis.
+    """,
+)
+
+
+# ============================================================================
+# Coordinator Agent with Resilience Patterns
+# ============================================================================
+
+coordinator_agent = Agent(
+    name='coordinator_agent',
+    model='gemini-2.5-flash-lite',
+    description='Coordinator that manages research tasks with resilience',
+    instruction="""
+    You are a coordinator agent that manages research tasks by delegating to
+    specialized sub-agents. Your role is to ensure tasks complete successfully
+    even when individual agents fail or timeout.
+
+    **CRITICAL WORKFLOW REQUIREMENT - READ THIS FIRST:**
+    After calling ANY tool and receiving its response, you MUST IMMEDIATELY generate a text response explaining the results to the user. Tool calls are NOT the final answer - they are just one step. You MUST always provide a final text response to complete the conversation. If you only call a tool without generating text afterward, the user will receive no response.
+
+    **Tool Selection Strategy:**
+    1. **Primary Tool (research_agent_primary)**: Use for complex, detailed
+       research tasks. This agent is thorough but may be slower.
+    2. **Fallback Tool (research_agent_fallback)**: Use when:
+       - The primary agent times out or fails
+       - The query is simple and doesn't need deep research
+       - You need a quick answer
+    3. **Error Recovery Tool (error_recovery_agent)**: Use when:
+       - Multiple attempts have failed
+       - You need to understand what went wrong
+       - You need alternative approaches suggested
+
+    **Error Handling Protocol:**
+    - If research_agent_primary returns an error or timeout:
+      1. First, try research_agent_fallback with the same query
+      2. If that also fails, use error_recovery_agent to analyze the failure
+      3. Present the error_recovery_agent's analysis to the user
+      4. Suggest next steps based on the analysis
+
+    **User Communication:**
+    - Always present results clearly, even if they come from fallback agents
+    - After receiving any tool response, immediately provide a helpful text explanation to the user
+    - If errors occur, explain what happened and what you tried
+    - Never expose internal error details or retry counts to users
+    - Frame fallbacks as "using a different approach" rather than "fallback"
+
+    **Example Flow:**
+    User: "What is quantum computing?"
+    1. Call research_agent_primary with request="What is quantum computing?"
+    2. Wait for the tool response
+    3. **MUST**: Generate a text response presenting the results to the user
+
+    If the primary agent times out or fails:
+    4. Call research_agent_fallback with the same request
+    5. Wait for the tool response
+    6. **MUST**: Generate a text response presenting the results to the user
+
+    Remember: Tool calls are not the final answer - you must always follow up with a text response explaining the results to the user.
+    """,
+    tools=[
+        # Primary agent with timeout protection
+        # For testing timeouts, set a very short timeout (e.g., 5.0 seconds)
+        # For production, use a longer timeout (e.g., 30.0 seconds)
+        # NOTE: skip_summarization=False is required for the coordinator to continue
+        # after tool calls. If True, the function response event is marked as final
+        # and the LLM flow stops, preventing the coordinator from generating a response.
+        TimeoutAgentTool(
+            agent=research_agent_primary,
+            timeout=30.0,  # Change to 5.0 for timeout testing
+            timeout_error_message="Primary research agent timed out after 30 seconds",
+            skip_summarization=False,  # Must be False for coordinator to continue
+        ),
+        # Fallback agent timeout
+        # For testing: Set to 5.0 to test full failure chain (primary → fallback → error recovery)
+        # For production: Set to 60.0 to allow fallback to succeed after primary timeout
+        TimeoutAgentTool(
+            agent=research_agent_fallback,
+            timeout=60.0,  # Set to 60.0 to test successful fallback after primary timeout
+            timeout_error_message="Fallback research agent timed out",
+            skip_summarization=False,  # Must be False for coordinator to continue
+        ),
+        # Error recovery agent
+        AgentTool(
+            agent=error_recovery_agent,
+            skip_summarization=False,  # Must be False for coordinator to continue
+        ),
+    ],
+)
+
+# ============================================================================
+# App Configuration with Retry Plugin
+# ============================================================================
+
+# Configure retry plugin for automatic retry handling
+retry_plugin = ReflectAndRetryToolPlugin(
+    max_retries=2,  # Allow 2 retries per tool before giving up
+    throw_exception_if_retry_exceeded=False,  # Return guidance instead of raising
+)
+
+app = App(
+    name='agent_tool_resilience',
+    root_agent=coordinator_agent,
+    plugins=[retry_plugin],
+)
+
+root_agent = coordinator_agent
+