From c6518b8d0f003d3d6ef0bda6b9ddafe934694b31 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 1 Dec 2023 20:47:13 -0800 Subject: [PATCH] fix(proxy_server.py): redact api keys from proxy load --- litellm/proxy/proxy_server.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 09ab175a18..344c9fab62 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -421,7 +421,12 @@ def load_router_config(router: Optional[litellm.Router], config_file_path: str): except Exception as e: raise Exception(f"Exception while reading Config: {e}") - print(f"Loaded config YAML:\n{json.dumps(config, indent=2)}") + printed_yaml = copy.deepcopy(config) + printed_yaml.pop("environment_variables") + for model in printed_yaml["model_list"]: + model["litellm_params"].pop("api_key", None) + + print(f"Loaded config YAML (api_key and environment_variables are not shown):\n{json.dumps(printed_yaml, indent=2)}") ## ENVIRONMENT VARIABLES environment_variables = config.get('environment_variables', None)