NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-14 01:41:25 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: call base class constructor in McpTool constructor.

Browse Source 
PiperOrigin-RevId: 759644369
This commit is contained in:
Wei Sun (Jack) 2025-05-16 09:27:44 -07:00 committed by Copybara-Service
parent dcb50effef
commit d0f117ebbc
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/google/adk/tools/mcp_tool/mcp_tool.py
View File

@ -17,7 +17,8 @@ from typing import Optional
from google.genai.types import FunctionDeclaration from google.genai.types import FunctionDeclaration
from typing_extensions import override from typing_extensions import override
from .mcp_session_manager import MCPSessionManager, retry_on_closed_resource from .mcp_session_manager import MCPSessionManager
from .mcp_session_manager import retry_on_closed_resource
# Attempt to import MCP Tool from the MCP library, and hints user to upgrade # Attempt to import MCP Tool from the MCP library, and hints user to upgrade
# their Python version to 3.10 if it fails. # their Python version to 3.10 if it fails.
@ -36,9 +37,9 @@ except ImportError as e:
raise e raise e
from ..base_tool import BaseTool
from ...auth.auth_credential import AuthCredential from ...auth.auth_credential import AuthCredential
from ...auth.auth_schemes import AuthScheme from ...auth.auth_schemes import AuthScheme
from ..base_tool import BaseTool
from ..openapi_tool.openapi_spec_parser.rest_api_tool import to_gemini_schema from ..openapi_tool.openapi_spec_parser.rest_api_tool import to_gemini_schema
from ..tool_context import ToolContext from ..tool_context import ToolContext
@ -79,8 +80,7 @@ class MCPTool(BaseTool):
raise ValueError("mcp_tool cannot be None") raise ValueError("mcp_tool cannot be None")
if mcp_session is None: if mcp_session is None:
raise ValueError("mcp_session cannot be None") raise ValueError("mcp_session cannot be None")
self.name = mcp_tool.name super().__init__(name=mcp_tool.name, description=mcp_tool.description or "")
self.description = mcp_tool.description if mcp_tool.description else ""
self._mcp_tool = mcp_tool self._mcp_tool = mcp_tool
self._mcp_session = mcp_session self._mcp_session = mcp_session
self._mcp_session_manager = mcp_session_manager self._mcp_session_manager = mcp_session_manager