brk_mcp 0.0.111

A bridge for LLMs to access BRK
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

use axum::Router;
use brk_interface::Interface;
use brk_rmcp::transport::{
    StreamableHttpServerConfig,
    streamable_http_server::{StreamableHttpService, session::local::LocalSessionManager},
};

use log::info;

use crate::MCP;

pub trait MCPRoutes {
    fn add_mcp_routes(self, interface: &'static Interface<'static>, mcp: bool) -> Self;
}

impl<T> MCPRoutes for Router<T>
where
    T: Clone + Send + Sync + 'static,
{
    fn add_mcp_routes(self, interface: &'static Interface<'static>, mcp: bool) -> Self {
        if !mcp {
            return self;
        }

        let service = StreamableHttpService::new(
            move || Ok(MCP::new(interface)),
            LocalSessionManager::default().into(),
            StreamableHttpServerConfig {
                stateful_mode: false,
                ..Default::default()
            },
        );

        info!("Setting MCP...");

        self.nest_service("/mcp", service)
    }
}