Abort if the session is rejected by the server

Author: blast-hardcheese

src/replit_river/error_schema.py

@@ -20,6 +20,9 @@
 # ERROR_CODE_UNKNOWN is the code for the RiverUnknownError
 ERROR_CODE_UNKNOWN = "UNKNOWN"
 
+# SESSION_STATE_MISMATCH is the code when the remote server rejects the session's state
+ERROR_CODE_SESSION_STATE_MISMATCH = "SESSION_STATE_MISMATCH"
+
 
 class RiverError(BaseModel):
     """Error message from the server."""

src/replit_river/v2/session.py

@@ -33,6 +33,7 @@
 )
 from replit_river.error_schema import (
     ERROR_CODE_CANCEL,
+    ERROR_CODE_SESSION_STATE_MISMATCH,
     ERROR_CODE_STREAM_CLOSED,
     ERROR_HANDSHAKE,
     RiverError,
@@ -314,6 +315,8 @@ async def websocket_closed_callback() -> None:
                         "river client get handshake response : %r", handshake_response
                     )  # noqa: E501
                     if not handshake_response.status.ok:
+                        if handshake_response.status.code == ERROR_CODE_SESSION_STATE_MISMATCH:
+                            await self.close()
                         raise RiverException(
                             ERROR_HANDSHAKE,
                             f"Handshake failed with code {handshake_response.status.code}: "  # noqa: E501