Coverage for agentos/orchestration/__init__.py: 97%

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

2 

3from agentos.orchestration.a2a_router import ( 

4 A2ARouter, 

5 TaskResult, 

6 TaskStatus, 

7) 

8from agentos.orchestration.a2a_router import ( 

9 AgentCard as RouterAgentCard, 

10) 

11from agentos.orchestration.a2a_router import ( 

12 Task as RouterTask, 

13) 

14from agentos.orchestration.graph import ( 

15 GraphEdge, 

16 GraphNode, 

17 GraphOrchestrator, 

18) 

19from agentos.orchestration.graph_executor import ( 

20 AgentGraph, 

21 GraphNodeState, 

22 GraphRecipe, 

23 GraphResult, 

24) 

25from agentos.orchestration.parallel import ( 

26 ParallelExecutor, 

27 RunResult, 

28) 

29from agentos.orchestration.task_decomposer import ( 

30 DecompositionStrategy, 

31 DecompositionTrace, 

32 TaskDAG, 

33 TaskDecomposer, 

34 TaskEdge, 

35 TaskNode, 

36 TaskNodeStatus, 

37 create_decomposer, 

38) 

39from agentos.swarm.coordinator import ( 

40 ConflictResolver, 

41 ConflictType, 

42 MessageBus, 

43 SwarmCoordinator, 

44 SwarmMessage, 

45 SwarmTask, 

46 SwarmTopology, 

47 TaskAllocator, 

48 TaskPriority, 

49) 

50from agentos.swarm.coordinator import ( 

51 SwarmAgentInfo as AgentInfo, 

52) 

53from agentos.swarm.coordinator import ( 

54 SwarmAgentRole as AgentRole, 

55) 

56from agentos.swarm.coordinator import ( 

57 TaskStatus as SwarmTaskStatus, 

58) 

59 

60# Distributed orchestration (optional: requires ray) 

61try: 

62 from agentos.orchestration.distributed import ( 

63 AgentPlacementSpec, 

64 CrossNodeBus, 

65 CrossNodeMailbox, 

66 DistSwarmConfig, 

67 DistSwarmCoordinator, 

68 DistTaskQueue, 

69 DistTaskRecord, 

70 DistTaskStatus, 

71 PlacementStrategy, 

72 RayAgentActor, 

73 quick_start, 

74 ) 

75 from agentos.orchestration.distributed import ( 

76 AgentStatus as DistAgentStatus, 

77 ) 

78 

79 _HAS_DISTRIBUTED = True 

80except ImportError: 

81 DistSwarmCoordinator = None # type: ignore 

82 DistSwarmConfig = None 

83 DistTaskQueue = None 

84 DistTaskRecord = None 

85 DistTaskStatus = None 

86 CrossNodeBus = None 

87 CrossNodeMailbox = None 

88 RayAgentActor = None 

89 AgentPlacementSpec = None 

90 DistAgentStatus = None 

91 PlacementStrategy = None 

92 quick_start = None 

93 _HAS_DISTRIBUTED = False 

94 

95__all__ = [ 

96 "GraphOrchestrator", 

97 "GraphNode", 

98 "GraphEdge", 

99 "A2ARouter", 

100 "RouterAgentCard", 

101 "RouterTask", 

102 "TaskResult", 

103 "TaskStatus", 

104 "AgentGraph", 

105 "GraphRecipe", 

106 "GraphNodeState", 

107 "GraphResult", 

108 "ParallelExecutor", 

109 "RunResult", 

110 # Swarm Coordinator v2 

111 "SwarmCoordinator", 

112 "AgentInfo", 

113 "AgentRole", 

114 "SwarmTask", 

115 "SwarmTopology", 

116 "TaskPriority", 

117 "SwarmTaskStatus", 

118 "TaskAllocator", 

119 "ConflictResolver", 

120 "ConflictType", 

121 "MessageBus", 

122 "SwarmMessage", 

123 # Task Decomposer v2 (v1.14.7) 

124 "TaskDecomposer", 

125 "TaskDAG", 

126 "TaskNode", 

127 "TaskEdge", 

128 "TaskNodeStatus", 

129 "DecompositionStrategy", 

130 "DecompositionTrace", 

131 "create_decomposer", 

132 # Distributed Orchestration (v1.14.2, optional) 

133 "DistSwarmCoordinator", 

134 "DistSwarmConfig", 

135 "DistTaskQueue", 

136 "DistTaskRecord", 

137 "DistTaskStatus", 

138 "CrossNodeBus", 

139 "CrossNodeMailbox", 

140 "RayAgentActor", 

141 "AgentPlacementSpec", 

142 "DistAgentStatus", 

143 "PlacementStrategy", 

144 "quick_start", 

145]