| 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
|
|
|
