diff --git a/src/modes/server.rs b/src/modes/server.rs
index 1128e36..aaa9a4a 100644
--- a/src/modes/server.rs
+++ b/src/modes/server.rs
@@ -59,7 +59,7 @@ async fn run_server(
         args: Arc::new(args.clone()),
     };
     
-    let app = api::api_router()
+    let app = Router::new()
         .layer(axum::middleware::from_fn(logging_middleware))
         .layer(
             ServiceBuilder::new()
diff --git a/src/modes/server/docs.rs b/src/modes/server/docs.rs
index 04f4511..cdc04e9 100644
--- a/src/modes/server/docs.rs
+++ b/src/modes/server/docs.rs
@@ -81,3 +81,10 @@ pub async fn handle_swagger_ui() -> Html<&'static str> {
     
     Html(html)
 }
+
+pub fn add_routes(router: Router<AppState>) -> Router<AppState> {
+    router
+        // Documentation endpoints
+        .route("/openapi.json", get(docs::handle_openapi))
+        .route("/swagger/", get(docs::handle_swagger_ui))
+}