Correctly handling a "not" type in input position

replit · Nov 29, 2024 · 76af9db · 76af9db
1 parent 502278f
commit 76af9db
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/replit_river/codegen/client.py b/replit_river/codegen/client.py
@@ -831,9 +831,14 @@ def __init__(self, client: river.Client[Any]):
                                 exclude_none=True,
                               )
                             """
-        if isinstance(
-            procedure.input, RiverConcreteType
-        ) and procedure.input.type not in ["object", "array"]:
+        if (
+            (
+                isinstance(procedure.input, RiverConcreteType)
+                and procedure.input.type not in ["object", "array"]
+            )
+            or isinstance(procedure.input, RiverNotType)
+            or procedure.input is None
+        ):
             render_input_method = "lambda x: x"
 
         assert (