pi_agent_rust 0.3.0

Native AI coding agent CLI - Rust port of Pi Agent
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

//! Event classification and timeout policy tests.

use super::*;

// ===================================================================
// #50: informational-vs-actionable event classification drives the
// per-event default timeout.
// ===================================================================

#[test]
fn informational_events_use_short_timeout() {
    let info_events = [
        ExtensionEventName::Startup,
        ExtensionEventName::AgentStart,
        ExtensionEventName::AgentEnd,
        ExtensionEventName::TurnStart,
        ExtensionEventName::TurnEnd,
        ExtensionEventName::MessageStart,
        ExtensionEventName::MessageUpdate,
        ExtensionEventName::MessageEnd,
        ExtensionEventName::ToolExecutionStart,
        ExtensionEventName::ToolExecutionUpdate,
        ExtensionEventName::ToolExecutionEnd,
        ExtensionEventName::SessionStart,
        ExtensionEventName::SessionSwitch,
        ExtensionEventName::SessionFork,
        ExtensionEventName::SessionCompact,
        ExtensionEventName::SessionTree,
        ExtensionEventName::SessionShutdown,
        ExtensionEventName::ModelSelect,
        ExtensionEventName::UserBash,
    ];
    for event in info_events {
        assert!(
            event.is_informational(),
            "{event} should be classified as informational"
        );
        assert_eq!(
            event.default_timeout_ms(),
            EXTENSION_INFO_EVENT_TIMEOUT_MS,
            "{event} should use the short default timeout"
        );
    }
}

#[test]
fn actionable_events_use_full_timeout() {
    // These events feed a decision (block/cancel/transform), so a
    // handler needs room to do real work before a verdict is expected.
    // ToolResult belongs here per its ExtensionEvent docstring
    // ("can modify result") — the dispatcher consumes handler-returned
    // changes to the tool result content.
    let actionable_events = [
        ExtensionEventName::Input,
        ExtensionEventName::BeforeAgentStart,
        ExtensionEventName::Context,
        ExtensionEventName::ToolCall,
        ExtensionEventName::ToolResult,
        ExtensionEventName::SessionBeforeSwitch,
        ExtensionEventName::SessionBeforeFork,
        ExtensionEventName::SessionBeforeCompact,
        ExtensionEventName::SessionBeforeTree,
        ExtensionEventName::ResourcesDiscover,
    ];
    for event in actionable_events {
        assert!(
            !event.is_informational(),
            "{event} should NOT be classified as informational"
        );
        assert_eq!(
            event.default_timeout_ms(),
            EXTENSION_EVENT_TIMEOUT_MS,
            "{event} should use the full default timeout"
        );
    }
}

#[test]
fn info_timeout_is_strictly_shorter_than_general_timeout() {
    assert!(
        EXTENSION_INFO_EVENT_TIMEOUT_MS < EXTENSION_EVENT_TIMEOUT_MS,
        "info timeout ({EXTENSION_INFO_EVENT_TIMEOUT_MS}ms) must be strictly shorter than general timeout ({EXTENSION_EVENT_TIMEOUT_MS}ms)"
    );
}