From bca9abeaef62d0bda51210aca04fa1df92161a4c Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Wed, 6 May 2026 15:03:19 -0700 Subject: [PATCH] feat(agent_session_endpoints): add module init exposing routers --- .../proxy/agent_session_endpoints/__init__.py | 35 +++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 litellm/proxy/agent_session_endpoints/__init__.py diff --git a/litellm/proxy/agent_session_endpoints/__init__.py b/litellm/proxy/agent_session_endpoints/__init__.py new file mode 100644 index 00000000000..24b347167e5 --- /dev/null +++ b/litellm/proxy/agent_session_endpoints/__init__.py @@ -0,0 +1,35 @@ +""" +Agent / Session / Run REST API surface — `/v2/agents`, `/v2/sessions`, +`/v2/sessions/{sid}/runs`. + +The `/v1/agents` namespace is owned by the existing A2A registry +(``litellm/proxy/agent_endpoints/``). To avoid collision, all endpoints +in this module mount under ``/v2/``. + +Exports the FastAPI routers consumed by ``proxy_server.py``: + + * ``agent_router`` — /v2/agents CRUD + * ``session_router`` — /v2/sessions CRUD + followup + conversation + * ``run_router`` — /v2/sessions/{sid}/runs (+ events SSE, cancel) + * ``internal_router`` — /v2/sessions/{sid}/internal/* (daemon callbacks) +""" + +from litellm.proxy.agent_session_endpoints.agent_endpoints import ( + router as agent_router, +) +from litellm.proxy.agent_session_endpoints.internal_endpoints import ( + router as internal_router, +) +from litellm.proxy.agent_session_endpoints.run_endpoints import ( + router as run_router, +) +from litellm.proxy.agent_session_endpoints.session_endpoints import ( + router as session_router, +) + +__all__ = [ + "agent_router", + "session_router", + "run_router", + "internal_router", +]