瀏覽代碼

fix:Improve MIME type detection for remote URL uploads using python-magic (#12693)

tags/0.15.3
yjc980121 9 月之前
父節點
當前提交
aad7e4dd1c
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 8 行新增0 行删除
  1. 8
    0
      api/controllers/common/helpers.py

+ 8
- 0
api/controllers/common/helpers.py 查看文件

@@ -7,6 +7,7 @@ from typing import Any
from uuid import uuid4

import httpx
import magic
from pydantic import BaseModel

from configs import dify_config
@@ -47,6 +48,13 @@ def guess_file_info_from_response(response: httpx.Response):
# If guessing fails, use Content-Type from response headers
mimetype = response.headers.get("Content-Type", "application/octet-stream")

# Use python-magic to guess MIME type if still unknown or generic
if mimetype == "application/octet-stream":
try:
mimetype = magic.from_buffer(response.content[:1024], mime=True)
except magic.MagicException:
pass

extension = os.path.splitext(filename)[1]

# Ensure filename has an extension

Loading…
取消
儲存