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

(fix) dynamoDB + hashed tokens bug

Browse source
This commit is contained in:
ishaan-jaff 2024-01-29 11:17:27 -08:00
parent 336c8af01e
commit 2e64ca9057
2 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_key_generate_dynamodb.py
View file

@ -116,6 +116,10 @@ def test_call_with_invalid_key(custom_db_client):
def test_call_with_invalid_model(custom_db_client):
# 3. Make a call to a key with an invalid model - expect to fail
from litellm._logging import verbose_proxy_logger
import logging
verbose_proxy_logger.setLevel(logging.DEBUG)
setattr(litellm.proxy.proxy_server, "custom_db_client", custom_db_client)
setattr(litellm.proxy.proxy_server, "master_key", "sk-1234")
try: