end

src/langbot/pkg/workflow/nodes/wait.py

@@ -5,11 +5,18 @@ 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"""
@@ -22,11 +29,22 @@ class WaitNode(WorkflowNode):
         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')}
+        return {'output': inputs.get('input'), 'waited_seconds': duration}