reverts

Author: airportyh

src/replit_river/client.py

@@ -21,6 +21,7 @@
 )
 
 from .rpc import (
+    ErrorType,
     InitType,
     RequestType,
     ResponseType,
@@ -79,7 +80,7 @@ async def send_rpc(
         request: RequestType,
         request_serializer: Callable[[RequestType], Any],
         response_deserializer: Callable[[Any], ResponseType],
-        error_deserializer: Callable[[Any], Any],
+        error_deserializer: Callable[[Any], ErrorType],
         timeout: timedelta,
     ) -> ResponseType:
         with _trace_procedure("rpc", service_name, procedure_name) as span_handle:
@@ -104,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], Any],
+        error_deserializer: Callable[[Any], ErrorType],
     ) -> ResponseType:
         with _trace_procedure("upload", service_name, procedure_name) as span_handle:
             session = await self._transport.get_or_create_session()
@@ -127,7 +128,7 @@ async def send_subscription(
         request: RequestType,
         request_serializer: Callable[[RequestType], Any],
         response_deserializer: Callable[[Any], ResponseType],
-        error_deserializer: Callable[[Any], Any],
+        error_deserializer: Callable[[Any], ErrorType],
     ) -> AsyncGenerator[Union[ResponseType, RiverError], None]:
         with _trace_procedure(
             "subscription", service_name, procedure_name
@@ -155,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], Any],
+        error_deserializer: Callable[[Any], ErrorType],
     ) -> AsyncGenerator[Union[ResponseType, RiverError], None]:
         with _trace_procedure("stream", service_name, procedure_name) as span_handle:
             session = await self._transport.get_or_create_session()