Skip to content

Commit ba2b35a

Browse files
wuliang229copybara-github
authored andcommitted
chore: allow "from google.adk.tools import AgentTool"
PiperOrigin-RevId: 784386985
1 parent 3568c92 commit ba2b35a

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

src/google/adk/tools/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414

1515

1616
from ..auth.auth_tool import AuthToolArguments
17+
from .agent_tool import AgentTool
1718
from .apihub_tool.apihub_toolset import APIHubToolset
1819
from .base_tool import BaseTool
1920
from .example_tool import ExampleTool
@@ -31,6 +32,7 @@
3132
from .vertex_ai_search_tool import VertexAiSearchTool
3233

3334
__all__ = [
35+
'AgentTool',
3436
'APIHubToolset',
3537
'AuthToolArguments',
3638
'BaseTool',

src/google/adk/tools/agent_tool.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@
2323

2424
from . import _automatic_function_calling_util
2525
from ..memory.in_memory_memory_service import InMemoryMemoryService
26-
from ..runners import Runner
2726
from ..sessions.in_memory_session_service import InMemorySessionService
2827
from ._forwarding_artifact_service import ForwardingArtifactService
2928
from .base_tool import BaseTool
@@ -103,6 +102,7 @@ async def run_async(
103102
tool_context: ToolContext,
104103
) -> Any:
105104
from ..agents.llm_agent import LlmAgent
105+
from ..runners import Runner
106106

107107
if self.skip_summarization:
108108
tool_context.actions.skip_summarization = True

0 commit comments

Comments
 (0)