AHjtdZddlmZddlZGddeZGddeZGdd eZGd d eZy) zv A2A Protocol Exceptions. Custom exception types for A2A protocol operations, following LiteLLM's exception pattern. )OptionalNceZdZdZ ddedededeedeejdeedeed eeffd Z d efd Z d efd Z xZ S)A2AErrorzm Base exception for A2A protocol errors. Follows the same pattern as LiteLLM's main exceptions. message status_code llm_providermodelresponselitellm_debug_info max_retries num_retriesc ||_d||_||_||_||_||_||_|xs6tj|jtjdd|_ t |1|jy)Nzlitellm.A2AError: POSTzhttps://litellm.ai)methodurl)rrequest) rrrr r r r httpxResponseRequestr super__init__) selfrrrr r r r r __class__s W/root/tools/cai/cai_env/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__3s||~r)i a2a_agentNNNNN) __name__ __module__ __qualname____doc__strintrrrrrr! __classcell__rs@rrr s'#-1,0%)%)''' ' } ' 5>>* '%SM'c]'c]'0#rrceZdZdZ d dedeedeedeejdeedeedeeffd Z xZ 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 rrr r r r r c B||_t| |dd|||||y)Nir")rrrr r r r r rrr) rrrr r r r r rs rrzA2AConnectionError.__init__As6 $1##  r)NNNNNN) r#r$r%r&r'rrrr(rr)r*s@rr,r,7s"#-1,0%)%)  c] }  5>>*  %SM  c] c]  rr,c neZdZdZ ddedeedeedeejdeef fd ZxZ S) 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 rrr r r c>||_t| |dd|||y)Nir")rrrr r r r.)rrrr r r rs rrzA2AAgentCardError.__init__bs0 $1  r)NNNN) r#r$r%r&r'rrrrr)r*s@rr0r0Xse"#-1,0   c] }  5>>*  %SM   rr0c FeZdZdZ ddededeedeeffd 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 localhost_urlbase_urloriginal_errorr cf||_||_||_d|d|d}t||||y)Nz,Agent card contains localhost/internal URL 'z'. Retrying with base URL 'z'.)rrr )r4r5r6rr)rr4r5r6r rrs rrzA2ALocalhostURLError.__init__sY+  ,;=/J''/j 4    r)NN) r#r$r%r&r'r Exceptionrr)r*s@rr3r3usI "/3#    !+  }   rr3) r&typingrrr8rr,r0r3rrr;sE  (y(V  B  :! -! r