diff --git a/src/google/adk/tools/mcp_tool/mcp_tool.py b/src/google/adk/tools/mcp_tool/mcp_tool.py index fac71037..6553bb2c 100644 --- a/src/google/adk/tools/mcp_tool/mcp_tool.py +++ b/src/google/adk/tools/mcp_tool/mcp_tool.py @@ -49,7 +49,7 @@ logger = logging.getLogger("google_adk." + __name__) class MCPTool(BaseTool): - """Turns a MCP Tool into a Vertex Agent Framework Tool. + """Turns an MCP Tool into an ADK Tool. Internally, the tool initializes from a MCP Tool, and uses the MCP Session to call the tool. @@ -63,9 +63,9 @@ class MCPTool(BaseTool): auth_scheme: Optional[AuthScheme] = None, auth_credential: Optional[AuthCredential] = None, ): - """Initializes a MCPTool. + """Initializes an MCPTool. - This tool wraps a MCP Tool interface and uses a session manager to + This tool wraps an MCP Tool interface and uses a session manager to communicate with the MCP server. Args: @@ -111,7 +111,7 @@ class MCPTool(BaseTool): Args: args: The arguments as a dict to pass to the tool. - tool_context: The tool context from upper level ADK agent. + tool_context: The tool context of the current invocation. Returns: Any: The response from the tool. @@ -119,6 +119,5 @@ class MCPTool(BaseTool): # Get the session from the session manager session = await self._mcp_session_manager.create_session() - # TODO(cheliu): Support passing tool context to MCP Server. response = await session.call_tool(self.name, arguments=args) return response