omx(team): auto-checkpoint worker-1 [1]
This commit is contained in:
parent
bc32639ce3
commit
ce02ace3a2
|
|
@ -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(),
|
||||
}
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -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;
|
||||
}
|
||||
}
|
||||
_ => {}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue