konduktor-nightly 0.1.0.dev20251128104812__py3-none-any.whl

konduktor/execution.py

@@ -0,0 +1,447 @@
+"""Execution layer."""
+
+import os
+import tempfile
+import typing
+from typing import Any, Dict, Optional
+
+import colorama
+
+if typing.TYPE_CHECKING:
+    import konduktor
+
+from konduktor import config, constants
+from konduktor import logging as konduktor_logging
+from konduktor.backends import DeploymentBackend, JobsetBackend
+from konduktor.data import data_utils
+from konduktor.data import registry as storage_registry
+from konduktor.data import storage as storage_lib
+from konduktor.utils import common_utils, exceptions, rich_utils, ux_utils
+
+logger = konduktor_logging.get_logger(__name__)
+
+
+def _execute(
+    task: 'konduktor.Task',
+    dryrun: bool = False,
+    detach_run: bool = False,
+) -> Optional[str]:
+    """Execute an task.
+
+    Args:
+      task: konduktor.Task
+      dryrun: bool; if True, only print the provision info (e.g., cluster
+        yaml).
+      stream_logs: bool; whether to stream all tasks' outputs to the client.
+      cluster_name: Name of the cluster to create/reuse.  If None,
+        auto-generate a name.
+
+    Returns:
+      workload_id: Optional[int]; the job ID of the submitted job. None if the
+        backend is not CloudVmRayBackend, or no job is submitted to
+        the cluster.
+    """
+    # (asaiacai): in the future we may support more backends but not likely
+    if task.serving:
+        backend = DeploymentBackend()  # type: ignore
+    else:
+        backend = JobsetBackend()  # type: ignore
+    # template the commands for syncing the contents within the shell command
+    # initialization of the pod
+    job_name = backend.execute(task, detach_run, dryrun=dryrun)
+
+    if dryrun:
+        logger.info('Dryrun finished.')
+        return None
+
+    # attach to head node output if detach_run is False
+    backend.post_execute()
+
+    return job_name
+
+
+def launch(
+    task: 'konduktor.Task',
+    dryrun: bool = False,
+    detach_run: bool = False,
+) -> Optional[str]:
+    """Launch a task
+
+    Args:
+        task: konduktor.Task
+        dryrun: if True, do not actually launch the task.
+        detach_run: If True, as soon as a job is submitted, return from this
+            function and do not stream execution logs.
+
+    Example:
+        .. code-block:: python
+
+            import konduktor
+            task = konduktor.Task(run='echo hello konduktor')
+            konduktor.launch(task)
+
+    Raises:
+      Other exceptions may be raised depending on the backend.
+
+    Returns:
+      workload_id: Optional[str]; the job ID of the submitted job.
+    """
+
+    maybe_translate_local_file_mounts_and_sync_up(task, 'job')
+
+    return _execute(
+        task=task,
+        dryrun=dryrun,
+        detach_run=detach_run,
+    )
+
+
+# (maybe translate local file mounts) and (sync up)
+def maybe_translate_local_file_mounts_and_sync_up(
+    task: 'konduktor.Task', task_type: str
+) -> None:
+    """Translates local->VM mounts into Storage->VM, then syncs up any Storage.
+
+    Eagerly syncing up local->Storage ensures Storage->VM would work at task
+    launch time.
+
+    If there are no local source paths to be translated, this function would
+    still sync up any storage mounts with local source paths (which do not
+    undergo translation).
+
+    When jobs.bucket or serve.bucket is not specified, an intermediate storage
+    dedicated for the job is created for the workdir and local file mounts and
+    the storage is deleted when the job finishes. We don't share the storage
+    between jobs, because jobs might have different resources requirements, and
+    sharing storage between jobs may cause egress costs or slower transfer
+    speeds.
+    """
+
+    # ================================================================
+    # Translate the workdir and local file mounts to cloud file mounts.
+    # ================================================================
+
+    def _sub_path_join(sub_path: Optional[str], path: str) -> str:
+        if sub_path is None:
+            return path
+        return os.path.join(sub_path, path).strip('/')
+
+    # We use uuid to generate a unique run id for the job, so that the bucket/
+    # subdirectory name is unique across different jobs/services.
+    # We should not use common_utils.get_usage_run_id() here, because when
+    # Python API is used, the run id will be the same across multiple
+    # jobs.launch/serve.up calls after the sky is imported.
+    run_id = common_utils.get_usage_run_id()[:4]
+    user_hash = common_utils.get_user_hash()
+    original_file_mounts = task.file_mounts if task.file_mounts else {}
+    original_storage_mounts = task.storage_mounts if task.storage_mounts else {}
+
+    copy_mounts = task.get_local_to_remote_file_mounts()
+    if copy_mounts is None:
+        copy_mounts = {}
+
+    has_local_source_paths_file_mounts = bool(copy_mounts)
+    has_local_source_paths_workdir = task.workdir is not None
+
+    msg = None
+    if has_local_source_paths_workdir and has_local_source_paths_file_mounts:
+        msg = 'workdir and file_mounts with local source paths'
+    elif has_local_source_paths_file_mounts:
+        msg = 'file_mounts with local source paths'
+    elif has_local_source_paths_workdir:
+        msg = 'workdir'
+    if msg:
+        logger.info(
+            ux_utils.starting_message(f'Translating {msg} to ' 'cloud Storage...')
+        )
+        rich_utils.force_update_status(
+            ux_utils.spinner_message(f'Translating {msg} to cloud Storage...')
+        )
+
+    # Get the bucket name for the workdir and file mounts,
+    # we store all these files in same bucket from config.
+    bucket_wth_prefix = config.get_nested((task_type, 'bucket'), None)
+    store_kwargs: Dict[str, Any] = {}
+    if bucket_wth_prefix is None:
+        store_type = sub_path = None
+        storage_account_name = region = None
+        bucket_name = constants.FILE_MOUNTS_BUCKET_NAME.format(
+            username=common_utils.get_cleaned_username(), user_hash=user_hash, id=run_id
+        )
+    else:
+        (store_type, bucket_name, sub_path, storage_account_name, region) = (
+            storage_lib.StoreType.get_fields_from_store_url(bucket_wth_prefix)
+        )
+        if storage_account_name is not None:
+            store_kwargs['storage_account_name'] = storage_account_name
+        if region is not None:
+            store_kwargs['region'] = region
+    # Step 1: Translate the workdir to SkyPilot storage.
+    new_storage_mounts = {}
+    if task.workdir is not None:
+        workdir = task.workdir
+        task.workdir = None
+        if (
+            constants.KONDUKTOR_REMOTE_WORKDIR in original_file_mounts
+            or constants.KONDUKTOR_REMOTE_WORKDIR in original_storage_mounts
+        ):
+            raise ValueError(
+                f'Cannot mount {constants.KONDUKTOR_REMOTE_WORKDIR} as both the '
+                'workdir and file_mounts contains it as the target.'
+            )
+        bucket_sub_path = _sub_path_join(
+            sub_path,
+            constants.FILE_MOUNTS_WORKDIR_SUBPATH.format(
+                task_name=task.name, run_id=run_id
+            ),
+        )
+        stores = None
+        if store_type is not None:
+            stores = [store_type]
+
+        storage_obj = storage_lib.Storage(
+            name=bucket_name,
+            source=workdir,
+            persistent=False,
+            mode=storage_lib.StorageMode.COPY,
+            stores=stores,
+            # Set `_is_sky_managed` to False when `bucket_with_prefix` is
+            # specified, so that the storage is not deleted when job finishes,
+            # but only the sub path is deleted.
+            # _is_sky_managed=bucket_wth_prefix is None,
+            _is_sky_managed=False,
+            _bucket_sub_path=bucket_sub_path,
+        )
+        new_storage_mounts[constants.KONDUKTOR_REMOTE_WORKDIR] = storage_obj
+        # Check of the existence of the workdir in file_mounts is done in
+        # the task construction.
+        logger.info(
+            f'  {colorama.Style.DIM}Workdir: {workdir!r} '
+            f'-> storage: {bucket_name!r}.{colorama.Style.RESET_ALL}'
+        )
+
+    # Step 2: Translate the local file mounts with folder in src to SkyPilot
+    # storage.
+    # TODO(zhwu): Optimize this by:
+    # 1. Use the same bucket for all the mounts.
+    # 2. When the src is the same, use the same bucket.
+    copy_mounts_with_file_in_src = {}
+    for i, (dst, src) in enumerate(copy_mounts.items()):
+        assert task.file_mounts is not None
+        task.file_mounts.pop(dst)
+        if os.path.isfile(os.path.abspath(os.path.expanduser(src))):
+            copy_mounts_with_file_in_src[dst] = src
+            continue
+        bucket_sub_path = _sub_path_join(
+            sub_path,
+            constants.FILE_MOUNTS_SUBPATH.format(
+                task_name=task.name, i=i, run_id=run_id
+            ),
+        )
+        stores = None
+        if store_type is not None:
+            stores = [store_type]
+        storage_obj = storage_lib.Storage(
+            name=bucket_name,
+            source=src,
+            persistent=False,
+            mode=storage_lib.StorageMode.COPY,
+            stores=stores,
+            # _is_sky_managed=not bucket_wth_prefix,
+            _is_sky_managed=False,
+            _bucket_sub_path=bucket_sub_path,
+        )
+        new_storage_mounts[dst] = storage_obj
+        logger.info(
+            f'  {colorama.Style.DIM}Folder : {src!r} '
+            f'-> storage: {bucket_name!r}.{colorama.Style.RESET_ALL}'
+        )
+
+    # Step 3: Translate local file mounts with file in src to SkyPilot storage.
+    # Hard link the files in src to a temporary directory, and upload folder.
+    file_mounts_tmp_subpath = _sub_path_join(
+        sub_path,
+        constants.FILE_MOUNTS_TMP_SUBPATH.format(task_name=task.name, run_id=run_id),
+    )
+    base_tmp_dir = os.path.expanduser(constants.FILE_MOUNTS_LOCAL_TMP_BASE_PATH)
+    os.makedirs(base_tmp_dir, exist_ok=True)
+    with tempfile.TemporaryDirectory(dir=base_tmp_dir) as temp_path:
+        local_fm_path = os.path.join(
+            temp_path, constants.FILE_MOUNTS_LOCAL_TMP_DIR.format(id=run_id)
+        )
+        os.makedirs(local_fm_path, exist_ok=True)
+        file_mount_remote_tmp_dir = constants.FILE_MOUNTS_REMOTE_TMP_DIR.format(
+            task_type
+        )
+        if copy_mounts_with_file_in_src:
+            src_to_file_id = {}
+            for i, src in enumerate(set(copy_mounts_with_file_in_src.values())):
+                src_to_file_id[src] = i
+                os.link(
+                    os.path.abspath(os.path.expanduser(src)),
+                    os.path.join(local_fm_path, f'file-{i}'),
+                )
+            stores = None
+            if store_type is not None:
+                stores = [store_type]
+            storage_obj = storage_lib.Storage(
+                name=bucket_name,
+                source=local_fm_path,
+                persistent=False,
+                # mode=storage_lib.StorageMode.MOUNT
+                mode=storage_lib.StorageMode.COPY,
+                stores=stores,
+                # _is_sky_managed=not bucket_wth_prefix,
+                _is_sky_managed=False,
+                _bucket_sub_path=file_mounts_tmp_subpath,
+            )
+
+            new_storage_mounts[file_mount_remote_tmp_dir] = storage_obj
+            if file_mount_remote_tmp_dir in original_storage_mounts:
+                with ux_utils.print_exception_no_traceback():
+                    raise ValueError(
+                        'Failed to translate file mounts, due to the default '
+                        f'destination {file_mount_remote_tmp_dir} '
+                        'being taken.'
+                    )
+            sources = list(src_to_file_id.keys())
+            sources_str = '\n    '.join(sources)
+            logger.info(
+                f'  {colorama.Style.DIM}Files (listed below) '
+                f' -> storage: {bucket_name}:'
+                f'\n    {sources_str}{colorama.Style.RESET_ALL}'
+            )
+
+        rich_utils.force_update_status(
+            ux_utils.spinner_message('Uploading translated local files/folders')
+        )
+        task.update_storage_mounts(new_storage_mounts)
+
+        # Step 4: Upload storage from sources
+        # Upload the local source to a bucket. The task will not be executed
+        # locally, so we need to upload the files/folders to the bucket manually
+        # here before sending the task to the remote jobs controller.  This will
+        # also upload any storage mounts that are not translated.  After
+        # sync_storage_mounts, we will also have file_mounts in the task, but
+        # these aren't used since the storage_mounts for the same paths take
+        # precedence.
+        if task.storage_mounts:
+            # There may be existing (non-translated) storage mounts, so log this
+            # whenever task.storage_mounts is non-empty.
+            rich_utils.force_update_status(
+                ux_utils.spinner_message(
+                    'Uploading local sources to storage[/]  '
+                    '[dim]View storages: sky storage ls'
+                )
+            )
+        try:
+            task.sync_storage_mounts()
+        except (ValueError, exceptions.NoCloudAccessError) as e:
+            if 'No enabled cloud for storage' in str(e) or isinstance(
+                e, exceptions.NoCloudAccessError
+            ):
+                data_src = None
+                if has_local_source_paths_file_mounts:
+                    data_src = 'file_mounts'
+                if has_local_source_paths_workdir:
+                    if data_src:
+                        data_src += ' and workdir'
+                    else:
+                        data_src = 'workdir'
+                store_enabled_clouds = ', '.join(storage_registry._STORE_ENABLED_CLOUDS)
+                with ux_utils.print_exception_no_traceback():
+                    raise exceptions.NotSupportedError(
+                        f'Unable to use {data_src} - no cloud with object '
+                        'store support is enabled. Please enable at least one '
+                        'cloud with object store support '
+                        f'({store_enabled_clouds}) by running `sky check`, or '
+                        f'remove {data_src} from your task.'
+                        '\nHint: If you do not have any cloud access, you may '
+                        'still download data and code over the network using '
+                        'curl or other tools in the `setup` section of the '
+                        'task.'
+                    ) from None
+
+    # Step 5: Add the file download into the file mounts, such as
+    #  /original-dst: s3://spot-fm-file-only-bucket-name/file-0
+    new_file_mounts = {}
+    if copy_mounts_with_file_in_src:
+        # file_mount_remote_tmp_dir will only exist when there are files in
+        # the src for copy mounts.
+        storage_obj = task.storage_mounts[file_mount_remote_tmp_dir]
+        assert storage_obj.stores, (storage_obj.__dict__, task.to_yaml_config())
+        curr_store_type = list(storage_obj.stores.keys())[0]
+        store_object = storage_obj.stores[curr_store_type]
+        assert store_object is not None, (storage_obj.__dict__, task.to_yaml_config())
+        bucket_url = storage_lib.StoreType.get_endpoint_url(store_object, bucket_name)
+        bucket_url += f'/{file_mounts_tmp_subpath}'
+        for dst, src in copy_mounts_with_file_in_src.items():
+            file_id = src_to_file_id[src]
+            new_file_mounts[dst] = bucket_url + f'/file-{file_id}'
+    task.update_file_mounts(new_file_mounts)
+
+    # Step 6: Replace the source field that is local path in all storage_mounts
+    # with bucket URI and remove the name field.
+    for storage_obj in task.storage_mounts.values():
+        if storage_obj.source is not None and not data_utils.is_cloud_store_url(
+            storage_obj.source
+        ):
+            # Need to replace the local path with bucket URI, and remove the
+            # name field, so that the storage mount can work on the jobs
+            # controller.
+            store_types = list(storage_obj.stores.keys())
+            assert len(store_types) == 1, (
+                'We only support one store type for now.',
+                storage_obj.stores,
+            )
+            curr_store_type = store_types[0]
+            store_object = storage_obj.stores[curr_store_type]
+            assert store_object is not None and storage_obj.name is not None, (
+                store_object,
+                storage_obj.name,
+            )
+            storage_obj.source = storage_lib.StoreType.get_endpoint_url(
+                store_object, storage_obj.name
+            )
+            storage_obj.force_delete = True
+
+    # Step 7: Convert all `MOUNT` mode storages which don't specify a source
+    # to specifying a source. If the source is specified with a local path,
+    # it was handled in step 6.
+    updated_mount_storages = {}
+    for storage_path, storage_obj in task.storage_mounts.items():
+        if storage_obj.mode == storage_lib.StorageMode.MOUNT and not storage_obj.source:
+            # Construct source URL with first store type and storage name
+            # E.g., s3://my-storage-name
+            store_types = list(storage_obj.stores.keys())
+            assert len(store_types) == 1, (
+                'We only support one store type for now.',
+                storage_obj.stores,
+            )
+            curr_store_type = store_types[0]
+            store_object = storage_obj.stores[curr_store_type]
+            assert store_object is not None and storage_obj.name is not None, (
+                store_object,
+                storage_obj.name,
+            )
+            source = storage_lib.StoreType.get_endpoint_url(
+                store_object, storage_obj.name
+            )
+            assert store_object is not None and storage_obj.name is not None, (
+                store_object,
+                storage_obj.name,
+            )
+            new_storage = storage_lib.Storage.from_yaml_config(
+                {
+                    'source': source,
+                    'persistent': storage_obj.persistent,
+                    'mode': storage_lib.StorageMode.MOUNT.value,
+                    # We enable force delete to allow the controller to delete
+                    # the object store in case persistent is set to False.
+                    '_force_delete': True,
+                }
+            )
+            updated_mount_storages[storage_path] = new_storage
+    task.update_storage_mounts(updated_mount_storages)
+    if msg:
+        logger.info(ux_utils.finishing_message('Uploaded local files/folders.'))

