omx(team): auto-checkpoint worker-1 [1]

This commit is contained in:
bellman
2026-05-15 10:29:59 +09:00
parent bc32639ce3
commit ce02ace3a2
2 changed files with 11 additions and 12 deletions

View File

@@ -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(),
}
}

View File

@@ -845,8 +845,7 @@ impl MessageStream {
StreamEvent::MessageDelta(MessageDeltaEvent { usage, .. }) => {
self.latest_usage = Some(usage.clone());
}
StreamEvent::MessageStop(_) => {
if !self.usage_recorded {
StreamEvent::MessageStop(_) if !self.usage_recorded => {
if let (Some(prompt_cache), Some(usage)) =
(&self.prompt_cache, self.latest_usage.as_ref())
{
@@ -858,7 +857,6 @@ impl MessageStream {
}
self.usage_recorded = true;
}
}
_ => {}
}
}