returning None is alright

Author: blast-hardcheese

replit_river/codegen/client.py

@@ -919,10 +919,6 @@ def __init__(self, client: river.Client[Any]):
             parse_output_method = "lambda x: None"
 
         if procedure.type == "rpc":
-            control_flow_keyword = "return "
-            if output_type == "None":
-                control_flow_keyword = ""
-
             current_chunks.extend(
                 [
                     reindent(
@@ -932,7 +928,7 @@ async def {name}(
               self,
               input: {render_type_expr(input_type)},
             ) -> {render_type_expr(output_type)}:
-              {control_flow_keyword}await self.client.send_rpc(
+              return await self.client.send_rpc(
                 {repr(schema_name)},
                 {repr(name)},
                 input,
@@ -967,9 +963,6 @@ async def {name}(
                 ]
             )
         elif procedure.type == "upload":
-            control_flow_keyword = "return "
-            if output_type == "None":
-                control_flow_keyword = ""
             if init_type:
                 assert render_init_method, "Expected an init renderer!"
                 current_chunks.extend(
@@ -982,7 +975,7 @@ async def {name}(
               init: {init_type},
               inputStream: AsyncIterable[{render_type_expr(input_type)}],
             ) -> {output_type}:
-              {control_flow_keyword}await self.client.send_upload(
+              return await self.client.send_upload(
                 {repr(schema_name)},
                 {repr(name)},
                 init,
@@ -1006,7 +999,7 @@ async def {name}(
               self,
               inputStream: AsyncIterable[{render_type_expr(input_type)}],
             ) -> {render_type_expr(output_or_error_type)}:
-              {control_flow_keyword}await self.client.send_upload(
+              return await self.client.send_upload(
                 {repr(schema_name)},
                 {repr(name)},
                 None,