From 98f2e2c7298200acc80ea00655660b12479a5d19 Mon Sep 17 00:00:00 2001 From: Yongtao Huang Date: Thu, 27 Mar 2025 15:42:12 +0800 Subject: [PATCH] Fix wrong allowed extensions (#16893) --- api/core/file/upload_file_parser.py | 4 +--- api/services/audio_service.py | 4 ++-- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/api/core/file/upload_file_parser.py b/api/core/file/upload_file_parser.py index 062a0b6d22..1f81d351b6 100644 --- a/api/core/file/upload_file_parser.py +++ b/api/core/file/upload_file_parser.py @@ -4,12 +4,10 @@ import time from typing import Optional from configs import dify_config +from constants import IMAGE_EXTENSIONS from core.helper.url_signer import UrlSigner from extensions.ext_storage import storage -IMAGE_EXTENSIONS = ["jpg", "jpeg", "png", "webp", "gif", "svg"] -IMAGE_EXTENSIONS.extend([ext.upper() for ext in IMAGE_EXTENSIONS]) - class UploadFileParser: @classmethod diff --git a/api/services/audio_service.py b/api/services/audio_service.py index 294dfe4c8c..a259f5a4c4 100644 --- a/api/services/audio_service.py +++ b/api/services/audio_service.py @@ -5,6 +5,7 @@ from typing import Optional from werkzeug.datastructures import FileStorage +from constants import AUDIO_EXTENSIONS from core.model_manager import ModelManager from core.model_runtime.entities.model_entities import ModelType from models.model import App, AppMode, AppModelConfig, Message @@ -18,7 +19,6 @@ from services.errors.audio import ( FILE_SIZE = 30 FILE_SIZE_LIMIT = FILE_SIZE * 1024 * 1024 -ALLOWED_EXTENSIONS = ["mp3", "mp4", "mpeg", "mpga", "m4a", "wav", "webm", "amr"] logger = logging.getLogger(__name__) @@ -44,7 +44,7 @@ class AudioService: raise NoAudioUploadedServiceError() extension = file.mimetype - if extension not in [f"audio/{ext}" for ext in ALLOWED_EXTENSIONS]: + if extension not in [f"audio/{ext}" for ext in AUDIO_EXTENSIONS]: raise UnsupportedAudioTypeServiceError() file_content = file.read()