test: jwt token

Author: vancauwe

README.md

@@ -99,7 +99,8 @@ the output should be: `{"data":{"type":"test","id":"1","attributes":{"message":"
 ```bash
 curl -X GET "http://localhost:8000/v1/inference/epfl/neo4j" \
   -H "accept: application/json" \
-  -H "Authorization: Bearer YOUR_TOKEN"
+  -H "Authorization: Bearer YOUR_TOKEN" \
+
 ```
 
 #### Inference

src/inference_api.py

@@ -23,14 +23,20 @@ def verify_jwt(credentials: HTTPAuthorizationCredentials = Depends(security)):
     print(token)
     header = jwt.get_unverified_header(token)
     print(header)
-    try:
-        claims = jwt.decode(token, JWT_SECRET_KEY, algorithms=[JWT_ALGORITHM])
-        if claims.get("service") != "airflow":
-            raise HTTPException(status_code=403, detail="Invalid service identity")
-        return claims
-    except JWTError as e:
-        print("JWT decode error:", e)
-        raise HTTPException(status_code=403, detail="Invalid token")
+    claims = jwt.decode(token, JWT_SECRET_KEY, algorithms=[JWT_ALGORITHM])
+    if claims.get("service") != "airflow":
+        raise HTTPException(status_code=403, detail="Invalid service identity")
+    else:
+        print("valid claims")
+    return claims
+    # try:
+    #     claims = jwt.decode(token, JWT_SECRET_KEY, algorithms=[JWT_ALGORITHM])
+    #     if claims.get("service") != "airflow":
+    #         raise HTTPException(status_code=403, detail="Invalid service identity")
+    #     return claims
+    # except JWTError as e:
+    #     print("JWT decode error:", e)
+    #     raise HTTPException(status_code=403, detail="Invalid token")
 
 
 @app.get("/v1/test")