kodegen_tools_introspection 0.5.1

//! Introspection Category HTTP Server
//!
//! Serves introspection tools via HTTP/HTTPS transport using kodegen_server_http.

use anyhow::Result;
use kodegen_server_http::{run_http_server, Managers, RouterSet, register_tool};
use rmcp::handler::server::router::{prompt::PromptRouter, tool::ToolRouter};

#[tokio::main]
async fn main() -> Result<()> {
    run_http_server("introspection", |_config, tracker| {
        let tracker = tracker.clone();
        Box::pin(async move {
        let tool_router = ToolRouter::new();
        let prompt_router = PromptRouter::new();
        let managers = Managers::new();

        // Register all 2 introspection tools
        let (tool_router, prompt_router) = register_tool(
            tool_router,
            prompt_router,
            kodegen_tools_introspection::InspectUsageStatsTool::new(tracker.clone()),
        );

        let (tool_router, prompt_router) = register_tool(
            tool_router,
            prompt_router,
            kodegen_tools_introspection::InspectToolCallsTool::new(),
        );

        Ok(RouterSet::new(tool_router, prompt_router, managers))
        })
    })
    .await
}