This is not async

Author: blast-hardcheese

src/replit_river/v2/client_transport.py

@@ -80,7 +80,7 @@ async def _retry_connection(self) -> Session:
         logger.debug("Triggering get_or_create_session")
         return await self.get_or_create_session()
 
-    async def _delete_session(self, session: Session) -> None:
+    def _delete_session(self, session: Session) -> None:
         if self._session is session:
             self._session = None
         else:

src/replit_river/v2/session.py

@@ -109,7 +109,7 @@ class ResultError(TypedDict):
 trace_propagator = TraceContextTextMapPropagator()
 trace_setter = TransportMessageTracingSetter()
 
-CloseSessionCallback: TypeAlias = Callable[["Session"], Coroutine[Any, Any, Any]]
+CloseSessionCallback: TypeAlias = Callable[["Session"], None]
 RetryConnectionCallback: TypeAlias = Callable[
     [],
     Coroutine[Any, Any, Any],
@@ -472,7 +472,7 @@ async def close(
 
         # Clear the session in transports
         # This will get us GC'd, so this should be the last thing.
-        await self._close_session_callback(self)
+        self._close_session_callback(self)
 
     def _start_buffered_message_sender(
         self,