wired into cli

block · Jan 8, 2025 · 2533514 · 2533514
1 parent df01349
commit 2533514
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/crates/goose-cli/src/commands/mcp.rs b/crates/goose-cli/src/commands/mcp.rs
@@ -1,14 +1,15 @@
 use anyhow::Result;
-use goose_mcp::DeveloperRouter;
+use goose_mcp::{DeveloperRouter, JetBrainsRouter};
 use mcp_server::router::RouterService;
-use mcp_server::{ByteTransport, Server};
+use mcp_server::{ByteTransport, Server, BoundedService};
 use tokio::io::{stdin, stdout};
 
 pub async fn run_server(name: &str) -> Result<()> {
     tracing::info!("Starting MCP server");
 
-    let router = match name {
-        "developer" => Some(RouterService(DeveloperRouter::new())),
+    let router: Option<Box<dyn BoundedService>> = match name {
+        "developer" => Some(Box::new(RouterService(DeveloperRouter::new()))),
+        "jetbrains" => Some(Box::new(RouterService(JetBrainsRouter::new()))),
         _ => None,
     };