o ưi @sdZddlmZmZmZddlmZddlmZddl m Z ddl m Z Gddde Z d ee d efd d Z ddeeefdeded efddZ ddeeefded efddZdS)z4 Common utilities for A2A (Agent-to-Agent) Protocol )AnyDictList) BaseModel)convert_content_list_to_str)BaseLLMException)AllMessageValuesc s:eZdZdZifdededeeefffdd ZZ S)A2AErrorz&Base exception for A2A protocol errors status_codemessageheaderscstj|||ddS)N)r r r )super__init__)selfr r r  __class__T/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/a2a/common_utils.pyrs  zA2AError.__init__) __name__ __module__ __qualname____doc__intstrrrr __classcell__rrrrr s r messagesreturncCs~g}|D]5}t|d}t|tr|dd}nt|tr%|dd}nt|dd}|r9||d|qd|S)at Convert OpenAI messages to a single prompt string for A2A agent. Formats each message as "{role}: {content}" and joins with newlines to preserve conversation history. Handles both string and list content. Args: messages: List of OpenAI-format messages Returns: Formatted prompt string with full conversation context )r roleuserz:  )r isinstancerZ model_dumpgetdictappendjoin)rZconversation_partsmsgZ content_textrrrrconvert_messages_to_prompts     r& r depth max_depthcCs|dus||kr dS|dg}g}|D]&}|ddkr'||ddqd|vr:t||d|}|r:||qd|S)aE Extract text content from A2A message parts. Args: message: A2A message dict with 'parts' containing text parts depth: Current recursion depth (internal use) max_depth: Maximum recursion depth to prevent infinite loops Returns: Concatenated text from all text parts Npartskindtext )r!r#extract_text_from_a2a_messager$)r r(r)r+Z text_partspartZ nested_textrrrr0?s   r0 response_dictc Cs|di}t|ts dSd|vrt|d|dS|d}|r&t|d|dS|d}|r9t|tr9t|d|dS|di}t|trR|d}|rRt|d|dS|d g}|rkt|dkrk|d}t|d|dSdS) a Extract text content from A2A response result. Args: response_dict: A2A response dict with 'result' containing message max_depth: Maximum recursion depth to prevent infinite loops Returns: Text from response message parts resultr*r+r)r(r)r artifactstatus artifacts)r!r r"r0len) r2r)r3r r4r5Zstatus_messager6Zfirst_artifactrrrextract_text_from_a2a_response_s6        r8N)rr')r')rtypingrrrZpydanticrZ8litellm.litellm_core_utils.prompt_templates.common_utilsrZ)litellm.llms.base_llm.chat.transformationrZlitellm.types.llms.openairr rr&rr0r8rrrrs4    !  !