Abstraction for browser automation tasks

Author: SentienceDEV

predicate/agents/__init__.py

@@ -8,15 +8,35 @@
 Agent types:
 - PredicateBrowserAgent: Single-executor agent with manual step definitions
 - PlannerExecutorAgent: Two-tier agent with LLM-generated plans
+
+Task abstractions:
+- AutomationTask: Generic task model for browser automation
+- TaskCategory: Task category hints for heuristics selection
+
+Heuristics:
+- HeuristicHint: Planner-generated hints for element selection
+- ComposableHeuristics: Dynamic heuristics composition
+
+Recovery:
+- RecoveryState: Checkpoint tracking for rollback recovery
+- RecoveryCheckpoint: Individual recovery checkpoint
 """
 
+from .automation_task import (
+    AutomationTask,
+    ExtractionSpec,
+    SuccessCriteria,
+    TaskCategory,
+)
 from .browser_agent import (
     CaptchaConfig,
     PermissionRecoveryConfig,
     PredicateBrowserAgent,
     PredicateBrowserAgentConfig,
     VisionFallbackConfig,
 )
+from .composable_heuristics import ComposableHeuristics
+from .heuristic_spec import COMMON_HINTS, HeuristicHint, get_common_hint
 from .planner_executor_agent import (
     ExecutorOverride,
     IntentHeuristics,
@@ -35,14 +55,25 @@
     normalize_plan,
     validate_plan_smoothness,
 )
+from .recovery import RecoveryCheckpoint, RecoveryState
 
 __all__ = [
+    # Automation Task
+    "AutomationTask",
+    "ExtractionSpec",
+    "SuccessCriteria",
+    "TaskCategory",
     # Browser Agent
     "CaptchaConfig",
     "PermissionRecoveryConfig",
     "PredicateBrowserAgent",
     "PredicateBrowserAgentConfig",
     "VisionFallbackConfig",
+    # Heuristics
+    "COMMON_HINTS",
+    "ComposableHeuristics",
+    "HeuristicHint",
+    "get_common_hint",
     # Planner + Executor Agent
     "ExecutorOverride",
     "IntentHeuristics",
@@ -60,5 +91,7 @@
     "StepStatus",
     "normalize_plan",
     "validate_plan_smoothness",
+    # Recovery
+    "RecoveryCheckpoint",
+    "RecoveryState",
 ]
-