Distribute STREAM_CLOSED_BIT into the versioned sessions

Author: blast-hardcheese

src/replit_river/client_session.py

@@ -34,7 +34,6 @@
 
 from .rpc import (
     ACK_BIT,
-    STREAM_CLOSED_BIT,
     STREAM_OPEN_BIT,
     ErrorType,
     InitType,
@@ -45,6 +44,9 @@
 logger = logging.getLogger(__name__)
 
 
+STREAM_CLOSED_BIT = 0x0004  # Synonymous with the cancel bit in v2
+
+
 class ClientSession(Session):
     def __init__(
         self,

src/replit_river/common_session.py

@@ -7,7 +7,7 @@
 from opentelemetry.trace import Span
 
 from replit_river.messages import FailedSendingMessageException
-from replit_river.rpc import ACK_BIT, STREAM_CLOSED_BIT, TransportMessage
+from replit_river.rpc import ACK_BIT, TransportMessage
 from replit_river.seq_manager import InvalidMessageException
 
 logger = logging.getLogger(__name__)

src/replit_river/rpc.py

@@ -48,7 +48,6 @@
 ]
 ACK_BIT = 0x0001
 STREAM_OPEN_BIT = 0x0002
-STREAM_CLOSED_BIT = 0x0004  # Synonymous with the cancel bit in v2
 
 # these codes are retriable
 # if the server sends a response with one of these codes,

src/replit_river/server_session.py

@@ -22,16 +22,19 @@
 
 from .rpc import (
     ACK_BIT,
-    STREAM_CLOSED_BIT,
     STREAM_OPEN_BIT,
     GenericRpcHandlerBuilder,
     TransportMessage,
     TransportMessageTracingSetter,
 )
 
 
+STREAM_CLOSED_BIT = 0x0004  # Synonymous with the cancel bit in v2
+
+
 logger = logging.getLogger(__name__)
 
+
 trace_propagator = TraceContextTextMapPropagator()
 trace_setter = TransportMessageTracingSetter()
 

src/replit_river/v2/client_session.py

@@ -27,7 +27,6 @@
 )
 from replit_river.rpc import (
     ACK_BIT,
-    STREAM_CLOSED_BIT,
     STREAM_OPEN_BIT,
 )
 from replit_river.seq_manager import (
@@ -38,6 +37,10 @@
 from replit_river.session import Session
 from replit_river.transport_options import MAX_MESSAGE_BUFFER_SIZE, TransportOptions
 
+
+STREAM_CLOSED_BIT = 0x0004  # Synonymous with the cancel bit in v2
+
+
 logger = logging.getLogger(__name__)