feat(organization): expose PATCH /v2/organization/{organization_id} in the OpenAPI schema

This commit is contained in:
ryan-crabbe-berri 2026-09-04 13:03:18 -07:00
parent 62087c5d5f
commit 3e4a884b25
2 changed files with 14 additions and 1 deletions

View file

@ -764,7 +764,6 @@ async def handle_update_object_permission(
tags=["organization management"],
dependencies=[Depends(user_api_key_auth)],
response_model=LiteLLM_OrganizationTableWithMembers,
include_in_schema=False,
)
async def update_organization_v2(
organization_id: str,

View file

@ -726,6 +726,20 @@ async def _run_update_organization_v2(
return mock_prisma_client
def test_v2_update_route_is_public_in_openapi():
"""PATCH /v2/organization/{organization_id} is a public route: hiding it again (include_in_schema=False)
would drop it from openapi.json, /docs, and the generated UI API types."""
from fastapi import FastAPI
from litellm.proxy.management_endpoints.organization_endpoints import router
app = FastAPI()
app.include_router(router)
v2_path = app.openapi()["paths"].get("/v2/organization/{organization_id}")
assert v2_path is not None
assert "patch" in v2_path
@pytest.mark.asyncio
async def test_v2_update_clears_tpm_limit_and_metadata(monkeypatch):
"""A cleared tpm_limit is written to the budget row as None; a cleared metadata is written as {}."""