Add on_session_created callback option #1710
Merged
+68
−2
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Motivation and Context
Currently, there's no way to retrieve the session ID from the SSE client. The session ID is embedded in the
endpoint URL sent by the server, but it's not exposed to callers. Users who need the session ID (e.g., for
reconnection with
mcp-session-idheader) would have to resort to intercepting raw HTTP bytes via custom transports.This change adds a simple callback mechanism that's invoked when the endpoint event is received, passing the
extracted session ID.
How Has This Been Tested?
test_sse_client_on_session_createdthat verifies the callback is invoked with a valid sessionID
Breaking Changes
None. The new
on_session_createdparameter is optional and defaults toNone.Types of changes
Checklist
Additional context
Usage example: