uPjhdZddlZGddeZGddeZGddeZGd d eZy) zv A2A Protocol Exceptions. Custom exception types for A2A protocol operations, following LiteLLM's exception pattern. NceZdZdZ ddededededzdejdzdedzd edzd edzffd Zd efd Z d efdZ xZ S)A2AErrorzm Base exception for A2A protocol errors. Follows the same pattern as LiteLLM's main exceptions. Nmessage status_code llm_providermodelresponselitellm_debug_info max_retries num_retriesc ||_d||_||_||_||_||_||_|xs6tj|jtjdd|_ t |1|jy)Nzlitellm.A2AError: POSTzhttps://litellm.ai)methodurl)rrequest) rrrrr r r httpxResponseRequestr super__init__) selfrrrrr r r r __class__s d/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/a2a_protocol/exceptions.pyrzA2AError.__init__s'+G95 ( "4&&  ENN((MM5IJ%   &returnc|j}|jr|d|jdz }|jr|d|jz }|S)Nz LiteLLM Retried: z timesz, LiteLLM Max Retries: )rr r )r_messages r__str__zA2AError.__str__)sV<<    ,T-=-=,>fE EH    1$2B2B1CD DHrc"|jS)N)r)rs r__repr__zA2AError.__repr__1s||~r)i a2a_agentNNNNN) __name__ __module__ __qualname____doc__strintrrrrr __classcell__rs@rrr s' *.)-"&"&''' ' Tz ' ..4' ' $J'4Z'4Z'0#rrceZdZdZ d dededzdedzdej dzdedzdedzd edzffd ZxZ S) A2AConnectionErrorz Raised when connection to an A2A agent fails. This typically occurs when: - The agent is unreachable - The agent card contains a localhost/internal URL - Network issues prevent connection Nrrrr r r r c B||_t| |dd|||||y)Nir!)rrrrr r r r rrr) rrrrr r r r rs rrzA2AConnectionError.__init__?s6 $1##  r)NNNNNN) r"r#r$r%r&rrr'rr(r)s@rr+r+5s *.)-"&"&  4Z Tz  ..4'   $J  4Z 4Z  rr+c neZdZdZ d dededzdedzdej dzdedzf fd ZxZS) A2AAgentCardErrorz Raised when there's an issue with the agent card. This includes: - Failed to fetch agent card - Invalid agent card format - Missing required fields Nrrrr r c>||_t| |dd|||y)Nir!)rrrrr r r-)rrrrr r rs rrzA2AAgentCardError.__init__`s0 $1  r)NNNN) r"r#r$r%r&rrrr(r)s@rr/r/Vse *.)-   4Z Tz  ..4'   $J   rr/c FeZdZdZ ddedededzdedzffd ZxZS) A2ALocalhostURLErrora Raised when an agent card contains a localhost/internal URL. Many A2A agents are deployed with agent cards that contain internal URLs like "http://0.0.0.0:8001/" or "http://localhost:8000/". This error indicates that the URL needs to be corrected and the request should be retried. Attributes: localhost_url: The localhost/internal URL found in the agent card base_url: The public base URL that should be used instead original_error: The original connection error that was raised N localhost_urlbase_urloriginal_errorrcf||_||_||_d|d|d}t||||y)Nz,Agent card contains localhost/internal URL 'z'. Retrying with base URL 'z'.)rrr)r3r4r5rr)rr3r4r5rrrs rrzA2ALocalhostURLError.__init__sO+  ,@Ojksjttvw   r)NN)r"r#r$r%r& Exceptionrr(r)s@rr2r2ssI ",0    "D(  Tz   rr2)r%rr7rr+r/r2rrr9sB  (y(V  B  : - r