UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-06 16:59:42 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Merge pull request #6109 from JoeyShapiro/dev

Browse Source 
fix: disallow empty files
This commit is contained in:
Timothy Jaeryang Baek 2024-10-20 20:41:56 -07:00 committed by GitHub
commit de1e29eed0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/open_webui/apps/webui/routers/files.py
View File

@ -47,6 +47,12 @@ def upload_file(file: UploadFile = File(...), user=Depends(get_verified_user)):
file_path = f"{UPLOAD_DIR}/{filename}" file_path = f"{UPLOAD_DIR}/{filename}"
contents = file.file.read() contents = file.file.read()
if len(contents) == 0:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.EMPTY_CONTENT,
)
with open(file_path, "wb") as f: with open(file_path, "wb") as f:
f.write(contents) f.write(contents)
f.close() f.close()

5
src/lib/components/workspace/Knowledge/Collection.svelte
View File

@ -124,6 +124,11 @@
itemId: tempItemId itemId: tempItemId
}; };
if (fileItem.size == 0) {
toast.error($i18n.t('File must have content to be uploaded'));
return null;
}
knowledge.files = [...(knowledge.files ?? []), fileItem]; knowledge.files = [...(knowledge.files ?? []), fileItem];
// Check if the file is an audio file and transcribe/convert it to text file // Check if the file is an audio file and transcribe/convert it to text file