systemprompt-api 0.1.18

HTTP API server and gateway for systemprompt.io OS
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

use axum::extract::Request;
use axum::middleware::Next;
use axum::response::Response;
use systemprompt_models::RequestContext;

pub async fn inject_trace_header(request: Request, next: Next) -> Response {
    let trace_id = request
        .extensions()
        .get::<RequestContext>()
        .map(|ctx| ctx.trace_id().as_str().to_string());

    let mut response = next.run(request).await;

    if let Some(id) = trace_id {
        if let Ok(header_value) = id.parse() {
            response.headers_mut().insert("x-trace-id", header_value);
        }
    }

    response
}