From 5edc493da9525f18dea84cd49342e131db44cb87 Mon Sep 17 00:00:00 2001
From: Liang Wu <wuliang@google.com>
Date: Fri, 18 Jul 2025 09:29:45 -0700
Subject: [PATCH] chore: allow "from google.adk.tools import AgentTool"

PiperOrigin-RevId: 784606041
---
 src/google/adk/tools/__init__.py   | 2 ++
 src/google/adk/tools/agent_tool.py | 5 ++---
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/google/adk/tools/__init__.py b/src/google/adk/tools/__init__.py
index c562d7f4..d93c15d1 100644
--- a/src/google/adk/tools/__init__.py
+++ b/src/google/adk/tools/__init__.py
@@ -14,6 +14,7 @@
 
 
 from ..auth.auth_tool import AuthToolArguments
+from .agent_tool import AgentTool
 from .apihub_tool.apihub_toolset import APIHubToolset
 from .base_tool import BaseTool
 from .example_tool import ExampleTool
@@ -31,6 +32,7 @@ from .url_context_tool import url_context
 from .vertex_ai_search_tool import VertexAiSearchTool
 
 __all__ = [
+    'AgentTool',
     'APIHubToolset',
     'AuthToolArguments',
     'BaseTool',
diff --git a/src/google/adk/tools/agent_tool.py b/src/google/adk/tools/agent_tool.py
index 2bbe8a9c..2638d79d 100644
--- a/src/google/adk/tools/agent_tool.py
+++ b/src/google/adk/tools/agent_tool.py
@@ -23,15 +23,12 @@ from typing_extensions import override
 
 from . import _automatic_function_calling_util
 from ..memory.in_memory_memory_service import InMemoryMemoryService
-from ..runners import Runner
-from ..sessions.in_memory_session_service import InMemorySessionService
 from ._forwarding_artifact_service import ForwardingArtifactService
 from .base_tool import BaseTool
 from .tool_context import ToolContext
 
 if TYPE_CHECKING:
   from ..agents.base_agent import BaseAgent
-  from ..agents.llm_agent import LlmAgent
 
 
 class AgentTool(BaseTool):
@@ -103,6 +100,8 @@ class AgentTool(BaseTool):
       tool_context: ToolContext,
   ) -> Any:
     from ..agents.llm_agent import LlmAgent
+    from ..runners import Runner
+    from ..sessions.in_memory_session_service import InMemorySessionService
 
     if self.skip_summarization:
       tool_context.actions.skip_summarization = True