Merge pull request #1560 from pipecat-ai/bot_left_signalling

Bot left signalling message
This commit is contained in:
Filipi da Silva Fuchter
2025-04-11 16:08:01 -03:00
committed by GitHub
2 changed files with 12 additions and 4 deletions

View File

@@ -308,7 +308,7 @@ class SmallWebRTCClient:
if self.is_connected and not self.is_closing:
logger.info(f"Disconnecting to Small WebRTC")
self._closing = True
await self._webrtc_connection.close()
await self._webrtc_connection.disconnect()
await self._handle_client_disconnected()
async def send_message(self, frame: TransportMessageFrame | TransportMessageUrgentFrame):

View File

@@ -44,6 +44,10 @@ class RenegotiateMessage(BaseModel):
type: Literal["renegotiate"] = "renegotiate"
class PeerLeftMessage(BaseModel):
type: Literal["peerLeft"] = "peerLeft"
class SignallingMessage:
Inbound = Union[TrackStatusMessage] # in case we need to add new messages in the future
outbound = Union[RenegotiateMessage]
@@ -226,7 +230,7 @@ class SmallWebRTCConnection(BaseObject):
logger.debug("Closing old peer connection")
# removing the listeners to prevent the bot from closing
self._pc.remove_all_listeners()
await self.close()
await self._close()
# we are initializing a new peer connection in this case.
self._initialize()
@@ -271,7 +275,11 @@ class SmallWebRTCConnection(BaseObject):
else:
logger.warning("Video transceiver not found. Cannot replace video track.")
async def close(self):
async def disconnect(self):
self.send_app_message({"type": SIGNALLING_TYPE, "message": PeerLeftMessage().model_dump()})
await self._close()
async def _close(self):
if self._pc:
await self._pc.close()
self._message_queue.clear()
@@ -296,7 +304,7 @@ class SmallWebRTCConnection(BaseObject):
await self._call_event_handler(state)
if state == "failed":
logger.warning("Connection failed, closing peer connection.")
await self.close()
await self._close()
# Despite the fact that aiortc provides this listener, they don't have a status for "disconnected"
# So, there is no advantage in looking at self._pc.connectionState