agentic-tools-core 0.1.3

Core traits and types for agentic-tools library family
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

//! OpenAI function calling schema renderer.

use schemars::Schema;
use serde_json::Value;
use serde_json::json;

/// Render a tool as an OpenAI function definition.
///
/// Output format:
/// ```json
/// {
///   "type": "function",
///   "function": {
///     "name": "...",
///     "description": "...",
///     "strict": true,
///     "parameters": { ... }
///   }
/// }
/// ```
pub fn render_function(name: &str, description: &str, parameters: &Schema, strict: bool) -> Value {
    let params = serde_json::to_value(parameters).expect("Schema serialization must succeed");
    json!({
        "type": "function",
        "function": {
            "name": name,
            "description": description,
            "strict": strict,
            "parameters": params
        }
    })
}

#[cfg(test)]
mod tests {
    use super::*;

    #[derive(schemars::JsonSchema)]
    #[allow(dead_code)]
    struct TestInput {
        message: String,
    }

    #[test]
    fn test_render_function() {
        let schema = schemars::schema_for!(TestInput);
        let rendered = render_function("greet", "Greet someone", &schema, true);

        assert_eq!(rendered["type"], "function");
        assert_eq!(rendered["function"]["name"], "greet");
        assert_eq!(rendered["function"]["description"], "Greet someone");
        assert_eq!(rendered["function"]["strict"], true);
        assert!(rendered["function"]["parameters"].is_object());
    }

    #[test]
    fn test_render_function_non_strict() {
        let schema = schemars::schema_for!(TestInput);
        let rendered = render_function("greet", "Greet someone", &schema, false);

        assert_eq!(rendered["function"]["strict"], false);
    }
}