Jyong
hace 11 meses
Se han modificado 1 ficheros con 3 adiciones y 0 borrados
+ 3
- 0
api/core/helper/ssrf_proxy.py
Ver fichero
| ) | ) | ||||
| retries = 0 | retries = 0 | ||||
| stream = kwargs.pop("stream", False) | |||||
| while retries <= max_retries: | while retries <= max_retries: | ||||
| try: | try: | ||||
| if dify_config.SSRF_PROXY_ALL_URL: | if dify_config.SSRF_PROXY_ALL_URL: | ||||
| response = client.request(method=method, url=url, **kwargs) | response = client.request(method=method, url=url, **kwargs) | ||||
| if response.status_code not in STATUS_FORCELIST: | if response.status_code not in STATUS_FORCELIST: | ||||
| if stream: | |||||
| return response.iter_bytes() | |||||
| return response | return response | ||||
| else: | else: | ||||
| logging.warning(f"Received status code {response.status_code} for URL {url} which is in the force list") | logging.warning(f"Received status code {response.status_code} for URL {url} which is in the force list") |
Cargando…