scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-05 02:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 49

test(proxy): mock get_configured_token_limits in model_info tests

Browse source 
Co-Authored-By: Ishaan Jaffer <155045088+ishaan-berri@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-07-18 17:41:40 +00:00
parent 23d990b2e5
commit bebb7b9ba4
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/test_litellm/proxy/management_endpoints/test_model_management_endpoints.py
View file

@ -1727,6 +1727,7 @@ class TestModelInfoEndpoint:
"gpt-3.5-turbo",
]
mock_router.get_model_access_groups.return_value = {}
mock_router.get_configured_token_limits.return_value = (None, None)
mock_get_key_models.return_value = ["gpt-4", "claude-3"]
mock_get_team_models.return_value = ["gpt-3.5-turbo"]
mock_get_complete_models.return_value = [
@ -1812,6 +1813,7 @@ class TestModelInfoEndpoint:
# Setup mocks
mock_router.get_model_names.return_value = ["team-model-1"]
mock_router.get_model_access_groups.return_value = {}
mock_router.get_configured_token_limits.return_value = (None, None)
mock_get_key_models.return_value = []
mock_get_team_models.return_value = ["team-model-1"]
mock_get_complete_models.return_value = ["team-model-1"]