05b794f29b
Signed-off-by: Sy03 <1370724210@qq.com> Signed-off-by: lishunyang12 <lishunyang03@gmail.com> Signed-off-by: Yueqian Lin <linyueqian@outlook.com> Co-authored-by: lishunyang12 <lishunyang03@gmail.com> Co-authored-by: Yueqian Lin <linyueqian@outlook.com> Co-authored-by: StormMelody <22371184@zju.edu.cn> Co-authored-by: Gao Han <hgaoaf@connect.ust.hk> Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
15 lines
513 B
Python
15 lines
513 B
Python
import pytest
|
|
|
|
from vllm_omni.experimental.fullduplex.engine.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"
|