certora-prover-cli 0.2.1__tar.gz → 0.2.2__tar.gz

{certora_prover_cli-0.2.1 → certora_prover_cli-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: certora-prover-cli
-Version: 0.2.1
+Version: 0.2.2
 Summary: A Python API and CLI for parsing outputs from the Certora Prover
 Project-URL: Homepage, https://github.com/Certora/ProverCLI
 Project-URL: Repository, https://github.com/Certora/ProverCLI

{certora_prover_cli-0.2.1 → certora_prover_cli-0.2.2}/src/prover_output_utility/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '0.2.1'
-__version_tuple__ = version_tuple = (0, 2, 1)
+__version__ = version = '0.2.2'
+__version_tuple__ = version_tuple = (0, 2, 2)
 
 __commit_id__ = commit_id = None

{certora_prover_cli-0.2.1 → certora_prover_cli-0.2.2}/src/prover_output_utility/api/main.py

@@ -18,6 +18,8 @@ from pathlib import Path
 from typing import Any, Dict, List, Optional
 
 import requests
+from requests.adapters import HTTPAdapter
+from urllib3.util.retry import Retry
 
 from ..auth import ProverAuth
 from ..aws_auth import AWSAuth
@@ -45,6 +47,49 @@ from .tree_parser import TreeParser
 from .url_utils import extract_job_id, extract_job_identifier
 
 
+#: Seconds a request may go without progress before it is abandoned. requests measures this per
+#: socket read rather than over the whole call, so a large download that keeps streaming is
+#: unaffected; it fires when the peer goes quiet. The headroom over the SigV4 fetcher's 30 is for
+#: the outputs archive, where the server has a lot to assemble before the first byte arrives.
+_REQUEST_TIMEOUT_S = 60
+
+#: A silent peer that resets or 5xx's mid-transfer is transient — the same request usually succeeds
+#: moments later. Retry it with exponential backoff (``backoff_factor`` doubles the wait each attempt:
+#: ~0.5s, 1s, 2s), capped at a few attempts so an unrecoverable endpoint still fails promptly rather
+#: than looping. Only the default idempotent methods are retried (urllib3 excludes POST), so a retry
+#: never re-runs a non-idempotent call.
+_MAX_RETRIES = 3
+_BACKOFF_FACTOR = 0.5
+
+
+class _TimedSession(requests.Session):
+    """A session whose requests give up on a silent peer and retry a transient failure.
+
+    requests applies no timeout of its own, so an unresponsive endpoint holds its caller for as
+    long as the socket stays open, and a caller that handed the call to a thread cannot get that
+    thread back. Every ``get``/``post`` on a session goes through ``request``, so the timeout default
+    belongs here rather than at each call site. A caller that passes its own timeout keeps it. The
+    mounted adapter additionally retries transient connection/read failures and 502/503/504 with
+    exponential backoff.
+    """
+
+    def __init__(self) -> None:
+        super().__init__()
+        retry = Retry(
+            total=_MAX_RETRIES,
+            backoff_factor=_BACKOFF_FACTOR,
+            status_forcelist=(502, 503, 504),
+            raise_on_status=False,  # leave status-code handling to the callers
+        )
+        adapter = HTTPAdapter(max_retries=retry)
+        self.mount("https://", adapter)
+        self.mount("http://", adapter)
+
+    def request(self, *args, **kwargs) -> requests.Response:
+        kwargs.setdefault("timeout", _REQUEST_TIMEOUT_S)
+        return super().request(*args, **kwargs)
+
+
 # Per-job locks for thread-safe fetch operations
 _fetch_locks: dict[str, threading.Lock] = {}
 _fetch_locks_lock = threading.Lock()
@@ -158,7 +203,7 @@ class ProverOutputAPI:
 
     def _setup_session(self, force_relogin: bool = False):
         """Setup session with authentication cookies."""
-        self.session = requests.Session()
+        self.session = _TimedSession()
         # Set up authentication
         self.session.cookies = self.auth.get_auth_cookies(force_relogin=force_relogin)
         self.data_fetcher = DataFetcher(self.session, api_instance=self, api_base_url=self.api_base_url)