mockforge-intelligence 0.3.234

AI-powered behavior, response generation, and behavioral cloning for MockForge
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

//! Behavioral cloning HTTP handlers, backed by an injected recording store.
//!
//! Probability models, learned sequences and edge amplification have separate
//! modules. The recorder implements storage downstream, avoiding a core cycle.

mod amplification;
mod probability;
mod probability_data;
mod sequences;

pub use amplification::{apply_amplification, get_rare_edges, ApplyAmplificationRequest};
pub use probability::{
    build_probability_model, get_probability_model, list_probability_models, sample_latency,
    sample_status_code, BuildProbabilityModelRequest, ProbabilityModelResponse,
};
pub use sequences::{
    discover_sequences, generate_sequence_scenario, get_sequence, list_sequences,
    DiscoverSequencesRequest, SequenceDiscoveryResponse,
};

use crate::behavioral_cloning::storage::BehavioralCloningStateProvider;

/// Create router for behavioral cloning endpoints
pub fn behavioral_cloning_router<S: BehavioralCloningStateProvider>(state: S) -> axum::Router {
    use axum::routing::{get, post};
    use axum::Router;

    Router::new()
        // Probability model endpoints
        .route(
            "/api/v1/behavioral-cloning/probability-models",
            post(build_probability_model::<S>).get(list_probability_models::<S>),
        )
        .route(
            "/api/v1/behavioral-cloning/probability-models/{endpoint}/{method}",
            get(get_probability_model::<S>),
        )
        .route(
            "/api/v1/behavioral-cloning/probability-models/{endpoint}/{method}/sample/status-code",
            post(sample_status_code::<S>),
        )
        .route(
            "/api/v1/behavioral-cloning/probability-models/{endpoint}/{method}/sample/latency",
            post(sample_latency::<S>),
        )
        // Sequence endpoints
        .route(
            "/api/v1/behavioral-cloning/sequences",
            get(list_sequences::<S>),
        )
        .route(
            "/api/v1/behavioral-cloning/sequences/discover",
            post(discover_sequences::<S>),
        )
        .route(
            "/api/v1/behavioral-cloning/sequences/{sequence_id}",
            get(get_sequence::<S>),
        )
        .route(
            "/api/v1/behavioral-cloning/sequences/{sequence_id}/scenario",
            post(generate_sequence_scenario::<S>),
        )
        // Amplification endpoints
        .route(
            "/api/v1/behavioral-cloning/amplification/apply",
            post(apply_amplification::<S>),
        )
        .route(
            "/api/v1/behavioral-cloning/amplification/rare-edges/{endpoint}/{method}",
            get(get_rare_edges::<S>),
        )
        .with_state(state)
}