bitrouter-api 0.22.0

BitRouter API - reusable warp server components for AI model routing
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

//! Warp filter for the `GET /v1/routes` endpoint.
//!
//! Returns the list of configured model routes, including the virtual model
//! name, provider, and API protocol for each route.

use std::sync::Arc;

use bitrouter_core::routers::routing_table::RoutingTable;
use warp::Filter;

/// Creates a warp filter for `GET /v1/routes`.
pub fn routes_filter<T>(
    table: Arc<T>,
) -> impl Filter<Extract = (impl warp::Reply,), Error = warp::Rejection> + Clone
where
    T: RoutingTable + Send + Sync + 'static,
{
    warp::path!("v1" / "routes")
        .and(warp::get())
        .and(warp::any().map(move || table.clone()))
        .map(handle_list_routes)
}

fn handle_list_routes<T: RoutingTable>(table: Arc<T>) -> impl warp::Reply {
    let entries = table.list_routes();
    let routes: Vec<serde_json::Value> = entries
        .into_iter()
        .map(|e| {
            serde_json::json!({
                "name": e.name,
                "provider": e.provider,
                "protocol": e.protocol,
            })
        })
        .collect();
    warp::reply::json(&serde_json::json!({ "routes": routes }))
}