tiny-loop 0.5.1

Minimal AI agent framework
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

use tiny_loop::tool::{ToolArgs, tool};

#[tool(name = "custom_tool")]
async fn my_function(#[serde(rename = "customParam")] param: String) -> String {
    format!("Got: {}", param)
}

#[derive(Clone)]
#[allow(dead_code)]
struct MyService;

#[tool]
impl MyService {
    #[name = "custom_method_one"]
    async fn method_one(self, #[serde(rename = "customParam")] param: String) -> String {
        format!("Method one: {}", param)
    }

    #[name = "custom_method_two"]
    async fn method_two(self, param: String) -> String {
        format!("Method two: {}", param)
    }
}

#[test]
fn test_custom_tool_name() {
    assert_eq!(CustomToolArgs::TOOL_NAME, "custom_tool");
}

#[test]
fn test_custom_method_names() {
    assert_eq!(CustomMethodOneArgs::TOOL_NAME, "custom_method_one");
    assert_eq!(CustomMethodTwoArgs::TOOL_NAME, "custom_method_two");
}

#[test]
fn test_serde_rename_function() {
    let json = r#"{"customParam": "test"}"#;
    let args: CustomToolArgs = serde_json::from_str(json).unwrap();
    assert_eq!(args.param, "test");
}

#[test]
fn test_serde_rename_method() {
    let json = r#"{"customParam": "test"}"#;
    let args: CustomMethodOneArgs = serde_json::from_str(json).unwrap();
    assert_eq!(args.param, "test");
}