jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix: correctly match http/https URLs in image upload file (#24180)

Browse source
This commit is contained in:
Yongtao Huang 2025-08-19 22:44:22 +08:00 • committed by GitHub
commit 6b5c2bea4d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 34 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

25
api/tests/unit_tests/core/tools/utils/test_web_reader_tool.py Normal file
View file

@ -0,0 +1,25 @@
from core.tools.utils.web_reader_tool import get_image_upload_file_ids
def test_get_image_upload_file_ids():
# should extract id from https + file-preview
content = "![image](https://example.com/a/b/files/abc123/file-preview)"
assert get_image_upload_file_ids(content) == ["abc123"]
# should extract id from http + image-preview
content = "![image](http://host/files/xyz789/image-preview)"
assert get_image_upload_file_ids(content) == ["xyz789"]
# should not match invalid scheme 'htt://'
content = "![image](htt://host/files/bad/file-preview)"
assert get_image_upload_file_ids(content) == []
# should extract multiple ids in order
content = """
some text
![image](https://h/files/id1/file-preview)
middle
![image](http://h/files/id2/image-preview)
end
"""
assert get_image_upload_file_ids(content) == ["id1", "id2"]