Crate smolagents_rs

SmolAgents is a Rust library for building and running agents that can use tools and code.

It is inspired by Hugging Face’s smolagents library and provides a simple interface for building and running agents.

It is designed to be used in a CLI application, but can be used in any Rust application.

Example usage:

use smolagents_rs::agents::{Agent, FunctionCallingAgent};
use smolagents_rs::models::openai::OpenAIServerModel;
use smolagents_rs::tools::{AnyTool, DuckDuckGoSearchTool, VisitWebsiteTool};
fn main() {
    let tools: Vec<Box<dyn AnyTool>> = vec![
        Box::new(DuckDuckGoSearchTool::new()),
        Box::new(VisitWebsiteTool::new()),
    ];
    let model = OpenAIServerModel::new(Some("https://api.openai.com/v1/chat/completions"), Some("gpt-4o-mini"), None, None);
    let mut agent = FunctionCallingAgent::new(model, tools, None, None, None, None).unwrap();
    let _result = agent
        .run("Who has the most followers on Twitter?", false, true)
        .unwrap();
}

Code Agent:

To use the code agent simply enable the code-agent feature.

use smolagents_rs::agents::{Agent, CodeAgent};
use smolagents_rs::models::openai::OpenAIServerModel;
use smolagents_rs::tools::{AnyTool, DuckDuckGoSearchTool, VisitWebsiteTool};
fn main() {
    let tools: Vec<Box<dyn AnyTool>> = vec![
        Box::new(DuckDuckGoSearchTool::new()),
        Box::new(VisitWebsiteTool::new()),
    ];
    let model = OpenAIServerModel::new(Some("https://api.openai.com/v1/chat/completions"), Some("gpt-4o-mini"), None, None);
    let mut agent = CodeAgent::new(model, tools, None, None, None, None).unwrap();
    let _result = agent
        .run("Who has the most followers on Twitter?", false, true)
        .unwrap();
}

Re-exports

pub use agents::*;

Modules

agents

This module contains the agents that can be used to solve tasks.

errors

local_python_interpreter

models

prompts

This module contains the prompts for the agents.

tools

This module contains the tools that can be used in an agent. These are the default tools that are available. You can also implement your own tools by implementing the Tool trait.