konduktor-nightly 0.1.0.dev20250915104603__py3-none-any.whl → 0.1.0.dev20251107104752__py3-none-any.whl

konduktor/__init__.py

@@ -11,7 +11,7 @@ from konduktor.task import Task
 __all__ = ['launch', 'Resources', 'Task', 'Serving']
 
 # Replaced with the current commit when building the wheels.
-_KONDUKTOR_COMMIT_SHA = '5beb52b28b01054af032813e670311b459b932b3'
+_KONDUKTOR_COMMIT_SHA = '5ceef9b8f579ac23f7a2bd863820aaa2341055e3'
 os.makedirs(os.path.expanduser('~/.konduktor'), exist_ok=True)
 
 
@@ -45,5 +45,5 @@ def _get_git_commit():
 
 
 __commit__ = _get_git_commit()
-__version__ = '1.0.0.dev0.1.0.dev20250915104603'
+__version__ = '1.0.0.dev0.1.0.dev20251107104752'
 __root_dir__ = os.path.dirname(os.path.abspath(__file__))

konduktor/backends/constants.py

@@ -18,3 +18,4 @@ STOP_USERNAME_LABEL = 'trainy.ai/stop-username'
 SECRET_BASENAME_LABEL = 'trainy.ai/secret-basename'
 SECRET_KIND_LABEL = 'trainy.ai/secret-kind'
 SECRET_OWNER_LABEL = 'trainy.ai/secret-owner'
+ROOT_NAME = 'trainy.ai/root-name'

konduktor/backends/deployment.py

@@ -54,8 +54,19 @@ def _wait_for_all_ready(namespace: str, name: str):
         except ApiException:
             services_map = {}
 
-        autoscaler = deployment_utils.get_autoscaler(namespace, name)
-        autoscalers_map = {name: autoscaler} if autoscaler else {}
+        autoscalers_map = {}
+        try:
+            autoscaler_obj = deployment_utils.get_autoscaler(namespace, name)
+            if autoscaler_obj:
+                # detect aibrix vs general from deployment labels
+                labels = (deployment.metadata.labels or {}) if deployment else {}
+                is_aibrix = deployment_utils.AIBRIX_NAME_LABEL in labels
+                if is_aibrix:
+                    autoscalers_map[name] = {'kpa': autoscaler_obj}
+                else:
+                    autoscalers_map[name] = {'hpa': autoscaler_obj}
+        except ApiException:
+            pass
 
         status = deployment_utils.get_model_status(
             name, deployments_map, services_map, autoscalers_map
@@ -144,6 +155,7 @@ class DeploymentBackend(backend.Backend):
         namespace = kubernetes_utils.get_kube_config_context_namespace(context)
 
         if not dryrun and task.serving:
+            logger.debug(f'[DEBUG] Creating deployment for task: {task.name}')
             deployment_utils.create_deployment(
                 namespace=namespace,
                 task=task,
@@ -151,25 +163,30 @@ class DeploymentBackend(backend.Backend):
                 dryrun=dryrun,
             )
 
+            logger.debug(f'[DEBUG] Creating service for task: {task.name}')
             deployment_utils.create_service(
                 namespace=namespace,
                 task=task,
                 dryrun=dryrun,
             )
 
-            deployment_utils.create_autoscaler(
+            # Create podautoscaler for non-general deployments
+            logger.debug(f'[DEBUG] Creating podautoscaler for task: {task.name}')
+            deployment_utils.create_pod_autoscaler(
                 namespace=namespace,
                 task=task,
                 dryrun=dryrun,
             )
 
-            # Apoxy resources for general deployments only when endpoint: trainy
-            if deployment_utils.get_endpoint_type_from_config() == 'trainy':
-                deployment_utils.create_apoxy_resources(
-                    namespace=namespace,
-                    task=task,
-                    dryrun=dryrun,
-                )
+            # HTTP Add-on resources for general deployments
+            logger.debug(
+                f'[DEBUG] Creating HTTP Add-on resources for task: {task.name}'
+            )
+            deployment_utils.create_http_addon_resources(
+                namespace=namespace,
+                task=task,
+                dryrun=dryrun,
+            )
 
         if not dryrun and not detach_run:
             with ux_utils.print_exception_no_traceback():