scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-04 02:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 71

fix: guard against model_info=None in router_strategy callbacks

Browse source
This commit is contained in:
Arya Maddel 2026-05-05 20:48:42 +05:30
parent a67b7a7e87
commit 9a88526f3b
3 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/router_strategy/least_busy.py
View file

@ -34,7 +34,7 @@ class LeastBusyLoggingHandler(CustomLogger):
model_group = kwargs["litellm_params"]["metadata"].get(
"model_group", None
)
id = kwargs["litellm_params"].get("model_info", {}).get("id", None)
id = (kwargs["litellm_params"].get("model_info") or {}).get("id", None)
if model_group is None or id is None:
return
elif isinstance(id, int):
@ -62,7 +62,7 @@ class LeastBusyLoggingHandler(CustomLogger):
"model_group", None
)
id = kwargs["litellm_params"].get("model_info", {}).get("id", None)
id = (kwargs["litellm_params"].get("model_info") or {}).get("id", None)
if model_group is None or id is None:
return
elif isinstance(id, int):
@ -95,7 +95,7 @@ class LeastBusyLoggingHandler(CustomLogger):
model_group = kwargs["litellm_params"]["metadata"].get(
"model_group", None
)
id = kwargs["litellm_params"].get("model_info", {}).get("id", None)
id = (kwargs["litellm_params"].get("model_info") or {}).get("id", None)
if model_group is None or id is None:
return
elif isinstance(id, int):
@ -129,7 +129,7 @@ class LeastBusyLoggingHandler(CustomLogger):
"model_group", None
)
id = kwargs["litellm_params"].get("model_info", {}).get("id", None)
id = (kwargs["litellm_params"].get("model_info") or {}).get("id", None)
if model_group is None or id is None:
return
elif isinstance(id, int):
@ -163,7 +163,7 @@ class LeastBusyLoggingHandler(CustomLogger):
model_group = kwargs["litellm_params"]["metadata"].get(
"model_group", None
)
id = kwargs["litellm_params"].get("model_info", {}).get("id", None)
id = (kwargs["litellm_params"].get("model_info") or {}).get("id", None)
if model_group is None or id is None:
return
elif isinstance(id, int):

4
litellm/router_strategy/lowest_cost.py
View file

@ -30,7 +30,7 @@ class LowestCostLoggingHandler(CustomLogger):
"model_group", None
)
id = kwargs["litellm_params"].get("model_info", {}).get("id", None)
id = (kwargs["litellm_params"].get("model_info") or {}).get("id", None)
if model_group is None or id is None:
return
elif isinstance(id, int):
@ -114,7 +114,7 @@ class LowestCostLoggingHandler(CustomLogger):
"model_group", None
)
id = kwargs["litellm_params"].get("model_info", {}).get("id", None)
id = (kwargs["litellm_params"].get("model_info") or {}).get("id", None)
if model_group is None or id is None:
return
elif isinstance(id, int):

2
litellm/router_strategy/lowest_tpm_rpm.py
View file

@ -40,7 +40,7 @@ class LowestTPMLoggingHandler(CustomLogger):
"model_group", None
)
id = kwargs["litellm_params"].get("model_info", {}).get("id", None)
id = (kwargs["litellm_params"].get("model_info") or {}).get("id", None)
if model_group is None or id is None:
return
elif isinstance(id, int):