Merge pull request #39794 from BerriAI/litellm_v2_org_update_public

feat(organization): expose PATCH /v2/organization/{organization_id} in the OpenAPI schema
This commit is contained in:
ryan-crabbe-berri 2026-09-04 13:32:52 -07:00 • committed by GitHub
commit 189bd857f4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 10 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,16 @@ 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 litellm.proxy.proxy_server import get_openapi_schema
v2_path = get_openapi_schema()["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 {}."""