agentai 0.4.0

AgentAI is a Rust library designed to simplify the creation of AI agents.
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

//! OpenRouter Example
//!
//! This example uses OpenRouter through the `open_router::` namespace of the model name.
//! The API key is read from the `OPEN_ROUTER_API_KEY` environment variable.

use agentai::tool::buildin::CurrentDateAndTimeToolBox;
use agentai::{Agent, AgentResult};
use anyhow::Result;
use log::{LevelFilter, info};
use simplelog::{ColorChoice, Config, TermLogger, TerminalMode};

const SYSTEM: &str = "You are helpful assistant. Use tools when you need the current date or time.";

#[tokio::main]
async fn main() -> Result<()> {
    TermLogger::init(
        LevelFilter::Info,
        Config::default(),
        TerminalMode::Mixed,
        ColorChoice::Auto,
    )?;
    info!("Starting AgentAI");

    let question = "What day of the week is it today?";

    info!("Question: {}", question);

    // The `open_router::` prefix sends the model to OpenRouter
    let agent = Agent::new("open_router::nvidia/nemotron-3.5-super-vl:free", SYSTEM)?
        .with_toolbox(CurrentDateAndTimeToolBox {});

    let result: AgentResult<String> = agent.run(question).await?;

    info!("Answer: {}", result.answer);

    Ok(())
}