Struct AutoForestBuilder

Source

pub struct AutoForestBuilder { /* private fields */ }

Expand description

Builder for AutoForest

Implementations§

Source §

impl AutoForestBuilder

Source

pub fn new(config: Config) -> Self

Creates a new AutoForestBuilder with the given config

Source

pub fn with_tools(self, tools: Vec<Box<dyn Tool>>) -> Self

Sets the tools available to the forest

Examples found in repository ?

examples/auto_forest_demo.rs (lines 27-30)

15async fn main() -> helios_engine::Result<()> {
16    println!("🌲 AutoForest - Intelligent Agent Orchestration Demo\n");
17    println!("====================================================\n");
18
19    // Step 1: Load or create configuration
20    println!("📋 Creating configuration...");
21    let config = Config::builder().temperature(0.7).max_tokens(2048).build();
22    println!("✓ Configuration ready\n");
23
24    // Step 2: Create AutoForest with available tools
25    println!("🔧 Initializing AutoForest with tools...");
26    let mut auto_forest = AutoForest::new(config)
27        .with_tools(vec![
28            Box::new(CalculatorTool),
29            // Additional tools could be added here
30        ])
31        .build()
32        .await?;
33    println!("✓ AutoForest initialized\n");
34
35    // Step 3: Define a complex task
36    let task = r#"
37    I need to analyze a business problem:
38    1. A company has Q3 revenue data showing mixed results across 5 product categories
39    2. I need to understand which categories are underperforming
40    3. Calculate the performance metrics for each category
41    4. Identify trends and predict Q4 performance
42    5. Recommend which categories need attention
43    "#;
44
45    println!("📝 Submitting task to AutoForest...");
46    println!("Task: {}\n", task.trim());
47
48    // Step 4: Execute the task
49    println!("🚀 AutoForest is orchestrating agent deployment...\n");
50    let result = auto_forest.execute_task(task).await?;
51
52    // Step 5: Display the orchestration plan
53    println!("📊 Orchestration Plan Generated:");
54    println!("================================\n");
55
56    if let Some(plan) = auto_forest.orchestration_plan() {
57        println!("Task: {}", plan.task.trim());
58        println!("Number of Agents Spawned: {}", plan.num_agents);
59        println!("Planning Reasoning: {}\n", plan.reasoning);
60
61        println!("Agent Configurations:");
62        println!("-----------------");
63        for (i, agent_config) in plan.agents.iter().enumerate() {
64            println!(
65                "Agent {}: {} ({})",
66                i + 1,
67                agent_config.name,
68                agent_config.role
69            );
70            println!("  Prompt: {}", agent_config.system_prompt);
71            if !agent_config.tool_indices.is_empty() {
72                println!("  Tools: {:?}", agent_config.tool_indices);
73            }
74            println!();
75        }
76
77        println!("Task Breakdown:");
78        println!("--------------");
79        for (agent_name, subtask) in &plan.task_breakdown {
80            println!("• {} → {}", agent_name, subtask);
81        }
82        println!();
83    }
84
85    // Step 6: Display spawned agents
86    println!("🤖 Spawned Agents:");
87    println!("-----------------");
88    let spawned = auto_forest.spawned_agents();
89    println!("Total agents created: {}\n", spawned.len());
90
91    for (i, spawned_agent) in spawned.iter().enumerate() {
92        println!("Agent {}: {}", i + 1, spawned_agent.config.name);
93        println!("  Role: {}", spawned_agent.config.role);
94        if let Some(result) = &spawned_agent.result {
95            println!("  Result available: Yes ({} chars)", result.len());
96        } else {
97            println!("  Result: Pending");
98        }
99        println!();
100    }
101
102    // Step 7: Display the final synthesized result
103    println!("📈 Final Synthesized Result:");
104    println!("==========================\n");
105    println!("{}\n", result);
106
107    // Step 8: Demonstrate a second task with different complexity
108    println!("\n🔄 Executing a second task with different complexity...\n");
109
110    let simple_task = "Calculate the average of these numbers: 100, 200, 300, 400, 500. Then tell me what percentage each is of the total.";
111
112    println!("Task: {}\n", simple_task);
113    println!("🚀 Processing...\n");
114
115    let result2 = auto_forest.execute_task(simple_task).await?;
116
117    println!("Result:\n{}\n", result2);
118
119    if let Some(plan) = auto_forest.orchestration_plan() {
120        println!(
121            "This task resulted in {} agent(s) - a simpler orchestration plan.",
122            plan.num_agents
123        );
124        println!("Reasoning: {}\n", plan.reasoning);
125    }
126
127    println!("✅ AutoForest demo completed!");
128    println!("=============================");
129    println!("\n💡 Key Features Demonstrated:");
130    println!("  • Automatic agent spawning based on task complexity");
131    println!("  • Specialized agent prompt generation");
132    println!("  • Task breakdown across agents");
133    println!("  • Result aggregation and synthesis");
134    println!("  • Flexible orchestration for varying task complexity");
135
136    Ok(())
137}

