From ce02ace3a28223ef80613a0312f4812dc37aa3f6 Mon Sep 17 00:00:00 2001
From: bellman <bellman@bellmanui-MacBookAir.local>
Date: Fri, 15 May 2026 10:29:59 +0900
Subject: [PATCH] omx(team): auto-checkpoint worker-1 [1]

---
 rust/crates/api/benches/request_building.rs |  1 +
 rust/crates/api/src/providers/anthropic.rs  | 22 ++++++++++-----------
 2 files changed, 11 insertions(+), 12 deletions(-)

diff --git a/rust/crates/api/benches/request_building.rs b/rust/crates/api/benches/request_building.rs
index 1a33dcbf..23493477 100644
--- a/rust/crates/api/benches/request_building.rs
+++ b/rust/crates/api/benches/request_building.rs
@@ -76,6 +76,7 @@ fn create_sample_request(message_count: usize) -> MessageRequest {
         presence_penalty: None,
         stop: None,
         reasoning_effort: None,
+        extra_body: std::collections::BTreeMap::new(),
     }
 }
 
diff --git a/rust/crates/api/src/providers/anthropic.rs b/rust/crates/api/src/providers/anthropic.rs
index 253a5ec2..51e10b44 100644
--- a/rust/crates/api/src/providers/anthropic.rs
+++ b/rust/crates/api/src/providers/anthropic.rs
@@ -845,19 +845,17 @@ impl MessageStream {
             StreamEvent::MessageDelta(MessageDeltaEvent { usage, .. }) => {
                 self.latest_usage = Some(usage.clone());
             }
-            StreamEvent::MessageStop(_) => {
-                if !self.usage_recorded {
-                    if let (Some(prompt_cache), Some(usage)) =
-                        (&self.prompt_cache, self.latest_usage.as_ref())
-                    {
-                        let record = prompt_cache.record_usage(&self.request, usage);
-                        *self
-                            .last_prompt_cache_record
-                            .lock()
-                            .unwrap_or_else(std::sync::PoisonError::into_inner) = Some(record);
-                    }
-                    self.usage_recorded = true;
+            StreamEvent::MessageStop(_) if !self.usage_recorded => {
+                if let (Some(prompt_cache), Some(usage)) =
+                    (&self.prompt_cache, self.latest_usage.as_ref())
+                {
+                    let record = prompt_cache.record_usage(&self.request, usage);
+                    *self
+                        .last_prompt_cache_record
+                        .lock()
+                        .unwrap_or_else(std::sync::PoisonError::into_inner) = Some(record);
                 }
+                self.usage_recorded = true;
             }
             _ => {}
         }