mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-21 00:21:49 +00:00
b604e2b20c
About 35,000 fixes ruff marks safe across 32 rules (UP006/UP045/UP007 modern annotations, UP032 f-strings, SIM114/SIM118, RET501, and friends), removal of the 1,296 typing imports the rewrite orphaned, and hand fixes for what the fixers could not see: five star-import freeloaders of typing names, two F823 late-import annotations, the /get/config/list introspection crash on types.UnionType, redundant function-local RoleMappings imports in ui_sso.py that shadowed the module-level name once the annotation lost its quotes, and one FURB168 tautology. B009/B010/PIE804/RUF019 are excluded on purpose: their safe fixes rewrite getattr/setattr/**-splat/key-in-dict escape hatches into forms basedpyright then rejects (283 new errors measured), so their budgets stay at base values. ruff-strict-budget.json drops by 39,579 this commit (39,968 across the branch) with 28 rules at an actual 0 and 9 more sharply down. type-discipline-budget.json ratchets LIT002/LIT006/LIT009 down; LIT001 moves to the now-honest total: the checker matches the spelling `set` but not the alias `Set`, so the 160 typing.Set annotations rewritten to set[...] were always mutable-set annotations and only now count.
145 lines
4.1 KiB
Python
145 lines
4.1 KiB
Python
"""
|
|
A2A Protocol Exceptions.
|
|
|
|
Custom exception types for A2A protocol operations, following LiteLLM's exception pattern.
|
|
"""
|
|
|
|
import httpx
|
|
|
|
|
|
class A2AError(Exception):
|
|
"""
|
|
Base exception for A2A protocol errors.
|
|
|
|
Follows the same pattern as LiteLLM's main exceptions.
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
message: str,
|
|
status_code: int = 500,
|
|
llm_provider: str = "a2a_agent",
|
|
model: str | None = None,
|
|
response: httpx.Response | None = None,
|
|
litellm_debug_info: str | None = None,
|
|
max_retries: int | None = None,
|
|
num_retries: int | None = None,
|
|
):
|
|
self.status_code = status_code
|
|
self.message = f"litellm.A2AError: {message}"
|
|
self.llm_provider = llm_provider
|
|
self.model = model
|
|
self.litellm_debug_info = litellm_debug_info
|
|
self.max_retries = max_retries
|
|
self.num_retries = num_retries
|
|
self.response = response or httpx.Response(
|
|
status_code=self.status_code,
|
|
request=httpx.Request(method="POST", url="https://litellm.ai"),
|
|
)
|
|
super().__init__(self.message)
|
|
|
|
def __str__(self) -> str:
|
|
_message = self.message
|
|
if self.num_retries:
|
|
_message += f" LiteLLM Retried: {self.num_retries} times"
|
|
if self.max_retries:
|
|
_message += f", LiteLLM Max Retries: {self.max_retries}"
|
|
return _message
|
|
|
|
def __repr__(self) -> str:
|
|
return self.__str__()
|
|
|
|
|
|
class A2AConnectionError(A2AError):
|
|
"""
|
|
Raised when connection to an A2A agent fails.
|
|
|
|
This typically occurs when:
|
|
- The agent is unreachable
|
|
- The agent card contains a localhost/internal URL
|
|
- Network issues prevent connection
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
message: str,
|
|
url: str | None = None,
|
|
model: str | None = None,
|
|
response: httpx.Response | None = None,
|
|
litellm_debug_info: str | None = None,
|
|
max_retries: int | None = None,
|
|
num_retries: int | None = None,
|
|
):
|
|
self.url = url
|
|
super().__init__(
|
|
message=message,
|
|
status_code=503,
|
|
llm_provider="a2a_agent",
|
|
model=model,
|
|
response=response,
|
|
litellm_debug_info=litellm_debug_info,
|
|
max_retries=max_retries,
|
|
num_retries=num_retries,
|
|
)
|
|
|
|
|
|
class A2AAgentCardError(A2AError):
|
|
"""
|
|
Raised when there's an issue with the agent card.
|
|
|
|
This includes:
|
|
- Failed to fetch agent card
|
|
- Invalid agent card format
|
|
- Missing required fields
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
message: str,
|
|
url: str | None = None,
|
|
model: str | None = None,
|
|
response: httpx.Response | None = None,
|
|
litellm_debug_info: str | None = None,
|
|
):
|
|
self.url = url
|
|
super().__init__(
|
|
message=message,
|
|
status_code=404,
|
|
llm_provider="a2a_agent",
|
|
model=model,
|
|
response=response,
|
|
litellm_debug_info=litellm_debug_info,
|
|
)
|
|
|
|
|
|
class A2ALocalhostURLError(A2AConnectionError):
|
|
"""
|
|
Raised when an agent card contains a localhost/internal URL.
|
|
|
|
Many A2A agents are deployed with agent cards that contain internal URLs
|
|
like "http://0.0.0.0:8001/" or "http://localhost:8000/". This error
|
|
indicates that the URL needs to be corrected and the request should be retried.
|
|
|
|
Attributes:
|
|
localhost_url: The localhost/internal URL found in the agent card
|
|
base_url: The public base URL that should be used instead
|
|
original_error: The original connection error that was raised
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
localhost_url: str,
|
|
base_url: str,
|
|
original_error: Exception | None = None,
|
|
model: str | None = None,
|
|
):
|
|
self.localhost_url = localhost_url
|
|
self.base_url = base_url
|
|
self.original_error = original_error
|
|
|
|
message = f"Agent card contains localhost/internal URL '{localhost_url}'. Retrying with base URL '{base_url}'."
|
|
super().__init__(
|
|
message=message,
|
|
url=localhost_url,
|
|
model=model,
|
|
)
|