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

Fix wrong allowed extensions (#16893)

Browse Source
This commit is contained in:
Yongtao Huang 2025-03-27 15:42:12 +08:00 committed by GitHub
parent 395fdc4960
commit 98f2e2c729
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/file/upload_file_parser.py
View File

@ -4,12 +4,10 @@ import time
from typing import Optional from typing import Optional
from configs import dify_config from configs import dify_config
from constants import IMAGE_EXTENSIONS
from core.helper.url_signer import UrlSigner from core.helper.url_signer import UrlSigner
from extensions.ext_storage import storage 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: class UploadFileParser:
@classmethod @classmethod

4
api/services/audio_service.py
View File

@ -5,6 +5,7 @@ from typing import Optional
from werkzeug.datastructures import FileStorage from werkzeug.datastructures import FileStorage
from constants import AUDIO_EXTENSIONS
from core.model_manager import ModelManager from core.model_manager import ModelManager
from core.model_runtime.entities.model_entities import ModelType from core.model_runtime.entities.model_entities import ModelType
from models.model import App, AppMode, AppModelConfig, Message from models.model import App, AppMode, AppModelConfig, Message
@ -18,7 +19,6 @@ from services.errors.audio import (
FILE_SIZE = 30 FILE_SIZE = 30
FILE_SIZE_LIMIT = FILE_SIZE * 1024 * 1024 FILE_SIZE_LIMIT = FILE_SIZE * 1024 * 1024
ALLOWED_EXTENSIONS = ["mp3", "mp4", "mpeg", "mpga", "m4a", "wav", "webm", "amr"]
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -44,7 +44,7 @@ class AudioService:
raise NoAudioUploadedServiceError() raise NoAudioUploadedServiceError()
extension = file.mimetype 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() raise UnsupportedAudioTypeServiceError()
file_content = file.read() file_content = file.read()