[BUG FIX]: prevent EngineCore crash when Qwen TTS Base task is missing ref_text (#2203)
Signed-off-by: Yueqian Lin <70319226+linyueqian@users.noreply.github.com> Co-authored-by: Yueqian Lin <70319226+linyueqian@users.noreply.github.com>
This commit is contained in:
1 parent
a41174e083
commit
001f2e32e2
3 files changed
+37
-5
No files matched your search
@@ -752,6 +752,26 @@ class TestTTSMethods:
|
||||
)
|
||||
assert speech_server._validate_tts_request(req) is None
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"ref_text",
|
||||
[None, "", " "],
|
||||
ids=["none", "empty", "whitespace"],
|
||||
)
|
||||
def test_validate_base_task_missing_ref_text_returns_400(self, speech_server, ref_text):
|
||||
"""Regression: Base task without ref_text must return 400, not crash EngineCore.
|
||||
|
||||
See https://github.com/vllm-project/vllm-omni/pull/2203
|
||||
"""
|
||||
req = OpenAICreateSpeechRequest(
|
||||
input="Hello",
|
||||
task_type="Base",
|
||||
ref_audio="data:audio/wav;base64,abc",
|
||||
ref_text=ref_text,
|
||||
)
|
||||
result = speech_server._validate_tts_request(req)
|
||||
assert result is not None, f"ref_text={ref_text!r} should be rejected"
|
||||
assert "ref_text" in result
|
||||
|
||||
def test_validate_tts_request_customvoice_no_speakers(self, speech_server):
|
||||
"""CustomVoice on a model with no speakers returns 400 instead of crashing engine."""
|
||||
req = OpenAICreateSpeechRequest(input="Hello", task_type="CustomVoice")
|
||||
|
||||
Reference in new issue
Block a user