refactor(agent): use hive_claude::TokenUsage directly, drop the duplicate
This commit is contained in:
parent
c9de2eeb97
commit
66f5b8720d
7 changed files with 16 additions and 48 deletions
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Reference in a new issue