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

DBSpendUpdateWriter

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-27 16:43:18 -07:00
parent 072be44a54
commit a0fd508de4
2 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/proxy/db/db_spend_update_writer.py
View file

@ -7,9 +7,8 @@ from typing import Any, Optional, Union
import litellm import litellm
from litellm._logging import verbose_proxy_logger from litellm._logging import verbose_proxy_logger
from litellm.proxy._types import LiteLLM_UserTable, SpendLogsPayload from litellm.proxy._types import LiteLLM_UserTable, SpendLogsPayload
from litellm.proxy.proxy_server import hash_token
from litellm.proxy.spend_tracking.spend_tracking_utils import get_logging_payload from litellm.proxy.spend_tracking.spend_tracking_utils import get_logging_payload
from litellm.proxy.utils import PrismaClient, ProxyUpdateSpend from litellm.proxy.utils import PrismaClient, ProxyUpdateSpend, hash_token
class DBSpendUpdateWriter: class DBSpendUpdateWriter:

9
tests/litellm/proxy/spend_tracking/test_spend_management_endpoints.py
View file

@ -416,7 +416,8 @@ class TestSpendLogsPayload:
# litellm._turn_on_debug() # litellm._turn_on_debug()
with patch.object( with patch.object(
litellm.proxy.proxy_server, "_set_spend_logs_payload" litellm.proxy.db.db_spend_update_writer.DBSpendUpdateWriter,
"_set_spend_logs_payload",
) as mock_client, patch.object(litellm.proxy.proxy_server, "prisma_client"): ) as mock_client, patch.object(litellm.proxy.proxy_server, "prisma_client"):
response = await litellm.acompletion( response = await litellm.acompletion(
model="gpt-4o", model="gpt-4o",
@ -509,7 +510,8 @@ class TestSpendLogsPayload:
client = AsyncHTTPHandler() client = AsyncHTTPHandler()
with patch.object( with patch.object(
litellm.proxy.proxy_server, "_set_spend_logs_payload" litellm.proxy.db.db_spend_update_writer.DBSpendUpdateWriter,
"_set_spend_logs_payload",
) as mock_client, patch.object( ) as mock_client, patch.object(
litellm.proxy.proxy_server, "prisma_client" litellm.proxy.proxy_server, "prisma_client"
), patch.object( ), patch.object(
@ -604,7 +606,8 @@ class TestSpendLogsPayload:
) )
with patch.object( with patch.object(
litellm.proxy.proxy_server, "_set_spend_logs_payload" litellm.proxy.db.db_spend_update_writer.DBSpendUpdateWriter,
"_set_spend_logs_payload",
) as mock_client, patch.object( ) as mock_client, patch.object(
litellm.proxy.proxy_server, "prisma_client" litellm.proxy.proxy_server, "prisma_client"
), patch.object( ), patch.object(