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

Merge branch 'main' into litellm_fix_service_account_behavior

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-01 12:04:28 -07:00 • committed by GitHub
commit 4ddca7a79c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
18 changed files with 518 additions and 194 deletions
Download patch file Download diff file Expand all files Collapse all files

13
tests/proxy_unit_tests/test_proxy_utils.py
View file

@ -1485,21 +1485,18 @@ from litellm.proxy.utils import ProxyUpdateSpend
async def test_end_user_transactions_reset():
# Setup
mock_client = MagicMock()
mock_client.end_user_list_transactions = {"1": 10.0} # Bad log
end_user_list_transactions = {"1": 10.0} # Bad log
mock_client.db.tx = AsyncMock(side_effect=Exception("DB Error"))
# Call function - should raise error
with pytest.raises(Exception):
await ProxyUpdateSpend.update_end_user_spend(
n_retry_times=0, prisma_client=mock_client, proxy_logging_obj=MagicMock()
n_retry_times=0,
prisma_client=mock_client,
proxy_logging_obj=MagicMock(),
end_user_list_transactions=end_user_list_transactions,
)
# Verify cleanup happened
assert (
mock_client.end_user_list_transactions == {}
), "Transactions list should be empty after error"
@pytest.mark.asyncio
async def test_spend_logs_cleanup_after_error():
# Setup test data