Making cancel sentinel classes private

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

View File

@@ -20,7 +20,7 @@ from pipecat.utils.asyncio.task_manager import BaseTaskManager
@dataclass
class WatchdogPriorityCancelSentinel:
class _WatchdogPriorityCancelSentinel:
def __lt__(self, other):
return True
@@ -62,7 +62,7 @@ class WatchdogPriorityQueue(asyncio.PriorityQueue):
else:
get_result = await super().get()
if isinstance(get_result, WatchdogPriorityCancelSentinel):
if isinstance(get_result, _WatchdogPriorityCancelSentinel):
logger.debug(
"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
task termination.
"""
super().put_nowait(WatchdogPriorityCancelSentinel())
super().put_nowait(_WatchdogPriorityCancelSentinel())
async def _watchdog_get(self):
"""Get item from queue while periodically resetting watchdog timer."""

View File

@@ -20,7 +20,7 @@ from pipecat.utils.asyncio.task_manager import BaseTaskManager
@dataclass
class WatchdogQueueCancelSentinel:
class _WatchdogQueueCancelSentinel:
pass
@@ -61,7 +61,7 @@ class WatchdogQueue(asyncio.Queue):
else:
get_result = await super().get()
if isinstance(get_result, WatchdogQueueCancelSentinel):
if isinstance(get_result, _WatchdogQueueCancelSentinel):
logger.debug(
"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
task termination.
"""
super().put_nowait(WatchdogQueueCancelSentinel())
super().put_nowait(_WatchdogQueueCancelSentinel())
async def _watchdog_get(self):
"""Get item from queue while periodically resetting watchdog timer."""