use anyhow::Result;
use kodegen_config::CATEGORY_INTROSPECTION;
use kodegen_server_http::{ServerBuilder, Managers, RouterSet, register_tool};
use rmcp::handler::server::router::{prompt::PromptRouter, tool::ToolRouter};
#[tokio::main]
async fn main() -> Result<()> {
ServerBuilder::new()
.category(CATEGORY_INTROSPECTION)
.register_tools(|| async {
let tool_router = ToolRouter::new();
let prompt_router = PromptRouter::new();
let managers = Managers::new();
let (tool_router, prompt_router) = register_tool(
tool_router,
prompt_router,
kodegen_tools_introspection::InspectUsageStatsTool::new(),
);
let (tool_router, prompt_router) = register_tool(
tool_router,
prompt_router,
kodegen_tools_introspection::InspectToolCallsTool::new(),
);
Ok(RouterSet::new(tool_router, prompt_router, managers))
})
.run()
.await
}