From 4f3b02360e6a0ff23cfa821782f7fea2d3d7a90b Mon Sep 17 00:00:00 2001 From: ryan-crabbe-berri Date: Fri, 4 Sep 2026 11:21:37 -0700 Subject: [PATCH] test(organization): type the legacy update helper's request body precisely --- .../proxy/management_endpoints/test_organization_endpoints.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_litellm/proxy/management_endpoints/test_organization_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_organization_endpoints.py index 5c2a0bdde3d..dc500df6fd6 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_organization_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_organization_endpoints.py @@ -963,7 +963,9 @@ async def test_v2_serializes_model_max_budget_on_budget_write(monkeypatch): assert json.loads(written) == {"gpt-4o": {"max_budget": 10}} -async def _run_legacy_update_organization(monkeypatch, *, body: dict, existing_budget_id: str): +async def _run_legacy_update_organization( + monkeypatch: pytest.MonkeyPatch, *, body: dict[str, object], existing_budget_id: str +) -> AsyncMock: from litellm.proxy._types import LitellmUserRoles, UserAPIKeyAuth from litellm.proxy.management_endpoints import organization_endpoints from litellm.proxy.management_endpoints.organization_endpoints import update_organization