Support for http request injection propagation to tools (#816)

This commit is contained in:
ihrpr 2025-05-28 15:59:14 +01:00 • committed by GitHub
commit 70014a2bbb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -81,10 +81,12 @@ class RequestResponder(Generic[ReceiveRequestT, SendResultT]):
ReceiveNotificationT
]""",
on_complete: Callable[["RequestResponder[ReceiveRequestT, SendResultT]"], Any],
message_metadata: MessageMetadata = None,
) -> None:
self.request_id = request_id
self.request_meta = request_meta
self.request = request
self.message_metadata = message_metadata
self._session = session
self._completed = False
self._cancel_scope = anyio.CancelScope()
@ -365,6 +367,7 @@ class BaseSession(
request=validated_request,
session=self,
on_complete=lambda r: self._in_flight.pop(r.request_id, None),
message_metadata=message.metadata,
)
self._in_flight[responder.request_id] = responder