feat: allow Bearer as valid token type on model service authenticat…

…ion (#2583)
lablup · Jul 31, 2024 · fb196c8 · fb196c8
1 parent 8adbc7c
commit fb196c8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/changes/2583.feature.md b/changes/2583.feature.md
@@ -0,0 +1 @@
+Allow `Bearer` as valid token type on model service authentication
diff --git a/src/ai/backend/wsproxy/proxy/frontend/http/abc.py b/src/ai/backend/wsproxy/proxy/frontend/http/abc.py
@@ -48,7 +48,7 @@ def ensure_credential(self, request: web.Request, circuit: Circuit) -> None:
                 if not auth_header:
                     raise InvalidCredentials("E20006: Authorization header not provided")
                 auth_type, auth_key = auth_header.split(" ", maxsplit=2)
-                if auth_type == "BackendAI":
+                if auth_type in ("BackendAI", "Bearer"):
                     token = auth_key
                 else:
                     raise InvalidCredentials(
@@ -64,7 +64,7 @@ def ensure_credential(self, request: web.Request, circuit: Circuit) -> None:
                 except jwt.PyJWTError as e:
                     raise InvalidCredentials from e
 
-                if decoded.get("id") != circuit.id:
+                if decoded.get("id") != str(circuit.id):
                     raise InvalidCredentials("E20008: Authorization token mismatch")
 
     async def initialize_backend(self, circuit: Circuit, routes: list[RouteInfo]) -> HTTPBackend:
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		Allow `Bearer` as valid token type on model service authentication