More examples

Hide additional examples

examples/ultra_simple.rs (line 83)

9async fn main() -> helios_engine::Result<()> {
10    println!("🚀 Ultra Simple Helios Example\n");
11
12    // ========== SIMPLEST AGENT CREATION ==========
13    println!("1️⃣  Creating an agent - shortest possible syntax:\n");
14
15    // One-liner: Create agent with auto config
16    let mut agent = Agent::builder("Helper")
17        .auto_config()
18        .prompt("You are helpful and concise.")
19        .build()
20        .await?;
21
22    println!("✓ Agent created!\n");
23
24    // ========== SIMPLEST CHAT ==========
25    println!("2️⃣  Asking questions - simplest possible:\n");
26
27    // Use .ask() instead of .chat() for more natural syntax
28    let answer = agent.ask("What is 2+2?").await?;
29    println!("Q: What is 2+2?\nA: {}\n", answer);
30
31    // ========== SIMPLEST CONFIG ==========
32    println!("3️⃣  Creating config with shortest syntax:\n");
33
34    // Ultra-short config creation
35    let _config = Config::builder()
36        .m("gpt-4") // .m() is shorthand for .model()
37        .key("your-api-key") // .key() is shorthand for .api_key()
38        .temp(0.8) // .temp() is shorthand for .temperature()
39        .tokens(1024) // .tokens() is shorthand for .max_tokens()
40        .build();
41
42    println!("✓ Config created with ultra-short syntax!\n");
43
44    // ========== SIMPLEST AGENT WITH TOOLS ==========
45    println!("4️⃣  Agent with tools - simplest way:\n");
46
47    let mut calc_agent = Agent::builder("Calculator")
48        .auto_config()
49        .prompt("You are a math expert.")
50        .with_tool(Box::new(CalculatorTool)) // Add single tool
51        .build()
52        .await?;
53
54    let result = calc_agent.ask("Calculate 15 * 7 + 5").await?;
55    println!("Q: Calculate 15 * 7 + 5\nA: {}\n", result);
56
57    // ========== SIMPLEST QUICK AGENT ==========
58    println!("5️⃣  Quick agent - one method call:\n");
59
60    // Agent::quick() creates agent in ONE LINE with auto config!
61    let mut quick_agent = Agent::quick("QuickBot").await?;
62    let quick_answer = quick_agent.ask("Say hello!").await?;
63    println!("Response: {}\n", quick_answer);
64
65    // ========== SIMPLEST CHAT MESSAGES ==========
66    println!("6️⃣  Creating messages - super short syntax:\n");
67
68    use helios_engine::ChatMessage;
69
70    // Short aliases for message creation
71    let _sys_msg = ChatMessage::sys("You are helpful"); // .sys() not .system()
72    let _user_msg = ChatMessage::msg("Hello there"); // .msg() not .user()
73    let _reply_msg = ChatMessage::reply("Hi! How can I help?"); // .reply() not .assistant()
74
75    println!("✓ Messages created with ultra-short syntax!\n");
76
77    // ========== SHORTEST AUTOFOREST ==========
78    println!("7️⃣  AutoForest - simplest multi-agent orchestration:\n");
79
80    use helios_engine::AutoForest;
81
82    let mut forest = AutoForest::new(Config::builder().m("gpt-4").build())
83        .with_tools(vec![Box::new(CalculatorTool)])
84        .build()
85        .await?;
86
87    // Use .run() for shortest syntax
88    let forest_result = forest.run("Analyze these numbers: 10, 20, 30, 40").await?;
89    println!("Forest Result:\n{}\n", forest_result);
90
91    // ========== COMPARISON TABLE ==========
92    println!("📊 Syntax Comparison - Short vs Long:\n");
93    println!("┌─────────────────────┬──────────────────────┬─────────────────┐");
94    println!("│ Operation           │ Short Syntax         │ Long Syntax      │");
95    println!("├─────────────────────┼──────────────────────┼─────────────────┤");
96    println!("│ Create Agent        │ Agent::quick()       │ Agent::builder()│");
97    println!("│ Ask Question        │ .ask()               │ .chat()          │");
98    println!("│ System Prompt       │ .prompt()            │ .system_prompt() │");
99    println!("│ Config Model        │ .m()                 │ .model()         │");
100    println!("│ Config Key          │ .key()               │ .api_key()       │");
101    println!("│ Config Temp         │ .temp()              │ .temperature()   │");
102    println!("│ Config Tokens       │ .tokens()            │ .max_tokens()    │");
103    println!("│ System Message      │ ChatMessage::sys()   │ ChatMessage::system()");
104    println!("│ User Message        │ ChatMessage::msg()   │ ChatMessage::user()");
105    println!("│ Assistant Message   │ ChatMessage::reply() │ ChatMessage::assistant()");
106    println!("│ AutoForest Execute  │ .run()               │ .execute_task()  │");
107    println!("└─────────────────────┴──────────────────────┴─────────────────┘\n");
108
109    println!("✅ All examples completed!");
110    println!("💡 Tip: Mix and match short and long syntax based on your preference!");
111
112    Ok(())
113}

