phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +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

19
litellm/tests/test_router.py
View file

@ -19,6 +19,25 @@ import os, httpx
load_dotenv()
def test_router_sensitive_keys():
try:
router = Router(
model_list=[
{
"model_name": "gpt-3.5-turbo", # openai model name
"litellm_params": { # params for litellm completion/embedding call
"model": "azure/chatgpt-v-2",
"api_key": "special-key",
},
"model_info": {"id": 12345},
},
],
)
except Exception as e:
print(f"error msg - {str(e)}")
assert "special-key" not in str(e)
@pytest.mark.parametrize("num_retries", [None, 2])
@pytest.mark.parametrize("max_retries", [None, 4])
def test_router_num_retries_init(num_retries, max_retries):