mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-06 08:16:43 +00:00
feat(organization): expose PATCH /v2/organization/{organization_id} in the OpenAPI schema
This commit is contained in:
parent
62087c5d5f
commit
3e4a884b25
2 changed files with 14 additions and 1 deletions
|
|
@ -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,
|
||||
|
|
|
|||
|
|
@ -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 {}."""
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue