replit-river 0.17.10__py3-none-any.whl → 0.17.12__py3-none-any.whl

replit_river/client_session.py

@@ -99,6 +99,8 @@ class ClientSession(Session):
             try:
                 await self._handle_messages_from_ws()
             except ConnectionClosed:
+                if self._should_abort_streams_after_transport_failure():
+                    await self.close()
                 if self._retry_connection_callback:
                     self._task_manager.create_task(self._retry_connection_callback())
 
@@ -106,10 +108,13 @@ class ClientSession(Session):
                 logger.debug("ConnectionClosed while serving", exc_info=True)
             except FailedSendingMessageException:
                 # Expected error if the connection is closed.
+                if self._should_abort_streams_after_transport_failure():
+                    await self.close()
                 logger.debug(
                     "FailedSendingMessageException while serving", exc_info=True
                 )
             except Exception:
+                await self.close()
                 logger.exception("caught exception at message iterator")
         except ExceptionGroup as eg:
             _, unhandled = eg.split(lambda e: isinstance(e, ConnectionClosed))

replit_river/codegen/client.py

@@ -405,7 +405,11 @@ def encode_type(
                                 | UnionTypeExpr
                                 | LiteralType
                             ) = other
-                            raise ValueError(f"What does it mean to have {_o2} here?")
+                            raise ValueError(
+                                f"What does it mean to have {
+                                    render_type_expr(_o2)
+                                } here?"
+                            )
         if permit_unknown_members:
             union = _make_open_union_type_expr(any_of)
         else:
