Coverage for agentos/swarm/__init__.py: 100%

1"""Swarm module - Swarm coordinator, collaboration patterns, v1.9.8 tool registry + routing""" 

2 

3from agentos.swarm.coordinator import ( 

4 SmartSwarmCoordinator, 

5 SwarmCoordinator, # backward-compat alias 

6 SwarmTopology, 

7 SwarmMessage, 

8 SwarmResult, 

9 ExecutionMode, 

10 # Migrated from orchestration/swarm_coordinator.py (v1.16.2) 

11 SwarmAgentRole, 

12 TaskPriority, 

13 TaskStatus, 

14 AgentInfo, 

15 SwarmTask, 

16 TaskAllocator, 

17 ConflictResolver, 

18 ConflictType, 

19) 

20from agentos.swarm.patterns import ( 

21 SwarmPatterns, 

22 Topology, 

23 CollaborationConfig, 

24 CollaborationResult, 

25) 

26from agentos.swarm.task_decomposer import ( 

27 TaskDecomposer, 

28 SubTask, 

29 Decomposition, 

30) 

31from agentos.swarm.result_fusion import ( 

32 ResultFusion, 

33 FusedResult, 

34) 

35from agentos.swarm.eval_feedback_loop import ( 

36 EvalFeedbackLoop, 

37 LoopResult, 

38 FeedbackSignal, 

39 RetryConfig, 

40) 

41from agentos.swarm.code_sandbox import ( 

42 CodeSandbox, 

43 SandboxResult, 

44 TestCase, 

45 CodeFeedbackExtractor, 

46) 

47from agentos.swarm.human_loop import ( 

48 HITLManager, 

49 HITLConfig, 

50 Breakpoint, 

51 BreakpointType, 

52 HumanDecision, 

53) 

54from agentos.swarm.agent_monitor import ( 

55 AgentMonitor, 

56 QualityGate, 

57 MonitorReport, 

58 GateResult, 

59 GateStatus, 

60 GateAction, 

61 output_not_empty, 

62 output_length_range, 

63 no_error_output, 

64 contains_keywords, 

65 latency_max, 

66 confidence_min, 

67) 

68from agentos.swarm.execution_trace import ( 

69 ExecutionTrace, 

70 TraceSpan, 

71 TraceEvent, 

72 TraceCollector, 

73) 

74from agentos.swarm.agent_memory import ( 

75 AgentMemory, 

76 WorkingMemory, 

77 ShortTermMemory, 

78 LongTermMemory, 

79 ContextWindowManager, 

80 ContextBudget, 

81 MemoryEntry, 

82) 

83from agentos.swarm.tool_registry import ( 

84 ToolRegistry, 

85 ToolRouter, 

86 ToolExecutor, 

87 ToolSchema, 

88 ToolParam, 

89 ToolCategory, 

90 RoutingDecision, 

91 RoutingContext, 

92 ToolExecutionError, 

93 create_tool, 

94) 

95 

96__all__ = [ 

97 # Coordinator 

98 "SmartSwarmCoordinator", 

99 "SwarmCoordinator", 

100 "SwarmTopology", 

101 "SwarmMessage", 

102 "SwarmResult", 

103 "ExecutionMode", 

104 # Migrated from orchestration (v1.16.2) 

105 "SwarmAgentRole", 

106 "TaskPriority", 

107 "TaskStatus", 

108 "AgentInfo", 

109 "SwarmTask", 

110 "TaskAllocator", 

111 "ConflictResolver", 

112 "ConflictType", 

113 # Patterns 

114 "SwarmPatterns", 

115 "Topology", 

116 "CollaborationConfig", 

117 "CollaborationResult", 

118 # Task Decomposer 

119 "TaskDecomposer", 

120 "SubTask", 

121 "Decomposition", 

122 # Result Fusion 

123 "ResultFusion", 

124 "FusedResult", 

125 # Eval Feedback Loop 

126 "EvalFeedbackLoop", 

127 "LoopResult", 

128 "FeedbackSignal", 

129 "RetryConfig", 

130 # Code Sandbox (v1.9.5) 

131 "CodeSandbox", 

132 "SandboxResult", 

133 "TestCase", 

134 "CodeFeedbackExtractor", 

135 # Human-in-the-Loop (v1.9.5) 

136 "HITLManager", 

137 "HITLConfig", 

138 "Breakpoint", 

139 "BreakpointType", 

140 "HumanDecision", 

141 # Agent Monitor (v1.9.6) 

142 "AgentMonitor", 

143 "QualityGate", 

144 "MonitorReport", 

145 "GateResult", 

146 "GateStatus", 

147 "GateAction", 

148 "output_not_empty", 

149 "output_length_range", 

150 "no_error_output", 

151 "contains_keywords", 

152 "latency_max", 

153 "confidence_min", 

154 # Execution Trace (v1.9.6) 

155 "ExecutionTrace", 

156 "TraceSpan", 

157 "TraceEvent", 

158 "TraceCollector", 

159 # Agent Memory (v1.9.7) 

160 "AgentMemory", 

161 "WorkingMemory", 

162 "ShortTermMemory", 

163 "LongTermMemory", 

164 "ContextWindowManager", 

165 "ContextBudget", 

166 "MemoryEntry", 

167 # Tool Registry & Router (v1.9.8) 

168 "ToolRegistry", 

169 "ToolRouter", 

170 "ToolExecutor", 

171 "ToolSchema", 

172 "ToolParam", 

173 "ToolCategory", 

174 "RoutingDecision", 

175 "RoutingContext", 

176 "ToolExecutionError", 

177 "create_tool", 

178]