feat: use env_vars for TIMEOUT_STREAMING

Qiskit · Oct 23, 2024 · ca8d281 · ca8d281
1 parent fbd66dc
commit ca8d281
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/client/qiskit_serverless/core/constants.py b/client/qiskit_serverless/core/constants.py
@@ -23,8 +23,9 @@
 
 # request timeout
 REQUESTS_TIMEOUT: int = 30
-REQUESTS_STREAMING_TIMEOUT: int = 60
 REQUESTS_TIMEOUT_OVERRIDE = "REQUESTS_TIMEOUT_OVERRIDE"
+REQUESTS_STREAMING_TIMEOUT: int = 60
+REQUESTS_STREAMING_TIMEOUT_OVERRIDE = "REQUESTS_TIMEOUT_OVERRIDE"
 
 # gateway
 ENV_GATEWAY_PROVIDER_HOST = "ENV_GATEWAY_PROVIDER_HOST"
@@ -49,9 +50,15 @@
 REQUESTS_TIMEOUT = int(
     os.getenv(REQUESTS_TIMEOUT_OVERRIDE, default=str(REQUESTS_TIMEOUT))
 )
+
+REQUESTS_STREAMING_TIMEOUT = int(
+    os.getenv(REQUESTS_TIMEOUT_OVERRIDE, default=str(REQUESTS_STREAMING_TIMEOUT))
+)
+
 MAX_ARTIFACT_FILE_SIZE_MB = int(
     os.getenv(
-        MAX_ARTIFACT_FILE_SIZE_MB_OVERRIDE, default=str(MAX_ARTIFACT_FILE_SIZE_MB)
+        MAX_ARTIFACT_FILE_SIZE_MB_OVERRIDE, default=str(
+            MAX_ARTIFACT_FILE_SIZE_MB)
     )
 )
 IBM_SERVERLESS_HOST_URL = os.getenv(