From dc5c23c675bfc9976f8c7a2af60d06cb9a50d4d6 Mon Sep 17 00:00:00 2001 From: "Xiang (Sean) Zhou" Date: Fri, 9 May 2025 07:38:37 -0700 Subject: [PATCH] fix predicate class name in MCP toolset PiperOrigin-RevId: 756761193 --- src/google/adk/tools/mcp_tool/mcp_toolset.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/google/adk/tools/mcp_tool/mcp_toolset.py b/src/google/adk/tools/mcp_tool/mcp_toolset.py index f0c8a30..3e25812 100644 --- a/src/google/adk/tools/mcp_tool/mcp_toolset.py +++ b/src/google/adk/tools/mcp_tool/mcp_toolset.py @@ -14,16 +14,14 @@ from contextlib import AsyncExitStack import sys -from types import TracebackType from typing import List from typing import Optional from typing import override from typing import TextIO -from typing import Type from ...agents.readonly_context import ReadonlyContext -from ..base_toolset import BaseToolPredicate from ..base_toolset import BaseToolset +from ..base_toolset import ToolPredicate from .mcp_session_manager import MCPSessionManager from .mcp_session_manager import retry_on_closed_resource from .mcp_session_manager import SseServerParams @@ -69,7 +67,7 @@ class MCPToolset(BaseToolset): *, connection_params: StdioServerParameters | SseServerParams, errlog: TextIO = sys.stderr, - tool_predicate: Optional[BaseToolPredicate] = None, + tool_predicate: Optional[ToolPredicate] = None, ): """Initializes the MCPToolset.