huskies: merge 1015

server/src/agents/token_usage.rs

@@ -5,7 +5,7 @@ use std::path::Path;
 use chrono::Utc;
 use serde::{Deserialize, Serialize};
 
-use super::TokenUsage;
+use super::{AgentModel, TokenUsage};
 
 /// A single token usage record persisted to disk.
 #[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
@@ -14,7 +14,7 @@ pub struct TokenUsageRecord {
     pub agent_name: String,
     pub timestamp: String,
     #[serde(default)]
-    pub model: Option<String>,
+    pub model: Option<AgentModel>,
     pub usage: TokenUsage,
 }
 
@@ -75,7 +75,7 @@ pub fn read_all(project_root: &Path) -> Result<Vec<TokenUsageRecord>, String> {
 pub fn build_record(
     story_id: &str,
     agent_name: &str,
-    model: Option<String>,
+    model: Option<AgentModel>,
     usage: TokenUsage,
 ) -> TokenUsageRecord {
     TokenUsageRecord {