|
import json
|
|
from typing import Any
|
|
|
|
import requests
|
|
from tenacity import retry, retry_if_exception, stop_after_attempt, wait_exponential
|
|
|
|
from openhands.utils.http_session import HttpSession
|
|
from openhands.utils.tenacity_stop import stop_if_should_exit
|
|
|
|
|
|
class RequestHTTPError(requests.HTTPError):
|
|
"""Exception raised when an error occurs in a request with details."""
|
|
|
|
def __init__(self, *args, detail=None, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.detail = detail
|
|
|
|
def __str__(self) -> str:
|
|
s = super().__str__()
|
|
if self.detail is not None:
|
|
s += f'\nDetails: {self.detail}'
|
|
return s
|
|
|
|
|
|
def is_retryable_error(exception):
|
|
return (
|
|
isinstance(exception, requests.HTTPError)
|
|
and exception.response.status_code == 429
|
|
)
|
|
|
|
|
|
@retry(
|
|
retry=retry_if_exception(is_retryable_error),
|
|
stop=stop_after_attempt(3) | stop_if_should_exit(),
|
|
wait=wait_exponential(multiplier=1, min=4, max=60),
|
|
)
|
|
def send_request(
|
|
session: HttpSession,
|
|
method: str,
|
|
url: str,
|
|
timeout: int = 10,
|
|
**kwargs: Any,
|
|
) -> requests.Response:
|
|
response = session.request(method, url, timeout=timeout, **kwargs)
|
|
try:
|
|
response.raise_for_status()
|
|
except requests.HTTPError as e:
|
|
try:
|
|
_json = response.json()
|
|
except (requests.exceptions.JSONDecodeError, json.decoder.JSONDecodeError):
|
|
_json = None
|
|
finally:
|
|
response.close()
|
|
raise RequestHTTPError(
|
|
e,
|
|
response=e.response,
|
|
detail=_json.get('detail') if _json is not None else None,
|
|
) from e
|
|
return response
|
|
|
