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

feat - allow looking up model_id in model info

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-04 13:13:42 -07:00
parent 4606b020b5
commit 4af5531834
2 changed files with 51 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

21
litellm/proxy/common_utils/openai_endpoint_utils.py Normal file
View file

@ -0,0 +1,21 @@
"""
Contains utils used by OpenAI compatible endpoints
"""
def remove_sensitive_info_from_deployment(deployment_dict: dict) -> dict:
"""
Removes sensitive information from a deployment dictionary.
Args:
deployment_dict (dict): The deployment dictionary to remove sensitive information from.
Returns:
dict: The modified deployment dictionary with sensitive information removed.
"""
deployment_dict["litellm_params"].pop("api_key", None)
deployment_dict["litellm_params"].pop("vertex_credentials", None)
deployment_dict["litellm_params"].pop("aws_access_key_id", None)
deployment_dict["litellm_params"].pop("aws_secret_access_key", None)
return deployment_dict