Gets the default value for the config timeout from the env variables.

Author: mrDzurb

ads/aqua/common/utils.py

@@ -33,7 +33,12 @@
 from ads.common.object_storage_details import ObjectStorageDetails
 from ads.common.oci_resource import SEARCH_TYPE, OCIResource
 from ads.common.utils import copy_file, get_console_link, upload_to_os
-from ads.config import AQUA_SERVICE_MODELS_BUCKET, CONDA_BUCKET_NS, TENANCY_OCID
+from ads.config import (
+    AQUA_SERVICE_MODELS_BUCKET,
+    CONDA_BUCKET_NS,
+    TENANCY_OCID,
+    THREADED_DEFAULT_TIMEOUT,
+)
 from ads.model import DataScienceModel, ModelVersionSet
 
 logger = logging.getLogger("ads.aqua")
@@ -196,7 +201,7 @@ def read_file(file_path: str, **kwargs) -> str:
         return UNKNOWN
 
 
-@threaded(timeout=5)
+@threaded(timeout=THREADED_DEFAULT_TIMEOUT)
 def load_config(file_path: str, config_file_name: str, **kwargs) -> dict:
     artifact_path = f"{file_path.rstrip('/')}/{config_file_name}"
     if artifact_path.startswith("oci://"):

ads/common/decorator/threaded.py

@@ -12,6 +12,8 @@
 
 from git import Optional
 
+from ads.config import THREADED_DEFAULT_TIMEOUT
+
 logger = logging.getLogger(__name__)
 
 # Create a global thread pool with a maximum of 10 threads
@@ -81,7 +83,7 @@ def wrapper(*args, **kwargs):
             """
             future = thread_pool.submit(func, *args, **kwargs)
             try:
-                return future.result(timeout=timeout)
+                return future.result(timeout=timeout or THREADED_DEFAULT_TIMEOUT)
             except concurrent.futures.TimeoutError as ex:
                 logger.debug(
                     f"The function '{func.__name__}' "

ads/config.py

@@ -8,6 +8,7 @@
 import inspect
 import os
 from typing import Dict, Optional
+
 from ads.common.config import DEFAULT_CONFIG_PATH, DEFAULT_CONFIG_PROFILE, Config, Mode
 
 OCI_ODSC_SERVICE_ENDPOINT = os.environ.get("OCI_ODSC_SERVICE_ENDPOINT")
@@ -85,6 +86,8 @@
 AQUA_SERVICE_NAME = "aqua"
 DATA_SCIENCE_SERVICE_NAME = "data-science"
 
+THREADED_DEFAULT_TIMEOUT = os.environ.get("THREADED_DEFAULT_TIMEOUT", 5)
+
 
 def export(
     uri: Optional[str] = DEFAULT_CONFIG_PATH,