Преглед на файлове

fix httpx doesn't support stream parameter (#10859)

tags/0.12.0
Jyong преди 11 месеца
родител
ревизия
ee1f14621a
No account linked to committer's email address
променени са 1 файла, в които са добавени 3 реда и са изтрити 0 реда
  1. 3
    0
      api/core/helper/ssrf_proxy.py

+ 3
- 0
api/core/helper/ssrf_proxy.py Целия файл

@@ -39,6 +39,7 @@ def make_request(method, url, max_retries=SSRF_DEFAULT_MAX_RETRIES, **kwargs):
)

retries = 0
stream = kwargs.pop("stream", False)
while retries <= max_retries:
try:
if dify_config.SSRF_PROXY_ALL_URL:
@@ -52,6 +53,8 @@ def make_request(method, url, max_retries=SSRF_DEFAULT_MAX_RETRIES, **kwargs):
response = client.request(method=method, url=url, **kwargs)

if response.status_code not in STATUS_FORCELIST:
if stream:
return response.iter_bytes()
return response
else:
logging.warning(f"Received status code {response.status_code} for URL {url} which is in the force list")

Loading…
Отказ
Запис