|
24 | 24 | )
|
25 | 25 |
|
26 | 26 | try:
|
27 |
| - from tenacity import AsyncRetrying, RetryCallState, RetryError, Retrying, retry, wait_exponential |
| 27 | + from tenacity import RetryCallState, RetryError, retry, wait_exponential |
28 | 28 | except ImportError as _import_error:
|
29 | 29 | raise ImportError(
|
30 | 30 | 'Please install `tenacity` to use the retries utilities, '
|
|
34 | 34 | from collections.abc import Awaitable, Callable
|
35 | 35 | from datetime import datetime, timezone
|
36 | 36 | from email.utils import parsedate_to_datetime
|
37 |
| -from typing import TYPE_CHECKING, Any, NoReturn, cast |
| 37 | +from typing import TYPE_CHECKING, Any, cast |
38 | 38 |
|
39 | 39 | from typing_extensions import TypedDict
|
40 | 40 |
|
@@ -157,18 +157,7 @@ def __init__(
|
157 | 157 | config: RetryConfig,
|
158 | 158 | wrapped: BaseTransport | None = None,
|
159 | 159 | validate_response: Callable[[Response], Any] | None = None,
|
160 |
| - **kwargs: NoReturn, |
161 | 160 | ):
|
162 |
| - # TODO: Remove the following checks (and **kwargs) during v1 release |
163 |
| - if 'controller' in kwargs: # pragma: no cover |
164 |
| - raise TypeError('The `controller` argument has been renamed to `config`, and now requires a `RetryConfig`.') |
165 |
| - if kwargs: # pragma: no cover |
166 |
| - raise TypeError(f'Unexpected keyword arguments: {", ".join(kwargs)}') |
167 |
| - if isinstance(config, Retrying): # pragma: no cover |
168 |
| - raise ValueError( |
169 |
| - 'Passing a Retrying instance is no longer supported; the `config` argument must be a `pydantic_ai.retries.RetryConfig`.' |
170 |
| - ) |
171 |
| - |
172 | 161 | self.config = config
|
173 | 162 | self.wrapped = wrapped or HTTPTransport()
|
174 | 163 | self.validate_response = validate_response
|
@@ -245,18 +234,7 @@ def __init__(
|
245 | 234 | config: RetryConfig,
|
246 | 235 | wrapped: AsyncBaseTransport | None = None,
|
247 | 236 | validate_response: Callable[[Response], Any] | None = None,
|
248 |
| - **kwargs: NoReturn, |
249 | 237 | ):
|
250 |
| - # TODO: Remove the following checks (and **kwargs) during v1 release |
251 |
| - if 'controller' in kwargs: # pragma: no cover |
252 |
| - raise TypeError('The `controller` argument has been renamed to `config`, and now requires a `RetryConfig`.') |
253 |
| - if kwargs: # pragma: no cover |
254 |
| - raise TypeError(f'Unexpected keyword arguments: {", ".join(kwargs)}') |
255 |
| - if isinstance(config, AsyncRetrying): # pragma: no cover |
256 |
| - raise ValueError( |
257 |
| - 'Passing an AsyncRetrying instance is no longer supported; the `config` argument must be a `pydantic_ai.retries.RetryConfig`.' |
258 |
| - ) |
259 |
| - |
260 | 238 | self.config = config
|
261 | 239 | self.wrapped = wrapped or AsyncHTTPTransport()
|
262 | 240 | self.validate_response = validate_response
|
|
0 commit comments