embacle-server 0.38.0

Unified OpenAI-compatible REST API + MCP server for embacle LLM runners
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

// ABOUTME: Axum router wiring OpenAI-compatible and MCP endpoints
// ABOUTME: Mounts completions, models, health, and MCP routes with optional auth middleware
//
// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2026 dravr.ai

use std::sync::Arc;

use axum::middleware;
use axum::routing::{get, post};
use axum::Router;
use dravr_tronc::mcp::transport::http::mcp_router as build_mcp_router;
use dravr_tronc::McpServer;

use crate::auth;
use crate::completions;
use crate::health;
use crate::models;
use crate::state::AppState;
use crate::uhp;

/// Build the application router with all endpoints
///
/// Routes:
/// - `POST /v1/chat/completions` — Chat completion (streaming and non-streaming)
/// - `GET /v1/models` — List available models
/// - `GET /health` — Provider health check
/// - `POST /mcp` — MCP Streamable HTTP (JSON-RPC 2.0, via dravr-tronc)
/// - `{UHP_BASE_PATH}/v1/*` — Unified Harness Protocol, default base `/uhp`
///
/// The auth middleware is applied to all routes. It only enforces
/// authentication when `EMBACLE_API_KEY` is set.
///
/// UHP is nested under its own base path rather than merged, because both
/// surfaces define `GET /v1/models` with incompatible bodies. It carries its
/// own auth and version layers: its refusals must be the UHP error envelope,
/// and its discovery endpoint must answer without a credential.
pub fn build(state: AppState) -> Router {
    let mcp_server = Arc::new(McpServer::new(
        "embacle-mcp",
        env!("CARGO_PKG_VERSION"),
        embacle_mcp::build_tool_registry(),
        Arc::clone(&state.shared),
    ));

    let mcp_router = build_mcp_router(mcp_server);

    let openai = Router::new()
        .route("/v1/chat/completions", post(completions::handle))
        .route("/v1/models", get(models::handle))
        .route("/health", get(health::handle))
        .with_state(state.clone())
        .merge(mcp_router)
        .layer(middleware::from_fn(auth::require_auth));

    let base = uhp::base_path();
    let uhp_routes = uhp::router(state);

    if base.is_empty() {
        openai.merge(uhp_routes)
    } else {
        openai.nest(&base, uhp_routes)
    }
}