1818from logging import getLogger
1919
2020from hishel import AsyncCacheTransport , AsyncInMemoryStorage , Controller
21- from httpx import AsyncHTTPTransport , Limits
21+ from httpx import AsyncBaseTransport , AsyncHTTPTransport , Limits
2222
2323from .config import CompositeCacheConfig , CompositeConfig
2424from .ratelimiting import AsyncRateLimitingTransport
@@ -34,7 +34,7 @@ class CompositeTransportFactory:
3434 def _create_common_transport_layers (
3535 cls ,
3636 config : CompositeConfig ,
37- base_transport : AsyncHTTPTransport | None = None ,
37+ base_transport : AsyncBaseTransport | None = None ,
3838 limits : Limits | None = None ,
3939 ):
4040 """Creates wrapped transports reused between different factory methods.
@@ -59,7 +59,7 @@ def _create_common_transport_layers(
5959 def create_ratelimiting_retry_transport (
6060 cls ,
6161 config : CompositeConfig ,
62- base_transport : AsyncHTTPTransport | None = None ,
62+ base_transport : AsyncBaseTransport | None = None ,
6363 limits : Limits | None = None ,
6464 ) -> AsyncRetryTransport :
6565 """Creates a retry transport, wrapping a rate limiting transport, wrapping an AsyncHTTPTransport."""
@@ -71,7 +71,7 @@ def create_ratelimiting_retry_transport(
7171 def create_cached_ratelimiting_retry_transport (
7272 cls ,
7373 config : CompositeCacheConfig ,
74- base_transport : AsyncHTTPTransport | None = None ,
74+ base_transport : AsyncBaseTransport | None = None ,
7575 limits : Limits | None = None ,
7676 ) -> AsyncCacheTransport :
7777 """Creates a retry transport, wrapping a rate limiting transport, wrapping a cache transport, wrapping an AsyncHTTPTransport."""
0 commit comments