frozenleaves/vllm-dev
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: add usage to TranscriptionResponse (text and json response_format) (#23576)

Browse Source 
Signed-off-by: Guillaume Calmettes <gcalmettes@scaleway.com>
This commit is contained in:
Guillaume Calmettes
2025-08-26 14:26:26 +02:00
committed by GitHub
parent 384dd1b0a8
commit ebd5a77bb5
3 changed files with 32 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/entrypoints/openai/test_transcription_validation.py
View File

@ -69,8 +69,11 @@ async def test_basic_audio(mary_had_lamb, model_name):
language="en",
response_format="text",
temperature=0.0)
out = json.loads(transcription)['text']
assert "Mary had a little lamb," in out
out = json.loads(transcription)
out_text = out['text']
out_usage = out['usage']
assert "Mary had a little lamb," in out_text
assert out_usage["seconds"] == 16, out_usage["seconds"]
@pytest.mark.asyncio
@ -116,9 +119,12 @@ async def test_long_audio_request(mary_had_lamb, client):
language="en",
response_format="text",
temperature=0.0)
out = json.loads(transcription)['text']
counts = out.count("Mary had a little lamb")
out = json.loads(transcription)
out_text = out['text']
out_usage = out['usage']
counts = out_text.count("Mary had a little lamb")
assert counts == 10, counts
assert out_usage["seconds"] == 161, out_usage["seconds"]
@pytest.mark.asyncio