chore: implement type vars

Author: BobTheBuidler

web3/module.py

@@ -5,11 +5,10 @@
     Coroutine,
     Dict,
     Optional,
-    Sequence,
     Tuple,
     TypeVar,
     Union,
-    cast,
+    overload,
 )
 
 from eth_abi.codec import (
@@ -27,13 +26,14 @@
 )
 from web3.method import (
     Method,
+    RequestArgs,
+    ResponseFormatters,
 )
 from web3.providers.persistent import (
     PersistentConnectionProvider,
 )
 from web3.types import (
     FormattedEthSubscriptionResponse,
-    RPCEndpoint,
     RPCResponse,
 )
 
@@ -58,34 +58,50 @@ def apply_result_formatters(
 TReturn = TypeVar("TReturn")
 
 
+@overload
+def retrieve_request_information_for_batching(
+    w3: "AsyncWeb3",
+    module: "Module",
+    method: Method[Callable[..., Any]],
+) -> Callable[..., Coroutine[Any, Any, Tuple[RequestArgs, ResponseFormatters]]]:
+    ...
+    
+@overload
+def retrieve_request_information_for_batching(
+    w3: "Web3",
+    module: "Module",
+    method: Method[Callable[..., Any]],
+) -> Callable[..., Tuple[RequestArgs, ResponseFormatters]]:
+    ...
+    
 @curry
 def retrieve_request_information_for_batching(
     w3: Union["AsyncWeb3", "Web3"],
     module: "Module",
     method: Method[Callable[..., Any]],
 ) -> Union[
-    Callable[..., Tuple[Tuple[RPCEndpoint, Any], Sequence[Any]]],
-    Callable[..., Coroutine[Any, Any, Tuple[Tuple[RPCEndpoint, Any], Sequence[Any]]]],
+    Callable[..., Tuple[RequestArgs, ResponseFormatters]],
+    Callable[..., Coroutine[Any, Any, Tuple[RequestArgs, ResponseFormatters]]],
 ]:
     async def async_inner(
         *args: Any, **kwargs: Any
-    ) -> Tuple[Tuple[RPCEndpoint, Any], Sequence[Any]]:
+    ) -> Tuple[RequestArgs, ResponseFormatters]:
         (method_str, params), response_formatters = method.process_params(
             module, *args, **kwargs
         )
         if isinstance(w3.provider, PersistentConnectionProvider):
             w3.provider._request_processor.cache_request_information(
-                None, cast(RPCEndpoint, method_str), params, response_formatters
+                None, method_str, params, response_formatters
             )
-        return (cast(RPCEndpoint, method_str), params), response_formatters
+        return (method_str, params), response_formatters
 
     def inner(
         *args: Any, **kwargs: Any
-    ) -> Tuple[Tuple[RPCEndpoint, Any], Sequence[Any]]:
+    ) -> Tuple[RequestArgs, ResponseFormatters]:
         (method_str, params), response_formatters = method.process_params(
             module, *args, **kwargs
         )
-        return (cast(RPCEndpoint, method_str), params), response_formatters
+        return (method_str, params), response_formatters
 
     return async_inner if module.is_async else inner
 
@@ -142,7 +158,7 @@ async def caller(
 
         if isinstance(async_w3.provider, PersistentConnectionProvider):
             return await async_w3.manager.socket_request(
-                cast(RPCEndpoint, method_str),
+                method_str,
                 params,
                 response_formatters=response_formatters,
             )