|
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