scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-16 23:41:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

test(router): annotate return types of team cooldown test helpers

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yassin 2026-09-13 09:50:12 +00:00
parent 9080f0904a
commit e41b3bd13f
2 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_litellm/router_utils/test_cooldown_handlers.py
View file

@ -440,10 +440,8 @@ class TestRoutingGroupCooldownAlternatives:
class TestTeamModelCooldownAlternatives:
def _router(self, team_deployments: int, blocked_ids: frozenset[str] = frozenset()):
from litellm import Router
return Router(
def _router(self, team_deployments: int, blocked_ids: frozenset[str] = frozenset()) -> litellm.Router:
return litellm.Router(
model_list=[
{
"model_name": f"model_name_team-1_{i}",

4
tests/test_litellm/test_router.py
View file

@ -897,7 +897,9 @@ def test_arouter_test_team_model():
def test_team_model_has_alternatives():
def team_deployment(deployment_id: str, team_id: str, public_model_name: str, blocked: bool = False):
def team_deployment(
deployment_id: str, team_id: str, public_model_name: str, blocked: bool = False
) -> DeploymentTypedDict:
return {
"model_name": f"model_name_{team_id}_{deployment_id}",
"litellm_params": {"model": "openai/gpt-4o-mini", "api_key": "sk-test"},