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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

use agentai::{Agent, AgentError, AgentResult, Prompt};
use anyhow::Result;
use log::{LevelFilter, info};
use simplelog::{ColorChoice, Config, TermLogger, TerminalMode};

const SYSTEM: &str = "You are helpful assistant";

// The project logo, published on GitHub and read from the repository files
const LOGO_URL: &str =
    "https://raw.githubusercontent.com/AdamStrojek/rust-agentai/master/assets/logo-docs.png";
const LOGO_PNG: &[u8] = include_bytes!(concat!(
    env!("CARGO_MANIFEST_DIR"),
    "/../../assets/logo-docs.png"
));

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

    // The model has to support images (vision)
    // The provider comes from the model name, for example `claude-haiku-5-5` or
    // `omlx::Qwen3.8-27B-4bit`; its API key and endpoint come from the provider's environment
    // variables, for example `ANTHROPIC_API_KEY` or `OMLX_ENDPOINT`.
    let model = std::env::var("AGENTAI_MODEL").unwrap_or("claude-haiku-5-5".to_string());

    let agent = Agent::new(&model, SYSTEM)?;

    // An image from its bytes: it is sent base64 encoded inside the request
    let prompt =
        Prompt::text("Describe this logo in one sentence.").with_image_bytes("image/png", LOGO_PNG);
    let result: AgentResult<String> = agent.run(prompt).await?;
    info!("Logo from bytes: {}", result.answer);

    // The same image by URL: the provider downloads it. Not every provider accepts image URLs
    // (local servers often take only base64 data); those reject the request with HTTP 400.
    let prompt = Prompt::text("What colours does this logo use?").with_image_url(LOGO_URL);
    match agent.run::<String>(prompt).await {
        Ok(result) => info!("Logo from URL: {}", result.answer),
        Err(AgentError::Llm(err)) if err.status() == Some(400) => {
            info!("This provider does not accept image URLs, send the bytes instead: {err}")
        }
        Err(err) => return Err(err.into()),
    }

    Ok(())
}