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

test fixes

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-31 19:33:10 -07:00
parent 811f488ca3
commit aa8261af89
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/proxy/hooks/proxy_track_cost_callback.py
View file

@ -13,7 +13,6 @@ from litellm.litellm_core_utils.core_helpers import (
from litellm.litellm_core_utils.litellm_logging import StandardLoggingPayloadSetup from litellm.litellm_core_utils.litellm_logging import StandardLoggingPayloadSetup
from litellm.proxy._types import UserAPIKeyAuth from litellm.proxy._types import UserAPIKeyAuth
from litellm.proxy.auth.auth_checks import log_db_metrics from litellm.proxy.auth.auth_checks import log_db_metrics
from litellm.proxy.db.db_spend_update_writer import DBSpendUpdateWriter
from litellm.proxy.utils import ProxyUpdateSpend from litellm.proxy.utils import ProxyUpdateSpend
from litellm.types.utils import ( from litellm.types.utils import (
StandardLoggingPayload, StandardLoggingPayload,

3
tests/proxy_unit_tests/test_unit_test_proxy_hooks.py
View file

@ -24,9 +24,10 @@ async def test_disable_spend_logs():
"litellm.proxy.proxy_server.prisma_client", mock_prisma_client "litellm.proxy.proxy_server.prisma_client", mock_prisma_client
): ):
from litellm.proxy.db.db_spend_update_writer import DBSpendUpdateWriter from litellm.proxy.db.db_spend_update_writer import DBSpendUpdateWriter
db_spend_update_writer = DBSpendUpdateWriter()
# Call update_database with disable_spend_logs=True # Call update_database with disable_spend_logs=True
await DBSpendUpdateWriter.update_database( await db_spend_update_writer.update_database(
token="fake-token", token="fake-token",
response_cost=0.1, response_cost=0.1,
user_id="user123", user_id="user123",