met 1.0.0-beta-preview

Metrale Engine: pure Rust LLM inference server (HTTP API, scheduler, TUI, CLI).
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

// SPDX-License-Identifier: MIT OR Apache-2.0

//! 2026-09-26: Render tests for the Laguna Jinja template.
//!
//! Owner: server (tokenizer) tests.
//! Invariants: none beyond the types.

use serde_json::json;

use crate::tokenizer::normalize_tool_call_arguments;

fn render_laguna_template(
    messages: &[serde_json::Value],
    tools: Option<&[serde_json::Value]>,
    enable_thinking: bool,
) -> String {
    let raw = std::fs::read_to_string(concat!(
        env!("CARGO_MANIFEST_DIR"),
        "/../../jinja-templates/laguna.jinja"
    ))
    .expect("bundled Laguna template must be present");
    let env = crate::tokenizer::jinja_helpers::build_jinja_env(&raw).expect("template compiles");
    let tmpl = env.get_template("chat").unwrap();
    let messages = normalize_tool_call_arguments(messages);
    tmpl.render(minijinja::context! {
        messages => minijinja::Value::from_serialize(&messages),
        tools => tools.map(minijinja::Value::from_serialize).unwrap_or(minijinja::Value::UNDEFINED),
        add_generation_prompt => true,
        enable_thinking => enable_thinking,
        disable_tool_steering => false,
    })
    .expect("template renders")
}

#[test]
fn laguna_template_renders_native_tool_round_trip() {
    let messages = vec![json!({
        "role": "assistant",
        "content": "",
        "tool_calls": [{
            "function": {
                "name": "Bash",
                "arguments": "{\"command\":\"pwd\"}"
            }
        }]
    })];
    let rendered = render_laguna_template(&messages, None, true);
    assert!(rendered.contains(
        "<assistant><think></think><tool_call>Bash<arg_key>command</arg_key><arg_value>pwd</arg_value></tool_call></assistant>"
    ));
    assert!(rendered.ends_with("<assistant><think>"));
}

#[test]
fn laguna_template_uses_checkpoint_tool_json_and_reasoning_controls() {
    let messages = vec![json!({"role": "user", "content": "weather"})];
    let tools = vec![json!({
        "type": "function",
        "function": {
            "name": "get_weather",
            "description": "Get weather",
            "parameters": {
                "type": "object",
                "properties": {"city": {"type": "string"}},
                "required": ["city"]
            }
        }
    })];

    let no_think = render_laguna_template(&messages, Some(&tools), false);
    assert!(no_think.contains(
        r#"{"type": "function", "function": {"name": "get_weather", "description": "Get weather", "parameters": {"type": "object", "properties": {"city": {"type": "string"}}, "required": ["city"]}}}"#
    ));
    assert!(no_think.ends_with("<assistant></think>"));

    let think = render_laguna_template(&messages, Some(&tools), true);
    assert!(think.ends_with("<assistant><think>"));
}