Bläddra i källkod

Fix wrong allowed extensions (#16893)

tags/1.2.0
Yongtao Huang 7 månader sedan
förälder
incheckning
98f2e2c729
Inget konto är kopplat till bidragsgivarens mejladress
2 ändrade filer med 3 tillägg och 5 borttagningar
  1. 1
    3
      api/core/file/upload_file_parser.py
  2. 2
    2
      api/services/audio_service.py

+ 1
- 3
api/core/file/upload_file_parser.py Visa fil

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

+ 2
- 2
api/services/audio_service.py Visa fil



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


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__)


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()

Laddar…
Avbryt
Spara