swarms-rs 0.2.0

The Enterprise-Grade Production-Ready Multi-Agent Orchestration Framework in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

use std::env;

use anyhow::{Context, Result};
use swarms_rs::{llm::provider::openai::OpenAI, structs::agent::Agent};
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};

#[tokio::main]
async fn main() -> Result<()> {
    dotenv::dotenv().ok();
    tracing_subscriber::registry()
        .with(tracing_subscriber::EnvFilter::from_default_env())
        .with(
            tracing_subscriber::fmt::layer()
                .with_line_number(true)
                .with_file(true),
        )
        .init();

    // Check for required environment variables
    let api_key = env::var("OPENAI_API_KEY")
        .context("OPENAI_API_KEY environment variable not set. Please set it in your .env file or environment.")?;

    // Create OpenAI client with error handling
    let client = OpenAI::new(api_key).set_model("gpt-4o-mini");

    // Build agent with error handling
    let agent = client
        .agent_builder()
        .system_prompt("You are a helpful assistant.")
        .agent_name("SwarmsAgent")
        .user_name("User")
        .enable_autosave()
        .max_loops(1)
        .save_state_dir("./temp/")
        .enable_plan("Split the task into subtasks.".to_owned())
        .build();

    // Run agent with error handling
    let response = agent
        .run("What is the meaning of life?".to_owned())
        .await
        .context("Failed to run agent")?;

    println!("{response}");

    Ok(())
}