upload dir should respect langflow flag
This commit is contained in:
parent
38e3007964
commit
f761eab1b4
2 changed files with 25 additions and 8 deletions
|
|
@ -45,7 +45,7 @@ async def upload(request: Request, document_service, session_manager):
|
||||||
return JSONResponse({"error": error_msg}, status_code=500)
|
return JSONResponse({"error": error_msg}, status_code=500)
|
||||||
|
|
||||||
|
|
||||||
async def upload_path(request: Request, task_service, session_manager):
|
async def upload_path(request: Request, task_service, session_manager, langflow_file_service):
|
||||||
"""Upload all files from a directory path"""
|
"""Upload all files from a directory path"""
|
||||||
payload = await request.json()
|
payload = await request.json()
|
||||||
base_dir = payload.get("path")
|
base_dir = payload.get("path")
|
||||||
|
|
@ -74,13 +74,29 @@ async def upload_path(request: Request, task_service, session_manager):
|
||||||
owner_name = user.name
|
owner_name = user.name
|
||||||
owner_email = user.email
|
owner_email = user.email
|
||||||
|
|
||||||
task_id = await task_service.create_upload_task(
|
from config.settings import DISABLE_INGEST_WITH_LANGFLOW
|
||||||
owner_user_id,
|
|
||||||
file_paths,
|
# Use same logic as single file uploads - respect the Langflow setting
|
||||||
jwt_token=jwt_token,
|
if DISABLE_INGEST_WITH_LANGFLOW:
|
||||||
owner_name=owner_name,
|
# Use direct DocumentFileProcessor (no Langflow)
|
||||||
owner_email=owner_email,
|
task_id = await task_service.create_upload_task(
|
||||||
)
|
owner_user_id,
|
||||||
|
file_paths,
|
||||||
|
jwt_token=jwt_token,
|
||||||
|
owner_name=owner_name,
|
||||||
|
owner_email=owner_email,
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
# Use Langflow pipeline for processing
|
||||||
|
task_id = await task_service.create_langflow_upload_task(
|
||||||
|
user_id=owner_user_id,
|
||||||
|
file_paths=file_paths,
|
||||||
|
langflow_file_service=langflow_file_service,
|
||||||
|
session_manager=session_manager,
|
||||||
|
jwt_token=jwt_token,
|
||||||
|
owner_name=owner_name,
|
||||||
|
owner_email=owner_email,
|
||||||
|
)
|
||||||
|
|
||||||
return JSONResponse(
|
return JSONResponse(
|
||||||
{"task_id": task_id, "total_files": len(file_paths), "status": "accepted"},
|
{"task_id": task_id, "total_files": len(file_paths), "status": "accepted"},
|
||||||
|
|
|
||||||
|
|
@ -558,6 +558,7 @@ async def create_app():
|
||||||
upload.upload_path,
|
upload.upload_path,
|
||||||
task_service=services["task_service"],
|
task_service=services["task_service"],
|
||||||
session_manager=services["session_manager"],
|
session_manager=services["session_manager"],
|
||||||
|
langflow_file_service=services["langflow_file_service"],
|
||||||
)
|
)
|
||||||
),
|
),
|
||||||
methods=["POST"],
|
methods=["POST"],
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue