fix: boost to support bearer auth

av · Sep 25, 2024 · fbe9ff4 · fbe9ff4
1 parent 0062761
commit fbe9ff4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/boost/src/main.py b/boost/src/main.py
@@ -22,8 +22,10 @@
 async def get_api_key(api_key_header: str = Security(auth_header)):
   if len(BOOST_AUTH) == 0:
     return
-  if api_key_header in BOOST_AUTH:
-    return api_key_header
+  # Bearer/plain versions
+  value = api_key_header.replace("Bearer ", "").replace("bearer ", "")
+  if value in BOOST_AUTH:
+    return value
   raise HTTPException(status_code=403, detail="Unauthorized")
 
 

diff --git a/http-catalog/boost.http b/http-catalog/boost.http
@@ -12,7 +12,7 @@ GET {{host}}/v1/models
 
 POST {{host}}/v1/chat/completions
 Content-Type: application/json
-Authorization: sk-fake
+Authorization: Bearer sk-boost
 
 {
   "model": "klmbr-llama3.1:8b",
@@ -30,7 +30,7 @@ Authorization: sk-fake
 
 POST {{host}}/v1/chat/completions
 Content-Type: application/json
-Authorization: sk-fake
+Authorization: sk-boost
 
 {
   "model": "klmbr-llama3.1:8b",