mirror of
https://github.com/instructkr/claw-code.git
synced 2026-05-19 22:11:28 +08:00
omx(team): auto-checkpoint worker-1 [1]
This commit is contained in:
@@ -76,6 +76,7 @@ fn create_sample_request(message_count: usize) -> MessageRequest {
|
|||||||
presence_penalty: None,
|
presence_penalty: None,
|
||||||
stop: None,
|
stop: None,
|
||||||
reasoning_effort: None,
|
reasoning_effort: None,
|
||||||
|
extra_body: std::collections::BTreeMap::new(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -845,19 +845,17 @@ impl MessageStream {
|
|||||||
StreamEvent::MessageDelta(MessageDeltaEvent { usage, .. }) => {
|
StreamEvent::MessageDelta(MessageDeltaEvent { usage, .. }) => {
|
||||||
self.latest_usage = Some(usage.clone());
|
self.latest_usage = Some(usage.clone());
|
||||||
}
|
}
|
||||||
StreamEvent::MessageStop(_) => {
|
StreamEvent::MessageStop(_) if !self.usage_recorded => {
|
||||||
if !self.usage_recorded {
|
if let (Some(prompt_cache), Some(usage)) =
|
||||||
if let (Some(prompt_cache), Some(usage)) =
|
(&self.prompt_cache, self.latest_usage.as_ref())
|
||||||
(&self.prompt_cache, self.latest_usage.as_ref())
|
{
|
||||||
{
|
let record = prompt_cache.record_usage(&self.request, usage);
|
||||||
let record = prompt_cache.record_usage(&self.request, usage);
|
*self
|
||||||
*self
|
.last_prompt_cache_record
|
||||||
.last_prompt_cache_record
|
.lock()
|
||||||
.lock()
|
.unwrap_or_else(std::sync::PoisonError::into_inner) = Some(record);
|
||||||
.unwrap_or_else(std::sync::PoisonError::into_inner) = Some(record);
|
|
||||||
}
|
|
||||||
self.usage_recorded = true;
|
|
||||||
}
|
}
|
||||||
|
self.usage_recorded = true;
|
||||||
}
|
}
|
||||||
_ => {}
|
_ => {}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user