feat: add message to ProgressNotification (#435)

Co-authored-by: ihrpr <inna.hrpr@gmail.com>
2025-05-13 10:40:15 -07:00 · 2025-05-13 10:40:15 -07:00 · 9e66f7ce89
commit 9e66f7ce89
parent c5782129bc
1 changed files with 387 additions and 12 deletions
--- a/src/mcp/client/session.py
+++ b/src/mcp/client/session.py
@ -168,7 +168,11 @@ class ClientSession(
        )

    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:
        """Send a progress notification."""
        await self.send_notification(
@ -179,6 +183,7 @@ class ClientSession(
                        progressToken=progress_token,
                        progress=progress,
                        total=total,
+                        message=message,
                    ),
                ),
            )