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

fix: ensure function call timeout task is always cancelled

Browse Source
This commit is contained in:
Om Chauhan
2026-02-02 08:38:54 +05:30
parent f453227ba3
commit 50dedf350d
1 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/pipecat/services/llm_service.py
View File

@@ -681,12 +681,11 @@ class LLMService(UserTurnCompletionLLMServiceMixin, AIService):
timeout_task = self.create_task(timeout_handler()) timeout_task = self.create_task(timeout_handler())
# Yield to the event loop so the timeout task coroutine gets entered
# before it could be cancelled. Without this, cancelling the task before
# it starts would leave the coroutine in a "never awaited" state.
await asyncio.sleep(0)
try: try:
# Yield to the event loop so the timeout task coroutine gets entered
# before it could be cancelled. Without this, cancelling the task before
# it starts would leave the coroutine in a "never awaited" state.
await asyncio.sleep(0)
if isinstance(item.handler, DirectFunctionWrapper): if isinstance(item.handler, DirectFunctionWrapper):
# Handler is a DirectFunctionWrapper # Handler is a DirectFunctionWrapper
await item.handler.invoke( await item.handler.invoke(
@@ -722,12 +721,12 @@ class LLMService(UserTurnCompletionLLMServiceMixin, AIService):
) )
await item.handler(params) await item.handler(params)
except Exception as e: except Exception as e:
# Cancel timeout task if it exists
if timeout_task and not timeout_task.done():
await self.cancel_task(timeout_task)
error_message = f"Error executing function call [{runner_item.function_name}]: {e}" error_message = f"Error executing function call [{runner_item.function_name}]: {e}"
logger.error(f"{self} {error_message}") logger.error(f"{self} {error_message}")
await self.push_error(error_msg=error_message, exception=e, fatal=False) await self.push_error(error_msg=error_message, exception=e, fatal=False)
finally:
if timeout_task and not timeout_task.done():
await self.cancel_task(timeout_task)
async def _cancel_function_call(self, function_name: Optional[str]): async def _cancel_function_call(self, function_name: Optional[str]):
cancelled_tasks = set() cancelled_tasks = set()