Formatting

Author: blast-hardcheese

src/replit_river/common_session.py

@@ -59,9 +59,7 @@ async def setup_heartbeat(
         await asyncio.sleep(heartbeat_ms / 1000)
         state = get_state()
         if not get_connected():
-            logger.debug(
-                "Websocket is not connected, not sending heartbeat"
-            )
+            logger.debug("Websocket is not connected, not sending heartbeat")
             continue
         if state != SessionState.ACTIVE:
             logger.debug(
@@ -140,7 +138,11 @@ async def buffered_message_sender(
         sent = False
         while msg := get_next_pending():
             ws = get_ws()
-            logger.debug("buffered_message_sender: Dequeued %r to send over %r", msg, ws)
+            logger.debug(
+                "buffered_message_sender: Dequeued %r to send over %r",
+                msg,
+                ws,
+            )
             if not ws:
                 break
             try:

src/replit_river/messages.py

@@ -17,7 +17,6 @@
     IgnoreMessageException,
     InvalidMessageException,
 )
-from replit_river.transport_options import TransportOptions
 
 logger = logging.getLogger(__name__)
 

src/replit_river/v2/client_session.py

@@ -5,7 +5,6 @@
 from typing import Any, AsyncGenerator, Callable, Literal, cast
 
 import nanoid
-import websockets
 from aiochannel import Channel
 from aiochannel.errors import ChannelClosed
 from opentelemetry.trace import Span
@@ -76,7 +75,11 @@ def __init__(
         )
 
         async def do_close_websocket() -> None:
-            logger.debug("do_close called, _ws_connected=%r, _ws_unwrapped=%r", self._ws_connected, self._ws_unwrapped)
+            logger.debug(
+                "do_close called, _ws_connected=%r, _ws_unwrapped=%r",
+                self._ws_connected,
+                self._ws_unwrapped,
+            )
             self._ws_connected = False
             if self._ws_unwrapped:
                 self._task_manager.create_task(self._ws_unwrapped.close())

src/replit_river/v2/client_transport.py

@@ -5,9 +5,8 @@
 
 import nanoid
 import websockets
-from pydantic import ValidationError
 import websockets.asyncio.client
-from websockets import WebSocketCommonProtocol
+from pydantic import ValidationError
 from websockets.asyncio.client import ClientConnection
 from websockets.exceptions import ConnectionClosed
 

src/replit_river/v2/session.py

@@ -4,7 +4,6 @@
 from typing import Any, Awaitable, Callable, Coroutine, TypeAlias
 
 import nanoid  # type: ignore
-import websockets
 from aiochannel import Channel
 from opentelemetry.trace import Span, use_span
 from opentelemetry.trace.propagation.tracecontext import TraceContextTextMapPropagator
@@ -162,9 +161,7 @@ async def _begin_close_session_countdown(self) -> None:
         self._close_session_after_time_secs = close_session_after_time_secs
         self._ws_connected = False
 
-    async def replace_with_new_websocket(
-        self, new_ws: ClientConnection
-    ) -> None:
+    async def replace_with_new_websocket(self, new_ws: ClientConnection) -> None:
         if self._ws_unwrapped and new_ws.id != self._ws_unwrapped.id:
             self._task_manager.create_task(
                 self._ws_unwrapped.close(