scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-08-28 05:25:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

test_no_duplicate_spend_logs

Browse source
This commit is contained in:
Ishaan Jaffer 2025-12-06 09:55:35 -08:00
parent ca7241188a
commit 71ea926f9a
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/test_litellm/responses/test_no_duplicate_spend_logs.py
View file

@ -5,8 +5,10 @@ This test verifies the fix for issue #15740 where kwargs.pop() was removing
the logging object before passing kwargs to internal acompletion() calls,
causing duplicate spend log entries for non-OpenAI providers.
"""
import sys
import asyncio
import os
import sys
import pytest
sys.path.insert(
@ -26,6 +28,7 @@ def test_logging_object_not_popped():
spend logs for non-OpenAI providers.
"""
import inspect
from litellm.responses import main as responses_module
# Get the source code of the responses function
@ -84,9 +87,11 @@ async def test_no_duplicate_spend_logs():
)
# Wait for async logging to complete using the logging worker's flush method
# This is more reliable than sleep() which can cause race conditions
# Then add a small delay to ensure callbacks finish executing
from litellm.litellm_core_utils.logging_worker import GLOBAL_LOGGING_WORKER
await GLOBAL_LOGGING_WORKER.flush()
# flush() empties the queue but callbacks may still be running
await asyncio.sleep(0.5)
# Verify that log_success_event was called exactly once
assert spend_logger.log_count == 1, (