add timeout

BWRC-AMS-ML-Discovery · Nov 15, 2023 · a2bf2e0 · a2bf2e0
1 parent f8a1f66
commit a2bf2e0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/Discovery/Client/discovery_client/__init__.py b/Discovery/Client/discovery_client/__init__.py
@@ -22,6 +22,7 @@
 # PyPi Imports
 import dotenv
 import httpx
+from httpx._types import TimeoutTypes
 
 # Workspace Imports
 # from discovery_shared.git import GitInfo
@@ -38,6 +39,7 @@ class Config:
 
     server_url: str = "localhost:8000"
     enable_https: bool = True
+    timeout: TimeoutTypes = 60.0
 
 
 # Create the module-scope configuration
@@ -102,6 +104,7 @@ def f(inp: rpc.input_type, *, rpc=rpc) -> rpc.return_type:
                     os.environ.get("DISCOVERY_USERNAME"),
                     os.environ.get("DISCOVERY_API_KEY"),
                 ),
+                timeout=config.timeout,
             )
             return rpc.return_type(**resp.json())