fix(router): payload limit was not correctly applied

huggingface · Jun 21, 2024 · 9e2efa7 · 9e2efa7
1 parent dbefcd8
commit 9e2efa7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/router/src/http/server.rs b/router/src/http/server.rs
@@ -1451,7 +1451,6 @@ pub async fn run(
 
     // Create router
     let mut app = Router::new()
-        .layer(DefaultBodyLimit::max(payload_limit))
         .merge(SwaggerUi::new("/docs").url("/api-doc/openapi.json", doc))
         // Base routes
         .route("/info", get(get_model_info))
@@ -1474,7 +1473,9 @@ pub async fn run(
         // AWS Sagemaker health route
         .route("/ping", get(health))
         // Prometheus metrics route
-        .route("/metrics", get(metrics));
+        .route("/metrics", get(metrics))
+        // Update payload limit
+        .layer(DefaultBodyLimit::max(payload_limit));
 
     #[cfg(feature = "google")]
     {