allow Any for error type so the existing error types can match them; type the type adapters properly

Author: airportyh

src/replit_river/client.py

@@ -80,7 +80,7 @@ async def send_rpc(
         request: RequestType,
         request_serializer: Callable[[RequestType], Any],
         response_deserializer: Callable[[Any], ResponseType],
-        error_deserializer: Callable[[Any], ErrorType],
+        error_deserializer: Callable[[Any], Any],
         timeout: timedelta,
     ) -> ResponseType:
         with _trace_procedure("rpc", service_name, procedure_name) as span_handle:
@@ -105,7 +105,7 @@ async def send_upload(
         init_serializer: Optional[Callable[[InitType], Any]],
         request_serializer: Callable[[RequestType], Any],
         response_deserializer: Callable[[Any], ResponseType],
-        error_deserializer: Callable[[Any], ErrorType],
+        error_deserializer: Callable[[Any], Any],
     ) -> ResponseType:
         with _trace_procedure("upload", service_name, procedure_name) as span_handle:
             session = await self._transport.get_or_create_session()
@@ -128,7 +128,7 @@ async def send_subscription(
         request: RequestType,
         request_serializer: Callable[[RequestType], Any],
         response_deserializer: Callable[[Any], ResponseType],
-        error_deserializer: Callable[[Any], ErrorType],
+        error_deserializer: Callable[[Any], Any],
     ) -> AsyncGenerator[Union[ResponseType, ErrorType], None]:
         with _trace_procedure(
             "subscription", service_name, procedure_name
@@ -156,7 +156,7 @@ async def send_stream(
         init_serializer: Optional[Callable[[InitType], Any]],
         request_serializer: Callable[[RequestType], Any],
         response_deserializer: Callable[[Any], ResponseType],
-        error_deserializer: Callable[[Any], ErrorType],
+        error_deserializer: Callable[[Any], Any],
     ) -> AsyncGenerator[Union[ResponseType, ErrorType], None]:
         with _trace_procedure("stream", service_name, procedure_name) as span_handle:
             session = await self._transport.get_or_create_session()

src/replit_river/codegen/client.py

@@ -775,7 +775,7 @@ def append_type_adapter_definition(
                 module_info,
                 [
                     FileContents(
-                        f"{type_adapter_name.value}: Any = "
+                        f"{type_adapter_name.value}: TypeAdapter[{rendered_type_expr}] = "
                         f"TypeAdapter({rendered_type_expr})"
                     )
                 ],