From 1c569fe3a7380eaf6baf2b2dc514a39a492564be Mon Sep 17 00:00:00 2001 From: Devon Stewart Date: Mon, 17 Mar 2025 14:31:35 -0700 Subject: [PATCH 1/2] client.send_stream always returns a "A | RiverError" --- src/replit_river/codegen/client.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/replit_river/codegen/client.py b/src/replit_river/codegen/client.py index 7126163d..1b82dd14 100644 --- a/src/replit_river/codegen/client.py +++ b/src/replit_river/codegen/client.py @@ -1086,6 +1086,7 @@ async def {name}( ] ) elif procedure.type == "stream": + output_or_error_type = UnionTypeExpr([output_or_error_type, TypeName("RiverError")]) if init_type: assert render_init_method, "Expected an init renderer!" current_chunks.extend( From 8cd8c8ca28fff3137548daca6013718555c17c79 Mon Sep 17 00:00:00 2001 From: Devon Stewart Date: Mon, 17 Mar 2025 14:33:03 -0700 Subject: [PATCH 2/2] client.send_subscription always returns a "A | RiverError" --- src/replit_river/codegen/client.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/src/replit_river/codegen/client.py b/src/replit_river/codegen/client.py index 1b82dd14..36ddbc76 100644 --- a/src/replit_river/codegen/client.py +++ b/src/replit_river/codegen/client.py @@ -1013,6 +1013,12 @@ async def {name}( ] ) elif procedure.type == "subscription": + output_or_error_type = UnionTypeExpr( + [ + output_or_error_type, + TypeName("RiverError"), + ] + ) current_chunks.extend( [ reindent( @@ -1086,7 +1092,12 @@ async def {name}( ] ) elif procedure.type == "stream": - output_or_error_type = UnionTypeExpr([output_or_error_type, TypeName("RiverError")]) + output_or_error_type = UnionTypeExpr( + [ + output_or_error_type, + TypeName("RiverError"), + ] + ) if init_type: assert render_init_method, "Expected an init renderer!" current_chunks.extend(