beta6

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name="replit-river"
-version="0.1.7-beta.5"
+version="0.1.7-beta.6"
 description="Replit river toolkit for Python"
 authors = ["Replit <[email protected]>"]
 license = "LICENSE"

replit_river/session.py

@@ -337,12 +337,6 @@ async def send_message(
             serviceName=service_name,
             procedureName=procedure_name,
         )
-        try:
-            await self._buffer.put(msg)
-        except Exception:
-            # We should close the session when there are too many messages in buffer
-            await self.close(True)
-            return
         try:
             await self._send_transport_message(
                 msg,
@@ -358,6 +352,14 @@ async def send_message(
             logging.error(
                 f"Failed sending message : {e}, waiting for retry from buffer"
             )
+        finally:
+            # We need to put this later to guarantee the ordering of message sent
+            try:
+                await self._buffer.put(msg)
+            except Exception:
+                # We should close the session when there are too many messages in buffer
+                await self.close(True)
+                return
 
     async def _send_responses_from_output_stream(
         self,