diff --git a/contributing/samples/human_tool_confirmation/agent.py b/contributing/samples/human_tool_confirmation/agent.py index 43e67dc9..e1ef5a51 100644 --- a/contributing/samples/human_tool_confirmation/agent.py +++ b/contributing/samples/human_tool_confirmation/agent.py @@ -13,6 +13,8 @@ # limitations under the License. from google.adk import Agent +from google.adk.apps import App +from google.adk.apps import ResumabilityConfig from google.adk.tools.function_tool import FunctionTool from google.adk.tools.tool_confirmation import ToolConfirmation from google.adk.tools.tool_context import ToolContext @@ -88,3 +90,12 @@ root_agent = Agent( ], generate_content_config=types.GenerateContentConfig(temperature=0.1), ) + +app = App( + name='human_tool_confirmation', + root_agent=root_agent, + # Set the resumability config to enable resumability. + resumability_config=ResumabilityConfig( + is_resumable=True, + ), +)