o ưi@sFdZddlmZmZddlZddlmZdededeefdd Z dS) 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. )AnyOptionalN)verbose_proxy_loggerdata route_typereturnc Csddlm}ddlm}m}|dd}t|tr|ds dS|dd}| |}|durDt d |d |||}|||d |j rLd |j vrat d |d |||}|||d |j d |d<t d|d|dtt|di|S)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)global_agent_registry)ROUTE_ENDPOINT_MAPPINGProxyModelNotFoundErrormodelza2a/Nz [A2A] Agent 'z' not found in registry)Zroute model_nameurlz' has no URL configuredZapi_basez[A2A] Routing z to )Z,litellm.proxy.agent_endpoints.agent_registryrZlitellm.proxy.route_llm_requestr r get isinstancestr startswithZget_agent_by_namererrorZagent_card_paramsdebuggetattrlitellm) rrrr r rZ agent_nameZagentZ route_namerr`/home/app/Keep/.python/lib/python3.10/site-packages/litellm/proxy/agent_endpoints/a2a_routing.pyroute_a2a_agent_requests$        r) __doc__typingrrrZlitellm._loggingrdictrrrrrrs