From 27cb5ac23839c53f92471d2efd9fe752d73d1c60 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Mon, 17 Jun 2024 18:19:11 -0300 Subject: [PATCH] chore: Update JWT decoding error handling in get_current_user_by_jwt function --- src/backend/base/langflow/services/auth/utils.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/src/backend/base/langflow/services/auth/utils.py b/src/backend/base/langflow/services/auth/utils.py index d0ece7562..49a1e4720 100644 --- a/src/backend/base/langflow/services/auth/utils.py +++ b/src/backend/base/langflow/services/auth/utils.py @@ -7,15 +7,14 @@ from cryptography.fernet import Fernet from fastapi import Depends, HTTPException, Security, status from fastapi.security import APIKeyHeader, APIKeyQuery, OAuth2PasswordBearer from jose import JWTError, jwt -from loguru import logger -from sqlmodel import Session -from starlette.websockets import WebSocket - from langflow.services.database.models.api_key.crud import check_key from langflow.services.database.models.api_key.model import ApiKey from langflow.services.database.models.user.crud import get_user_by_id, get_user_by_username, update_user_last_login_at from langflow.services.database.models.user.model import User from langflow.services.deps import get_session, get_settings_service +from loguru import logger +from sqlmodel import Session +from starlette.websockets import WebSocket oauth2_login = OAuth2PasswordBearer(tokenUrl="api/v1/login", auto_error=False) @@ -129,6 +128,7 @@ async def get_current_user_by_jwt( ) except JWTError as e: logger.error(f"JWT decoding error: {e}") + logger.exception(e) raise HTTPException( status_code=status.HTTP_401_UNAUTHORIZED, detail="Could not validate credentials", @@ -359,4 +359,3 @@ def decrypt_api_key(encrypted_api_key: str, settings_service=Depends(get_setting encoded_bytes = encrypted_api_key decrypted_key = fernet.decrypt(encoded_bytes).decode() return decrypted_key - return decrypted_key