Fix test failure

playbooks-ai · Jan 2, 2025 · f33630c · f33630c
1 parent 3fafcfa
commit f33630c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/website/api/session.py b/website/api/session.py
@@ -3,7 +3,7 @@
 import secrets
 from base64 import b64encode
 
-from cryptography.fernet import Fernet
+from cryptography.fernet import Fernet, InvalidToken
 from database import Base, SessionLocal, engine
 from fastapi import Request, Response
 from models import UserSession
@@ -71,7 +71,7 @@ async def get_or_create_session(self, request: Request, db) -> UserSession:
                 except Exception as e:
                     print(f"Database error: {e}")
                     # Fall through to create new session
-            except (json.JSONDecodeError, ValueError) as e:
+            except (json.JSONDecodeError, ValueError, InvalidToken) as e:
                 print(f"Session decoding error: {e}")
                 pass