From cb000e5a941143e0a0327a0f1080ddd4bdd4c978 Mon Sep 17 00:00:00 2001
From: "Xiang (Sean) Zhou" <seanzhougoogle@google.com>
Date: Thu, 15 May 2025 17:12:20 -0700
Subject: [PATCH] refactor: remove async root_agent handling as it no longer
 needed for MCP toolset

PiperOrigin-RevId: 759364382
---
 src/google/adk/cli/fast_api.py | 13 -------------
 1 file changed, 13 deletions(-)

diff --git a/src/google/adk/cli/fast_api.py b/src/google/adk/cli/fast_api.py
index 058a1ac..0b6d608 100644
--- a/src/google/adk/cli/fast_api.py
+++ b/src/google/adk/cli/fast_api.py
@@ -228,7 +228,6 @@ def get_fast_api_app(
 
   trace.set_tracer_provider(provider)
 
-  exit_stacks = []
   toolsets_to_close: set[BaseToolset] = set()
 
   @asynccontextmanager
@@ -237,9 +236,6 @@ def get_fast_api_app(
       async with lifespan(app) as lifespan_context:
         yield
 
-        if exit_stacks:
-          for stack in exit_stacks:
-            await stack.aclose()
         for toolset in toolsets_to_close:
           await toolset.close()
     else:
@@ -912,15 +908,6 @@ def get_fast_api_app(
     else:
       raise ValueError(f'Unable to find "root_agent" from {app_name}.')
 
-    # Handle an awaitable root agent and await for the actual agent.
-    if inspect.isawaitable(root_agent):
-      try:
-        agent, exit_stack = await root_agent
-        exit_stacks.append(exit_stack)
-        root_agent = agent
-      except Exception as e:
-        raise RuntimeError(f"error getting root agent, {e}") from e
-
     root_agent_dict[app_name] = root_agent
     toolsets_to_close.update(_get_all_toolsets(root_agent))
     return root_agent