uPjRUdZddlmZddlmZmZmZddlmZddl m Z erddl m Z dZ eed<d Zeed <d Zeed < dd lmZ ddlmZmZdedzdefdZeddddZeed<dZeed<d"dZdddedzfdZdddeddfdZdddeddfdZGd d!e Zy#e$rYYwxYw)#zr Custom A2A Card Resolver for LiteLLM. Extends the A2A SDK's card resolver to support multiple well-known paths. )MappingProxyType) TYPE_CHECKINGAnyFinal)verbose_logger)LOCALHOST_URL_PATTERNS) AgentCardN_A2ACardResolverz/.well-known/agent-card.jsonAGENT_CARD_WELL_KNOWN_PATHz/.well-known/agent.jsonPREV_AGENT_CARD_WELL_KNOWN_PATH)A2ACardResolver)r r urlreturncZ|sy|jtfdtDS)a Check if a URL is a localhost or internal URL. This detects common development URLs that are accidentally left in agent cards when deploying to production. Args: url: The URL to check Returns: True if the URL is localhost/internal Fc3&K|]}|v yw)N).0pattern url_lowers g/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/a2a_protocol/card_resolver.py z/is_localhost_or_internal_url..1sJw)#Js)loweranyr)rrs @ris_localhost_or_internal_urlrs( yy{I J3IJ JJJSONRPCz HTTP+JSONGRPC)jsonrpcz http+jsongrpc_CANONICAL_PROTOCOL_BINDINGSz0.3_LEGACY_PROTOCOL_VERSION agent_cardr c t|}|j||jD]S}tj |j j }|||j k(rB||_t|_U|S)a Canonicalize the supported interfaces of spec-adjacent agent cards. Some A2A servers (e.g. LangGraph Platform) serve agent cards with lowercase bindings like "jsonrpc", but a2a-sdk's ClientFactory matches bindings case-sensitively against its uppercase TransportProtocol constants and fails with "no compatible transports found." for spec-adjacent casings. The same servers also speak the A2A 0.3 JSON dialect ("kind"-discriminated payloads) while declaring protocolVersion "1.0", which a2a-sdk's strict v1 proto parsing rejects. A mis-cased binding fingerprints such a server, so its declared version is downgraded to 0.3 to route the SDK's ClientFactory onto its v0.3 compat transport, which speaks that dialect. ) typeCopyFromsupported_interfacesr getprotocol_bindingrr!protocol_version)r" normalized interface canonicals rnormalize_agent_card_interfacesr-?s)Z(*J #44> < @ @A[A[AaAaAc d   Y-G-G G %. "%= " > rcdt|dd}|r|St|dd}|rt|dddSy)z9Return the agent endpoint URL from the resolved SDK card.rNr&r)getattr)r"r interfacess rget_agent_card_urlr1Ys@UD1C   ,BDIJz!}eT22 rc|jddz}t|dr||_t|dd}|r ||d_yy)z4Set the agent endpoint URL on the resolved SDK card./rr&Nr)rstriphasattrrr/)r"rr*r0s rset_agent_card_urlr6esH 3#-Jz5!#  ,BDIJ& 1 rbase_urlct|dd}|r&t|r|jddz}||_t|dd}|r9t|ddd}|r't|r|jddz|d_|S)a Fix the agent card URL if it contains a localhost/internal address. Many A2A agents are deployed with agent cards that contain internal URLs like "http://0.0.0.0:8001/" or "http://localhost:8000/". This function replaces such URLs with the provided base_url. Args: agent_card: The agent card to fix base_url: The base URL to use as replacement Returns: The agent card with the URL fixed if necessary rNr3r&r)r/rr4r)r"r7card_url fixed_urlr0 interface_urls rfix_agent_card_urlr<psj%6H0:#??3/#5 "  ,BDIJ&z!}eTB 9-H ( 4s :JqM  rcLeZdZdZ ddedzdeeefdzddffd ZxZS) LiteLLMA2ACardResolverz Custom A2A card resolver that supports multiple well-known paths. Extends the base A2ACardResolver to try both: - /.well-known/agent-card.json (standard) - /.well-known/agent.json (previous/alternative) Nrelative_card_path http_kwargsrr cK|t|||d{Sttg}d}|D]>} t j d|j |t|||d{cS||td|j ddj|777#t$r.}t j d|j |||}Yd}~d}~wwxYww)a Fetch the agent card, trying multiple well-known paths. First tries the standard path, then falls back to the previous path. Args: relative_card_path: Optional path to the agent card endpoint. If None, tries both well-known paths. http_kwargs: Optional dictionary of keyword arguments to pass to httpx.get Returns: AgentCard from the A2A agent Raises: A2AClientHTTPError or A2AClientJSONError if both paths fail N)r?r@z(Attempting to fetch agent card from %s%sz(Failed to fetch agent card from %s%s: %sz Failed to fetch agent card from z. Tried paths: z, ) superget_agent_cardr r rdebugr7 Exceptionjoin)selfr?r@paths last_errorpathe __class__s rrCz%LiteLLMA2ACardResolver.get_agent_cards,  )/#5'0  ' +    D $$%OQUQ^Q^`de"W3'+ +4   ! :4==/Y]YbYbchYiXjkll; $$%OQUQ^Q^`dfgh  sJCBC5B"(B )B",3C B"" C+$CCCC)NN) __name__ __module__ __qualname____doc__strdictrrC __classcell__)rLs@rr>r>sL*.-14m$J4m#s(^d*4m  4m4mrr>)r"r rr )rPtypesrtypingrrrlitellm._loggingrlitellm.constantsr a2a.typesr r __annotations__r rQr a2a.clientr a2a.utils.constants ImportErrorboolrr r!r-r1r6r<r>rrrr^s #,,+4##"@C@'@@ >KcDjKTK*'7 'e#(%'4 ; 3: ';'S'T';#+>=m-=mi  sBB&%B&