vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Implement upload_file endpoint (#548)

Browse Source
This commit is contained in:
Kerem Yilmaz
2024-07-03 18:07:02 -07:00
committed by GitHub
parent 21b9eea446
commit 0a11e225d6
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
skyvern/forge/sdk/routes/agent_protocol.py
View File

@@ -753,8 +753,14 @@ async def update_organization(
async def validate_file_size(file: UploadFile) -> UploadFile: async def validate_file_size(file: UploadFile) -> UploadFile:
# Check the file size try:
if file.size > app.SETTINGS_MANAGER.MAX_UPLOAD_FILE_SIZE: file.file.seek(0, 2) # Move the pointer to the end of the file
size = file.file.tell() # Get the current position of the pointer, which represents the file size
file.file.seek(0) # Reset the pointer back to the beginning
except Exception as e:
raise HTTPException(status_code=500, detail="Could not determine file size.") from e
if size > app.SETTINGS_MANAGER.MAX_FILE_SIZE:
raise HTTPException( raise HTTPException(
status_code=413, status_code=413,
detail=f"File size exceeds the maximum allowed size ({app.SETTINGS_MANAGER.MAX_UPLOAD_FILE_SIZE} bytes)", detail=f"File size exceeds the maximum allowed size ({app.SETTINGS_MANAGER.MAX_UPLOAD_FILE_SIZE} bytes)",