defect-agent 0.1.0-alpha.2

Core agent runtime for defect: turn loop, context compaction, tools and session orchestration.
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
128
129

use super::*;

fn msg(role: Role, content: Vec<MessageContent>) -> Message {
    Message {
        role,
        content: content.into(),
    }
}

fn tool_use(id: &str) -> MessageContent {
    MessageContent::ToolUse {
        id: id.to_string(),
        name: "do_thing".to_string(),
        args: serde_json::json!({}),
    }
}

fn tool_result(id: &str) -> MessageContent {
    MessageContent::ToolResult {
        tool_use_id: id.to_string(),
        output: ToolResultBody::Text {
            text: "ok".to_string(),
        },
        is_error: false,
    }
}

fn text(t: &str) -> MessageContent {
    MessageContent::Text {
        text: t.to_string(),
    }
}

/// Valid paired sequence: returned as-is (same Arc, no rebuild).
#[test]
fn paired_sequence_unchanged() {
    let input = vec![
        msg(Role::User, vec![text("hi")]),
        msg(Role::Assistant, vec![tool_use("a")]),
        msg(Role::User, vec![tool_result("a")]),
    ];
    let out = sanitize_tool_pairing(input.clone());
    assert_eq!(out, input);
}

/// Orphan tool_use (no matching result): inject an error tool_result after the assistant
/// message.
#[test]
fn orphan_tool_use_gets_synthetic_result() {
    let input = vec![
        msg(Role::User, vec![text("hi")]),
        msg(Role::Assistant, vec![tool_use("a")]),
        // No result for `a` — the turn was interrupted mid-tool-execution.
    ];
    let out = sanitize_tool_pairing(input);
    assert_eq!(out.len(), 3);
    // The synthetic entry follows immediately after the assistant message.
    let MessageContent::ToolResult {
        tool_use_id,
        is_error,
        ..
    } = &out[2].content[0]
    else {
        panic!("expected synthetic tool_result");
    };
    assert_eq!(tool_use_id, "a");
    assert!(is_error);
    assert_eq!(out[2].role, Role::User);
}

/// Multiple orphans in the same assistant message: merge into a single user message
/// containing multiple results.
#[test]
fn multiple_orphans_in_one_assistant() {
    let input = vec![msg(Role::Assistant, vec![tool_use("a"), tool_use("b")])];
    let out = sanitize_tool_pairing(input);
    assert_eq!(out.len(), 2);
    assert_eq!(out[1].content.len(), 2);
    for (i, id) in ["a", "b"].iter().enumerate() {
        let MessageContent::ToolResult { tool_use_id, .. } = &out[1].content[i] else {
            panic!("expected tool_result");
        };
        assert_eq!(tool_use_id, id);
    }
}

/// Partial pairing: only fill the missing one; leave already paired ones unchanged.
#[test]
fn only_missing_one_is_filled() {
    let input = vec![
        msg(Role::Assistant, vec![tool_use("a")]),
        msg(Role::User, vec![tool_result("a")]),
        msg(Role::Assistant, vec![tool_use("b")]),
        // b has no result.
    ];
    let out = sanitize_tool_pairing(input);
    assert_eq!(out.len(), 4);
    let MessageContent::ToolResult { tool_use_id, .. } = &out[3].content[0] else {
        panic!("expected synthetic result for b");
    };
    assert_eq!(tool_use_id, "b");
}

/// A result is considered satisfied if it exists anywhere in the sequence (even if not
/// strictly adjacent); no duplicate is inserted.
#[test]
fn globally_satisfied_not_duplicated() {
    let input = vec![
        msg(Role::Assistant, vec![tool_use("a")]),
        msg(Role::User, vec![tool_result("a")]),
    ];
    let out = sanitize_tool_pairing(input.clone());
    assert_eq!(out, input);
    assert_eq!(out.len(), 2);
}

/// Empty sequence / no tool_use: returned unchanged.
#[test]
fn no_tool_use_unchanged() {
    let input = vec![
        msg(Role::User, vec![text("hi")]),
        msg(Role::Assistant, vec![text("hello")]),
    ];
    let out = sanitize_tool_pairing(input.clone());
    assert_eq!(out, input);

    let empty: Vec<Message> = Vec::new();
    assert!(sanitize_tool_pairing(empty).is_empty());
}