tPj pdZddlmZmZddlmZddlZddlmZddl m Z m Z  d de de d e dzd edzfd Zy) z A2A Agent Routing Handles routing for A2A agents (models with "a2a/" prefix). Looks up agents in the registry and injects their API base URL. )AnyFinal) HTTPExceptionN)verbose_proxy_logger)LitellmUserRolesUserAPIKeyAuthdata route_typeuser_api_key_dictreturncdKddlm}ddlm}ddlm}m}|jdd}t|tr|jdsy|d d}||d{} | 3tjd ||j||} || |d |duxrF|jtjk(xs'|jtjj k(} | s8|j#| j$| d{} | st'dd|d| j(rd| j(vr3tjd||j||} || |d | j(d|d<tj*d||dt-t.|di|S7P7w)z Route A2A agent requests directly to litellm with injected API base. Returns None if not an A2A request (allows normal routing to continue). r)AgentRequestHandler)get_agent_with_read_through)ROUTE_ENDPOINT_MAPPINGProxyModelNotFoundErrormodelza2a/Nz&[A2A] Agent '%s' not found in registryF)route model_name!retryable_with_model_read_through)agent_iduser_api_key_authizAgent 'zC' is not allowed for your key/team. Contact proxy admin for access.) status_codedetailurlz&[A2A] Agent '%s' has no URL configuredapi_basez[A2A] Routing %s to %s);litellm.proxy.agent_endpoints.auth.agent_permission_handlerr0litellm.proxy.common_utils.registry_read_throughrlitellm.proxy.route_llm_requestrrget isinstancestr startswithrerror user_roler PROXY_ADMINvalueis_agent_allowedrragent_card_paramsdebuggetattrlitellm) r r r rrrrr agent_nameagent route_nameis_admin is_alloweds n/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/agent_endpoints/a2a_routing.pyroute_a2a_agent_requestr5s "-J j# &j.C.CF.K#12J5Z@@E }""#KZX+// JG %J:qvww(t3##'7'C'CC M  & &*:*F*F*L*L L  "5"F"F^^/#G#    ,op   " "e53J3J&J""#KZX+// JG %J:qvww..u5D7T*EUV ,77zl , 4t 44AA s&AF0F+B'F0F.B(F0.F0)N)__doc__typingrrfastapirr.litellm._loggingrlitellm.proxy._typesrrdictr$r5rr4r=sW!1A 04@5 @5@5&,@5 4Z @5r<