modelcontextprotocol/python-sdk
Watch
1
0
Fork
You've already forked python-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add message to ProgressNotification (#435)

Browse source 
Co-authored-by: ihrpr <inna.hrpr@gmail.com>
This commit is contained in:
Akshey D 2025-05-13 10:40:15 -07:00 • committed by GitHub
commit 9e66f7ce89
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 387 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/mcp/shared/progress.py
View file

@ -43,11 +43,11 @@ class ProgressContext(
total: float | None
current: float = field(default=0.0, init=False)
async def progress(self, amount: float) -> None:
async def progress(self, amount: float, message: str | None = None) -> None:
self.current += amount
await self.session.send_progress_notification(
self.progress_token, self.current, total=self.total
self.progress_token, self.current, total=self.total, message=message
)

6
src/mcp/shared/session.py
View file

@ -401,7 +401,11 @@ class BaseSession(
"""
async def send_progress_notification(
self, progress_token: str | int, progress: float, total: float | None = None
self,
progress_token: str | int,
progress: float,
total: float | None = None,
message: str | None = None,
) -> None:
"""
Sends a progress notification for a request that is currently being