collab: Attach user_id to LLM spans (#16311)

This PR updates the LLM service to attach the user ID to the spans.

Release Notes:

- N/A
This commit is contained in:
Marshall Bowers 2024-08-15 15:49:12 -04:00 committed by GitHub
parent ff83e5b55a
commit 5e05821d18
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 1 deletions

View File

@ -138,7 +138,9 @@ async fn validate_api_token<B>(mut req: Request<B>, next: Next<B>) -> impl IntoR
));
}
tracing::Span::current().record("authn.jti", &claims.jti);
tracing::Span::current()
.record("user_id", claims.user_id)
.record("authn.jti", &claims.jti);
req.extensions_mut().insert(claims);
Ok::<_, Error>(next.run(req).await.into_response())

View File

@ -150,6 +150,7 @@ async fn main() -> Result<()> {
"http_request",
method = ?request.method(),
matched_path,
user_id = tracing::field::Empty,
authn.jti = tracing::field::Empty
)
})