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
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
103
104
105
106
107
108
109
110

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

/// First line of documentation
/// Second line of documentation
/// Third line of documentation
#[tool]
async fn multi_line_doc(param: String) -> String {
    format!("Got: {}", param)
}

#[tool]
async fn param_multi_line_doc(
    /// First line of param doc
    /// Second line of param doc
    /// Third line of param doc
    param: String,
) -> String {
    format!("Got: {}", param)
}

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

#[tool]
impl Service {
    /// First line of method doc
    /// Second line of method doc
    /// Third line of method doc
    async fn method_with_doc(
        self,
        /// First line of method param doc
        /// Second line of method param doc
        /// Third line of method param doc
        param: String,
    ) -> String {
        format!("Got: {}", param)
    }
}

#[test]
fn test_multi_line_doc_comment() {
    let desc = MultiLineDocArgs::TOOL_DESCRIPTION;
    assert!(
        desc.contains("First line of documentation"),
        "Description should contain first line"
    );
    assert!(
        desc.contains("Second line of documentation"),
        "Description should contain second line"
    );
    assert!(
        desc.contains("Third line of documentation"),
        "Description should contain third line"
    );
}

#[test]
fn test_multi_line_doc_comment_method() {
    let desc = MethodWithDocArgs::TOOL_DESCRIPTION;
    assert!(
        desc.contains("First line of method doc"),
        "Description should contain first line"
    );
    assert!(
        desc.contains("Second line of method doc"),
        "Description should contain second line"
    );
    assert!(
        desc.contains("Third line of method doc"),
        "Description should contain third line"
    );
}

#[test]
fn test_param_multi_line_doc() {
    let schema = schema_for!(ParamMultiLineDocArgs);
    let json = serde_json::to_string(&schema).unwrap();
    assert!(
        json.contains("First line of param doc"),
        "Schema should contain first line of param doc"
    );
    assert!(
        json.contains("Second line of param doc"),
        "Schema should contain second line of param doc"
    );
    assert!(
        json.contains("Third line of param doc"),
        "Schema should contain third line of param doc"
    );
}

#[test]
fn test_method_param_multi_line_doc() {
    let schema = schema_for!(MethodWithDocArgs);
    let json = serde_json::to_string(&schema).unwrap();
    assert!(
        json.contains("First line of method param doc"),
        "Schema should contain first line of method param doc"
    );
    assert!(
        json.contains("Second line of method param doc"),
        "Schema should contain second line of method param doc"
    );
    assert!(
        json.contains("Third line of method param doc"),
        "Schema should contain third line of method param doc"
    );
}