scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

code qa fixes

Browse source
This commit is contained in:
Ishaan Jaffer 2026-02-06 14:44:09 -08:00
parent 0505812c83
commit 01a6adcb1a
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
litellm/a2a_protocol/exception_mapping_utils.py
View file

@ -25,8 +25,11 @@ if TYPE_CHECKING:
# Runtime import
_A2AClient: Any = None
A2A_SDK_AVAILABLE = False
try:
from a2a.client import A2AClient as _A2AClient
A2A_SDK_AVAILABLE = True
except ImportError:
pass
@ -174,7 +177,16 @@ def handle_a2a_localhost_retry(
Returns:
A new A2A client with the fixed URL
Raises:
ImportError: If the A2A SDK is not installed
"""
if not A2A_SDK_AVAILABLE or _A2AClient is None:
raise ImportError(
"A2A SDK is required for localhost retry handling. "
"Install it with: pip install a2a"
)
request_type = "streaming " if is_streaming else ""
verbose_logger.warning(
f"A2A {request_type}request to '{error.localhost_url}' failed: {error.original_error}. "