tPjA (UdZddlmZddlmZddlmZmZddlm Z ddl m Z ddl m Z edZeezd zZeeefZed d hZeed <ehd Zeed<de defdZdegefdededefdZdedededefdZdedededefdZdedededefdZd edefd!Zd ededefd"Zd#edefd$Z d%ededededef d&Z!d%eded zfd'Z"d%edededefd(Z#d%edededefd)Z$d%ededefd*Z%d%eded zfd+Z&d%ededefd,Z'd%ededefd-Z(d%ededefd.Z)d%eded zfd/Z*d%edededefd0Z+d%edededefd1Z,d ededefd2Z-d%ed3ede fd4Z.d%ed3ede fd5Z/dededefd6Z0dedefd7Z1d8ed9eeded zfd:Z2dedefd;Z3ded edefd?Z4y )@a Normalize A2A JSON-RPC payloads to the protocol version LiteLLM serves for an agent. LiteLLM fronts upstream agents and lets an admin pin the protocol version it speaks to clients (``0.3`` or ``1.0``) per agent. Upstream responses may arrive in either wire shape, so every response, stream event, forwarded request and extended card is converted to the served version here. Conversion is shape-detecting (we infer the payload's current version rather than trusting a stored one) and best-effort: any failure falls back to returning the input unchanged so a conversion bug can never break an otherwise-valid response. The two wire shapes: - ``0.3``: JSON dump of the compat pydantic types, discriminated by a ``kind`` field (``message`` / ``task`` / ``status-update`` / ``artifact-update``). A send result is the bare object. - ``1.0``: protobuf JSON (``MessageToDict``), a oneof envelope keyed by ``message`` / ``task`` / ``statusUpdate`` / ``artifactUpdate`` with no ``kind``. A ``Task`` result is a bare object without ``kind``. )Callable) ModuleType)FinalLiteral) BaseModel)verbose_proxy_logger)normalize_protocol_version)0.31.0Nmessagetask_V1_SEND_ENVELOPE_KEYS>r r statusUpdateartifactUpdate_V1_STREAM_ENVELOPE_KEYSmodelreturnc*|jdddS)z>Dump a compat (0.3) pydantic model to its camelCase wire dict.Tjson)by_alias exclude_nonemode) model_dump)rs f/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/a2a/version_convert.py_dump_03r,s   T6  JJconvertfallbacklabelcp |S#t$r#}tjd|||cYd}~Sd}~wwxYw)z@Run a conversion, returning ``fallback`` unchanged if it raises.zA2A %s conversion failed: %sN) Exceptionrdebug)rrres r _best_effortr$1s7y ""#A5!Ls 5055contenttargetmethodc |jd|S|jd}t|ts|St|||t |jd}||ur|Si|d|iS)zuConvert a JSON-RPC response's ``result`` to ``target``. Errors and non-dict results pass through untouched. errorresultid)r' request_id)get isinstancedict_convert_result_as_request_id)r%r&r'r* converteds rnormalize_jsonrpc_responser3:sw {{7'KK)F fd #&vvfQ_`g`k`klp`qQrsIF +g +x ++reventr,c|jd|S|jd}t|ts|St|||}||ur|Si|d|iS)zDConvert a single streamed JSON-RPC event's ``result`` to ``target``.r)r*r,)r-r.r/_convert_stream_result)r4r&r,r*r2s rnormalize_stream_eventr8Ks` yy% IIh'F fd # -ffTIF )e )Xy ))rparamsservedc@|dk(rStfdddS)aDown-convert forwarded request ``params`` from the served version to 0.3. Upstream agents in this proxy pivot on 0.3 wire format, so when LiteLLM serves 1.0 the inbound params must be lowered before forwarding. A no-op when the served version is already 0.3. r ctS)N)r')_lower_request_params)r'r9srz*normalize_request_params..cs%fV<rzrequest params ()rr$)r9r:r's` `rnormalize_request_paramsrBYs0 < * rcardcNt|jd}||Sd|vrdSdS)anInfer the wire version of an agent card dict. ``protocolVersion`` is the authoritative indicator; semver values normalize to their major.minor (``"0.3.0"`` -> ``"0.3"``). Fall back to presence of ``supportedInterfaces`` (a 1.0-only field) only when the explicit field is absent or unrecognized; cards carrying neither signal are treated as 0.3. protocolVersionsupportedInterfacesr r )r r-)rC normalizeds r_detect_card_versionrHis63488z&normalize_agent_card..s 3D& Arz agent cardr@)r.r/rHr$)rCr&currents`` rnormalize_agent_cardrNwsI dD ! )$/G&&E/6Kt6S A4| \\rvaluec6t|ttfr|SdSrK)r.strint)rOs rr1r1susCj15;t;rr*cn|dk(rt|||S|dvr t||S|dk(r t||S|S)Nz message/sendr6) tasks/get tasks/cancel tasks/list)_convert_send_result _convert_task_convert_list_tasks_result)r*r&r'r,s rr0r0sK#FFzJJ ..VV,, )&&99 Mrc>d|vry|jtzryyNkindr r )keysrr*s r_detect_send_versionr_s"  {{}-- rcVt}||k(rStfddS)NctSrK)_send_result_tor,r*r&srr>z&_convert_send_result..s ;rz send resultr@)r_r$r*r&r,rMs``` rrWrWs4)&1G'V+ ; rcFddlm}m}m}m}m}m}|dk(rLt||} |j|j| t|nd| } ||| dS|j} ||| d t|| |jjS) Nr) MessageToDict ParseDictpb2_v10to_compat_send_message_responseto_core_send_message_response types_v03r r+r*)rootFpreserving_proto_field_nameTignore_unknown_fields)a2a.compat.v0_3.conversionsrfrgrhrirjrk_validate_message_or_taskSendMessageResponseSendMessageSuccessResponserQrrnr*) r*r&r,rfrgrhrirjrk compat_resultresponsepbs rrbrbs8K #7755&0&<3z?"$68   )( 3(-  ++-B fb5 3B CHHOO PPrcHdvrdnd}|k(rStfddS)Nr\r r ctSrK)_task_tor*r&srr>z_convert_task..s 8rr r@rAr*r&rMs`` rrXrXs.*0F*:G& 8& OOrc|jd}t|tr|sy|d}t|tsyd|vrdSdS)Ntasksrr\r r )r-r.listr/)r*rfirsts r_detect_list_tasks_versionrsF::g&E eT "%8E eT "eO5..rcRt}||k(rStfddS)NctSrK)_list_tasks_result_tor}srr>z,_convert_list_tasks_result..s%ff5rzlist tasks resultr@)rr$r~s`` rrYrYs4/7G'V+ 5! rc |jd}t|ts|Si|d|Dcgc] }t|tr t ||n|"c}iScc}w)Nr)r-r.rr/r|)r*r&ritems rrrs`::g&E eT "    Y^_QUJtT,B(4(L_ _s%Acddlm}m}m}m}m}m}|dk(r+||jj|}||dS|j} ||| dt|| S)Nr)rfrgrhto_compat_task to_core_taskrkr FroTrq) rsrfrgrhrrrkTaskmodel_validater) r*r&rfrgrhrrrkcorerys rr|r|sb"9>>#@#@#HITuEE B fb5 N2& ''rc>d|vry|jtzryyr[)r]rr^s r_detect_stream_versionrs"  {{}// rcVt}||k(rStfddS)NctSrK)_stream_result_torcsrr>z(_convert_stream_result..s!&&*=rz stream eventr@)rr$rds``` rr7r7 s4+F3G'V+ = rcddlm}m}m}m}m}m}|dk(rz'_lower_request_params..cs(9&AwOeOeOg:hjlmttrc\t|jdjSr)rCancelTaskRequestr9)rrgrhrs rr>z'_lower_request_params..ds+ )&Aw?X?X?Z*[]_ ` g grc\t|jdjSr)rSubscribeToTaskRequestr9)rrgrhrs rr>z'_lower_request_params..gs+ /y!WEcEcEe0fhj k r rrcntt|jdjSr)r(_flatten_create_push_notification_paramsTaskPushNotificationConfigr9)rrgrhrs rr>z'_lower_request_params..js; Bz'_lower_request_params..ts- ?y!W%Q%Q%STVX frc\t|jdjSr)r&ListTaskPushNotificationConfigsRequestr9)rrgrhrs rr>z'_lower_request_params..ys- @y!W%S%S%UVXZ frc\t|jdjSr)r'DeleteTaskPushNotificationConfigRequestr9)rrgrhrs rr>z'_lower_request_params..~s- By!W%T%T%VWY[ fr)rTrUztasks/resubscribez tasks/pushNotificationConfig/setz tasks/pushNotificationConfig/getz!tasks/pushNotificationConfig/listz#tasks/pushNotificationConfig/delete) _lower_list_tasks_paramsrsrgrhrrrrrrrr-r) r9r' loweringslowerrgrhrrrrrrrs @@@@@@@@@rr=r=Rs| '//   u   - - . 0 9!DID==(E } E&M ""rcNddlm}m}m}m}|j }t |||||d}t|jj|j}td|jD}t||} | |jdd|S| |d<|S)Nr)rfrgrhrkFroc3FK|]}t|jywrK)rQrO).0members r z+_lower_list_tasks_params..s'\fFLL(9'\s!status)rsrfrgrhrkListTasksRequestrrQ TaskStateNamer frozenset_proto_task_state_name_to_0_3r) r9rfrgrhrkprotor status_namevalid_0_3_values compat_statuss rrrs++-E 9fe$"5eLGW..33ELLABK''\ H[H['\\8FVWM Hd# N* Nrnamerc|jd}|dk(ry|jjdd}||vr|SdS)aMap a 1.0 protobuf ``TaskState`` enum name to its 0.3 wire string. The ``TASK_STATE_`` enum names line up with the 0.3 wire values once the prefix is dropped and underscores become dashes, so no private SDK mapping is needed. The result is validated against the 0.3 enum's own values; an unspecified or unrecognized state yields ``None`` so the status filter is dropped. TASK_STATE_ UNSPECIFIEDN_-) removeprefixrreplace)rrbase candidates rrrsJ##M2D }zz|++C5I!%559?4?rct|}|jdd}|jdd}||n|}t|ts|S|jdd}t|tr;|j dr*d|vr&|j dj dd|d<|jd dx}r d |vr||d <|j||S) zNMerge 1.x create envelope fields (parent/configId/config) into flat pb fields.configNpushNotificationConfigparentztasks/taskId/rconfigIdr+)r/rr.rQ startswithrsplitupdate)r9flatr push_confignestedr config_ids rrrsv,DHHXt,F":DAK$0FkF fd # HHXt,F&#6#4#4X#>8SWCW,,X6<rs *%!1C \ " #I  V  )9f*= >>"+,a"b%bKIK(K (2x<0HPSX`,,*,QT,Ya," *( *J *y *]e * X z c V^  =x =J = ]x ] ] ]<&