konduktor/kube_client.py

@@ -0,0 +1,237 @@
+import logging
+import os
+from typing import Any, Callable, List, Optional
+
+import kubernetes
+import urllib3
+
+from konduktor import config
+from konduktor import logging as konduktor_logging
+from konduktor.utils import annotations, ux_utils
+
+logger = konduktor_logging.get_logger(__name__)
+
+# Timeout to use for API calls
+API_TIMEOUT = 5
+DEFAULT_NAMESPACE = 'default'
+DEFAULT_IN_CLUSTER_REGION = 'in-cluster'
+# The name for the environment variable that stores the in-cluster context name
+# for Kubernetes clusters. This is used to associate a name with the current
+# context when running with in-cluster auth. If not set, the context name is
+# set to DEFAULT_IN_CLUSTER_REGION.
+IN_CLUSTER_CONTEXT_NAME_ENV_VAR = 'KONDUKTOR_IN_CLUSTER_CONTEXT_NAME'
+
+# Tracks the most recently selected/loaded context name.
+# None means no explicit context was resolved/loaded yet.
+_ACTIVE_CONTEXT: Optional[str] = None
+
+
+def _decorate_methods(obj: Any, decorator: Callable, decoration_type: str):
+    for attr_name in dir(obj):
+        attr = getattr(obj, attr_name)
+        # Skip methods starting with '__' since they are invoked through one
+        # of the main methods, which are already decorated.
+        if callable(attr) and not attr_name.startswith('__'):
+            continue
+    return obj
+
+
+def _api_logging_decorator(logger: str, level: int):
+    """Decorator to set logging level for API calls.
+
+    This is used to suppress the verbose logging from urllib3 when calls to the
+    Kubernetes API timeout.
+    """
+
+    def decorated_api(api):
+        def wrapped(*args, **kwargs):
+            obj = api(*args, **kwargs)
+            _decorate_methods(
+                obj, konduktor_logging.set_logging_level(logger, level), 'api_log'
+            )
+            return obj
+
+        return wrapped
+
+    return decorated_api
+
+
+def _load_config(context: Optional[str] = None):
+    urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
+
+    # If no context explicitly provided, prefer the configured allowed context
+    # (first element) when present. This ensures the client defaults to the
+    # user-specified context instead of kubeconfig's current-context.
+    effective_context = context
+    allowed_contexts: List[str] = config.get_nested(
+        ('kubernetes', 'allowed_contexts'), []
+    )
+
+    is_allowed_selected = False
+    if effective_context is None and allowed_contexts:
+        effective_context = allowed_contexts[0]
+        is_allowed_selected = True
+        logger.info(
+            'Detected kubernetes.allowed_contexts in config; using context: %s',
+            effective_context,
+        )
+
+    def _load_config_from_kubeconfig(context: Optional[str] = None):
+        try:
+            kubernetes.config.load_kube_config(context=context)
+        except kubernetes.config.config_exception.ConfigException as e:
+            # Improve error when a configured allowed context cannot be loaded
+            msg = str(e)
+            if is_allowed_selected and context is not None:
+                err_str = (
+                    'Configured Kubernetes context not usable: '
+                    f'kubernetes.allowed_contexts[0] = {context!r}. '
+                    'Please ensure this context exists and is valid in your '
+                    'kubeconfig (typically at ~/.kube/config).'
+                )
+            elif 'Expected key current-context' in msg:
+                err_str = (
+                    f'Failed to load Kubernetes configuration for {context!r}. '
+                    'Kubeconfig does not contain any valid context(s).\n'
+                )
+            else:
+                err_str = (
+                    f'Failed to load Kubernetes configuration for {context!r}. '
+                    'Please check if your kubeconfig file exists at '
+                    f'~/.kube/config and is valid.'
+                )
+            with ux_utils.print_exception_no_traceback():
+                raise ValueError(err_str) from None
+
+    global _ACTIVE_CONTEXT
+    if effective_context == in_cluster_context_name() or effective_context is None:
+        try:
+            # Load in-cluster config if running in a pod and context is None.
+            # Kubernetes set environment variables for service discovery do not
+            # show up in SkyPilot tasks. For now, we work around by using
+            # DNS name instead of environment variables.
+            # See issue: https://github.com/skypilot-org/skypilot/issues/2287
+            os.environ['KUBERNETES_SERVICE_HOST'] = 'kubernetes.default.svc'
+            os.environ['KUBERNETES_SERVICE_PORT'] = '443'
+            kubernetes.config.load_incluster_config()
+            _ACTIVE_CONTEXT = in_cluster_context_name()
+        except kubernetes.config.config_exception.ConfigException:
+            # If allowed_contexts was specified, do not fall back silently.
+            if is_allowed_selected:
+                _load_config_from_kubeconfig(effective_context)
+            else:
+                _load_config_from_kubeconfig()
+            # Best effort: set active context to current-context from kubeconfig
+            try:
+                _, current_ctx = kubernetes.config.list_kube_config_contexts()
+                _ACTIVE_CONTEXT = current_ctx.get('name') if current_ctx else None
+            except kubernetes.config.config_exception.ConfigException:
+                _ACTIVE_CONTEXT = None
+    else:
+        _load_config_from_kubeconfig(effective_context)
+        _ACTIVE_CONTEXT = effective_context
+
+
+@_api_logging_decorator('urllib3', logging.ERROR)
+@annotations.lru_cache(scope='request')
+def core_api(context: Optional[str] = None):
+    _load_config(context)
+    return kubernetes.client.CoreV1Api()
+
+
+@_api_logging_decorator('urllib3', logging.ERROR)
+@annotations.lru_cache(scope='request')
+def auth_api(context: Optional[str] = None):
+    _load_config(context)
+    return kubernetes.client.RbacAuthorizationV1Api()
+
+
+@_api_logging_decorator('urllib3', logging.ERROR)
+@annotations.lru_cache(scope='request')
+def networking_api(context: Optional[str] = None):
+    _load_config(context)
+    return kubernetes.client.NetworkingV1Api()
+
+
+@_api_logging_decorator('urllib3', logging.ERROR)
+@annotations.lru_cache(scope='request')
+def crd_api(context: Optional[str] = None):
+    _load_config(context)
+    return kubernetes.client.CustomObjectsApi()
+
+
+@_api_logging_decorator('urllib3', logging.ERROR)
+@annotations.lru_cache(scope='request')
+def node_api(context: Optional[str] = None):
+    _load_config(context)
+    return kubernetes.client.NodeV1Api()
+
+
+@_api_logging_decorator('urllib3', logging.ERROR)
+@annotations.lru_cache(scope='request')
+def apps_api(context: Optional[str] = None):
+    _load_config(context)
+    return kubernetes.client.AppsV1Api()
+
+
+@_api_logging_decorator('urllib3', logging.ERROR)
+@annotations.lru_cache(scope='request')
+def api_client(context: Optional[str] = None):
+    _load_config(context)
+    return kubernetes.client.ApiClient()
+
+
+@_api_logging_decorator('urllib3', logging.ERROR)
+@annotations.lru_cache(scope='request')
+def batch_api(context: Optional[str] = None):
+    _load_config(context)
+    return kubernetes.client.BatchV1Api()
+
+
+@_api_logging_decorator('urllib3', logging.ERROR)
+@annotations.lru_cache(scope='request')
+def crd_client(context: Optional[str] = None):
+    _load_config(context)
+    return kubernetes.client.CustomObjectsApi()
+
+
+@_api_logging_decorator('urllib3', logging.ERROR)
+@annotations.lru_cache(scope='request')
+def autoscaling_api(context: Optional[str] = None):
+    """Return the Kubernetes AutoscalingV2Api client."""
+    _load_config(context)
+    return kubernetes.client.AutoscalingV2Api()
+
+
+def api_exception():
+    return kubernetes.client.rest.ApiException
+
+
+def config_exception():
+    return kubernetes.config.config_exception.ConfigException
+
+
+def max_retry_error():
+    return urllib3.exceptions.MaxRetryError
+
+
+def stream():
+    return kubernetes.stream.stream
+
+
+def in_cluster_context_name() -> Optional[str]:
+    """Returns the name of the in-cluster context from the environment.
+
+    If the environment variable is not set, returns the default in-cluster
+    context name.
+    """
+    return os.environ.get(IN_CLUSTER_CONTEXT_NAME_ENV_VAR) or DEFAULT_IN_CLUSTER_REGION
+
+
+def get_active_context() -> Optional[str]:
+    """Returns the last context selected by the client loader.
+
+    This reflects the effective context used by the most recent client init.
+    May be None if no client has been initialized yet.
+    """
+    return _ACTIVE_CONTEXT