scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test(organization): type the legacy update helper's request body precisely

Browse source
This commit is contained in:
ryan-crabbe-berri 2026-09-04 11:21:37 -07:00
parent fb4cbccbf5
commit 4f3b02360e
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_litellm/proxy/management_endpoints/test_organization_endpoints.py
View file

@ -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