|
|
|
|
@ -56,8 +56,12 @@ def make_request(method, url, max_retries=SSRF_DEFAULT_MAX_RETRIES, **kwargs):
|
|
|
|
|
response = client.request(method=method, url=url, **kwargs)
|
|
|
|
|
elif dify_config.SSRF_PROXY_HTTP_URL and dify_config.SSRF_PROXY_HTTPS_URL:
|
|
|
|
|
proxy_mounts = {
|
|
|
|
|
"http://": httpx.HTTPTransport(proxy=dify_config.SSRF_PROXY_HTTP_URL),
|
|
|
|
|
"https://": httpx.HTTPTransport(proxy=dify_config.SSRF_PROXY_HTTPS_URL),
|
|
|
|
|
"http://": httpx.HTTPTransport(
|
|
|
|
|
proxy=dify_config.SSRF_PROXY_HTTP_URL, verify=HTTP_REQUEST_NODE_SSL_VERIFY
|
|
|
|
|
),
|
|
|
|
|
"https://": httpx.HTTPTransport(
|
|
|
|
|
proxy=dify_config.SSRF_PROXY_HTTPS_URL, verify=HTTP_REQUEST_NODE_SSL_VERIFY
|
|
|
|
|
),
|
|
|
|
|
}
|
|
|
|
|
with httpx.Client(mounts=proxy_mounts, verify=HTTP_REQUEST_NODE_SSL_VERIFY) as client:
|
|
|
|
|
response = client.request(method=method, url=url, **kwargs)
|
|
|
|
|
|