Gotcha

Author: blast-hardcheese

src/replit_river/v2/session.py

@@ -226,18 +226,20 @@ async def _do_ensure_connected[HandshakeMetadata](
 
                 try:
                     try:
-                        expectedSessionState = ExpectedSessionState(
-                            nextExpectedSeq=self.ack,
-                            nextSentSeq=self.seq,
-                        )
+                        next_seq = 0
+                        if self._send_buffer:
+                            next_seq = self._send_buffer[0].seq
                         handshake_request = ControlMessageHandshakeRequest[
                             HandshakeMetadata
                         ](  # noqa: E501
                             type="HANDSHAKE_REQ",
                             protocolVersion=protocol_version,
                             sessionId=self.session_id,
                             metadata=uri_and_metadata["metadata"],
-                            expectedSessionState=expectedSessionState,
+                            expectedSessionState=ExpectedSessionState(
+                                nextExpectedSeq=self.ack,
+                                nextSentSeq=next_seq,
+                            ),
                         )
                         stream_id = nanoid.generate()