scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-21 00:21:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 74

fix(a2a): narrow discovery status codes through a structural protocol

Browse source 
basedpyright cannot narrow the probe error through
isinstance(error, AgentCardResolutionError) while that class is imported
inside try/except ImportError, which left three reportAttributeAccessIssue
errors over the budget main now carries. A runtime-checkable Protocol with
the same status_code contract carries the narrowing instead, so the check
no longer depends on the possibly unbound SDK name and the import goes.
This commit is contained in:
mateo-berri 2026-09-17 17:17:21 -07:00
parent c29d1c2813
commit 44fff9297d
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/a2a_protocol/card_resolver.py
View file

@ -6,7 +6,7 @@ Extends the A2A SDK's card resolver to support multiple well-known paths.
from collections.abc import Mapping
from types import MappingProxyType
from typing import TYPE_CHECKING, Any, Final
from typing import TYPE_CHECKING, Any, Final, Protocol, runtime_checkable
from litellm._logging import verbose_logger
from litellm.a2a_protocol.exceptions import A2AAgentCardDiscoveryError
@ -24,7 +24,6 @@ AGENT_CARD_PATH_PARAM: Final = "agent_card_path"
try:
from a2a.client import A2ACardResolver as _A2ACardResolver
from a2a.client.errors import AgentCardResolutionError
from a2a.utils.constants import (
AGENT_CARD_WELL_KNOWN_PATH,
PREV_AGENT_CARD_WELL_KNOWN_PATH,
@ -33,11 +32,16 @@ except ImportError:
pass
@runtime_checkable
class _HasStatusCode(Protocol):
status_code: int | None
def _discovery_status_code(failures: tuple[tuple[str, Exception], ...]) -> int:
statuses: Final = tuple(
error.status_code
for _, error in failures
if isinstance(error, AgentCardResolutionError) and error.status_code is not None and error.status_code != 404
if isinstance(error, _HasStatusCode) and error.status_code is not None and error.status_code != 404
)
return statuses[0] if statuses else 404