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

Update to langflow service to user logger correctly

Browse source
This commit is contained in:
Edwin Jose 2025-09-08 12:13:02 -04:00
parent 5dd9959dd0
commit dcd44017ce
1 changed files with 22 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
src/services/langflow_file_service.py
View file

@ -142,18 +142,20 @@ class LangflowFileService:
Returns: Returns:
Combined result with upload info, ingestion result, and deletion status Combined result with upload info, ingestion result, and deletion status
""" """
self.logger.debug("[LF] Starting combined upload and ingest operation") logger.debug("[LF] Starting combined upload and ingest operation")
# Step 1: Upload the file # Step 1: Upload the file
try: try:
upload_result = await self.upload_user_file(file_tuple, jwt_token=jwt_token) upload_result = await self.upload_user_file(file_tuple, jwt_token=jwt_token)
self.logger.debug( logger.debug(
"[LF] Upload completed successfully", "[LF] Upload completed successfully",
file_id=upload_result.get("id"), extra={
file_path=upload_result.get("path"), "file_id": upload_result.get("id"),
"file_path": upload_result.get("path"),
}
) )
except Exception as e: except Exception as e:
self.logger.error("[LF] Upload failed during combined operation", error=str(e)) logger.error("[LF] Upload failed during combined operation", extra={"error": str(e)})
raise Exception(f"Upload failed: {str(e)}") raise Exception(f"Upload failed: {str(e)}")
# Step 2: Prepare for ingestion # Step 2: Prepare for ingestion
@ -165,7 +167,7 @@ class LangflowFileService:
final_tweaks = tweaks.copy() if tweaks else {} final_tweaks = tweaks.copy() if tweaks else {}
if settings: if settings:
self.logger.debug("[LF] Applying ingestion settings", settings=settings) logger.debug("[LF] Applying ingestion settings", extra={"settings": settings})
# Split Text component tweaks (SplitText-QIKhg) # Split Text component tweaks (SplitText-QIKhg)
if ( if (
@ -190,7 +192,7 @@ class LangflowFileService:
final_tweaks["OpenAIEmbeddings-joRJ6"] = {} final_tweaks["OpenAIEmbeddings-joRJ6"] = {}
final_tweaks["OpenAIEmbeddings-joRJ6"]["model"] = settings["embeddingModel"] final_tweaks["OpenAIEmbeddings-joRJ6"]["model"] = settings["embeddingModel"]
self.logger.debug("[LF] Final tweaks with settings applied", tweaks=final_tweaks) logger.debug("[LF] Final tweaks with settings applied", extra={"tweaks": final_tweaks})
# Step 3: Run ingestion # Step 3: Run ingestion
try: try:
@ -200,12 +202,14 @@ class LangflowFileService:
tweaks=final_tweaks, tweaks=final_tweaks,
jwt_token=jwt_token, jwt_token=jwt_token,
) )
self.logger.debug("[LF] Ingestion completed successfully") logger.debug("[LF] Ingestion completed successfully")
except Exception as e: except Exception as e:
self.logger.error( logger.error(
"[LF] Ingestion failed during combined operation", "[LF] Ingestion failed during combined operation",
error=str(e), extra={
file_path=file_path "error": str(e),
"file_path": file_path
}
) )
# Note: We could optionally delete the uploaded file here if ingestion fails # Note: We could optionally delete the uploaded file here if ingestion fails
raise Exception(f"Ingestion failed: {str(e)}") raise Exception(f"Ingestion failed: {str(e)}")
@ -217,16 +221,18 @@ class LangflowFileService:
if delete_after_ingest and file_id: if delete_after_ingest and file_id:
try: try:
self.logger.debug("[LF] Deleting file after successful ingestion", file_id=file_id) logger.debug("[LF] Deleting file after successful ingestion", extra={"file_id": file_id})
await self.delete_user_file(file_id) await self.delete_user_file(file_id)
delete_result = {"status": "deleted", "file_id": file_id} delete_result = {"status": "deleted", "file_id": file_id}
self.logger.debug("[LF] File deleted successfully") logger.debug("[LF] File deleted successfully")
except Exception as e: except Exception as e:
delete_error = str(e) delete_error = str(e)
self.logger.warning( logger.warning(
"[LF] Failed to delete file after ingestion", "[LF] Failed to delete file after ingestion",
error=delete_error, extra={
file_id=file_id "error": delete_error,
"file_id": file_id
}
) )
delete_result = {"status": "delete_failed", "file_id": file_id, "error": delete_error} delete_result = {"status": "delete_failed", "file_id": file_id, "error": delete_error}