huskies: merge 1015

2026-05-13 23:33:30 +00:00
parent 69b207872a
commit 5ed1438ab9
22 changed files with 227 additions and 73 deletions
@@ -9,7 +9,7 @@ use std::collections::HashMap;
 #[derive(Debug, Clone, PartialEq)]
 pub struct AgentTokenCost {
    pub agent_name: String,
-    pub model: Option<String>,
+    pub model: Option<crate::agents::AgentModel>,
    pub input_tokens: u64,
    pub output_tokens: u64,
    pub cache_creation_input_tokens: u64,
@@ -153,8 +153,11 @@ mod tests {
    #[test]
    fn aggregate_preserves_model_from_first_record() {
        let mut r = make_record("42_story_foo", "coder-1", 1.0);
-        r.model = Some("claude-sonnet".to_string());
+        r.model = Some(crate::agents::AgentModel::Sonnet);
        let summary = aggregate_for_story(&[r], "42_story_foo");
-        assert_eq!(summary.agents[0].model, Some("claude-sonnet".to_string()));
+        assert_eq!(
+            summary.agents[0].model,
+            Some(crate::agents::AgentModel::Sonnet)
+        );
    }
 }