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

Add UploadFileResponse to file upload API

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-27 23:35:30 -03:00
commit 9d808cc55d
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -4,6 +4,7 @@ from io import BytesIO
from fastapi import APIRouter, Depends, HTTPException, UploadFile
from fastapi.responses import StreamingResponse
from langflow.api.v1.schemas import UploadFileResponse
from langflow.services.deps import get_storage_service
from langflow.services.storage.service import StorageService
from langflow.services.storage.utils import build_content_type_from_extension
@ -18,7 +19,7 @@ async def upload_file(flow_id: str, file: UploadFile, storage_service: StorageSe
file_name = file.filename or hashlib.sha256(file_content).hexdigest()
folder = flow_id
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 UploadFileResponse(flowId=flow_id, file_path=f"{folder}/{file_name}")
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))