Add message queue for SSE messages POST endpoint (#459)

This commit is contained in:
Akash D 2025-05-06 17:10:43 -07:00 • committed by GitHub
commit 3b1b213a96
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -153,7 +153,7 @@ class StreamableHTTPTransport:
):
message.root.id = original_request_id
session_message = SessionMessage(message)
session_message = SessionMessage(message=message)
await read_stream_writer.send(session_message)
# Call resumption token callback if we have an ID
@ -286,7 +286,7 @@ class StreamableHTTPTransport:
try:
content = await response.aread()
message = JSONRPCMessage.model_validate_json(content)
session_message = SessionMessage(message)
session_message = SessionMessage(message=message)
await read_stream_writer.send(session_message)
except Exception as exc:
logger.error(f"Error parsing JSON response: {exc}")
@ -333,7 +333,7 @@ class StreamableHTTPTransport:
id=request_id,
error=ErrorData(code=32600, message="Session terminated"),
)
session_message = SessionMessage(JSONRPCMessage(jsonrpc_error))
session_message = SessionMessage(message=JSONRPCMessage(jsonrpc_error))
await read_stream_writer.send(session_message)
async def post_writer(