replit-river 0.17.9.dev0__tar.gz → 0.17.11__tar.gz

{replit_river-0.17.9.dev0 → replit_river-0.17.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: replit-river
-Version: 0.17.9.dev0
+Version: 0.17.11
 Summary: Replit river toolkit for Python
 Author-email: Replit <eng@replit.com>
 License: MIT License

{replit_river-0.17.9.dev0 → replit_river-0.17.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name="replit-river"
-version="0.17.9.dev0"
+version="0.17.11"
 description="Replit river toolkit for Python"
 authors = [{ name = "Replit", email = "eng@replit.com" }]
 license = { file = "LICENSE" }

{replit_river-0.17.9.dev0 → replit_river-0.17.11}/src/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-0.17.9.dev0 → replit_river-0.17.11}/src/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-0.17.9.dev0 → replit_river-0.17.11}/src/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