scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-14 09:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 214

test(utils): capture streamed sync speech usage when the audio ends, not at the headers

Browse source
This commit is contained in:
mateo-berri 2026-11-03 08:18:53 -08:00
parent cc7abd6084
commit 93ada24dfa
1 changed files with 19 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
tests/unit/test_utils.py
View file

@ -4212,6 +4212,25 @@ def test_speech_stream_is_logged_when_the_audio_ends(
assert audio.rest_sent_at is not None and _logged_end(logged) >= audio.rest_sent_at
@pytest.mark.usefixtures("local_model_cost_map")
def test_speech_stream_usage_is_captured_when_the_audio_ends(
respx_mock: respx.MockRouter, monkeypatch: pytest.MonkeyPatch
) -> None:
audio: Final = _SyncAudio(b"first", b"rest")
_mock_elevenlabs_audio(respx_mock, audio)
captured_at: Final[queue.Queue[float]] = queue.Queue()
monkeypatch.setattr(
litellm_logging.Logging, "usage_capture_callback", lambda *_: captured_at.put(time.time())
)
response: Final = litellm.speech(**_SPEECH_ARGS, stream=True)
captured_before_read: Final = captured_at.qsize()
assert response.read() == b"firstrest"
assert captured_before_read == 0
assert audio.rest_sent_at is not None and captured_at.get(timeout=5) >= audio.rest_sent_at
class TestCallbackAsyncSyncSeparation:
"""Test that LoggingCallbackManager auto-routes async callbacks to async lists."""