o ưi @sdZddlmZmZmZddlmZddlmZm Z ddl m Z m Z m Z mZddlmZer4ddlmZdZz ddlmZd ZWn eyKd ZYnwGd d d Z dd edeedeedeedef ddZ ddededddeddf ddZd S)z` A2A Protocol Exception Mapping Utils. Maps A2A SDK exceptions to LiteLLM A2A exception types. ) TYPE_CHECKINGAnyOptional)verbose_logger)fix_agent_card_urlis_localhost_or_internal_url)A2AAgentCardErrorA2AConnectionErrorA2AErrorA2ALocalhostURLError)CONNECTION_ERROR_PATTERNS) A2AClientFTNc@sVeZdZdZededefddZedeedefddZ ededefd d Z d S) A2AExceptionCheckerszA Helper class for checking various A2A error conditions. error_strreturncs,t|tsdS|tfddtDS)z Check if an error string indicates a connection error. Args: error_str: The error string to check Returns: True if the error indicates a connection issue Fc3|]}|vVqdSN.0patternZerror_str_lowerrc/home/app/Keep/.python/lib/python3.10/site-packages/litellm/a2a_protocol/exception_mapping_utils.py 8z;A2AExceptionCheckers.is_connection_error..) isinstancestrloweranyr )rrrris_connection_error)s z(A2AExceptionCheckers.is_connection_errorurlcCst|S)z Check if a URL is a localhost/internal URL. Args: url: The URL to check Returns: True if the URL is localhost/internal )r)r rrris_localhost_url:s z%A2AExceptionCheckers.is_localhost_urlcs4t|tsdS|gd}tfdd|DS)z Check if an error string indicates an agent card error. Args: error_str: The error string to check Returns: True if the error is related to agent card fetching/parsing F)z agent cardz agent-cardz .well-knownzcard not foundz invalid agentc3rrrrrrrr]rz;A2AExceptionCheckers.is_agent_card_error..)rrrr)rZagent_card_patternsrrris_agent_card_errorGs z(A2AExceptionCheckers.is_agent_card_errorN) __name__ __module__ __qualname____doc__ staticmethodrboolrrr!r"rrrrr$s roriginal_exceptioncard_urlapi_basemodelrcCstt|}|r|rt|rt|rt||||dt|r&t|||dt|r4t||p0||dt||d)as Map an A2A SDK exception to a LiteLLM A2A exception type. Args: original_exception: The original exception from the A2A SDK card_url: The URL from the agent card (if available) api_base: The original API base URL model: The model/agent name Returns: A mapped LiteLLM A2A exception Raises: A2ALocalhostURLError: If the error is a connection error to a localhost URL A2AConnectionError: If the error is a general connection error A2AAgentCardError: If the error is related to agent card issues A2AError: For other A2A-related errors ) localhost_urlbase_urloriginal_errorr,)messager r,)r0r,) rrr!rr r"rr r )r)r*r+r,rrrrmap_a2a_exception`s>  r1error agent_card a2a_client A2AClientType is_streamingc Csftrtdur td|rdnd}td|d|jd|jd|jd t||jt|j j |d S) a  Handle A2ALocalhostURLError by fixing the URL and creating a new client. This is called when we catch an A2ALocalhostURLError and want to retry with the corrected URL. Args: error: The localhost URL error agent_card: The agent card object to fix a2a_client: The current A2A client is_streaming: Whether this is a streaming request (for logging) Returns: A new A2A client with the fixed URL Raises: ImportError: If the A2A SDK is not installed NzRA2A SDK is required for localhost retry handling. Install it with: pip install a2az streaming zA2A z request to 'z ' failed: zF. Agent card contains localhost/internal URL. Retrying with base_url 'z'.) httpx_clientr3) A2A_SDK_AVAILABLE _A2AClient ImportErrorrwarningr-r/r.r _transportr8)r2r3r4r6Z request_typerrrhandle_a2a_localhost_retrys   r>)NNN)F)r&typingrrrZlitellm._loggingrZ"litellm.a2a_protocol.card_resolverrrZlitellm.a2a_protocol.exceptionsrr r r Zlitellm.constantsr Z a2a.clientr r5r9r:r;r Exceptionrr1r(r>rrrrsT     > C