client.send_subscription always returns a "A | RiverError"

blast-hardcheese · blast-hardcheese · commit 8cd8c8ca28ff · 2025-03-17T14:36:23.000-07:00
diff --git 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(

Original file line number	Diff line number	Diff line change
`@@ -1013,6 +1013,12 @@ async def {name}(`
`1013`	`1013`	`]`
`1014`	`1014`	`)`
`1015`	`1015`	`elif procedure.type == "subscription":`
	`1016`	`+ output_or_error_type = UnionTypeExpr(`
	`1017`	`+ [`
	`1018`	`+ output_or_error_type,`
	`1019`	`+ TypeName("RiverError"),`
	`1020`	`+ ]`
	`1021`	`+ )`
`1016`	`1022`	`current_chunks.extend(`
`1017`	`1023`	`[`
`1018`	`1024`	`reindent(`
`@@ -1086,7 +1092,12 @@ async def {name}(`
`1086`	`1092`	`]`
`1087`	`1093`	`)`
`1088`	`1094`	`elif procedure.type == "stream":`
`1089`		`- output_or_error_type = UnionTypeExpr([output_or_error_type, TypeName("RiverError")])`
	`1095`	`+ output_or_error_type = UnionTypeExpr(`
	`1096`	`+ [`
	`1097`	`+ output_or_error_type,`
	`1098`	`+ TypeName("RiverError"),`
	`1099`	`+ ]`
	`1100`	`+ )`
`1090`	`1101`	`if init_type:`
`1091`	`1102`	`assert render_init_method, "Expected an init renderer!"`
`1092`	`1103`	`current_chunks.extend(`