diff --git a/src/bin/axum_docs.rs b/src/bin/axum_docs.rs index 89e33e6..0faac2f 100644 --- a/src/bin/axum_docs.rs +++ b/src/bin/axum_docs.rs @@ -14,7 +14,7 @@ use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt}; use anyhow::Result; use mcp_server::router::RouterService; -use rust_doc_server::{jsonrpc_frame_codec::JsonRpcFrameCodec, DocRouter}; +use cratedocs_mcp::{jsonrpc_frame_codec::JsonRpcFrameCodec, DocRouter}; use std::sync::Arc; use tokio::{ io::{self, AsyncWriteExt}, @@ -131,7 +131,7 @@ async fn sse_handler(State(app): State) -> Sse futures::future::ok(Event::default().event("message").data(message)), Err(e) => futures::future::err(io::Error::new(io::ErrorKind::InvalidData, e)), }), diff --git a/src/bin/doc_server.rs b/src/bin/doc_server.rs index 95bc653..51ba1a3 100644 --- a/src/bin/doc_server.rs +++ b/src/bin/doc_server.rs @@ -1,7 +1,7 @@ use anyhow::Result; use mcp_server::router::RouterService; use mcp_server::{ByteTransport, Server}; -use rust_doc_server::DocRouter; +use cratedocs_mcp::DocRouter; use tokio::io::{stdin, stdout}; use tracing_appender::rolling::{RollingFileAppender, Rotation}; use tracing_subscriber::{self, EnvFilter}; diff --git a/src/docs.rs b/src/docs.rs index 305d13f..52b206a 100644 --- a/src/docs.rs +++ b/src/docs.rs @@ -41,6 +41,12 @@ pub struct DocRouter { cache: DocCache, } +impl Default for DocRouter { + fn default() -> Self { + Self::new() + } +} + impl DocRouter { pub fn new() -> Self { Self {