phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(router.py): enable settting 'order' for a deployment in model list

Browse source 
Allows user to control which model gets called first in model group
This commit is contained in:
Krrish Dholakia 2024-06-06 09:46:51 -07:00
parent 3c1d4179c2
commit 005128addc
3 changed files with 69 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/router.py
View file

@ -4048,6 +4048,12 @@ class Router:
for idx in reversed(invalid_model_indices):
_returned_deployments.pop(idx)
## ORDER FILTERING ## -> if user set 'order' in deployments, return deployments with lowest order (e.g. order=1 > order=2)
if len(_returned_deployments) > 0:
_returned_deployments = litellm.utils._get_order_filtered_deployments(
_returned_deployments
)
return _returned_deployments
def _common_checks_available_deployment(