From c166f9628c7596e5386b58aef202cd054cd4e89b Mon Sep 17 00:00:00 2001
From: Jeffrey Tang <jeff@predibase.com>
Date: Fri, 23 Feb 2024 15:52:26 -0600
Subject: [PATCH] builds

---
 router/src/server.rs | 26 ++++++++++++++++----------
 1 file changed, 16 insertions(+), 10 deletions(-)

diff --git a/router/src/server.rs b/router/src/server.rs
index ec1400177..b2b9d6016 100644
--- a/router/src/server.rs
+++ b/router/src/server.rs
@@ -117,11 +117,14 @@ async fn completions_v1(
     let req = req.0;
     let mut gen_req = CompatGenerateRequest::from(req);
 
-    match headers.get("Authorization") {
-        Some(x) => x.strip_prefix("Bearer :").and_then(|token| gen_req.parameters.api_token = token)
-        // TODO: Just for testing, don't merge
-        None => println!("headers: {headers}")
-    }
+    headers.get("Authorization").map_or((), |x| {
+        x.to_str().map_or((), |y| {
+            y.strip_prefix("Bearer :").map_or((), |token| {
+                println!("token!! {token}");
+                gen_req.parameters.api_token = Some(token.to_string());
+            })
+        })
+    });
 
     // default return_full_text given the pipeline_tag
     if gen_req.parameters.return_full_text.is_none() {
@@ -184,11 +187,14 @@ async fn chat_completions_v1(
     let req = req.0;
     let mut gen_req = CompatGenerateRequest::from(req);
 
-    match headers.get("Authorization") {
-        Some(x) => x.strip_prefix("Bearer :").and_then(|token| gen_req.parameters.api_token = token)
-        // TODO: Just for testing, don't merge
-        None => println!("headers: {headers}")
-    }
+    headers.get("Authorization").map_or((), |x| {
+        x.to_str().map_or((), |y| {
+            y.strip_prefix("Bearer :").map_or((), |token| {
+                println!("token!! {token}");
+                gen_req.parameters.api_token = Some(token.to_string());
+            })
+        })
+    });
 
     // default return_full_text given the pipeline_tag
     if gen_req.parameters.return_full_text.is_none() {