Skip to content

Commit e986721

Browse files
JohnJyongNov1c444
authored andcommitted
fix the ssrf of docx file extractor external images (#10237)
1 parent e5dcb7b commit e986721

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

api/core/rag/extractor/word_extractor.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
from docx import Document as DocxDocument
1515

1616
from configs import dify_config
17+
from core.helper import ssrf_proxy
1718
from core.rag.extractor.extractor_base import BaseExtractor
1819
from core.rag.models.document import Document
1920
from extensions.ext_database import db
@@ -86,7 +87,7 @@ def _extract_images_from_docx(self, doc, image_folder):
8687
image_count += 1
8788
if rel.is_external:
8889
url = rel.reltype
89-
response = requests.get(url, stream=True)
90+
response = ssrf_proxy.get(url, stream=True)
9091
if response.status_code == 200:
9192
image_ext = mimetypes.guess_extension(response.headers["Content-Type"])
9293
file_uuid = str(uuid.uuid4())

0 commit comments

Comments
 (0)