@@ -496,7 +500,7 @@ def encode_type(
             typeddict_encoder.append("None")
             return (NoneTypeExpr(), [], [], set())
         elif type.type == "Date":
-            typeddict_encoder.append("TODO: dstewart")
+            typeddict_encoder.append("datetime.datetime")
             return (LiteralType("datetime.datetime"), [], [], set())
         elif type.type == "array" and type.items:
             type_name, module_info, type_chunks, encoder_names = encode_type(

replit_river/session.py

@@ -286,6 +286,17 @@ class Session:
         if should_retry and self._retry_connection_callback:
             self._task_manager.create_task(self._retry_connection_callback())
 
+    def _should_abort_streams_after_transport_failure(self) -> bool:
+        return not self._transport_options.transparent_reconnect
+
+    def _abort_all_streams(self) -> None:
+        """Close all active stream channels, notifying any waiting consumers."""
+        if not self._streams:
+            return
+        for stream in self._streams.values():
+            stream.close()
+        self._streams.clear()
+
     async def close(self) -> None:
         """Close the session and all associated streams."""
         logger.info(
@@ -310,9 +321,7 @@ class Session:
 
             # TODO: unexpected_close should close stream differently here to
             # throw exception correctly.
-            for stream in self._streams.values():
-                stream.close()
-            self._streams.clear()
+            self._abort_all_streams()
 
             self._state = SessionState.CLOSED
 

{replit_river-0.17.10.dist-info → replit_river-0.17.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: replit-river
-Version: 0.17.10
+Version: 0.17.12
 Summary: Replit river toolkit for Python
 Author-email: Replit <eng@replit.com>
 License: MIT License
@@ -33,7 +33,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Internet
-Requires-Python: <4,>=3.11
+Requires-Python: <4,>=3.12
 Requires-Dist: aiochannel>=1.2.1
 Requires-Dist: grpcio-tools>=1.59.3
 Requires-Dist: grpcio>=1.59.3
@@ -43,7 +43,7 @@ Requires-Dist: opentelemetry-api>=1.28.2
 Requires-Dist: opentelemetry-sdk>=1.28.2
 Requires-Dist: protobuf>=5.28.3
 Requires-Dist: pydantic-core>=2.20.1
-Requires-Dist: pydantic==2.9.2
+Requires-Dist: pydantic<3,>=2.9.0
 Requires-Dist: websockets<14,>=13.0
 Description-Content-Type: text/markdown
 

{replit_river-0.17.10.dist-info → replit_river-0.17.12.dist-info}/RECORD

@@ -1,6 +1,6 @@
 replit_river/__init__.py,sha256=9NWf8QCC7U34WoV3KAvyM7PPrjVMJ7SUuG77cYHx8l0,528
 replit_river/client.py,sha256=tZKU7XRz2OgQctrSAafrthhC6Iy-d1cuf6FcAIe1etE,8534
-replit_river/client_session.py,sha256=-5zLTwlJCsPj6oL5psSuooQgbbheK3Ibi04GxkyYORE,19428
+replit_river/client_session.py,sha256=d0aDuZ5Ke4u5uN2_lcU6Q_bbVuDrwlN4LPDKNnHb3Qc,19687
 replit_river/client_transport.py,sha256=7ATTR6tnaT0H-NfWo5j6PvGXbSB9zCs5UsEuvcOdNiU,14588
 replit_river/common_session.py,sha256=iOV9gnhg2kcCHW0cR1thqtT6hWApP_83ErjMliMuDnM,3960
 replit_river/error_schema.py,sha256=dts5f3iEsKhCF56-N6yNsZSf45sV8ILcjix0BddJgD8,3985
@@ -13,13 +13,13 @@ replit_river/seq_manager.py,sha256=AUUdNIFARH0wmEsnl7wOylfSRG92iZUAR-8mTQzP9DY,2
 replit_river/server.py,sha256=XBmmmqlk96MAWmglzb8-a_z7Ju_tVm8ug9oOzVEckcA,3995
 replit_river/server_session.py,sha256=CZnEO0Q5dNIIVZfSkFAsU5ajPwDwg9cmPSfg_uV25GY,10143
 replit_river/server_transport.py,sha256=Xd3F98lFm2HRhmw9zgE_hmULY-75IlK8QRQe5ThThWM,12023
-replit_river/session.py,sha256=s-3zu8gXuBLHfYkLX89BLSvnlpOmnhfQhgYyEDKoPgE,14337
+replit_river/session.py,sha256=ig06WE8MTxUy8yQjgbgaz8kw6BNVLDrD_5J96Rxcf58,14671
 replit_river/task_manager.py,sha256=i-nHRG1qlvVFOqUn-UyrlROOIIMfXJtox7XZv5eduK8,4477
 replit_river/transport_options.py,sha256=Kx3n6_x6sA69WV52EUvm-GzJhDaaewUQzbtcdPlZntA,2155
 replit_river/websocket_wrapper.py,sha256=xrk421A5eXTH2bjAcNeBZN96zTO6xQsl0VS9qcwus6M,814
 replit_river/codegen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 replit_river/codegen/__main__.py,sha256=LyKe1OXIyev0BuUcu43iu6zBLJzeogQ8DocxDkALD-Q,153
-replit_river/codegen/client.py,sha256=ebcR9aQ_VZ00UP1h_a3Y_PwTQAMKXaPbDnZ9MAgLSjE,59810
+replit_river/codegen/client.py,sha256=EU3cuwDegkLI9V6SG0IWPPdLQvBl6dUyrSuu0JDYGpQ,59963
 replit_river/codegen/format.py,sha256=pLa9nWlw8hTOnjzZjMsMpgcmNYiLaF5VAHNQTwwJ7Zw,285
 replit_river/codegen/run.py,sha256=InJaUDJft80pY7v2l-LoXHr8HYg3Bmb_6XQvlqgvxv4,3170
 replit_river/codegen/schema.py,sha256=y3kzrEPgzW9gKMjqiF2souZk47mE_kuFkHlsxRdD560,5344
@@ -29,7 +29,7 @@ replit_river/v2/__init__.py,sha256=Fi4wP87qHOQwaWQmVbPJOOiGfkiZ6gTWg5bF5J8je5s,1
 replit_river/v2/client.py,sha256=PQ1qz15lFTdMOs0kktBdzMUEYD66OnJJN0loYFVjdn4,7241
 replit_river/v2/client_transport.py,sha256=swtsFoWIX8nnG5cBeHMkZSZsJ5SW-VbdNHBdLtldMgU,3445
 replit_river/v2/session.py,sha256=8EMKw8bevoIrMfQBKMl3oXFlOJ91lzjmuqsm3qZN8Ts,54147
-replit_river-0.17.10.dist-info/METADATA,sha256=LrMPCnZCsCa8ZQfuPQNkPcvhVnp7IcZCI7zwuk34OEk,4791
-replit_river-0.17.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-replit_river-0.17.10.dist-info/licenses/LICENSE,sha256=kyZ5E3kOH4dzPet_xD0w6xi04e-0fGT0HRMhKyLyLC0,1063
-replit_river-0.17.10.dist-info/RECORD,,
+replit_river-0.17.12.dist-info/METADATA,sha256=K3yvG8yAj4od7kQL3DxmFP3cOUoUWnx7kKHXB2XuZsM,4794
+replit_river-0.17.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+replit_river-0.17.12.dist-info/licenses/LICENSE,sha256=kyZ5E3kOH4dzPet_xD0w6xi04e-0fGT0HRMhKyLyLC0,1063
+replit_river-0.17.12.dist-info/RECORD,,