orign 0.2.3

A globally distributed container orchestrator
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
65
66
67
68
69
70
71
72
73
74
75
76

use crate::middleware::auth_middleware;
use crate::state::AppState;
use crate::handlers::v1::buffers::{
    create_buffer, delete_buffer, get_buffer,
    list_buffers, send_examples, train_buffer, update_buffer,
};
use crate::handlers::basic::{health_handler, root_handler};
use axum::{middleware, routing::delete, routing::get, routing::post, Router};
use tower_http::trace::{self, TraceLayer};
use tracing::Level;

pub fn create_routes(app_state: AppState) -> Router<AppState> {
    // Public routes that do not require authentication
    let public_routes = Router::new()
        .route("/", get(root_handler))
        .route("/health", get(health_handler));

    // Private routes that require authentication
    let private_routes = Router::new()
        ///
        // LLMs
        //
        // .route("/v1/llms", get(list_llms).post(create_llm))
        // .route("/v1/llms/:namespace/:name", get(get_llm).delete(delete_llm))
        // .route("/v1/llms/:namespace/:name/chat", post(chat_llm))
        // .route("/v1/llms/:namespace/:name/send", post(send_buffer_examples))
        // .route("/v1/llms/:namespace/:name/train", post(train_llm))
        //
        // Humans
        //
        // .route("/v1/humans", get(list_humans).post(create_human))
        // .route(
        //     "/v1/humans/:namespace/:name",
        //     get(get_human).delete(delete_human),
        // )
        // .route(
        //     "/v1/humans/:namespace/:name/feedback",
        //     post(request_human_feedback),
        // )
        // .route(
        //     "/v1/humans/:namespace/:name/feedback/:id",
        //     get(get_human_feedback),
        // )
        //
        // Buffers
        //
        .route("/v1/buffers", get(list_buffers).post(create_buffer))
        .route(
            "/v1/buffers/:namespace/:name",
            get(get_buffer).delete(delete_buffer).patch(update_buffer),
        )
        .route("/v1/buffers/:namespace/:name/examples", post(send_examples))
        .route("/v1/buffers/:namespace/:name/train", post(train_buffer))
        // .route(
        //     "/v1/buffers/:namespace/:name/trainings",
        //     get(list_trainings),
        // )
        // .route(
        //     "/v1/buffers/:namespace/:name/trainings/:id",
        //     get(get_training).delete(delete_training),
        // )
        //
        // Auth
        //
        .layer(middleware::from_fn_with_state(
            app_state.clone(),
            auth_middleware,
        ));

    // Combine public and private routes
    public_routes.merge(private_routes).layer(
        TraceLayer::new_for_http()
            .make_span_with(trace::DefaultMakeSpan::new().level(Level::INFO))
            .on_response(trace::DefaultOnResponse::new().level(Level::INFO)),
    )
}