uPj dZddlmZmZmZddlmZddlmZm Z ddl m Z m Z m Z mZddlmZerddlmZ ddlmZmZmZd ZGd d Z ddeded zded zded zdef dZ ddededddeddf dZy #e$r d Zd Zd Zd ZYEwxYw)z` A2A Protocol Exception Mapping Utils. Maps A2A SDK exceptions to LiteLLM A2A exception types. ) TYPE_CHECKINGAnyFinal)verbose_logger)is_localhost_or_internal_urlset_agent_card_url)A2AAgentCardErrorA2AConnectionErrorA2AErrorA2ALocalhostURLError)CONNECTION_ERROR_PATTERNS)Client)r ClientConfig create_clientTFNcdeZdZdZededefdZededzdefdZededefdZ y) A2AExceptionCheckerszA Helper class for checking various A2A error conditions. error_strreturncvt|tsy|jtfdtDS)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&K|]}|v ywN.0patternerror_str_lowers q/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/a2a_protocol/exception_mapping_utils.py z;A2AExceptionCheckers.is_connection_error..9sW'7o-W) isinstancestrloweranyr )rrs @ris_connection_errorz(A2AExceptionCheckers.is_connection_error*s0)S)!*!2W=VWWWurlNct|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&s ris_localhost_urlz%A2AExceptionCheckers.is_localhost_url;s,C00r%cvt|tsy|jgd}tfd|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 agentc3&K|]}|v ywrrrs rrz;A2AExceptionCheckers.is_agent_card_error..^sQ'7o-Qr)r r!r"r#)ragent_card_patternsrs @ris_agent_card_errorz(A2AExceptionCheckers.is_agent_card_errorHs:)S)!*!2& Q=PQQQr%) __name__ __module__ __qualname____doc__ staticmethodr!boolr$r(r,rr%rrr%syXsXtXX  1cDj 1T 1 1RsRtRRr%roriginal_exceptioncard_urlapi_basemodelrc@t|}|r;|r9tj|r$tj|rt ||||tj |rt |||tj|rt||xs||t||)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_errorr6)messager&r6)r;r6) r!rr(r$r r,r r r )r3r4r5r6rs rmap_a2a_exceptionr<as0-.I   1 1( ; 4 4Y ?""-   // :  // : $H    r%error agent_card a2a_client A2AClientType is_streamingcKts td| td|rdnd}tjd||j |j |jt||jt|dd}| tdt|t||  d{}||_ t|d d|_ ||_|S7&w) 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 zVA2A SDK is required for localhost retry handling. Install it with: pip install a2a-sdkNzsCannot retry A2A localhost URL fix: no agent card is available to rewrite, so the upstream URL cannot be corrected.z streaming zjA2A %srequest to '%s' failed: %s. Agent card contains localhost/internal URL. Retrying with base_url '%s'._litellm_httpx_clientz|Cannot retry A2A localhost URL fix: the client was not created by create_a2a_client, so no LiteLLM httpx client is attached.) httpx_client streaming) client_config_litellm_call_context)A2A_SDK_AVAILABLE ImportError RuntimeErrorrwarningr8r:r9rgetattrrrrDrH_litellm_agent_card)r=r>r?rA request_typerE new_clients rhandle_a2a_localhost_retryrQs0 rss @  +7,BLt    z5>>2 "*.EtLL I  ,"%" J(4J$'.+T(J$&0J" sB'C)C*'C)NNN)F)r0typingrrrlitellm._loggingr"litellm.a2a_protocol.card_resolverrrlitellm.a2a_protocol.exceptionsr r r r litellm.constantsr a2a.clientrr@rrrIrJr Exceptionr!r<r2rQrr%rrYs -,+ 82>>9R9R|  <!<Dj<Dj< : <  <F C CC C C  CE FLM s A88 BB