tiny-agent-rs 0.1.0

A lightweight, type-safe Rust agent library for LLM tool calling
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

use tiny_agent_rs::{
    tools::{CalculatorTool, JinaReaderTool},
    Agent, FunctionFactory,
};

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // Load keys from environment
    let api_key = std::env::var("OPENAI_API_KEY")?;
    let jina_key = std::env::var("JINA_API_KEY")?;

    let mut factory = FunctionFactory::new();
    factory.register_tool(CalculatorTool::new());
    factory.register_tool(JinaReaderTool::new(jina_key));

    let agent = Agent::new(api_key, factory).with_max_iterations(6);

    println!("=== Jina Reader Tool Demo ===\n");

    let task = "Use the jina_reader tool to fetch markdown for https://www.example.com, then summarize the content in one sentence.";
    println!("Task: {}\n", task);

    let result = agent.run_with_steps(task).await?;

    println!("{}", result.replay());

    println!("\n--- Detailed Explanation ---");
    println!("{}", result.explain());

    Ok(())
}