uPjo4dZddlZddlmZddlmZmZmZddlZddl m Z ddl m Z m Z ddlmZddlmZmZmZmZd d lmZmZmZd d lmZer ddlZdd lmZGdde Zy)z) A2A Protocol Transformation for LiteLLM N)Iterator) TYPE_CHECKINGAnyFinal)BaseModelResponseIterator) BaseConfigBaseLLMException)AllMessageValues)ChoicesMessage ModelResponseUsage)A2AErrorconvert_messages_to_promptextract_text_from_a2a_responseA2AModelResponseIterator)Loggingc<eZdZdZedededzdedzdeeefdzdeeefdeedzedzeeefdzff d Z dede efd Z d ededed e def d Z d%dedede ededededzdedzdefdZ d&dedzdedzdededede dzdefdZdede ededededef dZ d%dedej&dedddede edededddedzde dzdefdZ d'deezde de dzdefdZd eeefdeeeffd!Zd"ed#edeej6zdefd$Zy)( A2AConfigz Configuration for A2A (Agent-to-Agent) Protocol. Handles transformation between OpenAI and A2A JSON-RPC 2.0 formats. modelapi_baseNapi_keyheadersoptional_paramsreturncd|vr|jdddnd}|r| |||||fS ddlm}|j|}|r|'|jr|jj d}|j ro||j j d}||j j d}|r|}|j jD]\} } | d vs | |vs| || <|||fS#t$rYwxYw) a Resolve agent configuration from registry if model format is "a2a/". Extracts agent name from model string and looks up configuration in the agent registry (if available in proxy context). Args: model: Model string (e.g., "a2a/my-agent") api_base: Explicit api_base (takes precedence over registry) api_key: Explicit api_key (takes precedence over registry) headers: Explicit headers (takes precedence over registry) optional_params: Dict to merge additional litellm_params into Returns: Tuple of (api_base, api_key, headers) with registry values filled in /rNr)global_agent_registryurlrr)rrrr) split,litellm.proxy.agent_endpoints.agent_registryr!get_agent_by_nameagent_card_paramsgetlitellm_paramsitems ImportError) rrrrr agent_namer!agent agent_headerskeyvalues i/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/a2a/chat/transformation.py"resolve_agent_config_from_registryz,A2AConfig.resolve_agent_config_from_registry$s127:UlEKKQ/2 h2w7JwObWg- -  1BB:NE#(?(?$66::5AH''"'"6"6":":9"E/4/C/C/G/G /R (&3G',&:&:&@&@&B9 U&QQVYapVp38OC09 '))   sB0C,C, C,, C87C8c gdS)z*Return list of supported OpenAI parameters)stream temperature max_tokenstop_p)selfrs r0get_supported_openai_paramsz%A2AConfig.get_supported_openai_paramsbs   non_default_params drop_paramscV|jD]\}}|dk(s |dus||d<|S)z Map OpenAI parameters to A2A parameters. For A2A protocol, we need to map the stream parameter so transform_request can determine which JSON-RPC method to use. r3T)r))r8r;rrr<paramr/s r0map_openai_paramszA2AConfig.map_openai_paramsks@/446 2LE5 Ud],1) 2r:messagesr(c4d|vr d|vrd|d<|d||d<|S)a Validate environment and set headers for A2A requests. Args: headers: Request headers dict model: Model name messages: Messages list optional_params: Optional parameters litellm_params: LiteLLM parameters api_key: API key (optional for A2A) api_base: API base URL Returns: Updated headers dict z content-typez Content-Typezapplication/jsonzBearer Authorizationr7)r8rrr@rr(rrs r0validate_environmentzA2AConfig.validate_environments;4  (^7-J&8GN #  )0 ':GO $r:r3c>| td|jdS)a Get the complete A2A agent endpoint URL. A2A agents use JSON-RPC 2.0 at the base URL, not specific paths. The method (message/send or message/stream) is specified in the JSON-RPC request body, not in the URL. Args: api_base: Base URL of the A2A agent (e.g., "http://0.0.0.0:9999") api_key: API key (not used for URL construction) model: Model name (not used for A2A, agent determined by api_base) optional_params: Optional parameters litellm_params: LiteLLM parameters stream: Whether this is a streaming request (affects JSON-RPC method) Returns: Complete URL for the A2A endpoint (base URL) z%api_base is required for A2A providerr ) ValueErrorrstrip)r8rrrrr(r3s r0get_complete_urlzA2AConfig.get_complete_urls'6  DE Es##r:cttj}|s tdt |}dd|dgttjd}|j dd} | rdnd } d || d |id } | S) ab Transform OpenAI request to A2A JSON-RPC 2.0 format. Args: model: Model name messages: List of OpenAI messages optional_params: Optional parameters litellm_params: LiteLLM parameters headers: Request headers Returns: A2A JSON-RPC 2.0 request dict z3At least one message is required for A2A completionusertextkindrJroleparts messageIdr3Fzmessage/streamz message/sendz2.0message)jsonrpcidmethodparams)struuiduuid4rErr') r8rr@rr(r request_id full_context a2a_messager3rT request_datas r0transform_requestzA2AConfig.transform_requests,  - RS S9B %|<=TZZ\* (++He<,2( +.  r: raw_responsemodel_response logging_objLiteLLMLoggingObjr\encodingztiktoken.Encoding | None json_modec  |j} d| vrC| d}t|jd|j ddt |j t| }tdd t|d  g|_ ||_ | j d ttj|_ d dlm}|d|}|d|d}t%|dt'||||z|S#t$r3} t|jd| t |j d} ~ wwxYw#t$rY|SwxYw)ao Transform A2A JSON-RPC 2.0 response to OpenAI format. Args: model: Model name raw_response: HTTP response from A2A agent model_response: Model response object to populate logging_obj: Logging object request_data: Original request data messages: Original messages optional_params: Optional parameters litellm_params: LiteLLM parameters encoding: Encoding object api_key: API key json_mode: JSON mode flag Returns: Populated ModelResponse object zFailed to parse A2A response:  status_coderQrNerrorz A2A error: rQz Unknown errorstopr assistant)contentrN) finish_reasonindexrQrS) token_counterz gpt-3.5-turbo)rr@T)rrJcount_response_tokensusage) prompt_tokenscompletion_tokens total_tokens)json Exceptionrrfdictrr'rr r choicesrrVrWrXrS litellm.utilsrmsetattrr)r8rr^r_r`r\r@rr(rbrrc response_jsonergrJrmrprqs r0transform_responsezA2AConfig.transform_responsesrB #/#4#4#6M m #(1E(44%eii ?&K%LM\112 5]C $ $  "  %*--dC 4EF  3#0QY#ZM'4?QUmq'r  "/&7!.1B!B q (448<\112  j   s)C6=7D56 D2?.D--D25 EEstreaming_response sync_streamct|||S)a Get streaming iterator for A2A responses. Args: streaming_response: Streaming response iterator sync_stream: Whether this is a sync stream json_mode: JSON mode flag Returns: A2A streaming iterator )r|r}rcr)r8r|r}rcs r0get_model_response_iteratorz%A2AConfig.get_model_response_iteratorVs"(1#  r:rQc|jdd}|jdd}|dt|dgttjdS)z Convert OpenAI message to A2A message format. Args: message: OpenAI message dict Returns: A2A message dict rjrNrIrJrKrM)r'rVrWrX)r8rQrjrNs r0_openai_message_to_a2a_messagez(A2AConfig._openai_message_to_a2a_messagemsP!Y3kk&&1%s7|<=TZZ\*  r: error_messagerfclt|tjr t|n|}t |||S)z-Return appropriate error class for A2A errorsre) isinstancehttpxHeadersrur)r8rrfr headers_dicts r0get_error_classzA2AConfig.get_error_classs30:'5==/Qd7mW^ #!   r:)NN)N)F)__name__ __module__ __qualname____doc__ staticmethodrVrurtupler1listr9boolr?r rCrGr]rResponser r{rrrrintrr rr7r:r0rrs ;*;**;*t;*c3h$& ;* c3h ;* sTz3:tCH~'<< = ;*;*z  c       6##!!!'( !  !  !t!*! !T# $* $t $ $  $  $t  $  $D33'(3 3  3  3 3@#!%[[nn[& [ ) [  ['([[[-[t[$;[ [B"'  $sN  $;  #  . d38n cSVh & S s TTYTaTaMa fv r:r) rrWcollections.abcrtypingrrrr)litellm.llms.base_llm.base_model_iteratorr)litellm.llms.base_llm.chat.transformationrr litellm.types.llms.openair litellm.types.utilsr r r r common_utilsrrrstreaming_iteratorrtiktoken*litellm.litellm_core_utils.litellm_loggingrrarr7r:r0rsP $,, OR6FF 9Wk k r: