@@ -153,15 +153,15 @@ async def request[R](
153153 ) -> R :
154154 req = request_serialize (req )
155155 with anyio .fail_after (self .request_timeout ):
156- raw_resp = await self ._server .request (req )
156+ raw_resp = await self .get_server () .request (req )
157157 return response_deserialize (raw_resp , schema )
158158
159159 @override
160160 # @retry(stop=tenacity.stop_after_attempt(3), reraise=True)
161161 async def notify (self , msg : Notification ) -> None :
162162 noti = notification_serialize (msg )
163163 with anyio .fail_after (self .request_timeout ):
164- await self ._server .notify (noti )
164+ await self .get_server () .notify (noti )
165165
166166 async def _dispatch_server_requests (
167167 self , receiver : Receiver [ServerRequest ]
@@ -239,7 +239,7 @@ async def __asynccontextmanager__(self) -> AsyncGenerator[Self]:
239239 async with (
240240 asyncer .create_task_group () as tg ,
241241 channel [ServerRequest ].create () as (sender , receiver ),
242- self ._server .serve (workspace = self .get_workspace (), sender = sender ),
242+ self .get_server () .serve (workspace = self .get_workspace (), sender = sender ),
243243 ):
244244 # start to receive server requests here,
245245 # since server notification can be sent before `initialize`
0 commit comments