From 3f2de172aeb378813ddcdf96106bb48386b85ec6 Mon Sep 17 00:00:00 2001 From: Marshall Bowers Date: Tue, 8 Oct 2024 13:16:17 -0400 Subject: [PATCH] collab: Set cached token values when initially creating lifetime usage records (#18865) This PR fixes an issue where we weren't setting the cached token fields when initially creating a lifetime usage record. Release Notes: - N/A --- crates/collab/src/llm/db/queries/usages.rs | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/crates/collab/src/llm/db/queries/usages.rs b/crates/collab/src/llm/db/queries/usages.rs index 128a42bc58791d8fdb417e3053726907ee737fba..d703066913f8f5248ea37fc91bdf76b2c126212a 100644 --- a/crates/collab/src/llm/db/queries/usages.rs +++ b/crates/collab/src/llm/db/queries/usages.rs @@ -382,6 +382,10 @@ impl LlmDatabase { user_id: ActiveValue::set(user_id), model_id: ActiveValue::set(model.id), input_tokens: ActiveValue::set(input_token_count as i64), + cache_creation_input_tokens: ActiveValue::set( + cache_creation_input_tokens as i64, + ), + cache_read_input_tokens: ActiveValue::set(cache_read_input_tokens as i64), output_tokens: ActiveValue::set(output_token_count as i64), ..Default::default() }