1"""Orchestration module — Graph orchestrator, A2A routing, graph executor, parallel scheduler, swarm coordinator, distributed orchestration, task decomposer"""

2

3 from agentos.orchestration.graph import (

4 GraphOrchestrator,

5 GraphNode,

6 GraphEdge,

7)

8 from agentos.orchestration.a2a_router import (

9 A2ARouter,

10 AgentCard as RouterAgentCard,

11 Task as RouterTask,

12 TaskResult,

13 TaskStatus,

14)

15 from agentos.orchestration.graph_executor import (

16 AgentGraph,

17 GraphRecipe,

18 GraphNodeState,

19 GraphResult,

20)

21 from agentos.orchestration.parallel import (

22 ParallelExecutor,

23 RunResult,

24)

25 from agentos.orchestration.swarm_coordinator import (

26 SwarmCoordinator,

27 AgentInfo,

28 AgentRole,

29 SwarmTask,

30 SwarmTopology,

31 TaskPriority,

32 TaskStatus as SwarmTaskStatus,

33 TaskAllocator,

34 ConflictResolver,

35 ConflictType,

36 MessageBus,

37 SwarmMessage,

38)

39 from agentos.orchestration.task_decomposer import (

40 TaskDecomposer,

41 TaskDAG,

42 TaskNode,

43 TaskEdge,

44 TaskNodeStatus,

45 DecompositionStrategy,

46 DecompositionTrace,

47 create_decomposer,

48)

49

50# Distributed orchestration (optional: requires ray)

51 try:

52 from agentos.orchestration.distributed import (

53 DistSwarmCoordinator,

54 DistSwarmConfig,

55 DistTaskQueue,

56 DistTaskRecord,

57 DistTaskStatus,

58 CrossNodeBus,

59 CrossNodeMailbox,

60 RayAgentActor,

61 AgentPlacementSpec,

62 AgentStatus as DistAgentStatus,

63 PlacementStrategy,

64 quick_start,

65 )

66 _HAS_DISTRIBUTED = True

67 except ImportError:

68 DistSwarmCoordinator = None # type: ignore

69 DistSwarmConfig = None

70 DistTaskQueue = None

71 DistTaskRecord = None

72 DistTaskStatus = None

73 CrossNodeBus = None

74 CrossNodeMailbox = None

75 RayAgentActor = None

76 AgentPlacementSpec = None

77 DistAgentStatus = None

78 PlacementStrategy = None

79 quick_start = None

80 _HAS_DISTRIBUTED = False

81

82__all__ = [

83 "GraphOrchestrator",

84 "GraphNode",

85 "GraphEdge",

86 "A2ARouter",

87 "RouterAgentCard",

88 "RouterTask",

89 "TaskResult",

90 "TaskStatus",

91 "AgentGraph",

92 "GraphRecipe",

93 "GraphNodeState",

94 "GraphResult",

95 "ParallelExecutor",

96 "RunResult",

97 # Swarm Coordinator v2

98 "SwarmCoordinator",

99 "AgentInfo",

100 "AgentRole",

101 "SwarmTask",

102 "SwarmTopology",

103 "TaskPriority",

104 "SwarmTaskStatus",

105 "TaskAllocator",

106 "ConflictResolver",

107 "ConflictType",

108 "MessageBus",

109 "SwarmMessage",

110 # Task Decomposer v2 (v1.14.7)

111 "TaskDecomposer",

112 "TaskDAG",

113 "TaskNode",

114 "TaskEdge",

115 "TaskNodeStatus",

116 "DecompositionStrategy",

117 "DecompositionTrace",

118 "create_decomposer",

119 # Distributed Orchestration (v1.14.2, optional)

120 "DistSwarmCoordinator",

121 "DistSwarmConfig",

122 "DistTaskQueue",

123 "DistTaskRecord",

124 "DistTaskStatus",

125 "CrossNodeBus",

126 "CrossNodeMailbox",

127 "RayAgentActor",

128 "AgentPlacementSpec",

129 "DistAgentStatus",

130 "PlacementStrategy",

131 "quick_start",

132]