49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
|
# SPDX-License-Identifier: Apache-2.0
|
||
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from vllm_omni.engine.duplex.realtime_events import (
|
||
|
|
RealtimeProjectionState,
|
||
|
|
project_internal_event,
|
||
|
|
)
|
||
|
|
|
||
|
|
pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
|
||
|
|
|
||
|
|
|
||
|
|
def test_input_transcribed_is_a_user_transcript_event() -> None:
|
||
|
|
events = project_internal_event(
|
||
|
|
RealtimeProjectionState(session_id="s"),
|
||
|
|
{"type": "input.transcribed", "transcript": " 你好 "},
|
||
|
|
)
|
||
|
|
assert len(events) == 1
|
||
|
|
assert events[0].wire_type == "conversation.item.input_audio_transcription.completed"
|
||
|
|
assert events[0].transcript == "你好"
|
||
|
|
|
||
|
|
|
||
|
|
def test_input_transcribed_updates_the_committed_item() -> None:
|
||
|
|
state = RealtimeProjectionState(session_id="s")
|
||
|
|
project_internal_event(
|
||
|
|
state,
|
||
|
|
{
|
||
|
|
"type": "input.committed",
|
||
|
|
"realtime_item_id": "item_user_1",
|
||
|
|
"is_speech": True,
|
||
|
|
"message": {"role": "user"},
|
||
|
|
},
|
||
|
|
)
|
||
|
|
events = project_internal_event(
|
||
|
|
state,
|
||
|
|
{"type": "input.transcribed", "transcript": "你好", "realtime_item_id": "item_user_1"},
|
||
|
|
)
|
||
|
|
assert events[0].item_id == "item_user_1"
|
||
|
|
item = state.conversation_items["item_user_1"]
|
||
|
|
assert item["content"][0]["transcript"] == "你好"
|
||
|
|
|
||
|
|
|
||
|
|
def test_blank_input_transcribed_emits_nothing() -> None:
|
||
|
|
events = project_internal_event(
|
||
|
|
RealtimeProjectionState(session_id="s"),
|
||
|
|
{"type": "input.transcribed", "transcript": " "},
|
||
|
|
)
|
||
|
|
assert events == []
|