o ưi@sUdZddlmZmZmZmZddlmZddlm Z er#ddl m Z dZ ee d<dZee d <d Zee d <zdd lmZ dd lmZmZWn eyNYnwdeedefddZdddeddfddZGddde ZdS)zr Custom A2A Card Resolver for LiteLLM. Extends the A2A SDK's card resolver to support multiple well-known paths. ) TYPE_CHECKINGAnyDictOptional)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 urlreturncs&|sdS|tfddtDS)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 Fc3s|]}|vVqdS)N).0patternZ url_lowerrY/home/app/Keep/.python/lib/python3.10/site-packages/litellm/a2a_protocol/card_resolver.py 0sz/is_localhost_or_internal_url..)loweranyr)r rrris_localhost_or_internal_urls r agent_cardrbase_urlcCs0t|dd}|rt|r|dd}||_|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 r N/)getattrrrstripr )rrZcard_urlZ fixed_urlrrrfix_agent_card_url3s  rcsDeZdZdZ  d deedeeeefddffdd ZZ S) 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_kwargsrrc s|durtj||dIdHSttg}d}|D]=}ztd|j|tj||dIdHWStyU}ztd|j|d||}WYd}~qd}~ww|dur\|td|jdd|)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)rr z$Attempting to fetch agent card from z Failed to fetch agent card from z: z. Tried paths: z, ) superget_agent_cardr r rdebugr Exceptionjoin)selfrr pathsZ last_errorpathe __class__rrr"UsD   z%LiteLLMA2ACardResolver.get_agent_card)NN) __name__ __module__ __qualname____doc__rstrrrr" __classcell__rrr*rrLs r)r/typingrrrrZlitellm._loggingrZlitellm.constantsrZ a2a.typesrr __annotations__r r0r Z a2a.clientr Za2a.utils.constants ImportErrorboolrrrrrrrs$