swink-agent 0.13.2

Core scaffolding for running LLM-powered agentic loops
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

//! Tests for `display`.
#![cfg(test)]

use super::*;
use crate::types::{AssistantMessage, Cost, Usage, UserMessage};

#[test]
fn user_message_to_display() {
    let msg = LlmMessage::User(UserMessage {
        content: vec![ContentBlock::Text {
            text: "hello".to_string(),
        }],
        timestamp: 0,
        cache_hint: None,
    });
    let display = msg.to_display_messages();
    assert_eq!(display.len(), 1);
    assert_eq!(display[0].role, DisplayRole::User);
    assert_eq!(display[0].content, "hello");
    assert!(display[0].thinking.is_none());
}

#[test]
fn assistant_message_with_thinking() {
    let msg = LlmMessage::Assistant(AssistantMessage {
        content: vec![
            ContentBlock::Thinking {
                thinking: "reasoning".to_string(),
                signature: None,
            },
            ContentBlock::Text {
                text: "answer".to_string(),
            },
        ],
        provider: String::new(),
        model_id: String::new(),
        usage: Usage::default(),
        cost: Cost::default(),
        stop_reason: StopReason::Stop,
        error_message: None,
        error_kind: None,
        timestamp: 0,
        cache_hint: None,
    });
    let display = msg.to_display_messages();
    assert_eq!(display.len(), 1);
    assert_eq!(display[0].role, DisplayRole::Assistant);
    assert_eq!(display[0].content, "answer");
    assert_eq!(display[0].thinking.as_deref(), Some("reasoning"));
}

#[test]
fn assistant_error_message() {
    let msg = LlmMessage::Assistant(AssistantMessage {
        content: vec![],
        provider: String::new(),
        model_id: String::new(),
        usage: Usage::default(),
        cost: Cost::default(),
        stop_reason: StopReason::Error,
        error_message: Some("something broke".to_string()),
        error_kind: None,
        timestamp: 0,
        cache_hint: None,
    });
    let display = msg.to_display_messages();
    assert_eq!(display.len(), 1);
    assert_eq!(display[0].role, DisplayRole::Error);
    assert_eq!(display[0].content, "something broke");
}

#[test]
fn empty_tool_result_produces_no_messages() {
    let msg = LlmMessage::ToolResult(crate::types::ToolResultMessage {
        tool_call_id: "tc1".to_string(),
        content: vec![],
        is_error: false,
        timestamp: 0,
        details: serde_json::Value::Null,
        cache_hint: None,
    });
    let display = msg.to_display_messages();
    assert!(display.is_empty());
}