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

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

2 

3from agentos.orchestration.graph import ( 

4 GraphOrchestrator, 

5 GraphNode, 

6 GraphEdge, 

7) 

8from agentos.orchestration.a2a_router import ( 

9 A2ARouter, 

10 AgentCard as RouterAgentCard, 

11 Task as RouterTask, 

12 TaskResult, 

13 TaskStatus, 

14) 

15from agentos.orchestration.graph_executor import ( 

16 AgentGraph, 

17 GraphRecipe, 

18 GraphNodeState, 

19 GraphResult, 

20) 

21from agentos.orchestration.parallel import ( 

22 ParallelExecutor, 

23 RunResult, 

24) 

25from 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) 

39from 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) 

51try: 

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 

67except 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]