Skip to content

Commit

Permalink
fix(router.py): Fix model key/base leak in error message
Browse files Browse the repository at this point in the history
Fixes #5762
  • Loading branch information
krrishdholakia committed Sep 18, 2024
1 parent 2be3ffa commit e23559f
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions litellm/router.py
Original file line number Diff line number Diff line change
Expand Up @@ -4019,7 +4019,9 @@ def set_model_list(self, model_list: list):
_model_info=_model_info,
)

verbose_router_logger.debug(f"\nInitialized Model List {self.model_list}")
verbose_router_logger.debug(
f"\nInitialized Model List {self.get_model_names()}"
)
self.model_names = [m["model_name"] for m in model_list]

def _add_deployment(self, deployment: Deployment) -> Deployment:
Expand Down Expand Up @@ -5030,7 +5032,7 @@ def _common_checks_available_deployment(
# return the first deployment where the `model` matches the specificed deployment name
return deployment_model, deployment
raise ValueError(
f"LiteLLM Router: Trying to call specific deployment, but Model:{model} does not exist in Model List: {self.model_list}"
f"LiteLLM Router: Trying to call specific deployment, but Model:{model} does not exist in Model List: {self.get_model_names()}"
)
elif model in self.get_model_ids():
deployment = self.get_model_info(id=model)
Expand Down

0 comments on commit e23559f

Please sign in to comment.