3030RawT = TypeVar ("RawT" )
3131
3232
33- class ParsedMessage (RootModel [JSONRPCMessage ], Generic [RawT ]):
33+ class MessageFrame (RootModel [JSONRPCMessage ], Generic [RawT ]):
3434 root : JSONRPCMessage
3535 raw : RawT | None = None
3636
3737 class Config :
3838 arbitrary_types_allowed = True
3939
4040
41- ReadStream = MemoryObjectReceiveStream [ParsedMessage [RawT ] | Exception ]
42- ReadStreamWriter = MemoryObjectSendStream [ParsedMessage [RawT ] | Exception ]
43- WriteStream = MemoryObjectSendStream [ParsedMessage [RawT ]]
44- WriteStreamReader = MemoryObjectReceiveStream [ParsedMessage [RawT ]]
41+ ReadStream = MemoryObjectReceiveStream [MessageFrame [RawT ] | Exception ]
42+ ReadStreamWriter = MemoryObjectSendStream [MessageFrame [RawT ] | Exception ]
43+ WriteStream = MemoryObjectSendStream [MessageFrame [RawT ]]
44+ WriteStreamReader = MemoryObjectReceiveStream [MessageFrame [RawT ]]
4545
4646SendRequestT = TypeVar ("SendRequestT" , ClientRequest , ServerRequest )
4747SendResultT = TypeVar ("SendResultT" , ClientResult , ServerResult )
@@ -242,7 +242,7 @@ async def send_request(
242242 # TODO: Support progress callbacks
243243
244244 await self ._write_stream .send (
245- ParsedMessage (JSONRPCMessage (jsonrpc_request ), None )
245+ MessageFrame (JSONRPCMessage (jsonrpc_request ), None )
246246 )
247247
248248 try :
@@ -280,15 +280,15 @@ async def send_notification(self, notification: SendNotificationT) -> None:
280280 )
281281
282282 await self ._write_stream .send (
283- ParsedMessage (JSONRPCMessage (jsonrpc_notification ))
283+ MessageFrame (JSONRPCMessage (jsonrpc_notification ))
284284 )
285285
286286 async def _send_response (
287287 self , request_id : RequestId , response : SendResultT | ErrorData
288288 ) -> None :
289289 if isinstance (response , ErrorData ):
290290 jsonrpc_error = JSONRPCError (jsonrpc = "2.0" , id = request_id , error = response )
291- await self ._write_stream .send (ParsedMessage (JSONRPCMessage (jsonrpc_error )))
291+ await self ._write_stream .send (MessageFrame (JSONRPCMessage (jsonrpc_error )))
292292 else :
293293 jsonrpc_response = JSONRPCResponse (
294294 jsonrpc = "2.0" ,
@@ -298,7 +298,7 @@ async def _send_response(
298298 ),
299299 )
300300 await self ._write_stream .send (
301- ParsedMessage (JSONRPCMessage (jsonrpc_response ))
301+ MessageFrame (JSONRPCMessage (jsonrpc_response ))
302302 )
303303
304304 async def _receive_loop (self ) -> None :
0 commit comments