5d7fdf9350
Signed-off-by: chickeyton <ngton2014@gmail.com> Signed-off-by: Gao Han <hgaoaf@connect.ust.hk> Co-authored-by: Claude Fable 5 <noreply@anthropic.com> Co-authored-by: Gao Han <hgaoaf@connect.ust.hk>
18 lines
609 B
Python
18 lines
609 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
|
|
|
|
import pytest
|
|
|
|
from vllm_omni.engine.duplex.intermediate import get_stream_request_key
|
|
|
|
pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
|
|
|
|
|
|
def test_get_stream_request_key_requires_stable_identifier():
|
|
with pytest.raises(ValueError, match="stable request id"):
|
|
get_stream_request_key({"ids": {"tts": [1, 2, 3]}})
|
|
|
|
|
|
def test_get_stream_request_key_accepts_global_request_id():
|
|
assert get_stream_request_key({"global_request_id": ["duplex-sid-stage0"]}) == "duplex-sid-stage0"
|