mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-11 22:51:28 +00:00
fix(router.py): fix test
This commit is contained in:
parent
75bcfbb76a
commit
2e21f1d07a
1 changed files with 5 additions and 0 deletions
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue