diff --git a/crates/collab/src/llm.rs b/crates/collab/src/llm.rs index f39d4a5ca588606167cb52e89e2b2acac44b13fd..8592a36bf55a2c219456585fcaae4198ecf0b5ab 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,