dv-pipecat-ai 0.0.74.dev770__py3-none-any.whl → 0.0.82.dev776__py3-none-any.whl

pipecat/utils/asyncio/watchdog_async_iterator.py

@@ -1,72 +0,0 @@
-#
-# Copyright (c) 2024–2025, Daily
-#
-# SPDX-License-Identifier: BSD 2-Clause License
-#
-
-import asyncio
-from typing import AsyncIterator, Optional
-
-from pipecat.utils.asyncio.task_manager import BaseTaskManager
-
-
-class WatchdogAsyncIterator:
-    """An asynchronous iterator that monitors activity and resets the current
-    task watchdog timer. This is necessary to avoid task watchdog timers to
-    expire while we are waiting to get an item from the iterator.
-
-    """
-
-    def __init__(
-        self,
-        async_iterable,
-        *,
-        manager: BaseTaskManager,
-        timeout: float = 2.0,
-    ):
-        self._async_iterable = async_iterable
-        self._manager = manager
-        self._timeout = timeout
-        self._iter: Optional[AsyncIterator] = None
-        self._current_anext_task: Optional[asyncio.Task] = None
-
-    def __aiter__(self):
-        return self
-
-    async def __anext__(self):
-        if not self._iter:
-            self._iter = await self._ensure_async_iterator(self._async_iterable)
-
-        if self._manager.task_watchdog_enabled:
-            return await self._watchdog_anext()
-        else:
-            return await self._iter.__anext__()
-
-    async def _watchdog_anext(self):
-        while True:
-            try:
-                if not self._current_anext_task:
-                    self._current_anext_task = asyncio.create_task(self._iter.__anext__())
-
-                item = await asyncio.wait_for(
-                    asyncio.shield(self._current_anext_task),
-                    timeout=self._timeout,
-                )
-
-                self._manager.task_reset_watchdog()
-
-                # The task has finish, so we will create a new one for th next item.
-                self._current_anext_task = None
-
-                return item
-            except asyncio.TimeoutError:
-                self._manager.task_reset_watchdog()
-            except StopAsyncIteration:
-                self._current_anext_task = None
-                raise
-
-    async def _ensure_async_iterator(self, obj) -> AsyncIterator:
-        aiter = obj.__aiter__()
-        if asyncio.iscoroutine(aiter):
-            aiter = await aiter
-        return aiter

pipecat/utils/asyncio/watchdog_event.py

@@ -1,42 +0,0 @@
-#
-# Copyright (c) 2024–2025, Daily
-#
-# SPDX-License-Identifier: BSD 2-Clause License
-#
-
-import asyncio
-
-from pipecat.utils.asyncio.task_manager import BaseTaskManager
-
-
-class WatchdogEvent(asyncio.Event):
-    """An asynchronous event that resets the current task watchdog timer. This
-    is necessary to avoid task watchdog timers to expire while we are waiting on
-    the event.
-
-    """
-
-    def __init__(
-        self,
-        manager: BaseTaskManager,
-        *,
-        timeout: float = 2.0,
-    ) -> None:
-        super().__init__()
-        self._manager = manager
-        self._timeout = timeout
-
-    async def wait(self):
-        if self._manager.task_watchdog_enabled:
-            return await self._watchdog_wait()
-        else:
-            return await super().wait()
-
-    async def _watchdog_wait(self):
-        while True:
-            try:
-                await asyncio.wait_for(super().wait(), timeout=self._timeout)
-                self._manager.task_reset_watchdog()
-                return True
-            except asyncio.TimeoutError:
-                self._manager.task_reset_watchdog()

pipecat/utils/asyncio/watchdog_priority_queue.py

@@ -1,48 +0,0 @@
-#
-# Copyright (c) 2024–2025, Daily
-#
-# SPDX-License-Identifier: BSD 2-Clause License
-#
-
-import asyncio
-
-from pipecat.utils.asyncio.task_manager import BaseTaskManager
-
-
-class WatchdogPriorityQueue(asyncio.PriorityQueue):
-    """An asynchronous priority queue that resets the current task watchdog
-    timer. This is necessary to avoid task watchdog timers to expire while we
-    are waiting to get an item from the queue.
-
-    """
-
-    def __init__(
-        self,
-        manager: BaseTaskManager,
-        *,
-        maxsize: int = 0,
-        timeout: float = 2.0,
-    ) -> None:
-        super().__init__(maxsize)
-        self._manager = manager
-        self._timeout = timeout
-
-    async def get(self):
-        if self._manager.task_watchdog_enabled:
-            return await self._watchdog_get()
-        else:
-            return await super().get()
-
-    def task_done(self):
-        if self._manager.task_watchdog_enabled:
-            self._manager.task_reset_watchdog()
-        super().task_done()
-
-    async def _watchdog_get(self):
-        while True:
-            try:
-                item = await asyncio.wait_for(super().get(), timeout=self._timeout)
-                self._manager.task_reset_watchdog()
-                return item
-            except asyncio.TimeoutError:
-                self._manager.task_reset_watchdog()

pipecat/utils/asyncio/watchdog_queue.py

@@ -1,48 +0,0 @@
-#
-# Copyright (c) 2024–2025, Daily
-#
-# SPDX-License-Identifier: BSD 2-Clause License
-#
-
-import asyncio
-
-from pipecat.utils.asyncio.task_manager import BaseTaskManager
-
-
-class WatchdogQueue(asyncio.Queue):
-    """An asynchronous queue that resets the current task watchdog timer. This
-    is necessary to avoid task watchdog timers to expire while we are waiting to
-    get an item from the queue.
-
-    """
-
-    def __init__(
-        self,
-        manager: BaseTaskManager,
-        *,
-        maxsize: int = 0,
-        timeout: float = 2.0,
-    ) -> None:
-        super().__init__(maxsize)
-        self._manager = manager
-        self._timeout = timeout
-
-    async def get(self):
-        if self._manager.task_watchdog_enabled:
-            return await self._watchdog_get()
-        else:
-            return await super().get()
-
-    def task_done(self):
-        if self._manager.task_watchdog_enabled:
-            self._manager.task_reset_watchdog()
-        super().task_done()
-
-    async def _watchdog_get(self):
-        while True:
-            try:
-                item = await asyncio.wait_for(super().get(), timeout=self._timeout)
-                self._manager.task_reset_watchdog()
-                return item
-            except asyncio.TimeoutError:
-                self._manager.task_reset_watchdog()