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

fix: strip llm credential before returning

Browse source
This commit is contained in:
John HU 2024-06-17 12:14:40 -07:00 • committed by GitHub
parent 092040afec
commit 3d16bd6eb8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/proxy_server.py
View file

@ -6057,8 +6057,11 @@ async def model_info_v2(
model_info[k] = v model_info[k] = v
_model["model_info"] = model_info _model["model_info"] = model_info
# don't return the api key / vertex credentials # don't return the api key / vertex credentials
# don't return the llm credentials
_model["litellm_params"].pop("api_key", None) _model["litellm_params"].pop("api_key", None)
_model["litellm_params"].pop("vertex_credentials", None) _model["litellm_params"].pop("vertex_credentials", None)
_model["litellm_params"].pop("aws_access_key_id", None)
_model["litellm_params"].pop("aws_secret_access_key", None)
verbose_proxy_logger.debug("all_models: %s", all_models) verbose_proxy_logger.debug("all_models: %s", all_models)
return {"data": all_models} return {"data": all_models}
@ -6570,8 +6573,11 @@ async def model_info_v1(
if k not in model_info: if k not in model_info:
model_info[k] = v model_info[k] = v
model["model_info"] = model_info model["model_info"] = model_info
# don't return the api key # don't return the llm credentials
model["litellm_params"].pop("api_key", None) model["litellm_params"].pop("api_key", None)
model["litellm_params"].pop("vertex_credentials", None)
model["litellm_params"].pop("aws_access_key_id", None)
model["litellm_params"].pop("aws_secret_access_key", None)
verbose_proxy_logger.debug("all_models: %s", all_models) verbose_proxy_logger.debug("all_models: %s", all_models)
return {"data": all_models} return {"data": all_models}