|
@@ -11,15 +11,6 @@ from configs import dify_config
|
|
|
|
|
|
SSRF_DEFAULT_MAX_RETRIES = dify_config.SSRF_DEFAULT_MAX_RETRIES
|
|
|
|
|
|
-proxy_mounts = (
|
|
|
- {
|
|
|
- "http://": httpx.HTTPTransport(proxy=dify_config.SSRF_PROXY_HTTP_URL),
|
|
|
- "https://": httpx.HTTPTransport(proxy=dify_config.SSRF_PROXY_HTTPS_URL),
|
|
|
- }
|
|
|
- if dify_config.SSRF_PROXY_HTTP_URL and dify_config.SSRF_PROXY_HTTPS_URL
|
|
|
- else None
|
|
|
-)
|
|
|
-
|
|
|
BACKOFF_FACTOR = 0.5
|
|
|
STATUS_FORCELIST = [429, 500, 502, 503, 504]
|
|
|
|
|
@@ -51,7 +42,11 @@ def make_request(method, url, max_retries=SSRF_DEFAULT_MAX_RETRIES, **kwargs):
|
|
|
if dify_config.SSRF_PROXY_ALL_URL:
|
|
|
with httpx.Client(proxy=dify_config.SSRF_PROXY_ALL_URL) as client:
|
|
|
response = client.request(method=method, url=url, **kwargs)
|
|
|
- elif proxy_mounts:
|
|
|
+ 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),
|
|
|
+ }
|
|
|
with httpx.Client(mounts=proxy_mounts) as client:
|
|
|
response = client.request(method=method, url=url, **kwargs)
|
|
|
else:
|