refactor(agent): use hive_claude::TokenUsage directly, drop the duplicate

This commit is contained in:
müde 2026-07-05 20:21:33 +02:00
commit 66f5b8720d
7 changed files with 16 additions and 48 deletions

View file

@ -268,8 +268,8 @@ impl TurnStats {
pub fn last_usage(
&self,
) -> (
Option<crate::events::TokenUsage>,
Option<crate::events::TokenUsage>,
Option<hive_claude::TokenUsage>,
Option<hive_claude::TokenUsage>,
) {
let conn = self.inner.lock().unwrap();
conn.query_row(
@ -285,19 +285,19 @@ impl TurnStats {
let g = |i: usize| -> rusqlite::Result<u64> {
Ok(u64::try_from(row.get::<_, i64>(i)?).unwrap_or(0))
};
let cost = crate::events::TokenUsage {
let cost = hive_claude::TokenUsage {
input_tokens: g(0)?,
output_tokens: g(1)?,
cache_read_input_tokens: g(2)?,
cache_creation_input_tokens: g(3)?,
};
let last = crate::events::TokenUsage {
let last = hive_claude::TokenUsage {
input_tokens: g(4)?,
output_tokens: g(5)?,
cache_read_input_tokens: g(6)?,
cache_creation_input_tokens: g(7)?,
};
let ctx = if last == crate::events::TokenUsage::default() {
let ctx = if last == hive_claude::TokenUsage::default() {
None
} else {
Some(last)