avoid error when /compact response has no token_usage (#2417)

This commit is contained in:
Odysseas YIAKOUMIS
2025-08-24 18:00:57 +04:00
parent c6a52d611c
commit dc56984853

View File

@@ -2810,22 +2810,15 @@ async fn drain_to_completed(
response_id: _,
token_usage,
}) => {
let token_usage = match token_usage {
Some(usage) => usage,
None => {
return Err(CodexErr::Stream(
"token_usage was None in ResponseEvent::Completed".into(),
None,
));
}
};
sess.tx_event
.send(Event {
id: sub_id.to_string(),
msg: EventMsg::TokenCount(token_usage),
})
.await
.ok();
if let Some(token_usage) = token_usage {
sess.tx_event
.send(Event {
id: sub_id.to_string(),
msg: EventMsg::TokenCount(token_usage),
})
.await
.ok();
}
return Ok(());
}
Ok(_) => continue,