synaptic-tools 0.4.0

Tool registry and execution: ToolRegistry, SerialToolExecutor, ParallelToolExecutor
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127

use serde_json::json;
use std::collections::HashMap;
use synaptic_core::{Tool, ToolDefinition};

struct SimpleTool;

#[async_trait::async_trait]
impl Tool for SimpleTool {
    fn name(&self) -> &'static str {
        "simple"
    }
    fn description(&self) -> &'static str {
        "A simple tool"
    }
    fn parameters(&self) -> Option<serde_json::Value> {
        Some(json!({
            "type": "object",
            "properties": {
                "input": {"type": "string"}
            },
            "required": ["input"]
        }))
    }
    async fn call(
        &self,
        _args: serde_json::Value,
    ) -> Result<serde_json::Value, synaptic_core::SynapticError> {
        Ok(json!("ok"))
    }
}

struct NoParamsTool;

#[async_trait::async_trait]
impl Tool for NoParamsTool {
    fn name(&self) -> &'static str {
        "no_params"
    }
    fn description(&self) -> &'static str {
        "Tool with no parameters"
    }
    async fn call(
        &self,
        _args: serde_json::Value,
    ) -> Result<serde_json::Value, synaptic_core::SynapticError> {
        Ok(json!("done"))
    }
}

#[test]
fn tool_definition_extras_field() {
    let mut extras = HashMap::new();
    extras.insert("cache_control".to_string(), json!({"type": "ephemeral"}));

    let def = ToolDefinition {
        name: "search".into(),
        description: "Search the web".into(),
        parameters: json!({"type": "object", "properties": {}}),
        extras: Some(extras),
    };

    assert_eq!(
        def.extras.as_ref().unwrap()["cache_control"]["type"],
        "ephemeral"
    );
}

#[test]
fn tool_definition_extras_none_by_default() {
    let tool = SimpleTool;
    let def = tool.as_tool_definition();
    assert!(def.extras.is_none());
}

#[test]
fn as_tool_definition_from_trait() {
    let tool = SimpleTool;
    let def = tool.as_tool_definition();
    assert_eq!(def.name, "simple");
    assert_eq!(def.description, "A simple tool");
    assert!(def.parameters["properties"]["input"].is_object());
}

#[test]
fn tool_definition_with_parameters_schema() {
    let tool = SimpleTool;
    let def = tool.as_tool_definition();
    let required = def.parameters["required"].as_array().unwrap();
    assert_eq!(required.len(), 1);
    assert_eq!(required[0], "input");
}

#[test]
fn tool_definition_default_parameters_when_none() {
    let tool = NoParamsTool;
    let def = tool.as_tool_definition();
    assert_eq!(def.parameters["type"], "object");
    assert!(def.parameters["properties"].is_object());
}

#[test]
fn tool_definition_serde_roundtrip() {
    let def = ToolDefinition {
        name: "calc".into(),
        description: "Calculator".into(),
        parameters: json!({"type": "object"}),
        extras: None,
    };
    let json = serde_json::to_string(&def).unwrap();
    let deserialized: ToolDefinition = serde_json::from_str(&json).unwrap();
    assert_eq!(def, deserialized);
}

#[test]
fn tool_definition_serde_with_extras() {
    let mut extras = HashMap::new();
    extras.insert("priority".to_string(), json!("high"));
    let def = ToolDefinition {
        name: "deploy".into(),
        description: "Deploy app".into(),
        parameters: json!({"type": "object"}),
        extras: Some(extras),
    };
    let json = serde_json::to_string(&def).unwrap();
    let deserialized: ToolDefinition = serde_json::from_str(&json).unwrap();
    assert_eq!(deserialized.extras.unwrap()["priority"], "high");
}