phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fix) use get_logging_payload

Browse source
This commit is contained in:
ishaan-jaff 2024-01-18 13:40:48 -08:00
parent e79707f42c
commit fab53c236d
2 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/proxy/proxy_server.py
View file

@ -72,6 +72,7 @@ from litellm.proxy.utils import (
ProxyLogging, ProxyLogging,
_cache_user_row, _cache_user_row,
send_email, send_email,
get_logging_payload,
) )
from litellm.proxy.secret_managers.google_kms import load_google_kms from litellm.proxy.secret_managers.google_kms import load_google_kms
import pydantic import pydantic
@ -646,7 +647,7 @@ async def update_database(
async def _insert_spend_log_to_db(): async def _insert_spend_log_to_db():
# Helper to generate payload to log # Helper to generate payload to log
verbose_proxy_logger.debug("inserting spend log to db") verbose_proxy_logger.debug("inserting spend log to db")
payload = litellm.utils.get_logging_payload( payload = get_logging_payload(
kwargs=kwargs, kwargs=kwargs,
response_obj=completion_response, response_obj=completion_response,
start_time=start_time, start_time=start_time,

4
litellm/tests/test_key_generate_dynamodb.py
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 # 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, "custom_db_client", custom_db_client)
setattr(litellm.proxy.proxy_server, "master_key", "sk-1234") 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: try:
async def test(): async def test():