chore: initial public snapshot for github upload

llm-gateway-competitors/litellm-wheel-src/litellm/proxy/agent_endpoints/a2a_routing.py

@@ -0,0 +1,53 @@
+"""
+A2A Agent Routing
+
+Handles routing for A2A agents (models with "a2a/<agent-name>" prefix).
+Looks up agents in the registry and injects their API base URL.
+"""
+
+from typing import Any, Optional
+
+import litellm
+from litellm._logging import verbose_proxy_logger
+
+
+def route_a2a_agent_request(data: dict, route_type: str) -> Optional[Any]:
+    """
+    Route A2A agent requests directly to litellm with injected API base.
+
+    Returns None if not an A2A request (allows normal routing to continue).
+    """
+    # Import here to avoid circular imports
+    from litellm.proxy.agent_endpoints.agent_registry import global_agent_registry
+    from litellm.proxy.route_llm_request import (
+        ROUTE_ENDPOINT_MAPPING,
+        ProxyModelNotFoundError,
+    )
+
+    model_name = data.get("model", "")
+
+    # Check if this is an A2A agent request
+    if not isinstance(model_name, str) or not model_name.startswith("a2a/"):
+        return None
+
+    # Extract agent name (e.g., "a2a/my-agent" -> "my-agent")
+    agent_name = model_name[4:]
+
+    # Look up agent in registry
+    agent = global_agent_registry.get_agent_by_name(agent_name)
+    if agent is None:
+        verbose_proxy_logger.error(f"[A2A] Agent '{agent_name}' not found in registry")
+        route_name = ROUTE_ENDPOINT_MAPPING.get(route_type, route_type)
+        raise ProxyModelNotFoundError(route=route_name, model_name=model_name)
+
+    # Get API base URL from agent config
+    if not agent.agent_card_params or "url" not in agent.agent_card_params:
+        verbose_proxy_logger.error(f"[A2A] Agent '{agent_name}' has no URL configured")
+        route_name = ROUTE_ENDPOINT_MAPPING.get(route_type, route_type)
+        raise ProxyModelNotFoundError(route=route_name, model_name=model_name)
+
+    # Inject API base and route to litellm
+    data["api_base"] = agent.agent_card_params["url"]
+    verbose_proxy_logger.debug(f"[A2A] Routing {model_name} to {data['api_base']}")
+
+    return getattr(litellm, f"{route_type}")(**data)