client.send_subscription always returns a "A | RiverError"

blast-hardcheese · blast-hardcheese · commit 689c034b468d · 2025-03-17T14:33:03.000-07:00
diff --git a/src/replit_river/codegen/client.py b/src/replit_river/codegen/client.py
@@ -1013,6 +1013,7 @@ async def {name}(
                 ]
             )
         elif procedure.type == "subscription":
+            output_or_error_type = UnionTypeExpr([output_or_error_type, TypeName("RiverError")])
             current_chunks.extend(
                 [
                     reindent(

Original file line number	Diff line number	Diff line change
`@@ -1013,6 +1013,7 @@ async def {name}(`
`1013`	`1013`	`]`
`1014`	`1014`	`)`
`1015`	`1015`	`elif procedure.type == "subscription":`
	`1016`	`+ output_or_error_type = UnionTypeExpr([output_or_error_type, TypeName("RiverError")])`
`1016`	`1017`	`current_chunks.extend(`
`1017`	`1018`	`[`
`1018`	`1019`	`reindent(`