mobius-gateway 0.15.0

Headless authenticated gateway for möbius frontends
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99

use mobius::backend::checkpoint::Checkpoint;
use mobius::protocol::{SessionContext, TokenUsage};

use super::*;
use mobius::backend::checkpoint::ExecutionStats;

mod bots;
mod lifecycle;
mod projection;
mod replay;
mod swarm_delivery;
mod swarm_management;

pub(crate) async fn ensure_test_bot(
    gateway: &GatewayHost,
) -> std::result::Result<crate::wire::BotRecord, Rejection> {
    let state = gateway.state.lock().await;
    if let Some(bot) = state.bots.bots().map_err(internal)?.into_iter().next() {
        return Ok(bot);
    }
    let mut config = state
        .config
        .lock()
        .map_err(|_| internal("gateway configuration lock is poisoned"))?;
    if config.bot_defaults.is_none() {
        let next = config
            .registering_provider(
                AgentComposition::default().provider,
                "Test".into(),
                Default::default(),
                Vec::new(),
                Vec::new(),
            )
            .map_err(invalid_config)?;
        state.store.save(&next).map_err(internal)?;
        *config = next;
    }
    let composition = config
        .bot_defaults
        .as_ref()
        .expect("provider registration installs Bot defaults")
        .config
        .clone();
    drop(config);
    state
        .bots
        .create_bot("Test Bot", "Own gateway test work.", composition)
        .map_err(invalid_bot)
}

async fn enable_test_collaboration(
    gateway: &GatewayHost,
    bot: crate::wire::BotRecord,
) -> crate::wire::BotRecord {
    let mut config = bot.config.config.clone();
    config.middleware.set_setting(
        "bots",
        "collaboration",
        Some(mobius::protocol::FrontendSettingValue::String(
            "swarm".into(),
        )),
    );
    gateway
        .update_bot(
            &bot.id,
            bot.config.revision,
            &bot.name,
            &bot.description,
            bot.tint,
            config,
        )
        .await
        .expect("enable collaboration")
}

pub(crate) async fn create_test_session(
    gateway: &GatewayHost,
    workspace: &Path,
) -> std::result::Result<HostHandle, Rejection> {
    let bot = ensure_test_bot(gateway).await?;
    gateway.create_session(workspace, &bot.id).await
}

pub(crate) async fn create_distinct_test_session(
    gateway: &GatewayHost,
    workspace: &Path,
    handle: &str,
) -> std::result::Result<(HostHandle, crate::wire::BotRecord), Rejection> {
    let template = ensure_test_bot(gateway).await?;
    let bot = {
        let state = gateway.state.lock().await;
        state
            .bots
            .create_bot(handle, "Own distinct test work.", template.config.config)
            .map_err(invalid_bot)?
    };
    let host = gateway.create_session(workspace, &bot.id).await?;
    Ok((host, bot))
}