Skip to main content

aster/agents/
mod.rs

1mod agent;
2pub(crate) mod chatrecall_extension;
3pub(crate) mod code_execution_extension;
4pub mod execute_commands;
5pub mod extension;
6pub mod extension_malware_check;
7pub mod extension_manager;
8pub mod extension_manager_extension;
9pub mod final_output_tool;
10pub mod identity;
11mod large_response_handler;
12pub mod mcp_client;
13pub mod moim;
14pub mod platform_tools;
15pub mod prompt_manager;
16mod reply_parts;
17pub mod retry;
18mod schedule_tool;
19pub(crate) mod skills_extension;
20pub mod subagent_execution_tool;
21pub mod subagent_handler;
22mod subagent_task_config;
23pub mod subagent_tool;
24pub(crate) mod todo_extension;
25mod tool_execution;
26pub mod types;
27
28/// SubAgent 调度器模块
29///
30/// 基于 Anthropic 最佳实践实现的 SubAgent 调度系统,提供:
31/// - Orchestrator-Worker 模式的任务分发
32/// - 上下文继承、压缩和隔离
33/// - 并行执行和依赖管理
34/// - 结果聚合和摘要生成
35pub mod subagent_scheduler;
36
37// ============================================================================
38// ============================================================================
39
40/// Agent context management module
41///
42/// Provides context creation, inheritance, compression, filtering,
43/// persistence, and isolation capabilities for agents.
44pub mod context;
45
46/// Agent communication module
47///
48/// Provides inter-agent communication including message bus,
49/// shared state management, and agent coordination.
50pub mod communication;
51
52/// Parallel execution module
53///
54/// Provides parallel agent execution with dependency management,
55/// retry logic, and agent resource pooling.
56pub mod parallel;
57
58/// Agent monitoring module
59///
60/// Provides metrics collection, alert management, and
61/// performance analysis for agent execution.
62pub mod monitor;
63
64/// Agent resume module
65///
66/// Provides state persistence, checkpoint management,
67/// and agent resume capabilities.
68pub mod resume;
69
70/// Specialized agents module
71///
72/// Provides specialized agent implementations including
73/// Explore agent and Plan agent.
74pub mod specialized;
75
76/// Unified error handling module
77///
78/// Provides comprehensive error handling including error recording,
79/// timeout handling, and retry mechanisms.
80pub mod error_handling;
81
82// ============================================================================
83// Core Agent Exports
84// ============================================================================
85
86pub use agent::{Agent, AgentEvent};
87pub use execute_commands::COMPACT_TRIGGERS;
88pub use extension::ExtensionConfig;
89pub use extension_manager::ExtensionManager;
90pub use identity::AgentIdentity;
91pub use prompt_manager::PromptManager;
92pub use subagent_task_config::TaskConfig;
93pub use types::{FrontendTool, RetryConfig, SessionConfig, SuccessCheck};
94
95// ============================================================================
96// Context Module Re-exports
97// ============================================================================
98
99pub use context::{
100    // Core context types
101    AgentContext,
102    AgentContextError,
103    // Context manager
104    AgentContextManager,
105    AgentContextResult,
106    // Context operations
107    CompressionResult,
108    ContextFilter,
109    // Context inheritance
110    ContextInheritanceConfig,
111    ContextInheritanceType,
112    // Context isolation
113    ContextIsolation,
114    ContextMetadata,
115    ContextUpdate,
116    FileContext,
117    ResourceUsage,
118    SandboxRestrictions,
119    SandboxState,
120    SandboxedContext,
121    ToolExecutionResult,
122};
123
124// ============================================================================
125// Communication Module Re-exports
126// ============================================================================
127
128pub use communication::{
129    // Coordinator
130    AgentCapabilities,
131    AgentCoordinator,
132    // Message bus
133    AgentMessage,
134    AgentMessageBus,
135    AgentStatus,
136    AssignmentCriteria,
137    CoordinatorError,
138    CoordinatorEvent,
139    CoordinatorResult,
140    CoordinatorStats,
141    DeadlockInfo,
142    DependencyLink,
143    LoadBalanceStrategy,
144    // Shared state
145    Lock,
146    MessageBusError,
147    MessageBusResult,
148    MessageBusStats,
149    MessagePriority,
150    MessageSubscription,
151    MessageTarget,
152    SharedStateError,
153    SharedStateManager,
154    SharedStateResult,
155    SharedStateStats,
156    StateEvent,
157    Task,
158    TaskResult,
159    TaskStatus as CoordinatorTaskStatus,
160    WatchHandle,
161};
162
163// ============================================================================
164// Parallel Module Re-exports
165// ============================================================================
166
167pub use parallel::{
168    // Pool
169    AgentPool,
170    // Executor
171    AgentResult,
172    AgentTask,
173    AgentWorker,
174    DependencyGraph,
175    ExecutionProgress,
176    ExecutorError,
177    ExecutorResult,
178    MergedResult,
179    ParallelAgentConfig,
180    ParallelAgentExecutor,
181    ParallelExecutionResult,
182    PoolError,
183    PoolResult,
184    PoolStatus,
185    TaskExecutionInfo,
186    TaskStatus as ExecutorTaskStatus,
187};
188
189// ============================================================================
190// Monitor Module Re-exports
191// ============================================================================
192
193pub use monitor::{
194    AgentExecutionStatus,
195    // Metrics
196    AgentMonitor,
197    AggregatedStats,
198    // Alerts
199    Alert,
200    AlertManager,
201    AlertSeverity,
202    AlertType,
203    // Analyzer
204    AnalysisThresholds,
205    Bottleneck,
206    BottleneckCategory,
207    ErrorRecord,
208    FullAgentMetrics,
209    MonitorConfig,
210    PerformanceAnalyzer,
211    PerformanceMetrics,
212    PerformanceRating,
213    PerformanceReport,
214    PerformanceScores,
215    Suggestion,
216    SuggestionPriority,
217    TokenUsage,
218    ToolCallMetric,
219};
220
221// ============================================================================
222// Resume Module Re-exports
223// ============================================================================
224
225pub use resume::{
226    // Resumer
227    AgentResumer,
228    // State manager
229    AgentState,
230    AgentStateManager,
231    AgentStateStatus,
232    Checkpoint,
233    ResumeOptions,
234    ResumePoint,
235    ResumePointInfo,
236    ResumerError,
237    ResumerResult,
238    StateManagerError,
239    StateManagerResult,
240    ToolCallRecord,
241};
242
243// ============================================================================
244// Specialized Module Re-exports
245// ============================================================================
246
247pub use specialized::{
248    // Plan agent
249    Alternative,
250    ArchitecturalDecision,
251    // Explore agent
252    CodeSnippet,
253    Complexity,
254    CriticalFile,
255    ExploreAgent,
256    ExploreError,
257    ExploreOptions,
258    ExploreResult,
259    ExploreResultData,
260    ExploreStats,
261    ModificationType,
262    PlanAgent,
263    PlanError,
264    PlanOptions,
265    PlanResult,
266    PlanResultData,
267    PlanStep,
268    RequirementsAnalysis,
269    Risk,
270    RiskCategory,
271    RiskSeverity,
272    ScopeDefinition,
273    StructureAnalysis,
274    ThoroughnessLevel,
275};
276
277// ============================================================================
278// Error Handling Module Re-exports
279// ============================================================================
280
281pub use error_handling::{
282    // Error handler
283    AgentError,
284    AgentErrorKind,
285    ErrorContext,
286    ErrorHandler,
287    // Retry handler
288    RetryHandler,
289    RetryResult,
290    RetryStrategy,
291    // Timeout handler
292    TimeoutConfig,
293    TimeoutEvent,
294    TimeoutHandler,
295    TimeoutStatus,
296    UnifiedErrorRecord,
297    UnifiedRetryConfig,
298};
299
300// ============================================================================
301// SubAgent Scheduler Module Re-exports
302// ============================================================================
303
304pub use subagent_scheduler::{
305    // 配置
306    SchedulerConfig,
307    // 类型
308    SchedulerError,
309    SchedulerEvent,
310    SchedulerExecutionResult,
311    SchedulerProgress,
312    SchedulerResult,
313    // 策略
314    SchedulingStrategy,
315    StrategySelector,
316    // 执行器
317    SubAgentExecutor,
318    SubAgentResult,
319    SubAgentScheduler,
320    SubAgentTask,
321    SubAgentTaskStatus,
322    // 摘要
323    SummaryGenerator,
324    TaskComplexity,
325    TokenUsage as SchedulerTokenUsage,
326};