harn-vm 0.7.55

Async bytecode virtual machine for the Harn programming language
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
128
129
130
131
132
133

use super::*;

#[test]
fn candidates_from_anthropic_shape() {
    let tools = serde_json::json!([
        {
            "name": "weather_lookup",
            "description": "Look up the weather",
            "input_schema": {
                "type": "object",
                "properties": {
                    "city": {"type": "string", "description": "City name"}
                }
            }
        }
    ]);
    let got = candidates_from_native(tools.as_array().unwrap());
    assert_eq!(got.len(), 1);
    assert_eq!(got[0].name, "weather_lookup");
    assert!(got[0].description.contains("weather"));
    assert_eq!(got[0].param_text, vec!["city: City name".to_string()]);
}

#[test]
fn candidates_from_openai_shape() {
    let tools = serde_json::json!([
        {
            "type": "function",
            "function": {
                "name": "execute_sql",
                "description": "Run a SQL query",
                "parameters": {
                    "type": "object",
                    "properties": {
                        "query": {"type": "string"}
                    }
                }
            }
        }
    ]);
    let got = candidates_from_native(tools.as_array().unwrap());
    assert_eq!(got.len(), 1);
    assert_eq!(got[0].name, "execute_sql");
    assert_eq!(got[0].param_text, vec!["query".to_string()]);
}

#[test]
fn in_tree_runs_bm25_by_default() {
    let candidates = vec![
        ToolCandidate {
            name: "weather".to_string(),
            description: String::new(),
            param_text: Vec::new(),
            tags: Vec::new(),
        },
        ToolCandidate {
            name: "cooking".to_string(),
            description: String::new(),
            param_text: Vec::new(),
            tags: Vec::new(),
        },
    ];
    let outcome = run_in_tree(InTreeStrategy::Bm25, "weather", &candidates, 5);
    assert_eq!(outcome.tool_names, vec!["weather"]);
}

#[test]
fn in_tree_rejects_empty_query() {
    let candidates = vec![ToolCandidate {
        name: "any".to_string(),
        description: String::new(),
        param_text: Vec::new(),
        tags: Vec::new(),
    }];
    let outcome = run_in_tree(InTreeStrategy::Bm25, "   ", &candidates, 5);
    assert!(outcome.tool_names.is_empty());
    assert!(outcome.diagnostic.is_some());
}

#[test]
fn mcp_server_tag_makes_server_name_searchable() {
    // Anthropic-shape tool with `_mcp_server` injected by
    // mcp_list_tools. A BM25 query for "github" should promote it
    // via the tag-in-corpus path even though name/description don't
    // contain the word.
    let tools = serde_json::json!([
        {
            "name": "create_issue",
            "description": "Create a tracking issue",
            "_mcp_server": "github",
            "input_schema": {"type": "object", "properties": {}}
        },
        {
            "name": "render_markdown",
            "description": "Render markdown to html",
            "input_schema": {"type": "object", "properties": {}}
        }
    ]);
    let candidates = candidates_from_native(tools.as_array().unwrap());
    assert_eq!(candidates.len(), 2);
    // First candidate carries the mcp tags.
    assert!(candidates[0].tags.contains(&"mcp:github".to_string()));
    assert!(candidates[0].tags.contains(&"github".to_string()));
    // Second has no tags.
    assert!(candidates[1].tags.is_empty());
    // BM25 with the query "github" should return create_issue only.
    let outcome = run_in_tree(InTreeStrategy::Bm25, "github", &candidates, 5);
    assert_eq!(outcome.tool_names, vec!["create_issue"]);
}

#[test]
fn search_outcome_into_tool_result_serializes() {
    let outcome = SearchOutcome {
        tool_names: vec!["a".to_string(), "b".to_string()],
        diagnostic: None,
    };
    let json = outcome.into_tool_result();
    assert_eq!(
        json,
        serde_json::json!({"tool_names": ["a", "b"]}),
        "result shape must be the minimal {{tool_names}} contract"
    );
}

#[test]
fn search_outcome_with_diagnostic() {
    let outcome = SearchOutcome::empty("nothing to see here");
    let json = outcome.into_tool_result();
    assert_eq!(
        json,
        serde_json::json!({"tool_names": [], "diagnostic": "nothing to see here"})
    );
}