From 8c67563055b15348fee1258220ea7a9f276ff452 Mon Sep 17 00:00:00 2001 From: "guangli.bao" Date: Mon, 24 Nov 2025 09:43:32 +0800 Subject: [PATCH 1/2] fix encode_audio with dict input failure Signed-off-by: guangli.bao --- src/guidellm/extras/audio.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/guidellm/extras/audio.py b/src/guidellm/extras/audio.py index 8d7e7de90..7b8466992 100644 --- a/src/guidellm/extras/audio.py +++ b/src/guidellm/extras/audio.py @@ -107,8 +107,9 @@ def _decode_audio( # noqa: C901, PLR0912 raise ValueError( f"Audio dict must contain either 'data' or 'url' keys, got {audio}" ) + audio_data = audio["data"] if "data" in audio else audio["url"] return _decode_audio( - audio=audio.get("data") or audio.get("url"), + audio=audio_data, sample_rate=sample_rate, max_duration=max_duration, ) From fd77ccceea2731eaef967dcf911e99aafc3a8e26 Mon Sep 17 00:00:00 2001 From: "guangli.bao" Date: Tue, 25 Nov 2025 09:43:05 +0800 Subject: [PATCH 2/2] fix based on comment Signed-off-by: guangli.bao --- src/guidellm/extras/audio.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/guidellm/extras/audio.py b/src/guidellm/extras/audio.py index 7b8466992..54a9d1a10 100644 --- a/src/guidellm/extras/audio.py +++ b/src/guidellm/extras/audio.py @@ -107,7 +107,11 @@ def _decode_audio( # noqa: C901, PLR0912 raise ValueError( f"Audio dict must contain either 'data' or 'url' keys, got {audio}" ) - audio_data = audio["data"] if "data" in audio else audio["url"] + audio_data = audio["data"] if "data" in audio else audio.get("url") + if audio_data is None: + raise ValueError( + f"Audio dict must contain either 'data' or 'url' keys, got {audio}" + ) return _decode_audio( audio=audio_data, sample_rate=sample_rate,