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

Refactor file upload and download functions

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-27 23:26:49 -03:00
commit b28da3f71a
1 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/backend/langflow/api/v1/files.py
View file

@ -4,7 +4,6 @@ from io import BytesIO
from fastapi import APIRouter, Depends, HTTPException, UploadFile from fastapi import APIRouter, Depends, HTTPException, UploadFile
from fastapi.responses import StreamingResponse from fastapi.responses import StreamingResponse
from langflow.services.deps import get_storage_service from langflow.services.deps import get_storage_service
from langflow.services.storage.service import StorageService from langflow.services.storage.service import StorageService
from langflow.services.storage.utils import build_content_type_from_extension from langflow.services.storage.utils import build_content_type_from_extension
@ -18,7 +17,7 @@ async def upload_file(flow_id: str, file: UploadFile, storage_service: StorageSe
file_content = await file.read() file_content = await file.read()
file_name = file.filename or hashlib.sha256(file_content).hexdigest() file_name = file.filename or hashlib.sha256(file_content).hexdigest()
folder = flow_id folder = flow_id
storage_service.save_file(folder=folder, file_name=file_name, data=file_content) storage_service.save_file(flow_id=folder, file_name=file_name, data=file_content)
return {"message": "File uploaded successfully", "file_path": f"{folder}/{file.filename}"} return {"message": "File uploaded successfully", "file_path": f"{folder}/{file.filename}"}
except Exception as e: except Exception as e:
raise HTTPException(status_code=500, detail=str(e)) raise HTTPException(status_code=500, detail=str(e))
@ -37,7 +36,7 @@ async def download_file(flow_id: str, file_name: str, storage_service: StorageSe
if not content_type: if not content_type:
raise HTTPException(status_code=500, detail=f"Content type not found for extension {extension}") raise HTTPException(status_code=500, detail=f"Content type not found for extension {extension}")
file_content = storage_service.get_file(folder=flow_id, file_name=file_name) file_content = storage_service.get_file(flow_id=flow_id, file_name=file_name)
return StreamingResponse(BytesIO(file_content), media_type=content_type) return StreamingResponse(BytesIO(file_content), media_type=content_type)
except Exception as e: except Exception as e:
raise HTTPException(status_code=500, detail=str(e)) raise HTTPException(status_code=500, detail=str(e))
@ -46,7 +45,7 @@ async def download_file(flow_id: str, file_name: str, storage_service: StorageSe
@router.get("/list/{flow_id}") @router.get("/list/{flow_id}")
async def list_files(flow_id: str, storage_service: StorageService = Depends(get_storage_service)): async def list_files(flow_id: str, storage_service: StorageService = Depends(get_storage_service)):
try: try:
files = storage_service.list_files(folder=flow_id) files = storage_service.list_files(flow_id=flow_id)
return {"files": files} return {"files": files}
except Exception as e: except Exception as e:
raise HTTPException(status_code=500, detail=str(e)) raise HTTPException(status_code=500, detail=str(e))
@ -55,7 +54,7 @@ async def list_files(flow_id: str, storage_service: StorageService = Depends(get
@router.delete("/delete/{flow_id}/{file_name}") @router.delete("/delete/{flow_id}/{file_name}")
async def delete_file(flow_id: str, file_name: str, storage_service: StorageService = Depends(get_storage_service)): async def delete_file(flow_id: str, file_name: str, storage_service: StorageService = Depends(get_storage_service)):
try: try:
storage_service.delete_file(folder=flow_id, file_name=file_name) storage_service.delete_file(flow_id=flow_id, file_name=file_name)
return {"message": f"File {file_name} deleted successfully"} return {"message": f"File {file_name} deleted successfully"}
except Exception as e: except Exception as e:
raise HTTPException(status_code=500, detail=str(e)) raise HTTPException(status_code=500, detail=str(e))