Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-01-28 01:18:39 +04:00
parent 8e2b0b6fd2
commit 665f95eda3
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/utils/files.py
+3
View File
@@ -18,6 +18,7 @@ from open_webui.storage.provider import Storage
from open_webui.models.chats import Chats from open_webui.models.chats import Chats
from open_webui.models.files import Files from open_webui.models.files import Files
from open_webui.routers.files import upload_file_handler from open_webui.routers.files import upload_file_handler
from open_webui.retrieval.web.utils import validate_url
import mimetypes import mimetypes
import base64 import base64
@@ -33,6 +34,8 @@ MARKDOWN_IMAGE_URL_PATTERN = re.compile(r"!\[(.*?)\]\((.+?)\)", re.IGNORECASE)
def get_image_base64_from_url(url: str) -> Optional[str]: def get_image_base64_from_url(url: str) -> Optional[str]:
try: try:
if url.startswith("http"): if url.startswith("http"):
# Validate URL to prevent SSRF attacks against local/private networks
validate_url(url)
# Download the image from the URL # Download the image from the URL
response = requests.get(url) response = requests.get(url)
response.raise_for_status() response.raise_for_status()