mirror of
https://github.com/dograh-hq/dograh.git
synced 2026-06-28 08:49:42 +02:00
feat: abort immediately on max call duration exceed
This commit is contained in:
parent
78ba62e185
commit
c586d02d5d
3 changed files with 27 additions and 5 deletions
19
api/tests/test_pipecat_engine_callbacks.py
Normal file
19
api/tests/test_pipecat_engine_callbacks.py
Normal file
|
|
@ -0,0 +1,19 @@
|
|||
from unittest.mock import AsyncMock
|
||||
|
||||
import pytest
|
||||
from pipecat.utils.enums import EndTaskReason
|
||||
|
||||
from api.services.workflow.pipecat_engine_callbacks import create_max_duration_callback
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_max_duration_callback_aborts_immediately():
|
||||
engine = AsyncMock()
|
||||
|
||||
callback = create_max_duration_callback(engine)
|
||||
await callback()
|
||||
|
||||
engine.end_call_with_reason.assert_awaited_once_with(
|
||||
EndTaskReason.CALL_DURATION_EXCEEDED.value,
|
||||
abort_immediately=True,
|
||||
)
|
||||
Loading…
Add table
Add a link
Reference in a new issue