scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

fix(aiohttp): properly map asyncio.CancelledError to httpx.ConnectError

Browse source
This commit is contained in:
Tanmay Sharma 2026-03-23 16:59:52 +05:30
parent 0cc7650e65
commit 51427aa772
1 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/llms/custom_httpx/aiohttp_transport.py
View file

@ -23,6 +23,8 @@ AIOHTTP_EXC_MAP: Dict = {
aiohttp.ServerTimeoutError: httpx.TimeoutException,
aiohttp.ConnectionTimeoutError: httpx.ConnectTimeout,
aiohttp.SocketTimeoutError: httpx.ReadTimeout,
# Asyncio cancellation - map to retryable ConnectError to trigger retry logic
asyncio.CancelledError: httpx.ConnectError,
# Proxy related exceptions
aiohttp.ClientProxyConnectionError: httpx.ProxyError,
# SSL related exceptions
@ -59,7 +61,13 @@ except ImportError:
def map_aiohttp_exceptions() -> typing.Iterator[None]:
try:
yield
except (Exception, asyncio.CancelledError) as exc:
except asyncio.CancelledError as exc:
# asyncio.CancelledError inherits from BaseException (not Exception) in Python 3.9+.
# Catch it explicitly before the Exception handler and map to a retryable
# ConnectError so the router can retry on alternative deployments.
message = f"Request cancelled during DNS resolution or connection: {str(exc)}"
raise httpx.ConnectError(message) from exc
except Exception as exc:
mapped_exc = None
for from_exc, to_exc in AIOHTTP_EXC_MAP.items():