Regenerating code

Author: blast-hardcheese

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

@@ -16,6 +16,12 @@
     Stream_MethodOutputTypeAdapter,
     encode_Stream_MethodInput,
 )
+from .emit_error import Emit_ErrorErrors, Emit_ErrorErrorsTypeAdapter
+
+intTypeAdapter: TypeAdapter[int] = TypeAdapter(int)
+
+
+boolTypeAdapter: TypeAdapter[bool] = TypeAdapter(bool)
 
 
 class Test_ServiceService:
@@ -40,3 +46,22 @@ async def stream_method(
                 x  # type: ignore[arg-type]
             ),
         )
+
+    async def emit_error(
+        self,
+        inputStream: AsyncIterable[int],
+    ) -> AsyncIterator[bool | Emit_ErrorErrors | RiverError]:
+        return self.client.send_stream(
+            "test_service",
+            "emit_error",
+            None,
+            inputStream,
+            None,
+            lambda x: x,
+            lambda x: boolTypeAdapter.validate_python(
+                x  # type: ignore[arg-type]
+            ),
+            lambda x: Emit_ErrorErrorsTypeAdapter.validate_python(
+                x  # type: ignore[arg-type]
+            ),
+        )

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

@@ -0,0 +1,45 @@
+# Code generated by river.codegen. DO NOT EDIT.
+from collections.abc import AsyncIterable, AsyncIterator
+import datetime
+from typing import (
+    Any,
+    Literal,
+    Mapping,
+    NotRequired,
+    TypedDict,
+)
+from typing_extensions import Annotated
+
+from pydantic import BaseModel, Field, TypeAdapter, WrapValidator
+from replit_river.error_schema import RiverError
+from replit_river.client import (
+    RiverUnknownError,
+    translate_unknown_error,
+    RiverUnknownValue,
+    translate_unknown_value,
+)
+
+import replit_river as river
+
+
+class Emit_ErrorErrorsOneOf_DATA_LOSS(RiverError):
+    code: Literal["DATA_LOSS"]
+    message: str
+
+
+class Emit_ErrorErrorsOneOf_UNEXPECTED_DISCONNECT(RiverError):
+    code: Literal["UNEXPECTED_DISCONNECT"]
+    message: str
+
+
+Emit_ErrorErrors = Annotated[
+    Emit_ErrorErrorsOneOf_DATA_LOSS
+    | Emit_ErrorErrorsOneOf_UNEXPECTED_DISCONNECT
+    | RiverUnknownError,
+    WrapValidator(translate_unknown_error),
+]
+
+
+Emit_ErrorErrorsTypeAdapter: TypeAdapter[Emit_ErrorErrors] = TypeAdapter(
+    Emit_ErrorErrors
+)