mirror of
https://github.com/dograh-hq/dograh.git
synced 2026-06-13 08:15:21 +02:00
b5a192557b
The UserIdleHandler injected its "are you still there?" and disconnect prompts as role="user" messages. These are agent-side directives, not user utterances, so they should be injected as role="system" to avoid polluting the conversation transcript with fake user turns and to read correctly by the LLM. Updated the realtime append tests to match. Also forward ports 3000 (UI) and 8000 (API) in the devcontainer so the running services are reachable from the host. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
59 lines
2 KiB
Python
59 lines
2 KiB
Python
from types import SimpleNamespace
|
|
from unittest.mock import AsyncMock
|
|
|
|
import pytest
|
|
from pipecat.frames.frames import LLMMessagesAppendFrame
|
|
from pipecat.services.openai.realtime import events
|
|
|
|
from api.services.pipecat.realtime.openai_realtime import (
|
|
DograhOpenAIRealtimeLLMService,
|
|
)
|
|
from api.services.workflow.pipecat_engine_callbacks import UserIdleHandler
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_openai_realtime_messages_append_frame_sends_conversation_item():
|
|
service = DograhOpenAIRealtimeLLMService(api_key="test")
|
|
service._api_session_ready = True
|
|
service.send_client_event = AsyncMock()
|
|
service._send_manual_response_create = AsyncMock()
|
|
|
|
await service._handle_messages_append(
|
|
LLMMessagesAppendFrame(
|
|
[{"role": "system", "content": "Are you still there?"}],
|
|
run_llm=True,
|
|
)
|
|
)
|
|
|
|
service.send_client_event.assert_awaited_once()
|
|
event = service.send_client_event.await_args.args[0]
|
|
assert isinstance(event, events.ConversationItemCreateEvent)
|
|
assert event.item.role == "system"
|
|
assert event.item.type == "message"
|
|
assert event.item.content == [
|
|
events.ItemContent(type="input_text", text="Are you still there?")
|
|
]
|
|
service._send_manual_response_create.assert_awaited_once()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_user_idle_handler_uses_realtime_append_path():
|
|
engine = SimpleNamespace(
|
|
llm=SimpleNamespace(),
|
|
end_call_with_reason=AsyncMock(),
|
|
)
|
|
aggregator = SimpleNamespace(push_frame=AsyncMock())
|
|
handler = UserIdleHandler(engine)
|
|
|
|
await handler.handle_idle(aggregator)
|
|
|
|
aggregator.push_frame.assert_awaited_once()
|
|
frame = aggregator.push_frame.await_args.args[0]
|
|
assert isinstance(frame, LLMMessagesAppendFrame)
|
|
assert frame.run_llm is True
|
|
assert frame.messages == [
|
|
{
|
|
"role": "system",
|
|
"content": "The user has been quiet. Politely and briefly ask if they're still there in the language that the user has been speaking so far.",
|
|
}
|
|
]
|