Source

pub async fn build(self) -> Result<AutoForest>

Builds the AutoForest orchestrator

Examples found in repository ?

examples/auto_forest_demo.rs (line 31)

15async fn main() -> helios_engine::Result<()> {
16    println!("🌲 AutoForest - Intelligent Agent Orchestration Demo\n");
17    println!("====================================================\n");
18
19    // Step 1: Load or create configuration
20    println!("📋 Creating configuration...");
21    let config = Config::builder().temperature(0.7).max_tokens(2048).build();
22    println!("✓ Configuration ready\n");
23
24    // Step 2: Create AutoForest with available tools
25    println!("🔧 Initializing AutoForest with tools...");
26    let mut auto_forest = AutoForest::new(config)
27        .with_tools(vec![
28            Box::new(CalculatorTool),
29            // Additional tools could be added here
30        ])
31        .build()
32        .await?;
33    println!("✓ AutoForest initialized\n");
34
35    // Step 3: Define a complex task
36    let task = r#"
37    I need to analyze a business problem:
38    1. A company has Q3 revenue data showing mixed results across 5 product categories
39    2. I need to understand which categories are underperforming
40    3. Calculate the performance metrics for each category
41    4. Identify trends and predict Q4 performance
42    5. Recommend which categories need attention
43    "#;
44
45    println!("📝 Submitting task to AutoForest...");
46    println!("Task: {}\n", task.trim());
47
48    // Step 4: Execute the task
49    println!("🚀 AutoForest is orchestrating agent deployment...\n");
50    let result = auto_forest.execute_task(task).await?;
51
52    // Step 5: Display the orchestration plan
53    println!("📊 Orchestration Plan Generated:");
54    println!("================================\n");
55
56    if let Some(plan) = auto_forest.orchestration_plan() {
57        println!("Task: {}", plan.task.trim());
58        println!("Number of Agents Spawned: {}", plan.num_agents);
59        println!("Planning Reasoning: {}\n", plan.reasoning);
60
61        println!("Agent Configurations:");
62        println!("-----------------");
63        for (i, agent_config) in plan.agents.iter().enumerate() {
64            println!(
65                "Agent {}: {} ({})",
66                i + 1,
67                agent_config.name,
68                agent_config.role
69            );
70            println!("  Prompt: {}", agent_config.system_prompt);
71            if !agent_config.tool_indices.is_empty() {
72                println!("  Tools: {:?}", agent_config.tool_indices);
73            }
74            println!();
75        }
76
77        println!("Task Breakdown:");
78        println!("--------------");
79        for (agent_name, subtask) in &plan.task_breakdown {
80            println!("• {} → {}", agent_name, subtask);
81        }
82        println!();
83    }
84
85    // Step 6: Display spawned agents
86    println!("🤖 Spawned Agents:");
87    println!("-----------------");
88    let spawned = auto_forest.spawned_agents();
89    println!("Total agents created: {}\n", spawned.len());
90
91    for (i, spawned_agent) in spawned.iter().enumerate() {
92        println!("Agent {}: {}", i + 1, spawned_agent.config.name);
93        println!("  Role: {}", spawned_agent.config.role);
94        if let Some(result) = &spawned_agent.result {
95            println!("  Result available: Yes ({} chars)", result.len());
96        } else {
97            println!("  Result: Pending");
98        }
99        println!();
100    }
101
102    // Step 7: Display the final synthesized result
103    println!("📈 Final Synthesized Result:");
104    println!("==========================\n");
105    println!("{}\n", result);
106
107    // Step 8: Demonstrate a second task with different complexity
108    println!("\n🔄 Executing a second task with different complexity...\n");
109
110    let simple_task = "Calculate the average of these numbers: 100, 200, 300, 400, 500. Then tell me what percentage each is of the total.";
111
112    println!("Task: {}\n", simple_task);
113    println!("🚀 Processing...\n");
114
115    let result2 = auto_forest.execute_task(simple_task).await?;
116
117    println!("Result:\n{}\n", result2);
118
119    if let Some(plan) = auto_forest.orchestration_plan() {
120        println!(
121            "This task resulted in {} agent(s) - a simpler orchestration plan.",
122            plan.num_agents
123        );
124        println!("Reasoning: {}\n", plan.reasoning);
125    }
126
127    println!("✅ AutoForest demo completed!");
128    println!("=============================");
129    println!("\n💡 Key Features Demonstrated:");
130    println!("  • Automatic agent spawning based on task complexity");
131    println!("  • Specialized agent prompt generation");
132    println!("  • Task breakdown across agents");
133    println!("  • Result aggregation and synthesis");
134    println!("  • Flexible orchestration for varying task complexity");
135
136    Ok(())
137}

