浏览代码

Fix wrong allowed extensions (#16893)

tags/1.2.0
Yongtao Huang 7 个月前
父节点
当前提交
98f2e2c729
没有帐户链接到提交者的电子邮件
共有 2 个文件被更改,包括 3 次插入5 次删除
  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 查看文件

@@ -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

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

正在加载...
取消
保存