Fix log out url part

Author: PaulAsjes

workos/session.py

@@ -30,6 +30,7 @@ def __init__(
 
         self.jwks = PyJWKClient(self.user_management.get_jwks_url())
 
+        # Algorithms are hardcoded for security reasons. See https://pyjwt.readthedocs.io/en/stable/algorithms.html#specifying-an-algorithm
         self.jwk_algorithms = ['RS256']
 
     def authenticate(
@@ -119,20 +120,19 @@ def refresh(self, options: Optional[Dict[str, Any]] = None) -> Union[
     def get_logout_url(self) -> str:
         auth_response = self.authenticate()
 
-        if not auth_response["authenticated"]:
-            raise ValueError(auth_response["reason"])
+        if not auth_response.authenticated:
+            raise ValueError(f"Failed to extract session ID for logout URL: {auth_response.reason}")
 
         return self.user_management.get_logout_url(
-            session_id=auth_response["session_id"]
+            session_id=auth_response.session_id
         )
 
     def is_valid_jwt(self, token: str) -> bool:
         try:
             signing_key = self.jwks.get_signing_key_from_jwt(token)
             jwt.decode(token, signing_key.key, algorithms=self.jwk_algorithms)
             return True
-        except jwt.exceptions.InvalidTokenError as error:
-            print("invalid token", error)
+        except jwt.exceptions.InvalidTokenError:
             return False
 
     @staticmethod

workos/types/user_management/session.py

@@ -18,6 +18,7 @@ class AuthenticateWithSessionCookieSuccessResponse(WorkOSModel):
     permissions: Optional[List[str]] = None
     user: User
     impersonator: Optional[Impersonator] = None
+    entitlements: Optional[List[str]] = None
 
 class AuthenticateWithSessionCookieErrorResponse(WorkOSModel):
     authenticated: bool = False