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 import APIRouter, Depends, HTTPException, UploadFile
from fastapi.responses import StreamingResponse from fastapi.responses import StreamingResponse
from langflow.api.v1.schemas import UploadFileResponse
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 +19,7 @@ async def upload_file(flow_id: str, file: UploadFile, storage_service: StorageSe
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(flow_id=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 UploadFileResponse(flowId=flow_id, file_path=f"{folder}/{file_name}")
except Exception as e: except Exception as e:
raise HTTPException(status_code=500, detail=str(e)) raise HTTPException(status_code=500, detail=str(e))