make_auth_interceptor (src/auth.rs)

LocalAI · backend/rust/kokoros/src/auth.rs · rust · MIT
auth, tag:embed-classified

Creates a gRPC auth interceptor from env token in Rust.

// Source: https://github.com/mudler/LocalAI | License: MIT | Copyright 2023-2025 Ettore Di Giacinto (mudler@localai.io)
// Provided via justclaw.it.com; keep this notice with the code.
use tonic::{Request, Status};

/// Returns an interceptor function if LOCALAI_GRPC_AUTH_TOKEN is set.
pub fn make_auth_interceptor(
) -> Option<impl Fn(Request<()>) -> Result<Request<()>, Status> + Clone> {
    let token = std::env::var("LOCALAI_GRPC_AUTH_TOKEN").ok()?;
    if token.is_empty() {
        return None;
    }
    let expected = format!("Bearer {}", token);
    Some(
        move |req: Request<()>| -> Result<Request<()>, Status> {
            let meta = req.metadata();
            match meta.get("authorization") {
                Some(val) => {
                    if val.as_bytes() == expected.a

... (truncated -- full source via MCP)

See the full source, get the GitHub permalink, and search for more like it.

Get a free API key