aljaz/mcp-python-sdk
1
0
Fork 0
mirror of https://github.com/aljazceru/mcp-python-sdk.git synced 2025-12-20 15:24:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

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
parent c5782129bc
commit 9e66f7ce89
9 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