Feat/increate request timeout for upload (#1519)

* feat: increase request timeout for upload * change download timeout * fix lint
Qiskit · Oct 22, 2024 · 0d09fcb · 0d09fcb
1 parent eb05950
commit 0d09fcb
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/client/qiskit_serverless/core/constants.py b/client/qiskit_serverless/core/constants.py
@@ -23,6 +23,7 @@
 
 # request timeout
 REQUESTS_TIMEOUT: int = 30
+REQUESTS_STREAMING_TIMEOUT: int = 60
 REQUESTS_TIMEOUT_OVERRIDE = "REQUESTS_TIMEOUT_OVERRIDE"
 
 # gateway

diff --git a/client/qiskit_serverless/core/files.py b/client/qiskit_serverless/core/files.py
@@ -33,7 +33,10 @@
 from opentelemetry import trace
 from tqdm import tqdm
 
-from qiskit_serverless.core.constants import REQUESTS_TIMEOUT
+from qiskit_serverless.core.constants import (
+    REQUESTS_STREAMING_TIMEOUT,
+    REQUESTS_TIMEOUT,
+)
 from qiskit_serverless.utils.json import safe_json_request
 
 
@@ -67,7 +70,7 @@ def download(
                 params={"file": file, "provider": provider},
                 stream=True,
                 headers={"Authorization": f"Bearer {self._token}"},
-                timeout=REQUESTS_TIMEOUT,
+                timeout=REQUESTS_STREAMING_TIMEOUT,
             ) as req:
                 req.raise_for_status()
 
@@ -95,7 +98,7 @@ def upload(self, file: str, provider: Optional[str] = None) -> Optional[str]:
                     data={"provider": provider},
                     stream=True,
                     headers={"Authorization": f"Bearer {self._token}"},
-                    timeout=REQUESTS_TIMEOUT,
+                    timeout=REQUESTS_STREAMING_TIMEOUT,
                 ) as req:
                     if req.ok:
                         return req.text