jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

🔧 chore(login.py): import get_current_active_user function to improve code readability and maintainability

Browse source 
✨ feat(login.py): add current_user dependency to refresh_token endpoint to ensure only authenticated users can refresh their tokens
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-25 16:29:24 -03:00
commit 89ea8e45f6
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/langflow/api/v1/login.py
View file

@ -9,6 +9,7 @@ from langflow.services.auth.utils import (
create_user_tokens, create_user_tokens,
create_refresh_token, create_refresh_token,
create_user_longterm_token, create_user_longterm_token,
get_current_active_user,
) )
from langflow.services.utils import get_settings_manager from langflow.services.utils import get_settings_manager
@ -49,7 +50,9 @@ async def auto_login(db: Session = Depends(get_session)):
@router.post("/refresh") @router.post("/refresh")
async def refresh_token(token: str): async def refresh_token(
token: str, current_user: Session = Depends(get_current_active_user)
):
if token: if token:
return create_refresh_token(token) return create_refresh_token(token)
else: else: