caretta 0.11.6

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

use crate::agent::tracker::TrackerInfo;
use crate::agent::types::{AgentEvent, ClaudeEvent, Config, should_use_event_model};
use crate::ui::components::{format_cost_usd, format_token_count, format_tokens_per_second};
use dioxus::prelude::*;

#[derive(Default)]
struct UsageSummary {
    input_tokens: u32,
    output_tokens: u32,
    last_output_tokens_per_second: Option<String>,
    estimated_cost_usd: f64,
}

fn summarize_usage(events: &[AgentEvent], config: &Config) -> UsageSummary {
    let mut summary = UsageSummary::default();
    let mut active_model = config.pricing_model_key();
    let has_configured_model = active_model.is_some();

    for event in events {
        match event {
            AgentEvent::Claude(ClaudeEvent::System {
                model: Some(model), ..
            }) if should_use_event_model(model, has_configured_model) => {
                active_model = Some(model.trim().to_string());
            }
            AgentEvent::Claude(ClaudeEvent::Result {
                duration_ms,
                input_tokens,
                output_tokens,
                ..
            }) => {
                let input_tokens = input_tokens.unwrap_or(0);
                let output_tokens = output_tokens.unwrap_or(0);

                summary.input_tokens = summary.input_tokens.saturating_add(input_tokens);
                summary.output_tokens = summary.output_tokens.saturating_add(output_tokens);
                if let Some(rate) =
                    duration_ms.and_then(|ms| format_tokens_per_second(output_tokens, ms))
                {
                    summary.last_output_tokens_per_second = Some(rate);
                }
                if let Some(model) = &active_model
                    && let Some(cost) =
                        config
                            .pricing
                            .estimate_cost_usd(model, input_tokens, output_tokens)
                {
                    summary.estimated_cost_usd += cost;
                }
            }
            _ => {}
        }
    }

    summary
}

#[component]
pub fn Statusbar(
    config: Signal<Config>,
    tracker_ids: Signal<Vec<TrackerInfo>>,
    issues: Signal<Vec<crate::agent::tracker::PendingIssue>>,
    events: Signal<Vec<crate::agent::types::AgentEvent>>,
    is_working: Signal<bool>,
    theme_name: String,
) -> Element {
    let working = *is_working.read();
    let issue_count = issues.read().len();
    let event_count = events.read().len();
    let usage = summarize_usage(&events.read(), &config.read());
    let total_tokens = usage.input_tokens.saturating_add(usage.output_tokens);
    let total_token_label = format_token_count(total_tokens);
    let estimated_cost_label =
        (usage.estimated_cost_usd > 0.0).then(|| format_cost_usd(usage.estimated_cost_usd));
    let tracker_nums = tracker_ids
        .read()
        .iter()
        .map(|t| format!("#{}", t.number))
        .collect::<Vec<_>>()
        .join(", ");

    rsx! {
        div { class: "statusbar",
            div { class: "statusbar-left",
                span { class: if working { "status-dot status-dot-active" } else { "status-dot status-dot-idle" } }
                span { if working { "Working" } else { "Ready" } }
                if !tracker_nums.is_empty() {
                    span { class: "status-sep", "|" }
                    span { "{tracker_nums}" }
                }
            }
            div { class: "statusbar-right",
                span { "{issue_count} issues" }
                span { class: "status-sep", "|" }
                span { "{event_count} events" }
                span { class: "status-sep", "|" }
                if total_tokens > 0 {
                    span { "{total_token_label} tok" }
                    span { class: "status-sep", "|" }
                }
                if let Some(rate) = usage.last_output_tokens_per_second {
                    span { "{rate} out tok/s" }
                    span { class: "status-sep", "|" }
                }
                if let Some(cost) = estimated_cost_label {
                    span { "{cost}" }
                    span { class: "status-sep", "|" }
                }
                span { "{config.read().agent}" }
                span { class: "status-sep", "|" }
                span { "{theme_name}" }
            }
        }
    }
}