apunkt/dograh
1
0
Fork 0
mirror of https://github.com/dograh-hq/dograh.git synced 2026-06-28 08:49:42 +02:00
Code Issues Projects Releases Packages Wiki Activity

chore: upgrade pipecat

Browse source
This commit is contained in:
Abhishek Kumar 2026-03-06 16:49:14 +05:30
parent 7b77721964
commit e34e4f8f3c
13 changed files with 45 additions and 58 deletions
Download patch file Download diff file Expand all files Collapse all files

11
api/services/workflow/pipecat_engine_custom_tools.py
View file

@ -128,14 +128,14 @@ class CustomToolManager:
function_name = schema["function"]["name"]
# Create and register the handler
handler, disable_timeout, cancel_on_interruption = self._create_handler(
handler, timeout_secs, cancel_on_interruption = self._create_handler(
tool, function_name
)
self._engine.llm.register_function(
function_name,
handler,
cancel_on_interruption=cancel_on_interruption,
disable_timeout=disable_timeout,
timeout_secs=timeout_secs,
)
logger.debug(
@ -156,21 +156,20 @@ class CustomToolManager:
Returns:
Async handler function for the tool
"""
# Whether to disable function call timeout
disable_timeout = False
timeout_secs: Optional[float] = None
cancel_on_interruption = True
if tool.category == ToolCategory.END_CALL.value:
cancel_on_interruption = False
handler = self._create_end_call_handler(tool, function_name)
elif tool.category == ToolCategory.TRANSFER_CALL.value:
disable_timeout = True
timeout_secs = 120.0
cancel_on_interruption = False
handler = self._create_transfer_call_handler(tool, function_name)
else:
handler = self._create_http_tool_handler(tool, function_name)
return handler, disable_timeout, cancel_on_interruption
return handler, timeout_secs, cancel_on_interruption
def _create_http_tool_handler(self, tool: Any, function_name: str):
"""Create a handler function for an HTTP API tool.