Skip to content

Commit a8ffd71

Browse files
committed
pass related request id on progress to allow this to trigger event id in sse
1 parent 79eb3c9 commit a8ffd71

File tree

3 files changed

+5
-4
lines changed

3 files changed

+5
-4
lines changed

src/mcp/server/fastmcp/server.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1088,6 +1088,7 @@ async def report_progress(self, progress: float, total: float | None = None, mes
10881088
progress=progress,
10891089
total=total,
10901090
message=message,
1091+
related_request_id=self.request_context.request_id,
10911092
)
10921093

10931094
async def read_resource(self, uri: str | AnyUrl) -> Iterable[ReadResourceContents]:

src/mcp/server/session.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -301,7 +301,7 @@ async def send_progress_notification(
301301
progress: float,
302302
total: float | None = None,
303303
message: str | None = None,
304-
related_request_id: str | None = None,
304+
related_request_id: types.RequestId | None = None,
305305
) -> None:
306306
"""Send a progress notification."""
307307
await self.send_notification(

tests/issues/test_176_progress_token.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,6 @@ async def test_progress_token_zero_first_call():
3636

3737
# Verify progress notifications
3838
assert mock_session.send_progress_notification.call_count == 3, "All progress notifications should be sent"
39-
mock_session.send_progress_notification.assert_any_call(progress_token=0, progress=0.0, total=10.0, message=None)
40-
mock_session.send_progress_notification.assert_any_call(progress_token=0, progress=5.0, total=10.0, message=None)
41-
mock_session.send_progress_notification.assert_any_call(progress_token=0, progress=10.0, total=10.0, message=None)
39+
mock_session.send_progress_notification.assert_any_call(progress_token=0, progress=0.0, total=10.0, message=None, related_request_id='test-request')
40+
mock_session.send_progress_notification.assert_any_call(progress_token=0, progress=5.0, total=10.0, message=None, related_request_id='test-request')
41+
mock_session.send_progress_notification.assert_any_call(progress_token=0, progress=10.0, total=10.0, message=None, related_request_id='test-request')

0 commit comments

Comments
 (0)