zag-agent 0.18.0

Core library for zag — a unified interface for AI coding agents
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
130
131

use super::*;
use crate::usage_limits::{UsageLimitConfig, UsageLimitProviderOverride};

#[test]
fn detects_bare_usage_limit_with_epoch() {
    let cfg = UsageLimitConfig::default();
    let line = "Claude AI usage limit reached|1760000400";
    let hit = detect_text(line, &cfg).expect("should detect");
    assert_eq!(hit.provider, "claude");
    assert_eq!(hit.scope, UsageLimitScope::Session);
    assert_eq!(hit.reset_at.unwrap().timestamp(), 1_760_000_400);
    assert_eq!(hit.raw, "Claude AI usage limit reached|1760000400");
}

#[test]
fn detects_weekly_scope() {
    let cfg = UsageLimitConfig::default();
    let line = "Claude AI weekly usage limit reached|1760000400";
    let hit = detect_text(line, &cfg).expect("should detect weekly");
    assert_eq!(hit.scope, UsageLimitScope::Weekly);
    assert_eq!(hit.reset_at.unwrap().timestamp(), 1_760_000_400);
}

#[test]
fn detects_global_scope() {
    let cfg = UsageLimitConfig::default();
    let line = "Claude AI global usage limit reached|1760000400";
    let hit = detect_text(line, &cfg).expect("should detect global");
    assert_eq!(hit.scope, UsageLimitScope::Global);
}

#[test]
fn detects_when_embedded_in_larger_text() {
    let cfg = UsageLimitConfig::default();
    let line = "I'm sorry, but Claude AI usage limit reached|1760000400 — try again later.";
    assert!(detect_text(line, &cfg).is_some());
}

#[test]
fn does_not_match_unrelated_text() {
    let cfg = UsageLimitConfig::default();
    assert!(detect_text("Hello world", &cfg).is_none());
    assert!(detect_text("Claude AI is great", &cfg).is_none());
    assert!(detect_text("usage limit reached", &cfg).is_none());
}

#[test]
fn respects_disabled_flag() {
    let cfg = UsageLimitConfig {
        enabled: false,
        ..Default::default()
    };
    let line = "Claude AI usage limit reached|1760000400";
    assert!(detect_text(line, &cfg).is_none());
}

#[test]
fn respects_per_provider_disabled() {
    let mut cfg = UsageLimitConfig::default();
    cfg.providers.insert(
        "claude".to_string(),
        UsageLimitProviderOverride {
            enabled: Some(false),
            ..Default::default()
        },
    );
    let line = "Claude AI usage limit reached|1760000400";
    assert!(detect_text(line, &cfg).is_none());
}

#[test]
fn extra_patterns_match_when_defaults_dont() {
    let mut cfg = UsageLimitConfig::default();
    cfg.providers.insert(
        "claude".to_string(),
        UsageLimitProviderOverride {
            extra_patterns: vec![r"hypothetical-new-limit-format=(\d+)".to_string()],
            ..Default::default()
        },
    );
    let line = "Server: hypothetical-new-limit-format=1760000400 sorry";
    let hit = detect_text(line, &cfg).expect("user pattern should match");
    assert_eq!(hit.provider, "claude");
}

#[test]
fn detects_api_retry_envelope() {
    let cfg = UsageLimitConfig::default();
    let value = serde_json::json!({
        "type": "system",
        "subtype": "api_retry",
        "attempt": 1,
        "max_retries": 5,
        "retry_delay_ms": 2000,
        "error_status": 429,
        "error": "rate_limit",
    });
    let hit = detect_json(&value, &cfg).expect("should detect api_retry rate limit");
    assert_eq!(hit.provider, "claude");
    assert!(hit.reset_at.is_none()); // structured envelope, no reset time
}

#[test]
fn detects_result_error_envelope() {
    let cfg = UsageLimitConfig::default();
    let value = serde_json::json!({
        "type": "result",
        "subtype": "error",
        "is_error": true,
        "error": "rate_limit",
    });
    assert!(detect_json(&value, &cfg).is_some());
}

#[test]
fn ignores_non_rate_limit_system_events() {
    let cfg = UsageLimitConfig::default();
    let value = serde_json::json!({
        "type": "system",
        "subtype": "init",
        "session_id": "abc",
    });
    assert!(detect_json(&value, &cfg).is_none());

    let other = serde_json::json!({
        "type": "system",
        "subtype": "api_retry",
        "error": "server_error",
    });
    assert!(detect_json(&other, &cfg).is_none());
}