More examples

Hide additional examples

examples/ultra_simple.rs (line 84)

9async fn main() -> helios_engine::Result<()> {
10    println!("🚀 Ultra Simple Helios Example\n");
11
12    // ========== SIMPLEST AGENT CREATION ==========
13    println!("1️⃣  Creating an agent - shortest possible syntax:\n");
14
15    // One-liner: Create agent with auto config
16    let mut agent = Agent::builder("Helper")
17        .auto_config()
18        .prompt("You are helpful and concise.")
19        .build()
20        .await?;
21
22    println!("✓ Agent created!\n");
23
24    // ========== SIMPLEST CHAT ==========
25    println!("2️⃣  Asking questions - simplest possible:\n");
26
27    // Use .ask() instead of .chat() for more natural syntax
28    let answer = agent.ask("What is 2+2?").await?;
29    println!("Q: What is 2+2?\nA: {}\n", answer);
30
31    // ========== SIMPLEST CONFIG ==========
32    println!("3️⃣  Creating config with shortest syntax:\n");
33
34    // Ultra-short config creation
35    let _config = Config::builder()
36        .m("gpt-4") // .m() is shorthand for .model()
37        .key("your-api-key") // .key() is shorthand for .api_key()
38        .temp(0.8) // .temp() is shorthand for .temperature()
39        .tokens(1024) // .tokens() is shorthand for .max_tokens()
40        .build();
41
42    println!("✓ Config created with ultra-short syntax!\n");
43
44    // ========== SIMPLEST AGENT WITH TOOLS ==========
45    println!("4️⃣  Agent with tools - simplest way:\n");
46
47    let mut calc_agent = Agent::builder("Calculator")
48        .auto_config()
49        .prompt("You are a math expert.")
50        .with_tool(Box::new(CalculatorTool)) // Add single tool
51        .build()
52        .await?;
53
54    let result = calc_agent.ask("Calculate 15 * 7 + 5").await?;
55    println!("Q: Calculate 15 * 7 + 5\nA: {}\n", result);
56
57    // ========== SIMPLEST QUICK AGENT ==========
58    println!("5️⃣  Quick agent - one method call:\n");
59
60    // Agent::quick() creates agent in ONE LINE with auto config!
61    let mut quick_agent = Agent::quick("QuickBot").await?;
62    let quick_answer = quick_agent.ask("Say hello!").await?;
63    println!("Response: {}\n", quick_answer);
64
65    // ========== SIMPLEST CHAT MESSAGES ==========
66    println!("6️⃣  Creating messages - super short syntax:\n");
67
68    use helios_engine::ChatMessage;
69
70    // Short aliases for message creation
71    let _sys_msg = ChatMessage::sys("You are helpful"); // .sys() not .system()
72    let _user_msg = ChatMessage::msg("Hello there"); // .msg() not .user()
73    let _reply_msg = ChatMessage::reply("Hi! How can I help?"); // .reply() not .assistant()
74
75    println!("✓ Messages created with ultra-short syntax!\n");
76
77    // ========== SHORTEST AUTOFOREST ==========
78    println!("7️⃣  AutoForest - simplest multi-agent orchestration:\n");
79
80    use helios_engine::AutoForest;
81
82    let mut forest = AutoForest::new(Config::builder().m("gpt-4").build())
83        .with_tools(vec![Box::new(CalculatorTool)])
84        .build()
85        .await?;
86
87    // Use .run() for shortest syntax
88    let forest_result = forest.run("Analyze these numbers: 10, 20, 30, 40").await?;
89    println!("Forest Result:\n{}\n", forest_result);
90
91    // ========== COMPARISON TABLE ==========
92    println!("📊 Syntax Comparison - Short vs Long:\n");
93    println!("┌─────────────────────┬──────────────────────┬─────────────────┐");
94    println!("│ Operation           │ Short Syntax         │ Long Syntax      │");
95    println!("├─────────────────────┼──────────────────────┼─────────────────┤");
96    println!("│ Create Agent        │ Agent::quick()       │ Agent::builder()│");
97    println!("│ Ask Question        │ .ask()               │ .chat()          │");
98    println!("│ System Prompt       │ .prompt()            │ .system_prompt() │");
99    println!("│ Config Model        │ .m()                 │ .model()         │");
100    println!("│ Config Key          │ .key()               │ .api_key()       │");
101    println!("│ Config Temp         │ .temp()              │ .temperature()   │");
102    println!("│ Config Tokens       │ .tokens()            │ .max_tokens()    │");
103    println!("│ System Message      │ ChatMessage::sys()   │ ChatMessage::system()");
104    println!("│ User Message        │ ChatMessage::msg()   │ ChatMessage::user()");
105    println!("│ Assistant Message   │ ChatMessage::reply() │ ChatMessage::assistant()");
106    println!("│ AutoForest Execute  │ .run()               │ .execute_task()  │");
107    println!("└─────────────────────┴──────────────────────┴─────────────────┘\n");
108
109    println!("✅ All examples completed!");
110    println!("💡 Tip: Mix and match short and long syntax based on your preference!");
111
112    Ok(())
113}