wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Making cancel sentinel classes private

Browse Source
This commit is contained in:
Filipi Fuchter
2025-07-01 17:09:05 -03:00
parent fccd48bfff
commit 721f662bbe
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/pipecat/utils/asyncio/watchdog_priority_queue.py
View File

@@ -20,7 +20,7 @@ from pipecat.utils.asyncio.task_manager import BaseTaskManager
@dataclass @dataclass
class WatchdogPriorityCancelSentinel: class _WatchdogPriorityCancelSentinel:
def __lt__(self, other): def __lt__(self, other):
return True return True
@@ -62,7 +62,7 @@ class WatchdogPriorityQueue(asyncio.PriorityQueue):
else: else:
get_result = await super().get() get_result = await super().get()
if isinstance(get_result, WatchdogPriorityCancelSentinel): if isinstance(get_result, _WatchdogPriorityCancelSentinel):
logger.debug( logger.debug(
"Received WatchdogPriorityCancelSentinel, throwing CancelledError to force cancelling" "Received WatchdogPriorityCancelSentinel, throwing CancelledError to force cancelling"
) )
@@ -91,7 +91,7 @@ class WatchdogPriorityQueue(asyncio.PriorityQueue):
forces the task to raise CancelledError when consumed, ensuring proper forces the task to raise CancelledError when consumed, ensuring proper
task termination. task termination.
""" """
super().put_nowait(WatchdogPriorityCancelSentinel()) super().put_nowait(_WatchdogPriorityCancelSentinel())
async def _watchdog_get(self): async def _watchdog_get(self):
"""Get item from queue while periodically resetting watchdog timer.""" """Get item from queue while periodically resetting watchdog timer."""

6
src/pipecat/utils/asyncio/watchdog_queue.py
View File

@@ -20,7 +20,7 @@ from pipecat.utils.asyncio.task_manager import BaseTaskManager
@dataclass @dataclass
class WatchdogQueueCancelSentinel: class _WatchdogQueueCancelSentinel:
pass pass
@@ -61,7 +61,7 @@ class WatchdogQueue(asyncio.Queue):
else: else:
get_result = await super().get() get_result = await super().get()
if isinstance(get_result, WatchdogQueueCancelSentinel): if isinstance(get_result, _WatchdogQueueCancelSentinel):
logger.debug( logger.debug(
"Received WatchdogQueueCancelFrame, throwing CancelledError to force cancelling" "Received WatchdogQueueCancelFrame, throwing CancelledError to force cancelling"
) )
@@ -90,7 +90,7 @@ class WatchdogQueue(asyncio.Queue):
forces the task to raise CancelledError when consumed, ensuring proper forces the task to raise CancelledError when consumed, ensuring proper
task termination. task termination.
""" """
super().put_nowait(WatchdogQueueCancelSentinel()) super().put_nowait(_WatchdogQueueCancelSentinel())
async def _watchdog_get(self): async def _watchdog_get(self):
"""Get item from queue while periodically resetting watchdog timer.""" """Get item from queue while periodically resetting watchdog timer."""