story-kit: merge 301_story_dedicated_token_usage_page_in_web_ui

2026-03-19 11:34:08 +00:00
parent 586d06b840
commit a6ac6497e9
8 changed files with 534 additions and 8 deletions
--- a/server/src/agents/pool.rs
+++ b/server/src/agents/pool.rs
@@ -507,8 +507,11 @@ impl AgentPool {
                        && let Some(agent) = agents.get(&key_clone)
                        && let Some(ref pr) = agent.project_root
                    {
+                        let model = config_clone
+                            .find_agent(&aname)
+                            .and_then(|a| a.model.clone());
                        let record = super::token_usage::build_record(
-                            &sid, &aname, usage.clone(),
+                            &sid, &aname, model, usage.clone(),
                        );
                        if let Err(e) = super::token_usage::append_record(pr, &record) {
                            slog_error!(