gmakstutis/openrag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Refactor user file upload and ingestion flow to streamline JWT token handling

Browse source 
This commit simplifies the handling of the JWT token in the upload_user_file and run_ingestion functions by removing unnecessary lines and ensuring the token is passed correctly to downstream services. This change enhances code readability and maintains the focus on robust async coding practices and well-documented code.
This commit is contained in:
Gabriel Luiz Freitas Almeida 2025-09-08 08:43:53 -03:00
parent 7b5589653a
commit deb39a6e5b
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
src/api/langflow_files.py
View file

@ -34,9 +34,7 @@ async def upload_user_file(
logger.debug("JWT token status", jwt_present=jwt_token is not None)
logger.debug("Calling langflow_file_service.upload_user_file")
result = await langflow_file_service.upload_user_file(
file_tuple, jwt_token=jwt_token
)
result = await langflow_file_service.upload_user_file(file_tuple, jwt_token)
logger.debug("Upload successful", result=result)
return JSONResponse(result, status_code=201)
except Exception as e:
@ -99,15 +97,20 @@ async def run_ingestion(
# (search parameters are for retrieval, not document processing)
logger.debug("Final tweaks with settings applied", tweaks=tweaks)
# Include user JWT if available
jwt_token = getattr(request.state, "jwt_token", None)
if jwt_token:
# Set auth context for downstream services
from auth_context import set_auth_context
user_id = getattr(request.state, "user_id", None)
set_auth_context(user_id, jwt_token)
result = await langflow_file_service.run_ingestion_flow(
file_paths=file_paths or [],
jwt_token=jwt_token,
session_id=session_id,
tweaks=tweaks,
jwt_token=jwt_token,
)
return JSONResponse(result)
except Exception as e: