3131RawT = TypeVar ("RawT" )
3232
3333
34- class ParsedMessage (RootModel [JSONRPCMessage ], Generic [RawT ]):
34+ class MessageFrame (RootModel [JSONRPCMessage ], Generic [RawT ]):
3535 root : JSONRPCMessage
3636 raw : RawT | None = None
3737
3838 class Config :
3939 arbitrary_types_allowed = True
4040
4141
42- ReadStream = MemoryObjectReceiveStream [ParsedMessage [RawT ] | Exception ]
43- ReadStreamWriter = MemoryObjectSendStream [ParsedMessage [RawT ] | Exception ]
44- WriteStream = MemoryObjectSendStream [ParsedMessage [RawT ]]
45- WriteStreamReader = MemoryObjectReceiveStream [ParsedMessage [RawT ]]
42+ ReadStream = MemoryObjectReceiveStream [MessageFrame [RawT ] | Exception ]
43+ ReadStreamWriter = MemoryObjectSendStream [MessageFrame [RawT ] | Exception ]
44+ WriteStream = MemoryObjectSendStream [MessageFrame [RawT ]]
45+ WriteStreamReader = MemoryObjectReceiveStream [MessageFrame [RawT ]]
4646
4747SendRequestT = TypeVar ("SendRequestT" , ClientRequest , ServerRequest )
4848SendResultT = TypeVar ("SendResultT" , ClientResult , ServerResult )
@@ -259,7 +259,7 @@ async def send_request(
259259 # TODO: Support progress callbacks
260260
261261 await self ._write_stream .send (
262- ParsedMessage (JSONRPCMessage (jsonrpc_request ), None )
262+ MessageFrame (JSONRPCMessage (jsonrpc_request ), None )
263263 )
264264
265265 try :
@@ -297,15 +297,15 @@ async def send_notification(self, notification: SendNotificationT) -> None:
297297 )
298298
299299 await self ._write_stream .send (
300- ParsedMessage (JSONRPCMessage (jsonrpc_notification ))
300+ MessageFrame (JSONRPCMessage (jsonrpc_notification ))
301301 )
302302
303303 async def _send_response (
304304 self , request_id : RequestId , response : SendResultT | ErrorData
305305 ) -> None :
306306 if isinstance (response , ErrorData ):
307307 jsonrpc_error = JSONRPCError (jsonrpc = "2.0" , id = request_id , error = response )
308- await self ._write_stream .send (ParsedMessage (JSONRPCMessage (jsonrpc_error )))
308+ await self ._write_stream .send (MessageFrame (JSONRPCMessage (jsonrpc_error )))
309309 else :
310310 jsonrpc_response = JSONRPCResponse (
311311 jsonrpc = "2.0" ,
@@ -315,7 +315,7 @@ async def _send_response(
315315 ),
316316 )
317317 await self ._write_stream .send (
318- ParsedMessage (JSONRPCMessage (jsonrpc_response ))
318+ MessageFrame (JSONRPCMessage (jsonrpc_response ))
319319 )
320320
321321 async def _receive_loop (self ) -> None :
0 commit comments