fix(router.py): fix test

This commit is contained in:
Krrish Dholakia 2025-08-13 16:27:13 -07:00
parent 90cef7c774
commit 5c2b045a36

View file

@ -5475,6 +5475,11 @@ class Router:
pass
## GET LITELLM MODEL INFO - raises exception, if model is not mapped
if model is None:
# Handle case where base_model is None (e.g., Azure models without base_model set)
# Use the original model from litellm_params
model = _model
if not model.startswith("{}/".format(custom_llm_provider)):
model_info_name = "{}/{}".format(custom_llm_provider, model)
else: