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
53
54
55
56
57
58
59

//! Tests for the options of the `#[tool(...)]` attribute.
#![cfg(feature = "macros")]

use agentai::tool::{Tool, ToolBox, ToolError, ToolResult, toolbox};

struct StrictToolBox;

#[toolbox]
impl StrictToolBox {
    /// Strict tool with parameters.
    #[tool(strict)]
    fn strict_tool(
        &self,
        /// City name
        city: String,
        /// Optional unit
        unit: Option<String>,
    ) -> ToolResult {
        Ok(format!("{city} {unit:?}"))
    }

    /// Explicitly non-strict tool.
    #[tool(strict = false)]
    fn not_strict(&self) -> ToolResult {
        Ok("ok".to_string())
    }

    /// Renamed strict tool.
    #[tool(name = "renamed", strict = true)]
    fn renamed_strict(&self) -> ToolResult {
        Ok("ok".to_string())
    }

    /// Tool without options.
    #[tool]
    fn plain(&self) -> ToolResult {
        Ok("ok".to_string())
    }
}

fn find<'a>(tools: &'a [Tool], name: &str) -> &'a Tool {
    tools
        .iter()
        .find(|t| t.name.as_str() == name)
        .unwrap_or_else(|| panic!("tool '{name}' not found"))
}

#[test]
fn strict_option_sets_the_strict_flag() -> Result<(), ToolError> {
    let tools = StrictToolBox.tools_definitions()?;

    assert_eq!(find(&tools, "strict_tool").strict, Some(true));
    assert!(find(&tools, "strict_tool").schema.is_some());
    assert_eq!(find(&tools, "not_strict").strict, Some(false));
    assert_eq!(find(&tools, "renamed").strict, Some(true));
    assert_eq!(find(&tools, "plain").strict, None);

    Ok(())
}