symposium-crate-sources-proxy 0.2.1

ACP proxy component providing Rust crate source code research tools
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

//! Rust Crate Sources Proxy
//!
//! An ACP proxy that provides the `rust_crate_query` MCP tool for researching
//! Rust crate sources via dedicated sub-agent sessions.

mod crate_sources_mcp;
mod eg;
mod research_agent;
mod state;

use anyhow::Result;
use sacp::component::Component;
use sacp::mcp_server::McpServiceRegistry;
use sacp::ProxyToConductor;
use state::ResearchState;
use std::sync::Arc;

/// Run the proxy as a standalone binary connected to stdio
pub async fn run() -> Result<()> {
    // Initialize tracing
    tracing_subscriber::fmt()
        .with_env_filter(
            tracing_subscriber::EnvFilter::try_from_default_env()
                .unwrap_or_else(|_| tracing_subscriber::EnvFilter::new("info")),
        )
        .init();

    tracing::info!("Starting rust-crate-sources-proxy");

    CrateSourcesProxy.serve(sacp_tokio::Stdio::new()).await?;

    Ok(())
}

/// A proxy which forwards all messages to its successor, adding access to the rust-crate-query MCP server.
pub struct CrateSourcesProxy;

impl Component for CrateSourcesProxy {
    async fn serve(self, client: impl Component) -> Result<(), sacp::Error> {
        // Create shared state for tracking active research sessions
        let state = Arc::new(ResearchState::new());

        // Create MCP service registry with the user-facing service
        let mcp_registry = McpServiceRegistry::new().with_mcp_server(
            "rust-crate-query",
            research_agent::build_server(state.clone()),
        )?;

        ProxyToConductor::builder()
            .name("rust-crate-sources-proxy")
            .with_handler(mcp_registry)
            .with_handler(research_agent::PermissionAutoApprover::new(state.clone()))
            .serve(client)
            .await
    }
}