phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(router.py): security fix - don't show api key in invalid model setup error message

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-29 16:14:57 -07:00
parent 8f0019c241
commit 1d18ca6a7d
2 changed files with 29 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/router.py
View file

@ -2610,8 +2610,17 @@ class Router:
if "azure" in model_name:
if api_base is None or not isinstance(api_base, str):
filtered_litellm_params = {
k: v
for k, v in model["litellm_params"].items()
if k != "api_key"
}
_filtered_model = {
"model_name": model["model_name"],
"litellm_params": filtered_litellm_params,
}
raise ValueError(
f"api_base is required for Azure OpenAI. Set it on your config. Model - {model}"
f"api_base is required for Azure OpenAI. Set it on your config. Model - {_filtered_model}"
)
azure_ad_token = litellm_params.get("azure_ad_token")
if azure_ad_token is not None: