lint

Author: airportyh

src/replit_river/codegen/client.py

@@ -159,11 +159,11 @@ def _make_open_union_type_expr(one_of: list[TypeExpression]) -> OpenUnionTypeExp
         return OpenUnionTypeExpr(
             UnionTypeExpr(one_of),
             fallback_type="RiverUnknownError"
-                if base_model == "RiverError"
-                else "RiverUnknownValue",
+            if base_model == "RiverError"
+            else "RiverUnknownValue",
             validator_function="translate_unknown_error"
-                if base_model == "RiverError"
-                else "translate_unknown_value",
+            if base_model == "RiverError"
+            else "translate_unknown_value",
         )
 
     encoder_name: TypeName | None = None  # defining this up here to placate mypy

src/replit_river/codegen/typing.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from typing import NewType, assert_never
+from typing import NewType, assert_never, cast
 
 ModuleName = NewType("ModuleName", str)
 ClassName = NewType("ClassName", str)
@@ -184,10 +184,11 @@ def render_type_expr(value: TypeExpression) -> str:
                     retval = "None"
             return retval
         case OpenUnionTypeExpr(inner):
+            open_union = cast(OpenUnionTypeExpr, value)
             return (
                 "Annotated["
-                f"{render_type_expr(inner)} | {value.fallback_type},"
-                f"WrapValidator({value.validator_function})"
+                f"{render_type_expr(inner)} | {open_union.fallback_type},"
+                f"WrapValidator({open_union.validator_function})"
                 "]"
             )
         case TypeName(name):