瀏覽代碼

fix file type not support when preview (#1841)

Co-authored-by: jyong <jyong@dify.ai>
Jyong 1 年之前
父節點
當前提交
08494058e9
共有 1 個文件被更改,包括 9 次插入1 次删除
  1. 9 1
      api/services/file_service.py

+ 9 - 1
api/services/file_service.py

@@ -133,7 +133,15 @@ class FileService:
 
         # extract text from file
         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()
 
         text = FileExtractor.load(upload_file, return_text=True)