predicate-runtime-python/predicate/agents/__init__.py at 6e8958c7a4cccdbc5d511e6aa2d45ef558215009 · PredicateSystems/predicate-runtime-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
"""
Agent-level orchestration helpers (snapshot-first, verification-first).

This package provides a "browser-use-like" agent surface built on top of:
- AgentRuntime (snapshots, verification, tracing)
- RuntimeAgent (execution loop and bounded vision fallback)

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 (
    ActionRecord,
    AuthBoundaryConfig,
    CheckoutDetectionConfig,
    ExecutorOverride,
    IntentHeuristics,
    ModalDismissalConfig,
    Plan,
    PlanStep,
    PlannerExecutorAgent,
    PlannerExecutorConfig,
    PredicateSpec,
    RecoveryNavigationConfig,
    RetryConfig,
    RunOutcome,
    SnapshotContext,
    SnapshotEscalationConfig,
    StepOutcome,
    StepStatus,
    StepwisePlanningConfig,
    normalize_plan,
    validate_plan_smoothness,
)
from .recovery import RecoveryCheckpoint, RecoveryState
from .agent_factory import (
    ConfigPreset,
    create_planner_executor_agent,
    get_config_preset,
)

__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
    "ActionRecord",
    "AuthBoundaryConfig",
    "CheckoutDetectionConfig",
    "ExecutorOverride",
    "IntentHeuristics",
    "ModalDismissalConfig",
    "Plan",
    "PlanStep",
    "PlannerExecutorAgent",
    "PlannerExecutorConfig",
    "PredicateSpec",
    "RecoveryNavigationConfig",
    "RetryConfig",
    "RunOutcome",
    "SnapshotContext",
    "SnapshotEscalationConfig",
    "StepOutcome",
    "StepStatus",
    "StepwisePlanningConfig",
    "normalize_plan",
    "validate_plan_smoothness",
    # Recovery
    "RecoveryCheckpoint",
    "RecoveryState",
    # Agent Factory
    "ConfigPreset",
    "create_planner_executor_agent",
    "get_config_preset",
]