fix: axum middlewares only handles routes that are registered

TabbyML · Nov 23, 2023 · 212f194 · 212f194
1 parent b8e0a63
commit 212f194
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/crates/tabby/src/serve.rs b/crates/tabby/src/serve.rs
@@ -2,6 +2,7 @@ use std::{sync::Arc, time::Duration};
 
 use axum::{routing, Router};
 use clap::Args;
+use hyper::StatusCode;
 use tabby_common::{
     api,
     api::{code::CodeSearch, event::EventLogger},
@@ -192,6 +193,13 @@ async fn api_router(
                     config.server.completion_timeout,
                 )))
         });
+    } else {
+        routers.push({
+            Router::new().route(
+                "/v1/completions",
+                routing::post(StatusCode::NOT_IMPLEMENTED),
+            )
+        })
     }
 
     if let Some(chat_state) = chat_state {
@@ -201,6 +209,13 @@ async fn api_router(
                 routing::post(routes::chat_completions).with_state(chat_state),
             )
         })
+    } else {
+        routers.push({
+            Router::new().route(
+                "/v1beta/chat/completions",
+                routing::post(StatusCode::NOT_IMPLEMENTED),
+            )
+        })
     }
 
     routers.push({