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

Refactor file download functions

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-29 14:42:51 -03:00
commit 37d48faf16
1 changed files with 2 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -49,9 +49,7 @@ async def upload_file(
@router.get("/download/{flow_id}/{file_name}") @router.get("/download/{flow_id}/{file_name}")
async def download_file( async def download_file(file_name: str, flow_id: str, storage_service: StorageService = Depends(get_storage_service)):
file_name: str, flow_id: str = Depends(get_flow_id), storage_service: StorageService = Depends(get_storage_service)
):
try: try:
extension = file_name.split(".")[-1] extension = file_name.split(".")[-1]
@ -75,9 +73,7 @@ async def download_file(
@router.get("/images/{flow_id}/{file_name}") @router.get("/images/{flow_id}/{file_name}")
async def download_image( async def download_image(file_name: str, flow_id: str, storage_service: StorageService = Depends(get_storage_service)):
file_name: str, flow_id: str = Depends(get_flow_id), storage_service: StorageService = Depends(get_storage_service)
):
try: try:
extension = file_name.split(".")[-1] extension = file_name.split(".")[-1]