ai-agents/scripts/workflow/__init__.py at 82e350093bdc4a347fca95d22ed63c08640524b0 · rjmurillo/ai-agents · 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
"""Workflow execution and chaining for agent pipelines.

Supports sequential chaining, parallel execution, and refinement loops.
"""

from scripts.workflow.coordinator import (
    CentralizedStrategy,
    CoordinationStrategy,
    HierarchicalStrategy,
    MeshStrategy,
    aggregate_subordinate_outputs,
    build_execution_plan,
    find_ready_steps,
    get_strategy,
)
from scripts.workflow.executor import WorkflowExecutor
from scripts.workflow.parallel import (
    AggregationStrategy,
    ParallelGroup,
    ParallelStepExecutor,
    can_parallelize,
    identify_parallel_groups,
    mark_parallel_steps,
)
from scripts.workflow.schema import (
    CoordinationMode,
    StepKind,
    StepRef,
    StepResult,
    WorkflowDefinition,
    WorkflowResult,
    WorkflowStatus,
    WorkflowStep,
)

__all__ = [
    "AggregationStrategy",
    "CentralizedStrategy",
    "CoordinationMode",
    "CoordinationStrategy",
    "HierarchicalStrategy",
    "MeshStrategy",
    "ParallelGroup",
    "ParallelStepExecutor",
    "StepKind",
    "StepRef",
    "StepResult",
    "WorkflowDefinition",
    "WorkflowExecutor",
    "WorkflowResult",
    "WorkflowStatus",
    "WorkflowStep",
    "aggregate_subordinate_outputs",
    "build_execution_plan",
    "can_parallelize",
    "find_ready_steps",
    "get_strategy",
    "identify_parallel_groups",
    "mark_parallel_steps",
]