diff --git a/crates/collab/src/llm.rs b/crates/collab/src/llm.rs index 4ec8b70ac0..cf254a42ee 100644 --- a/crates/collab/src/llm.rs +++ b/crates/collab/src/llm.rs @@ -141,7 +141,8 @@ async fn validate_api_token(mut req: Request, next: Next) -> impl IntoR tracing::Span::current() .record("user_id", claims.user_id) .record("login", claims.github_user_login.clone()) - .record("authn.jti", &claims.jti); + .record("authn.jti", &claims.jti) + .record("is_staff", &claims.is_staff); req.extensions_mut().insert(claims); Ok::<_, Error>(next.run(req).await.into_response()) @@ -558,6 +559,7 @@ impl Drop for TokenCountingStream { user_id = claims.user_id, login = claims.github_user_login, authn.jti = claims.jti, + is_staff = claims.is_staff, requests_this_minute = usage.requests_this_minute, tokens_this_minute = usage.tokens_this_minute, ); diff --git a/crates/collab/src/main.rs b/crates/collab/src/main.rs index 35a80b702e..a946d16620 100644 --- a/crates/collab/src/main.rs +++ b/crates/collab/src/main.rs @@ -154,7 +154,8 @@ async fn main() -> Result<()> { matched_path, user_id = tracing::field::Empty, login = tracing::field::Empty, - authn.jti = tracing::field::Empty + authn.jti = tracing::field::Empty, + is_staff = tracing::field::Empty ) }) .on_response(