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

pass the oauth2 token to the fastapi Security

Browse source
This commit is contained in:
Maryam Abdoli 2023-11-02 16:35:00 -04:00
commit 5686a6fd63
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/langflow/services/auth/utils.py
View file

@ -70,13 +70,12 @@ async def api_key_security(
async def get_current_user( async def get_current_user(
request: Request, token: str = Security(oauth2_login),
query_param: str = Security(api_key_query), query_param: str = Security(api_key_query),
header_param: str = Security(api_key_header), header_param: str = Security(api_key_header),
db: Session = Depends(get_session), db: Session = Depends(get_session),
) -> User: ) -> User:
try: try:
token = oauth2_login(request)
return await get_current_user_by_jwt(token, db) return await get_current_user_by_jwt(token, db)
except HTTPException: except HTTPException:
user = await api_key_security(query_param, header_param, db) user = await api_key_security(query_param, header_param, db)