[owl] Make Bearer Auth case-insensitive (#811)

API server: owl

- Make Bearer Auth case-insensitive
- Bug fix:  messages

Author: jiahuei

services/api/src/owl/utils/handlers.py

@@ -87,8 +87,10 @@ def make_response(
     headers["x-request-id"] = request.headers.get("x-request-id", "")
     request_headers = {k.lower(): v for k, v in request.headers.items()}
     token = request_headers.get("authorization", "")
-    if token.startswith("Bearer "):
-        request_headers["authorization"] = f"Bearer {mask_string(token[7:], include_len=False)}"
+    if token.lower().startswith("bearer "):
+        request_headers["authorization"] = (
+            f"{token[:6]} {mask_string(token[7:], include_len=False)}"
+        )
     else:
         request_headers["authorization"] = mask_string(token, include_len=False)
     response = ORJSONResponse(

services/api/src/owl/utils/lm.py

@@ -234,7 +234,7 @@ def _map_and_log_exception(
         messages: list[dict] | None = None,
         **hyperparams,
     ) -> Exception:
-        messages = [mask_content(m) for m in messages]
+        messages = [mask_content(m) for m in messages] if messages else None
         err_mssg = getattr(e, "message", str(e))
         if isinstance(e, JamaiException):
             return e

services/api/tests/routers/test_serving.py

@@ -310,6 +310,15 @@ def test_model_info(setup: ServingContext):
     assert isinstance(response, ModelInfoListResponse)
     assert len(response.data) == 4
 
+    # Test Bearer Auth case-insensitivity
+    response = JamAI(
+        user_id=setup.user_id,
+        project_id=setup.project_ids[0],
+        headers={"Authorization": f"beaRER    {ENV_CONFIG.service_key_plain}"},
+    ).model_info()
+    assert isinstance(response, ModelInfoListResponse)
+    assert len(response.data) == 4
+
     response = client.model_info(model=chat_model_id)
     assert len(response.data) == 1
     assert response.data[0].id == chat_model_id