diff --git a/litellm/router_strategy/lowest_latency.py b/litellm/router_strategy/lowest_latency.py index 7492662e3b..0449a843bd 100644 --- a/litellm/router_strategy/lowest_latency.py +++ b/litellm/router_strategy/lowest_latency.py @@ -52,7 +52,7 @@ class LowestLatencyLoggingHandler(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): @@ -204,7 +204,7 @@ class LowestLatencyLoggingHandler(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): @@ -273,7 +273,7 @@ class LowestLatencyLoggingHandler(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):