updated snapshots

Author: airportyh

tests/codegen/rpc/generated/test_service/rpc_method.py

@@ -12,7 +12,12 @@
 
 from pydantic import BaseModel, Field, TypeAdapter, WrapValidator
 from replit_river.error_schema import RiverError
-from replit_river.client import RiverUnknownError, translate_unknown_error
+from replit_river.client import (
+    RiverUnknownError,
+    translate_unknown_error,
+    RiverUnknownValue,
+    translate_unknown_value,
+)
 
 import replit_river as river
 

tests/codegen/snapshot/snapshots/test_basic_stream/test_service/stream_method.py

@@ -12,7 +12,12 @@
 
 from pydantic import BaseModel, Field, TypeAdapter, WrapValidator
 from replit_river.error_schema import RiverError
-from replit_river.client import RiverUnknownError, translate_unknown_error
+from replit_river.client import (
+    RiverUnknownError,
+    translate_unknown_error,
+    RiverUnknownValue,
+    translate_unknown_value,
+)
 
 import replit_river as river
 

tests/codegen/snapshot/snapshots/test_pathological_types/test_service/__init__.py

@@ -31,7 +31,7 @@
     encode_Pathological_MethodInputReq_Obj_Undefined,
 )
 
-boolTypeAdapter: TypeAdapter[Any] = TypeAdapter(bool)
+boolTypeAdapter: TypeAdapter[bool] = TypeAdapter(bool)
 
 
 class Test_ServiceService:

tests/codegen/snapshot/snapshots/test_pathological_types/test_service/pathological_method.py

@@ -12,7 +12,12 @@
 
 from pydantic import BaseModel, Field, TypeAdapter, WrapValidator
 from replit_river.error_schema import RiverError
-from replit_river.client import RiverUnknownValue, translate_unknown_value
+from replit_river.client import (
+    RiverUnknownError,
+    translate_unknown_error,
+    RiverUnknownValue,
+    translate_unknown_value,
+)
 
 import replit_river as river
 
@@ -473,6 +478,6 @@ class Pathological_MethodInput(TypedDict):
     undefined: NotRequired[None]
 
 
-Pathological_MethodInputTypeAdapter: TypeAdapter[Any] = TypeAdapter(
-    Pathological_MethodInput
+Pathological_MethodInputTypeAdapter: TypeAdapter[Pathological_MethodInput] = (
+    TypeAdapter(Pathological_MethodInput)
 )

tests/codegen/snapshot/snapshots/test_unknown_enum/enumService/needsEnum.py

@@ -12,7 +12,12 @@
 
 from pydantic import BaseModel, Field, TypeAdapter, WrapValidator
 from replit_river.error_schema import RiverError
-from replit_river.client import RiverUnknownError, translate_unknown_error
+from replit_river.client import (
+    RiverUnknownError,
+    translate_unknown_error,
+    RiverUnknownValue,
+    translate_unknown_value,
+)
 
 import replit_river as river
 
@@ -45,8 +50,10 @@ class NeedsenumErrorsOneOf_err_second(RiverError):
 
 
 NeedsenumErrors = Annotated[
-    Literal["err_first", "err_second"] | RiverUnknownValue,
-    WrapValidator(translate_unknown_value),
+    NeedsenumErrorsOneOf_err_first
+    | NeedsenumErrorsOneOf_err_second
+    | RiverUnknownError,
+    WrapValidator(translate_unknown_error),
 ]
 
 

tests/codegen/snapshot/snapshots/test_unknown_enum/enumService/needsEnumObject.py

@@ -12,7 +12,12 @@
 
 from pydantic import BaseModel, Field, TypeAdapter, WrapValidator
 from replit_river.error_schema import RiverError
-from replit_river.client import RiverUnknownError, translate_unknown_error
+from replit_river.client import (
+    RiverUnknownError,
+    translate_unknown_error,
+    RiverUnknownValue,
+    translate_unknown_value,
+)
 
 import replit_river as river
 
@@ -98,8 +103,8 @@ class NeedsenumobjectOutputFooOneOf_out_second(BaseModel):
 NeedsenumobjectOutputFoo = Annotated[
     NeedsenumobjectOutputFooOneOf_out_first
     | NeedsenumobjectOutputFooOneOf_out_second
-    | RiverUnknownError,
-    WrapValidator(translate_unknown_error),
+    | RiverUnknownValue,
+    WrapValidator(translate_unknown_value),
 ]