From 25335143f8dc697f31812350a3ef8c4a64448a87 Mon Sep 17 00:00:00 2001
From: Max Novich <mnovich@squareup.com>
Date: Tue, 7 Jan 2025 17:11:44 -0800
Subject: [PATCH] wired into cli

---
 crates/goose-cli/src/commands/mcp.rs | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/crates/goose-cli/src/commands/mcp.rs b/crates/goose-cli/src/commands/mcp.rs
index 58b5f1925..9080024e1 100644
--- 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,
     };