Disambiguating between seconds and milliseconds

Author: blast-hardcheese

src/replit_river/v2/session.py

@@ -1086,13 +1086,12 @@ async def _do_ensure_connected[HandshakeMetadata](
     last_error: Exception | None = None
     attempt_count = 0
 
-    handshake_deadline_ms = (
+    handshake_deadline = (
         get_current_time() + transport_options.handshake_timeout_ms / 1000
     )
 
     while (
-        rate_limiter.has_budget(client_id)
-        and get_current_time() < handshake_deadline_ms
+        rate_limiter.has_budget(client_id) and get_current_time() < handshake_deadline
     ):
         if (state := get_state()) in TerminalStates or state in ActiveStates:
             logger.info(f"_do_ensure_connected stopping due to state={state}")
@@ -1154,14 +1153,14 @@ async def websocket_closed_callback() -> None:
                     "Handshake failed, conn closed while sending response",
                 ) from e
 
-            if get_current_time() >= handshake_deadline_ms:
+            if get_current_time() >= handshake_deadline:
                 raise RiverException(
                     ERROR_HANDSHAKE,
                     "Handshake response timeout, closing connection",
                 )
 
             try:
-                timeout = handshake_deadline_ms - get_current_time() / 1000.0
+                timeout = handshake_deadline - get_current_time()
                 async with asyncio.timeout(timeout):
                     data = await ws.recv(decode=False)
             except ConnectionClosedOK: