From c5d64baba4ef772f72a311ad492328606c2ccc64 Mon Sep 17 00:00:00 2001 From: kurokobo Date: Sun, 23 Jun 2024 00:01:02 +0900 Subject: [PATCH] fix: correct typos (#5510) --- api/controllers/console/datasets/file.py | 4 ++-- api/services/file_service.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/api/controllers/console/datasets/file.py b/api/controllers/console/datasets/file.py index 23ab224731..c13bd45abb 100644 --- a/api/controllers/console/datasets/file.py +++ b/api/controllers/console/datasets/file.py @@ -14,7 +14,7 @@ from controllers.console.setup import setup_required from controllers.console.wraps import account_initialization_required, cloud_edition_billing_resource_check from fields.file_fields import file_fields, upload_config_fields from libs.login import login_required -from services.file_service import ALLOWED_EXTENSIONS, UNSTRUSTURED_ALLOWED_EXTENSIONS, FileService +from services.file_service import ALLOWED_EXTENSIONS, UNSTRUCTURED_ALLOWED_EXTENSIONS, FileService PREVIEW_WORDS_LIMIT = 3000 @@ -77,7 +77,7 @@ class FileSupportTypeApi(Resource): @account_initialization_required def get(self): etl_type = current_app.config['ETL_TYPE'] - allowed_extensions = UNSTRUSTURED_ALLOWED_EXTENSIONS if etl_type == 'Unstructured' else ALLOWED_EXTENSIONS + allowed_extensions = UNSTRUCTURED_ALLOWED_EXTENSIONS if etl_type == 'Unstructured' else ALLOWED_EXTENSIONS return {'allowed_extensions': allowed_extensions} diff --git a/api/services/file_service.py b/api/services/file_service.py index aad477f8a0..6c308a09df 100644 --- a/api/services/file_service.py +++ b/api/services/file_service.py @@ -21,7 +21,7 @@ IMAGE_EXTENSIONS = ['jpg', 'jpeg', 'png', 'webp', 'gif', 'svg'] IMAGE_EXTENSIONS.extend([ext.upper() for ext in IMAGE_EXTENSIONS]) ALLOWED_EXTENSIONS = ['txt', 'markdown', 'md', 'pdf', 'html', 'htm', 'xlsx', 'xls', 'docx', 'csv'] -UNSTRUSTURED_ALLOWED_EXTENSIONS = ['txt', 'markdown', 'md', 'pdf', 'html', 'htm', 'xlsx', 'xls', +UNSTRUCTURED_ALLOWED_EXTENSIONS = ['txt', 'markdown', 'md', 'pdf', 'html', 'htm', 'xlsx', 'xls', 'docx', 'csv', 'eml', 'msg', 'pptx', 'ppt', 'xml', 'epub'] PREVIEW_WORDS_LIMIT = 3000 @@ -36,7 +36,7 @@ class FileService: if len(filename) > 200: filename = filename.split('.')[0][:200] + '.' + extension etl_type = current_app.config['ETL_TYPE'] - allowed_extensions = UNSTRUSTURED_ALLOWED_EXTENSIONS + IMAGE_EXTENSIONS if etl_type == 'Unstructured' \ + allowed_extensions = UNSTRUCTURED_ALLOWED_EXTENSIONS + IMAGE_EXTENSIONS if etl_type == 'Unstructured' \ else ALLOWED_EXTENSIONS + IMAGE_EXTENSIONS if extension.lower() not in allowed_extensions: raise UnsupportedFileTypeError() @@ -139,7 +139,7 @@ class FileService: # extract text from file extension = upload_file.extension etl_type = current_app.config['ETL_TYPE'] - allowed_extensions = UNSTRUSTURED_ALLOWED_EXTENSIONS if etl_type == 'Unstructured' else ALLOWED_EXTENSIONS + allowed_extensions = UNSTRUCTURED_ALLOWED_EXTENSIONS if etl_type == 'Unstructured' else ALLOWED_EXTENSIONS if extension.lower() not in allowed_extensions: raise UnsupportedFileTypeError()