(fix) use get_logging_payload

This commit is contained in:
ishaan-jaff 2024-01-18 13:40:48 -08:00
parent 7bdf0a0a63
commit 4294657b99
2 changed files with 6 additions and 1 deletions

View file

@ -179,6 +179,10 @@ def test_call_with_key_over_budget(custom_db_client):
# 5. Make a call with a key over budget, expect to fail
setattr(litellm.proxy.proxy_server, "custom_db_client", custom_db_client)
setattr(litellm.proxy.proxy_server, "master_key", "sk-1234")
from litellm._logging import verbose_proxy_logger
import logging
verbose_proxy_logger.setLevel(logging.DEBUG)
try:
async def test():