uPj dZddlmZmZddlmZddlmZddlm Z ddl m Z Gdde Z d e e d efd Zdd eeefd eded efdZddeeefded efdZy)z4 Common utilities for A2A (Agent-to-Agent) Protocol )AnyFinal) BaseModel)convert_content_list_to_str)BaseLLMException)AllMessageValuesc >eZdZdZifdededeeefffd ZxZ S)A2AErrorz&Base exception for A2A protocol errors status_codemessageheadersc*t||||y)N)r r r )super__init__)selfr r r __class__s b/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/a2a/common_utils.pyrzA2AError.__init__s #  ) __name__ __module__ __qualname____doc__intstrdictrr __classcell__)rs@rr r s90 #%      c3h   rr messagesreturnc^g}|D]}t|}t|tr!|jj dd}n>t|t r|j dd}nt |j dd}|s|j |d|dj|S)ah 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 isinstancer model_dumpgetrappendjoin)rconversation_partsmsg content_textr s rconvert_messages_to_promptr+ s!# A23?  c9 %>>#''7D T "7766*D9==0D   % %b&? @ A 99' ((rr depth max_depthc(|||k\ry|jdg}g}|D]a}|jddk(r"|j|jdd9d|vs>t||dz|}|sQ|j|cdj|S)a= 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 partskindtext )r%r&extract_text_from_a2a_messager')r r,r-r0 text_partspart nested_texts rr5r5@s%9,;;w+E#%J/ 88F v %   dhhvr2 3 _7eaiSK!!+./ 88J r response_dictc|jdi}t|tsyd|vrt|d|S|jd}|rt|d|S|jd}|rt|trt|d|S|jdi}t|tr!|jd}|rt|d|S|jd g}|r!t |dkDr|d}t|d|Sy) 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/r0r)r,r-r artifactstatus artifacts)r%r#rr5len) r9r-r;r r<r=status_messager>first_artifacts rextract_text_from_a2a_responserB^s"%%h3F fd #&,V1 RRZZ *G,WASSjj,HJx.,XQ)TTJJx,F&$ & 9 5 0qT]^ ^zz+r2IS^a' )! ,^1PYZZ rN)r )rC)rtypingrrpydanticr8litellm.litellm_core_utils.prompt_templates.common_utilsr)litellm.llms.base_llm.chat.transformationrlitellm.types.llms.openairr listrr+rrr5rBrrrKsG6    ).>)?)C)@ 4S> # VY cf <1$sCx.1S1Z]1r