AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 03:39:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix file type not support when preview (#1841)

Browse Source 
Co-authored-by: jyong <jyong@dify.ai>
This commit is contained in:
Jyong 2023-12-26 15:06:44 +08:00 committed by GitHub
parent 9080ece3fb
commit 08494058e9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/services/file_service.py
View File

@ -133,7 +133,15 @@ class FileService:
# extract text from file # extract text from file
extension = upload_file.extension extension = upload_file.extension
if extension.lower() not in ALLOWED_EXTENSIONS: etl_type = current_app.config['ETL_TYPE']
if etl_type == 'Unstructured':
allowed_extensions = ['txt', 'markdown', 'md', 'pdf', 'html', 'htm', 'xlsx',
'docx', 'csv', 'eml', 'msg', 'pptx', 'ppt', 'xml',
'jpg', 'jpeg', 'png', 'webp', 'gif', 'svg']
else:
allowed_extensions = ['txt', 'markdown', 'md', 'pdf', 'html', 'htm', 'xlsx', 'docx', 'csv',
'jpg', 'jpeg', 'png', 'webp', 'gif', 'svg']
if extension.lower() not in allowed_extensions:
raise UnsupportedFileTypeError() raise UnsupportedFileTypeError()
text = FileExtractor.load(upload_file, return_text=True) text = FileExtractor.load(upload_file, return_text=True)