@@ -264,12 +264,14 @@ async def call_tool(
264264 name : str ,
265265 arguments : dict [str , Any ] | None = None ,
266266 on_resumption_token_update : ResumptionTokenUpdateCallback | None = None ,
267+ resumption_token : ResumptionToken | None = None ,
267268 ) -> types .CallToolResult :
268269 """Send a tools/call request."""
269270 metadata = None
270- if on_resumption_token_update :
271+ if on_resumption_token_update or resumption_token :
271272 metadata = ClientMessageMetadata (
272273 on_resumption_token_update = on_resumption_token_update ,
274+ resumption_token = resumption_token ,
273275 )
274276
275277 return await self .send_request (
@@ -283,27 +285,6 @@ async def call_tool(
283285 metadata = metadata ,
284286 )
285287
286- async def resume_tool (
287- self ,
288- resumption_token : ResumptionToken ,
289- ) -> types .CallToolResult :
290- """Send a tools/call request with resumtion token to resume the tool."""
291-
292- return await self .send_request (
293- types .ClientRequest (
294- types .CallToolRequest (
295- method = "tools/call" ,
296- params = types .CallToolRequestParams (
297- name = "resume_from_token" , arguments = {}
298- ),
299- )
300- ),
301- types .CallToolResult ,
302- metadata = ClientMessageMetadata (
303- resumption_token = resumption_token ,
304- ),
305- )
306-
307288 async def list_prompts (self ) -> types .ListPromptsResult :
308289 """Send a prompts/list request."""
309290 return await self .send_request (
0 commit comments