|
14 | 14 | from dataclasses import dataclass |
15 | 15 | from datetime import timedelta |
16 | 16 | from types import TracebackType |
17 | | -from typing import Any, TypeAlias, overload |
| 17 | +from typing import Any, TypeAlias |
18 | 18 |
|
19 | 19 | import anyio |
20 | 20 | from pydantic import BaseModel |
21 | | -from typing_extensions import Self, deprecated |
| 21 | +from typing_extensions import Self |
22 | 22 |
|
23 | 23 | import mcp |
24 | 24 | from mcp import types |
@@ -191,45 +191,21 @@ def tools(self) -> dict[str, types.Tool]: |
191 | 191 | """Returns the tools as a dictionary of names to tools.""" |
192 | 192 | return self._tools |
193 | 193 |
|
194 | | - @overload |
195 | 194 | async def call_tool( |
196 | 195 | self, |
197 | 196 | name: str, |
198 | | - arguments: dict[str, Any], |
199 | | - read_timeout_seconds: timedelta | None = None, |
200 | | - progress_callback: ProgressFnT | None = None, |
201 | | - *, |
202 | | - meta: dict[str, Any] | None = None, |
203 | | - ) -> types.CallToolResult: ... |
204 | | - |
205 | | - @overload |
206 | | - @deprecated("The 'args' parameter is deprecated. Use 'arguments' instead.") |
207 | | - async def call_tool( |
208 | | - self, |
209 | | - name: str, |
210 | | - *, |
211 | 197 | args: dict[str, Any], |
212 | 198 | read_timeout_seconds: timedelta | None = None, |
213 | 199 | progress_callback: ProgressFnT | None = None, |
214 | | - meta: dict[str, Any] | None = None, |
215 | | - ) -> types.CallToolResult: ... |
216 | | - |
217 | | - async def call_tool( |
218 | | - self, |
219 | | - name: str, |
220 | | - arguments: dict[str, Any] | None = None, |
221 | | - read_timeout_seconds: timedelta | None = None, |
222 | | - progress_callback: ProgressFnT | None = None, |
223 | 200 | *, |
224 | 201 | meta: dict[str, Any] | None = None, |
225 | | - args: dict[str, Any] | None = None, |
226 | 202 | ) -> types.CallToolResult: |
227 | 203 | """Executes a tool given its name and arguments.""" |
228 | 204 | session = self._tool_to_session[name] |
229 | 205 | session_tool_name = self.tools[name].name |
230 | 206 | return await session.call_tool( |
231 | 207 | session_tool_name, |
232 | | - arguments if args is None else args, |
| 208 | + args, |
233 | 209 | read_timeout_seconds=read_timeout_seconds, |
234 | 210 | progress_callback=progress_callback, |
235 | 211 | meta=meta, |
|
0 commit comments