Skip to content

Commit 2e21f1d

Browse files
fix(router.py): fix test
1 parent 75bcfbb commit 2e21f1d

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

litellm/router.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5475,6 +5475,11 @@ def get_router_model_info(
54755475
pass
54765476

54775477
## GET LITELLM MODEL INFO - raises exception, if model is not mapped
5478+
if model is None:
5479+
# Handle case where base_model is None (e.g., Azure models without base_model set)
5480+
# Use the original model from litellm_params
5481+
model = _model
5482+
54785483
if not model.startswith("{}/".format(custom_llm_provider)):
54795484
model_info_name = "{}/{}".format(custom_llm_provider, model)
54805485
else:

0 commit comments

Comments
 (0)