agentix 0.17.0

Multi-provider LLM client for Rust — streaming, non-streaming, tool calls, MCP, DeepSeek, OpenAI, Anthropic, Gemini
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

//! Example 07: Agentic loop with tools
//!
//! `agentix::agent()` drives the full LLM ↔ tool-call loop automatically.
//! It keeps calling the LLM until there are no more tool calls, yielding
//! typed `AgentEvent`s as it goes.
//!
//! Run with:
//!   OPENAI_API_KEY=sk-... cargo run --example 07_agent

use agentix::{AgentEvent, Message, Request, ToolBundle, UserContent, tool};
use futures::StreamExt;
use std::env;

// ── Tools ─────────────────────────────────────────────────────────────────────

struct Calculator;

#[tool]
impl agentix::Tool for Calculator {
    /// Add two numbers.
    /// a: first number
    /// b: second number
    async fn add(&self, a: f64, b: f64) -> f64 {
        a + b
    }

    /// Multiply two numbers.
    /// a: first number
    /// b: second number
    async fn multiply(&self, a: f64, b: f64) -> f64 {
        a * b
    }

    /// Divide a by b. Returns an error if b is zero.
    /// a: dividend
    /// b: divisor
    async fn divide(&self, a: f64, b: f64) -> Result<f64, String> {
        if b == 0.0 {
            Err("division by zero".into())
        } else {
            Ok(a / b)
        }
    }
}

// ── Main ──────────────────────────────────────────────────────────────────────

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let request = if let Ok(k) = env::var("DEEPSEEK_API_KEY") {
        Request::deepseek(k)
    } else if let Ok(k) = env::var("OPENAI_API_KEY") {
        Request::openai(k)
    } else {
        panic!("Set DEEPSEEK_API_KEY or OPENAI_API_KEY");
    }.system_prompt("You are a math assistant. Use your tools to compute exact results.");

    let http = reqwest::Client::new();
    let tools = ToolBundle::default() + Calculator;

    let history = vec![
        Message::User(vec![UserContent::Text { text: "What is (123 + 456) * 789 / 3?".into() }]),
    ];

    println!("Question: (123 + 456) * 789 / 3\n");

    let mut stream = agentix::agent(tools, http, request, history, Some(25_000));

    while let Some(event) = stream.next().await {
        match event {
            AgentEvent::Token(t) => print!("{t}"),
            AgentEvent::Reasoning(t) => print!("\x1b[2m{t}\x1b[0m"),
            AgentEvent::ToolCallStart(tc) => {
                println!("\n→ {}({})", tc.name, tc.arguments);
            }
            AgentEvent::ToolProgress { name, progress, .. } => {
                println!("  [{name}] {progress}");
            }
            AgentEvent::ToolResult { name, ref content, .. } => {
                let text = content.iter()
                    .filter_map(|p| if let agentix::Content::Text { text } = p { Some(text.as_str()) } else { None })
                    .collect::<Vec<_>>().join(" ");
                println!("← [{name}] = {text}");
            }
            AgentEvent::Usage(u) => {
                eprintln!("\n[tokens: {}]", u.total_tokens);
            }
            AgentEvent::Done(total) => {
                eprintln!("\n[total tokens: {}]", total.total_tokens);
            }
            AgentEvent::Warning(w) => eprintln!("\n[warn] {w}"),
            AgentEvent::Error(e) => {
                eprintln!("\n[error] {e}");
                break;
            }
            AgentEvent::ToolCallChunk(_) => {}
        }
    }

    println!();
    Ok(())
}