phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix - stop printing api_key in debug mode

Browse source
This commit is contained in:
Ishaan Jaff 2024-04-11 15:05:22 -07:00
parent 68e0c93246
commit 921f6d307e
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/router.py
View file

@ -1900,8 +1900,12 @@ class Router:
local_only=True, local_only=True,
) # cache for 1 hr ) # cache for 1 hr
else: else:
_api_key = api_key
if _api_key is not None and isinstance(_api_key, str):
# only show first 5 chars of api_key
_api_key = _api_key[:8] + "*" * 15
verbose_router_logger.debug( verbose_router_logger.debug(
f"Initializing Azure OpenAI Client for {model_name}, Api Base: {str(api_base)}, Api Key:{api_key}" f"Initializing Azure OpenAI Client for {model_name}, Api Base: {str(api_base)}, Api Key:{_api_key}"
) )
azure_client_params = { azure_client_params = {
"api_key": api_key, "api_key": api_key,
@ -1998,8 +2002,12 @@ class Router:
) # cache for 1 hr ) # cache for 1 hr
else: else:
_api_key = api_key
if _api_key is not None and isinstance(_api_key, str):
# only show first 5 chars of api_key
_api_key = _api_key[:8] + "*" * 15
verbose_router_logger.debug( verbose_router_logger.debug(
f"Initializing OpenAI Client for {model_name}, Api Base:{str(api_base)}, Api Key:{api_key}" f"Initializing OpenAI Client for {model_name}, Api Base:{str(api_base)}, Api Key:{_api_key}"
) )
cache_key = f"{model_id}_async_client" cache_key = f"{model_id}_async_client"
_client = openai.AsyncOpenAI( # type: ignore _client = openai.AsyncOpenAI( # type: ignore