end

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

@@ -5,12 +5,51 @@ Node metadata is loaded from: ../../templates/metadata/nodes/condition.yaml
 
 from __future__ import annotations
 
+import logging
+import re
+import signal
 from typing import Any
 
 from langbot_plugin.api.entities.builtin.workflow import ExecutionContext
 from ..node import WorkflowNode, workflow_node
 from ..safe_eval import safe_eval_with_vars
 
+logger = logging.getLogger(__name__)
+
+# 正则表达式超时限制（秒）
+_REGEX_TIMEOUT = 2
+
+
+class _RegexTimeoutError(Exception):
+    """正则表达式超时错误"""
+    pass
+
+
+def _handle_timeout(signum, frame):
+    """超时信号处理"""
+    raise _RegexTimeoutError('Regex match timed out')
+
+
+def _safe_regex_match(pattern: str, text: str) -> tuple[bool, str]:
+    """安全地执行正则表达式匹配，带有超时限制"""
+    # 设置超时信号
+    old_handler = signal.signal(signal.SIGALRM, _handle_timeout)
+    signal.setitimer(signal.ITIMER_REAL, _REGEX_TIMEOUT)
+
+    try:
+        result = bool(re.match(pattern, str(text)))
+        return result, ''
+    except _RegexTimeoutError:
+        logger.warning('Regex match timed out for pattern: %s', pattern[:50])
+        return False, 'Regex match timed out'
+    except re.error as e:
+        logger.warning('Invalid regex pattern: %s', e)
+        return False, f'Invalid regex: {e}'
+    finally:
+        signal.setitimer(signal.ITIMER_REAL, 0)
+        signal.signal(signal.SIGALRM, old_handler)
+
+
 @workflow_node('condition')
 class ConditionNode(WorkflowNode):
     """Condition node - branch based on condition"""
@@ -35,11 +74,11 @@ class ConditionNode(WorkflowNode):
         elif condition_type == 'empty':
             result = not bool(input_data)
         elif condition_type == 'regex':
-            import re
-
             left = self.get_config('left_value', '')
             pattern = self.get_config('right_value', '')
-            result = bool(re.match(pattern, str(left)))
+            result, error = _safe_regex_match(pattern, left)
+            if error:
+                return {'true': None, 'false': input_data, 'error': error}
 
         if result:
             return {'true': input_data, 'false': None}
@@ -50,7 +89,8 @@ class ConditionNode(WorkflowNode):
         try:
             local_vars = {'input': data, 'data': data, 'variables': context.variables}
             return bool(safe_eval_with_vars(expression, local_vars))
-        except Exception:
+        except Exception as e:
+            logger.warning('Expression evaluation error: %s', e)
             return False
 
     async def _evaluate_comparison(self, data: Any, context: ExecutionContext) -> bool: