From 9bd3dbcf284d80629bde23367d89f0825ee5e954 Mon Sep 17 00:00:00 2001 From: Marshall Bowers Date: Fri, 4 Apr 2025 14:33:23 -0400 Subject: [PATCH] collab: Include more information on some LLM usage log lines (#28116) This PR updates the `user rate limit` and `user usage` log lines to include some more information that will be useful for graphing in Axiom. Release Notes: - N/A --- crates/collab/src/llm.rs | 3 +++ 1 file changed, 3 insertions(+) diff --git a/crates/collab/src/llm.rs b/crates/collab/src/llm.rs index f39d4a5ca5..8592a36bf5 100644 --- a/crates/collab/src/llm.rs +++ b/crates/collab/src/llm.rs @@ -574,6 +574,7 @@ async fn check_usage_limit( is_staff = claims.is_staff, provider = provider.to_string(), model = model.name, + usage_measure = resource, requests_this_minute = usage.requests_this_minute, tokens_this_minute = usage.tokens_this_minute, input_tokens_this_minute = usage.input_tokens_this_minute, @@ -696,6 +697,8 @@ impl Drop for TokenCountingStream { login = claims.github_user_login, authn.jti = claims.jti, is_staff = claims.is_staff, + provider = provider.to_string(), + model = model, requests_this_minute = usage.requests_this_minute, tokens_this_minute = usage.tokens_this_minute, input_tokens_this_minute = usage.input_tokens_this_minute,