mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-04 12:56:02 +00:00
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
"""Wait Node - pause execution for a duration
|
|
|
|
Node metadata is loaded from: ../../templates/metadata/nodes/wait.yaml
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Any
|
|
|
|
from langbot_plugin.api.entities.builtin.workflow import ExecutionContext
|
|
from ..node import WorkflowNode, workflow_node
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# 最大等待时间（秒）
|
|
_MAX_WAIT_SECONDS = 300 # 5 分钟
|
|
|
|
|
|
@workflow_node('wait')
|
|
class WaitNode(WorkflowNode):
|
|
"""Wait node - pause execution for a duration"""
|
|
|
|
category = 'control'
|
|
|
|
async def execute(self, inputs: dict[str, Any], context: ExecutionContext) -> dict[str, Any]:
|
|
import asyncio
|
|
|
|
duration = self.get_config('duration', 1)
|
|
duration_type = self.get_config('duration_type', 'seconds')
|
|
|
|
# 转换为秒
|
|
if duration_type == 'minutes':
|
|
duration *= 60
|
|
elif duration_type == 'hours':
|
|
duration *= 3600
|
|
|
|
# 限制最大等待时间
|
|
if duration > _MAX_WAIT_SECONDS:
|
|
logger.warning('Wait duration %s exceeds maximum %s, capping to %s',
|
|
duration, _MAX_WAIT_SECONDS, _MAX_WAIT_SECONDS)
|
|
duration = _MAX_WAIT_SECONDS
|
|
|
|
# 确保 duration 为正数
|
|
duration = max(0, duration)
|
|
|
|
logger.info('Waiting for %.2f seconds', duration)
|
|
await asyncio.sleep(duration)
|
|
|
|
return {'output': inputs.get('input'), 'waited_seconds': duration}
|