From 690a5ec8485856210828466a17a9d36059a6cfc1 Mon Sep 17 00:00:00 2001 From: yucheng Date: Thu, 5 Nov 2026 03:21:04 -0800 Subject: [PATCH] test(proxy): leave the spend-log byte counter to the real writer --- tests/unit/proxy/hooks/test_proxy_track_cost_callback.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/unit/proxy/hooks/test_proxy_track_cost_callback.py b/tests/unit/proxy/hooks/test_proxy_track_cost_callback.py index e4f9a9df207..d80a9a782e4 100644 --- a/tests/unit/proxy/hooks/test_proxy_track_cost_callback.py +++ b/tests/unit/proxy/hooks/test_proxy_track_cost_callback.py @@ -4078,7 +4078,6 @@ async def test_async_post_call_failure_hook_scrubs_the_prompt_out_of_the_persist monkeypatch.setattr(litellm, "turn_off_message_logging", redact) spend_logs: Final = _SpendLogQueue() monkeypatch.setattr("litellm.proxy.proxy_server.prisma_client", spend_logs) - monkeypatch.setattr(PrismaClient, "spend_log_queue_bytes", 0) await ProxyDBLogger(spend_writer=DBSpendUpdateWriter).async_post_call_failure_hook( request_data={"model": "gpt-4o", "messages": [{"role": "user", "content": secret}], "metadata": {}},