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

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

2 

3from agentos.swarm.agent_memory import ( 

4 AgentMemory, 

5 ContextBudget, 

6 ContextWindowManager, 

7 LongTermMemory, 

8 MemoryEntry, 

9 ShortTermMemory, 

10 WorkingMemory, 

11) 

12from agentos.swarm.agent_monitor import ( 

13 AgentMonitor, 

14 GateAction, 

15 GateResult, 

16 GateStatus, 

17 MonitorReport, 

18 QualityGate, 

19 confidence_min, 

20 contains_keywords, 

21 latency_max, 

22 no_error_output, 

23 output_length_range, 

24 output_not_empty, 

25) 

26from agentos.swarm.code_sandbox import ( 

27 CodeFeedbackExtractor, 

28 CodeSandbox, 

29 SandboxResult, 

30 TestCase, 

31) 

32from agentos.swarm.coordinator import ( 

33 ConflictResolver, 

34 ConflictType, 

35 ExecutionMode, 

36 SmartSwarmCoordinator, 

37 SwarmAgentInfo, 

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

39 SwarmAgentRole, 

40 SwarmCoordinator, # backward-compat alias 

41 SwarmMessage, 

42 SwarmResult, 

43 SwarmTask, 

44 SwarmTopology, 

45 TaskAllocator, 

46 TaskPriority, 

47 TaskStatus, 

48) 

49from agentos.swarm.eval_feedback_loop import ( 

50 EvalFeedbackLoop, 

51 FeedbackSignal, 

52 LoopResult, 

53 RetryConfig, 

54) 

55from agentos.swarm.execution_trace import ( 

56 ExecutionTrace, 

57 TraceCollector, 

58 TraceEvent, 

59 TraceSpan, 

60) 

61from agentos.swarm.human_loop import ( 

62 Breakpoint, 

63 BreakpointType, 

64 HITLConfig, 

65 HITLManager, 

66 HumanDecision, 

67) 

68from agentos.swarm.patterns import ( 

69 CollaborationConfig, 

70 CollaborationResult, 

71 SwarmPatterns, 

72 Topology, 

73) 

74from agentos.swarm.result_fusion import ( 

75 FusedResult, 

76 ResultFusion, 

77) 

78from agentos.swarm.task_decomposer import ( 

79 Decomposition, 

80 SubTask, 

81 TaskDecomposer, 

82) 

83from agentos.swarm.tool_registry import ( 

84 RoutingContext, 

85 RoutingDecision, 

86 ToolCategory, 

87 ToolExecutionError, 

88 ToolExecutor, 

89 ToolParam, 

90 ToolRegistry, 

91 ToolRouter, 

92 ToolSchema, 

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 "SwarmAgentInfo", 

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]