From fab53c236d40aae83c0acd22ce6ebe34f2283044 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 18 Jan 2024 13:40:48 -0800 Subject: [PATCH] (fix) use get_logging_payload --- litellm/proxy/proxy_server.py | 3 ++- litellm/tests/test_key_generate_dynamodb.py | 4 ++++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 32e985113e..8145d19aa3 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -72,6 +72,7 @@ from litellm.proxy.utils import ( ProxyLogging, _cache_user_row, send_email, + get_logging_payload, ) from litellm.proxy.secret_managers.google_kms import load_google_kms import pydantic @@ -646,7 +647,7 @@ async def update_database( async def _insert_spend_log_to_db(): # Helper to generate payload to log verbose_proxy_logger.debug("inserting spend log to db") - payload = litellm.utils.get_logging_payload( + payload = get_logging_payload( kwargs=kwargs, response_obj=completion_response, start_time=start_time, diff --git a/litellm/tests/test_key_generate_dynamodb.py b/litellm/tests/test_key_generate_dynamodb.py index 09f699af7d..2cfa9c9531 100644 --- a/litellm/tests/test_key_generate_dynamodb.py +++ b/litellm/tests/test_key_generate_dynamodb.py @@ -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():