o ưi@s\dZddlmZddlZGdddeZGdddeZGdd d eZGd d d eZdS) zv A2A Protocol Exceptions. Custom exception types for A2A protocol operations, following LiteLLM's exception pattern. )OptionalNcseZdZdZ       ddedededeed eejd eed eed eeffd d Z defddZ defddZ Z S)A2AErrorzm Base exception for A2A protocol errors. Follows the same pattern as LiteLLM's main exceptions.  a2a_agentNmessage status_code llm_providermodelresponselitellm_debug_info max_retries num_retriesc sb||_d||_||_||_||_||_||_|p&tj|jtj dddd|_ t |jdS)Nzlitellm.A2AError: POSTzhttps://litellm.ai)methodurl)rrequest) rrrr r r r httpxResponseRequestr super__init__) selfrrrr r r r r  __class__V/home/app/Keep/.python/lib/python3.10/site-packages/litellm/a2a_protocol/exceptions.pyrs  zA2AError.__init__returncCs8|j}|jr|d|jd7}|jr|d|j7}|S)Nz LiteLLM Retried: z timesz, LiteLLM Max Retries: )rr r )r_messagerrr__str__+s zA2AError.__str__cCs|S)N)r)rrrr__repr__3szA2AError.__repr__)rrNNNNN) __name__ __module__ __qualname____doc__strintrrrrrr __classcell__rrrrr s8  rcsfeZdZdZ      d dedeedeedeejdeedeed eeffd d Z Z 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 Nrrr r r r r c s&||_tj|dd|||||ddS)Nir)rrrr r r r r rrr)rrrr r r r r rrrrAs  zA2AConnectionError.__init__)NNNNNN) r r!r"r#r$rrrr%rr&rrrrr'7s. r'c sReZdZdZ    d dedeedeedeejdeef fdd ZZ 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 Nrrr r r cs"||_tj|dd|||ddS)Nir)rrrr r r r()rrrr r r rrrrbs zA2AAgentCardError.__init__)NNNN) r r!r"r#r$rrrrr&rrrrr)Xs" r)c s@eZdZdZ  d dededeedeeffdd ZZS) 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_errorr cs:||_||_||_d|d|d}tj|||ddS)Nz,Agent card contains localhost/internal URL 'z'. Retrying with base URL 'z'.)rrr )r+r,r-rr)rr+r,r-r rrrrrs zA2ALocalhostURLError.__init__)NN) r r!r"r#r$r Exceptionrr&rrrrr*usr*) r#typingrrr.rr'r)r*rrrrs +!