Merge pull request #185 from pipecat-ai/aleix/add-start-recording

transport(daily): add start_recording, stop_recording and stop_dialout
This commit is contained in:
Aleix Conchillo Flaqué
2024-05-29 08:24:59 +08:00
committed by GitHub
2 changed files with 45 additions and 1 deletions

View File

@@ -7,6 +7,15 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
## [Unreleased] ## [Unreleased]
### Added
- Exposed Daily transport `on_app_message` event.
- Added Daily transport `on_call_state_updated` event.
- Added Daily transport `start_recording()`, `stop_recording` and
`stop_dialout`.
### Changed ### Changed
- Added `PipelineParams`. This replaces the `allow_interruptions` argument in - Added `PipelineParams`. This replaces the `allow_interruptions` argument in

View File

@@ -114,6 +114,7 @@ class DailyCallbacks(BaseModel):
on_left: Callable[[], None] on_left: Callable[[], None]
on_error: Callable[[str], None] on_error: Callable[[str], None]
on_app_message: Callable[[Any, str], None] on_app_message: Callable[[Any, str], None]
on_call_state_updated: Callable[[str], None]
on_dialin_ready: Callable[[str], None] on_dialin_ready: Callable[[str], None]
on_dialout_connected: Callable[[Any], None] on_dialout_connected: Callable[[Any], None]
on_dialout_stopped: Callable[[Any], None] on_dialout_stopped: Callable[[Any], None]
@@ -345,6 +346,15 @@ class DailyTransportClient(EventHandler):
def start_dialout(self, settings): def start_dialout(self, settings):
self._client.start_dialout(settings) self._client.start_dialout(settings)
def stop_dialout(self, participant_id):
self._client.stop_dialout(participant_id)
def start_recording(self, streaming_settings, stream_id, force_new):
self._client.start_recording(streaming_settings, stream_id, force_new)
def stop_recording(self, stream_id):
self._client.stop_recording(stream_id)
def capture_participant_transcription(self, participant_id: str, callback: Callable): def capture_participant_transcription(self, participant_id: str, callback: Callable):
if not self._params.transcription_enabled: if not self._params.transcription_enabled:
return return
@@ -381,6 +391,9 @@ class DailyTransportClient(EventHandler):
def on_app_message(self, message: Any, sender: str): def on_app_message(self, message: Any, sender: str):
self._callbacks.on_app_message(message, sender) self._callbacks.on_app_message(message, sender)
def on_call_state_updated(self, state: str):
self._callbacks.on_call_state_updated(state)
def on_dialin_ready(self, sip_endpoint: str): def on_dialin_ready(self, sip_endpoint: str):
self._callbacks.on_dialin_ready(sip_endpoint) self._callbacks.on_dialin_ready(sip_endpoint)
@@ -635,6 +648,7 @@ class DailyTransport(BaseTransport):
on_left=self._on_left, on_left=self._on_left,
on_error=self._on_error, on_error=self._on_error,
on_app_message=self._on_app_message, on_app_message=self._on_app_message,
on_call_state_updated=self._on_call_state_updated,
on_dialin_ready=self._on_dialin_ready, on_dialin_ready=self._on_dialin_ready,
on_dialout_connected=self._on_dialout_connected, on_dialout_connected=self._on_dialout_connected,
on_dialout_stopped=self._on_dialout_stopped, on_dialout_stopped=self._on_dialout_stopped,
@@ -657,6 +671,8 @@ class DailyTransport(BaseTransport):
# these handlers. # these handlers.
self._register_event_handler("on_joined") self._register_event_handler("on_joined")
self._register_event_handler("on_left") self._register_event_handler("on_left")
self._register_event_handler("on_app_message")
self._register_event_handler("on_call_state_updated")
self._register_event_handler("on_dialout_connected") self._register_event_handler("on_dialout_connected")
self._register_event_handler("on_dialout_stopped") self._register_event_handler("on_dialout_stopped")
self._register_event_handler("on_dialout_error") self._register_event_handler("on_dialout_error")
@@ -695,9 +711,18 @@ class DailyTransport(BaseTransport):
if self._output: if self._output:
await self._output.process_frame(frame, FrameDirection.DOWNSTREAM) await self._output.process_frame(frame, FrameDirection.DOWNSTREAM)
def start_dialout(self, settings): def start_dialout(self, settings=None):
self._client.start_dialout(settings) self._client.start_dialout(settings)
def stop_dialout(self, participant_id):
self._client.stop_dialout(participant_id)
def start_recording(self, streaming_settings=None, stream_id=None, force_new=None):
self._client.start_recording(streaming_settings, stream_id, force_new)
def stop_recording(self, stream_id=None):
self._client.stop_recording(stream_id)
def capture_participant_transcription(self, participant_id: str): def capture_participant_transcription(self, participant_id: str):
self._client.capture_participant_transcription( self._client.capture_participant_transcription(
participant_id, participant_id,
@@ -728,6 +753,10 @@ class DailyTransport(BaseTransport):
def _on_app_message(self, message: Any, sender: str): def _on_app_message(self, message: Any, sender: str):
if self._input: if self._input:
self._input.push_app_message(message, sender) self._input.push_app_message(message, sender)
self.on_app_message(message, sender)
def _on_call_state_updated(self, state: str):
self.on_call_state_updated(state)
async def _handle_dialin_ready(self, sip_endpoint: str): async def _handle_dialin_ready(self, sip_endpoint: str):
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
@@ -801,6 +830,12 @@ class DailyTransport(BaseTransport):
def on_left(self): def on_left(self):
pass pass
def on_app_message(self, message, sender):
pass
def on_call_state_updated(self, state):
pass
def on_dialout_connected(self, data): def on_dialout_connected(self, data):
pass pass