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
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

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

//! 2026-09-26: Unit tests for the chat message IR (`ir::message`).
//!
//! Owner: server (chat IR).
//! Invariants: none beyond the types.

use super::message::*;

#[test]
fn role_wire_roundtrip() {
    for (role, wire) in [
        (Role::System, "system"),
        (Role::User, "user"),
        (Role::Assistant, "assistant"),
        (Role::Tool, "tool"),
    ] {
        assert_eq!(role.as_wire(), wire);
        assert_eq!(Role::from_wire(wire), Some(role));
    }
    assert_eq!(Role::from_wire("function"), None);
}

#[test]
fn text_concatenates_text_parts_in_order_ignoring_images() {
    let m = Message {
        role: Role::User,
        content: vec![
            ContentPart::Text("a".into()),
            ContentPart::Image(ImageSource {
                data: ImageData::Base64("img".into()),
            }),
            ContentPart::Text("b".into()),
        ],
        tool_calls: Vec::new(),
        tool_call_id: None,
        name: None,
        reasoning: None,
        tool_error: false,
    };
    assert_eq!(m.text(), "ab");
    assert_eq!(m.media_kinds(), vec![MediaKind::Image]);
}

/// 2026-09-26: `media_kinds` keeps content order across modalities; text
/// parts between media items do not count.
#[test]
fn media_kinds_follow_content_order_across_modalities() {
    let m = Message {
        role: Role::User,
        content: vec![
            ContentPart::Video(VideoSource {
                data: ImageData::Base64("clip".into()),
            }),
            ContentPart::Text("compare".into()),
            ContentPart::Image(ImageSource {
                data: ImageData::Base64("still".into()),
            }),
            ContentPart::Video(VideoSource {
                data: ImageData::Base64("clip2".into()),
            }),
        ],
        tool_calls: Vec::new(),
        tool_call_id: None,
        name: None,
        reasoning: None,
        tool_error: false,
    };
    assert_eq!(
        m.media_kinds(),
        vec![MediaKind::Video, MediaKind::Image, MediaKind::Video],
        "grouping by modality here is the reordering defect"
    );
}

#[test]
fn assistant_tool_call_and_reasoning_are_first_class() {
    let m = Message {
        role: Role::Assistant,
        content: vec![ContentPart::Text("hi".into())],
        tool_calls: vec![ToolCall {
            id: "call_1".into(),
            name: "get_weather".into(),
            arguments: serde_json::json!({"city": "SF"}),
        }],
        tool_call_id: None,
        name: None,
        reasoning: Some(Reasoning {
            text: "let me think".into(),
        }),
        tool_error: false,
    };
    assert_eq!(m.role.as_wire(), "assistant");
    assert_eq!(m.tool_calls[0].name, "get_weather");
    assert_eq!(m.tool_calls[0].arguments["city"], "SF");
    assert_eq!(m.reasoning.as_ref().unwrap().text, "let me think");
    assert!(m.media_kinds().is_empty());
}

#[test]
fn tool_message_carries_error_flag_and_call_id() {
    let m = Message {
        role: Role::Tool,
        content: vec![ContentPart::Text("exit 127".into())],
        tool_calls: Vec::new(),
        tool_call_id: Some("call_1".into()),
        name: Some("bash".into()),
        reasoning: None,
        tool_error: true,
    };
    assert!(m.tool_error);
    assert_eq!(m.tool_call_id.as_deref(), Some("call_1"));
    assert_eq!(m.text(), "exit 127");
}