Patch circular import

Author: blast-hardcheese

src/replit_river/rate_limiter.py

@@ -4,12 +4,18 @@
 from contextvars import Context
 from typing import Literal
 
+from replit_river.error_schema import RiverException
 from replit_river.transport_options import ConnectionRetryOptions
-from replit_river.v2.client_transport import BudgetExhaustedException
 
 logger = logging.getLogger(__name__)
 
 
+class BudgetExhaustedException(RiverException):
+    def __init__(self, code: str, message: str, client_id: str) -> None:
+        super().__init__(code, message)
+        self.client_id = client_id
+
+
 class LeakyBucketRateLimit:
     """Asynchronous leaky bucket rate limiter.
 

src/replit_river/v2/client_transport.py

@@ -26,12 +26,6 @@ def __init__(self, code: str, message: str, client_id: str) -> None:
         self.client_id = client_id
 
 
-class BudgetExhaustedException(RiverException):
-    def __init__(self, code: str, message: str, client_id: str) -> None:
-        super().__init__(code, message)
-        self.client_id = client_id
-
-
 class ClientTransport(Generic[HandshakeMetadataType]):
     _session: Session | None