updates to generated test output

Author: airportyh

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

@@ -6,10 +6,18 @@
 from pydantic import TypeAdapter
 
 from replit_river.error_schema import RiverError
+
+RiverErrorTypeAdapter = TypeAdapter(RiverError)
 import replit_river as river
 
 
-from .rpc_method import Rpc_MethodInput, Rpc_MethodOutput, encode_Rpc_MethodInput
+from .rpc_method import (
+    Rpc_MethodInput,
+    Rpc_MethodInputTypeAdapter,
+    Rpc_MethodOutput,
+    Rpc_MethodOutputTypeAdapter,
+    encode_Rpc_MethodInput,
+)
 
 
 class Test_ServiceService:
@@ -26,10 +34,10 @@ async def rpc_method(
             "rpc_method",
             input,
             encode_Rpc_MethodInput,
-            lambda x: TypeAdapter(Rpc_MethodOutput).validate_python(
+            lambda x: Rpc_MethodOutputTypeAdapter.validate_python(
                 x  # type: ignore[arg-type]
             ),
-            lambda x: TypeAdapter(RiverError).validate_python(
+            lambda x: RiverErrorTypeAdapter.validate_python(
                 x  # type: ignore[arg-type]
             ),
             timeout,

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

@@ -39,5 +39,11 @@ class Rpc_MethodInput(TypedDict):
     data: str
 
 
+Rpc_MethodInputTypeAdapter = TypeAdapter(Rpc_MethodInput)  # type: ignore
+
+
 class Rpc_MethodOutput(BaseModel):
     data: str
+
+
+Rpc_MethodOutputTypeAdapter = TypeAdapter(Rpc_MethodOutput)  # type: ignore

tests/codegen/stream/generated/test_service/__init__.py

@@ -6,12 +6,16 @@
 from pydantic import TypeAdapter
 
 from replit_river.error_schema import RiverError
+
+RiverErrorTypeAdapter = TypeAdapter(RiverError)
 import replit_river as river
 
 
 from .stream_method import (
     Stream_MethodInput,
+    Stream_MethodInputTypeAdapter,
     Stream_MethodOutput,
+    Stream_MethodOutputTypeAdapter,
     encode_Stream_MethodInput,
 )
 
@@ -31,10 +35,10 @@ async def stream_method(
             inputStream,
             None,
             encode_Stream_MethodInput,
-            lambda x: TypeAdapter(Stream_MethodOutput).validate_python(
+            lambda x: Stream_MethodOutputTypeAdapter.validate_python(
                 x  # type: ignore[arg-type]
             ),
-            lambda x: TypeAdapter(RiverError).validate_python(
+            lambda x: RiverErrorTypeAdapter.validate_python(
                 x  # type: ignore[arg-type]
             ),
         )

tests/codegen/stream/generated/test_service/stream_method.py

@@ -39,5 +39,11 @@ class Stream_MethodInput(TypedDict):
     data: str
 
 
+Stream_MethodInputTypeAdapter = TypeAdapter(Stream_MethodInput)  # type: ignore
+
+
 class Stream_MethodOutput(BaseModel):
     data: str
+
+
+Stream_MethodOutputTypeAdapter = TypeAdapter(Stream_MethodOutput)  # type: ignore