diff --git a/api/core/file/remote_fetcher.py b/api/core/file/remote_fetcher.py index b5c067c5ba..5aa2d0538b 100644 --- a/api/core/file/remote_fetcher.py +++ b/api/core/file/remote_fetcher.py @@ -43,9 +43,7 @@ _UPLOAD_FILE_PATH_PATTERN = re.compile( r"^/files/(?P[a-fA-F0-9-]+)/(?Pfile-preview|image-preview)$" ) _TOOL_FILE_PATH_PATTERN = re.compile(r"^/files/tools/(?P[a-fA-F0-9-]+)\.(?P[^/]+)$") -_DATASOURCE_FILE_PATH_PATTERN = re.compile( - r"^/files/datasources/(?P[a-fA-F0-9-]+)\.(?P[^/]+)$" -) +_DATASOURCE_FILE_PATH_PATTERN = re.compile(r"^/files/datasources/(?P[a-fA-F0-9-]+)\.(?P[^/]+)$") _file_access_controller = DatabaseFileAccessController() diff --git a/api/tests/unit_tests/core/file/test_remote_fetcher.py b/api/tests/unit_tests/core/file/test_remote_fetcher.py index c5a45ceec6..7165606db5 100644 --- a/api/tests/unit_tests/core/file/test_remote_fetcher.py +++ b/api/tests/unit_tests/core/file/test_remote_fetcher.py @@ -127,10 +127,7 @@ def test_invalid_signature_delegates_to_ssrf_proxy(monkeypatch): proxy_response = httpx.Response(403, request=httpx.Request("GET", "http://localhost:5001/bad")) ssrf_get = MagicMock(return_value=proxy_response) monkeypatch.setattr(remote_fetcher.ssrf_proxy, "get", ssrf_get) - url = ( - f"http://localhost:5001/files/{UPLOAD_FILE_ID}/file-preview" - "?timestamp=1700000000&nonce=nonce&sign=bad" - ) + url = f"http://localhost:5001/files/{UPLOAD_FILE_ID}/file-preview?timestamp=1700000000&nonce=nonce&sign=bad" response = remote_fetcher.get(url, timeout=3)