Retry on protocol error (#73)

Retry request in batch client in the event of a httpx.ProtocolError exception. Prevents failures in rare events in which the server response causes httpx to raise protocol error

CHANGELOG.md

@@ -8,6 +8,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ## [Unreleased]
 
 - Improve upload speeds for files submitted with the batch client
+- Retry requests in batch client on httpx.ProtocolError
 
 ## [1.11.0] - 2023-08-25
 

requirements.txt

@@ -2,3 +2,4 @@ websockets>=10
 httpx[http2]~=0.23
 polling2~=0.5
 toml~=0.10.2
+tenacity~=8.2.3

speechmatics/batch_client.py

@@ -12,6 +12,7 @@
 
 import httpx
 from polling2 import poll
+from tenacity import retry, retry_if_exception_type, stop_after_attempt
 
 from speechmatics.exceptions import JobNotFoundException, TranscriptionError
 from speechmatics.helpers import get_version
@@ -166,11 +167,19 @@ def send_request(self, method: str, path: str, **kwargs) -> httpx.Response:
 
         :raises httpx.HTTPError: When a request fails, raises an HTTPError
         """
+
         # pylint: disable=no-member
-        with self.api_client.stream(method, path, **kwargs) as response:
-            response.read()
-            response.raise_for_status()
-            return response
+        @retry(
+            stop=stop_after_attempt(2),
+            retry=retry_if_exception_type(httpx.RemoteProtocolError),
+        )
+        def send():
+            with self.api_client.stream(method, path, **kwargs) as response:
+                response.read()
+                response.raise_for_status()
+                return response
+
+        return send()
 
     def list_jobs(self) -> List[Dict[str, Any]]:
         """