feat: Add optional CORS allow any option value in http server cli (#260)

huggingface · Jun 28, 2024 · 970922e · 970922e
1 parent 0361479
commit 970922e
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/router/src/http/server.rs b/router/src/http/server.rs
@@ -1534,11 +1534,15 @@ pub async fn run(
     // map to go inside the option and then map to parse from String to HeaderValue
     // Finally, convert to AllowOrigin
     let allow_origin: Option<AllowOrigin> = cors_allow_origin.map(|cors_allow_origin| {
-        AllowOrigin::list(
-            cors_allow_origin
-                .into_iter()
-                .map(|origin| origin.parse::<HeaderValue>().unwrap()),
-        )
+        if cors_allow_origin.iter().any(|origin| origin == "*") {
+            AllowOrigin::any()
+        } else {
+            AllowOrigin::list(
+                cors_allow_origin
+                    .into_iter()
+                    .map(|origin| origin.parse::<HeaderValue>().unwrap()),
+            )
+        }
     });
 
     let prom_handle = prom_builder