Skip to content

Commit 02f00c4

Browse files
committed
remove resume_tool
1 parent a346d6c commit 02f00c4

File tree

2 files changed

+8
-23
lines changed

2 files changed

+8
-23
lines changed

src/mcp/client/session.py

Lines changed: 3 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -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(

tests/shared/test_streamable_http.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1067,7 +1067,11 @@ async def run_tool():
10671067

10681068
# Resume the tool with the resumption token
10691069
assert captured_resumption_token is not None
1070-
result = await session.resume_tool(captured_resumption_token)
1070+
result = await session.call_tool(
1071+
"long_running_with_checkpoints",
1072+
{},
1073+
resumption_token=captured_resumption_token,
1074+
)
10711075

10721076
# We should get a complete result
10731077
assert len(result.content) == 1

0 commit comments

Comments
 (0)