mirrors/zed
1
0
Fork 0
mirror of https://github.com/zed-industries/zed.git synced 2025-02-07 11:11:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

collab: Attach user_id to LLM spans (#16311)

Browse source 
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
Show stats Download patch file Download diff file Expand all files Collapse all files

4
crates/collab/src/llm.rs
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); req.extensions_mut().insert(claims);
Ok::<_, Error>(next.run(req).await.into_response()) Ok::<_, Error>(next.run(req).await.into_response())

1
crates/collab/src/main.rs
View file

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