konduktor-nightly 0.1.0.dev20250606105011__py3-none-any.whl → 0.1.0.dev20250608104639__py3-none-any.whl

konduktor/__init__.py

@@ -14,7 +14,7 @@ __all__ = [
 ]
 
 # Replaced with the current commit when building the wheels.
-_KONDUKTOR_COMMIT_SHA = 'b5ac51935edc296ff721f11d112e64438b366608'
+_KONDUKTOR_COMMIT_SHA = 'b607feee1f4c97beba242d70181c1f72582e52d2'
 os.makedirs(os.path.expanduser('~/.konduktor'), exist_ok=True)
 
 
@@ -48,5 +48,5 @@ def _get_git_commit():
 
 
 __commit__ = _get_git_commit()
-__version__ = '1.0.0.dev0.1.0.dev20250606105011'
+__version__ = '1.0.0.dev0.1.0.dev20250608104639'
 __root_dir__ = os.path.dirname(os.path.abspath(__file__))

konduktor/backends/jobset_utils.py

@@ -41,6 +41,7 @@ JOBSET_USER_LABEL = 'trainy.ai/username'
 JOBSET_ACCELERATOR_LABEL = 'trainy.ai/accelerator'
 JOBSET_NUM_ACCELERATORS_LABEL = 'trainy.ai/num-accelerators'
 
+SECRET_BASENAME_LABEL = 'konduktor/basename'
 
 _JOBSET_METADATA_LABELS = {
     'jobset_name_label': JOBSET_NAME_LABEL,
@@ -141,10 +142,10 @@ def create_pod_spec(task: 'konduktor.Task') -> Dict[str, Any]:
         storage_secrets[store_scheme] = cloud_store._STORE.get_k8s_credential_name()
         sync_commands.append(cloud_store.make_sync_file_command(src, dst))
 
+    context = kubernetes_utils.get_current_kube_config_context_name()
+    namespace = kubernetes_utils.get_kube_config_context_namespace(context)
     tailscale_secret = config.get_nested(('tailscale', 'secret_name'), None)
     if tailscale_secret:
-        context = kubernetes_utils.get_current_kube_config_context_name()
-        namespace = kubernetes_utils.get_kube_config_context_namespace(context)
         secret_exist, err = kubernetes_utils.check_secret_exists(
             tailscale_secret, namespace, context
         )
@@ -165,10 +166,6 @@ def create_pod_spec(task: 'konduktor.Task') -> Dict[str, Any]:
         ):
             private_key, public_key = private_key_file.read(), public_key_file.read()
             user_hash = common_utils.get_user_hash()
-            context = kubernetes_utils.get_current_kube_config_context_name()
-            namespace = kubernetes_utils.get_kube_config_context_namespace(
-                context_name=context
-            )
             secret_name = f'konduktor-ssh-keys-{user_hash}'
             ok, result = kubernetes_utils.set_secret(
                 secret_name=secret_name,
@@ -187,6 +184,7 @@ def create_pod_spec(task: 'konduktor.Task') -> Dict[str, Any]:
     # Mount the user's secrets
     git_ssh_secret_name = None
     env_secret_envs = []
+    default_secrets = []
 
     context = kubernetes_utils.get_current_kube_config_context_name()
     namespace = kubernetes_utils.get_kube_config_context_namespace(context)
@@ -209,6 +207,10 @@ def create_pod_spec(task: 'konduktor.Task') -> Dict[str, Any]:
                     'valueFrom': {'secretKeyRef': {'name': secret_name, 'key': key}},
                 }
             )
+        elif kind == 'default':
+            secret_name = secret.metadata.name
+            basename = secret.metadata.labels.get(SECRET_BASENAME_LABEL, secret_name)
+            default_secrets.append({'k8s_name': secret_name, 'mount_name': basename})
 
     with tempfile.NamedTemporaryFile() as temp:
         common_utils.fill_template(
@@ -239,6 +241,8 @@ def create_pod_spec(task: 'konduktor.Task') -> Dict[str, Any]:
                 # Kinds of Secrets
                 # --kind git-ssh
                 'git_ssh': git_ssh_secret_name,
+                # --kind default
+                'default_secrets': default_secrets,
             },
             temp.name,
         )
@@ -318,7 +322,8 @@ def create_jobset(
         'template'
     ] = pod_spec  # noqa: E501
     try:
-        jobset = kube_client.crd_api().create_namespaced_custom_object(
+        context = kubernetes_utils.get_current_kube_config_context_name()
+        jobset = kube_client.crd_api(context=context).create_namespaced_custom_object(
             group=JOBSET_API_GROUP,
             version=JOBSET_API_VERSION,
             namespace=namespace,
@@ -348,7 +353,8 @@ def create_jobset(
 def list_jobset(namespace: str) -> Optional[Dict[str, Any]]:
     """Lists all jobsets in this namespace"""
     try:
-        response = kube_client.crd_api().list_namespaced_custom_object(
+        context = kubernetes_utils.get_current_kube_config_context_name()
+        response = kube_client.crd_api(context=context).list_namespaced_custom_object(
             group=JOBSET_API_GROUP,
             version=JOBSET_API_VERSION,
             namespace=namespace,
@@ -372,7 +378,8 @@ def list_jobset(namespace: str) -> Optional[Dict[str, Any]]:
 def get_jobset(namespace: str, job_name: str) -> Optional[Dict[str, Any]]:
     """Retrieves jobset in this namespace"""
     try:
-        response = kube_client.crd_api().get_namespaced_custom_object(
+        context = kubernetes_utils.get_current_kube_config_context_name()
+        response = kube_client.crd_api(context=context).get_namespaced_custom_object(
             group=JOBSET_API_GROUP,
             version=JOBSET_API_VERSION,
             namespace=namespace,
@@ -409,7 +416,8 @@ def delete_jobset(namespace: str, job_name: str) -> Optional[Dict[str, Any]]:
         Response from delete operation
     """
     try:
-        response = kube_client.crd_api().delete_namespaced_custom_object(
+        context = kubernetes_utils.get_current_kube_config_context_name()
+        response = kube_client.crd_api(context=context).delete_namespaced_custom_object(
             group=JOBSET_API_GROUP,
             version=JOBSET_API_VERSION,
             namespace=namespace,
@@ -446,7 +454,8 @@ def get_job(namespace: str, job_name: str) -> Optional[Dict[str, Any]]:
         # Get the job object using the job name
         # pattern {jobset-name}-workers-0-{worker_id}
         job_name = f'{job_name}-workers-0'
-        response = kube_client.batch_api().read_namespaced_job(
+        context = kubernetes_utils.get_current_kube_config_context_name()
+        response = kube_client.batch_api(context=context).read_namespaced_job(
             name=job_name, namespace=namespace
         )
         return response

konduktor/cli.py

@@ -853,12 +853,11 @@ def secret():
       konduktor secret create --kind git-ssh --from-file=~/.ssh/id_rsa my-ssh-name
       konduktor secret create --kind env --inline FOO=bar my-env-name
       konduktor delete my-ssh-name
-      konduktor list
+      konduktor secret list
 
     \b
     For details on COMMAND ARGS:
       konduktor secret create -h
-      konduktor secret delete -h
       konduktor secret list -h
     """
 

konduktor/templates/pod.yaml.j2

@@ -76,6 +76,10 @@ kubernetes:
           - name: GIT_SSH_COMMAND
             value: "ssh -i /run/konduktor/git-ssh-secret/gitkey -o StrictHostKeyChecking=no"
           {% endif %}
+          {% if default_secrets %}
+          - name: KONDUKTOR_DEFAULT_SECRETS
+            value: "/konduktor/default-secrets"
+          {% endif %}
           # these are for compatibility with skypilot
           - name: SKYPILOT_NODE_IPS
             value: "{{ node_hostnames }}"
@@ -96,6 +100,10 @@ kubernetes:
           - name: {{ secret_type }}-secret
             mountPath: /run/konduktor/{{ secret_type }}-secret
           {% endfor %}
+          {% for secret in default_secrets %}
+          - name: default-secret-{{ secret.mount_name }}
+            mountPath: /konduktor/default-secrets/{{ secret.mount_name }}
+          {% endfor %}
           {% if git_ssh %}
           - name: git-ssh-secret
             mountPath: /run/konduktor/git-ssh-secret
@@ -348,6 +356,11 @@ kubernetes:
         secret:
           secretName: {{ secret_name }}
       {% endfor %}
+      {% for secret in default_secrets %}
+      - name: default-secret-{{ secret.mount_name }}
+        secret:
+          secretName: {{ secret.k8s_name }}
+      {% endfor %}
       {% if git_ssh %}
       - name: git-ssh-secret
         secret:

{konduktor_nightly-0.1.0.dev20250606105011.dist-info → konduktor_nightly-0.1.0.dev20250608104639.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: konduktor-nightly
-Version: 0.1.0.dev20250606105011
+Version: 0.1.0.dev20250608104639
 Summary: GPU Cluster Health Management
 Author: Andrew Aikawa
 Author-email: asai@berkeley.edu

{konduktor_nightly-0.1.0.dev20250606105011.dist-info → konduktor_nightly-0.1.0.dev20250608104639.dist-info}/RECORD

@@ -1,4 +1,4 @@
-konduktor/__init__.py,sha256=6RoR_j8vjqRC1nwAEhIsUnNMR-X7DiYN2kcQNdz7llk,1540
+konduktor/__init__.py,sha256=8nTGii1ywhKw3UAf0UutnAzRtL9TPS4dOmcBOZ8nKYo,1540
 konduktor/adaptors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 konduktor/adaptors/aws.py,sha256=s47Ra-GaqCQibzVfmD0pmwEWHif1EGO5opMbwkLxTCU,8244
 konduktor/adaptors/common.py,sha256=ZIqzjx77PIHUwpjfAQ1uX8B2aX78YMuGj4Bppd-MdyM,4183
@@ -7,9 +7,9 @@ konduktor/authentication.py,sha256=_mVy3eqoKohicHostFiGwG1-2ybxP-l7ouofQ0LRlCY,4
 konduktor/backends/__init__.py,sha256=1Q6sqqdeMYarpTX_U-QVywJYf7idiUTRsyP-E4BQSOw,129
 konduktor/backends/backend.py,sha256=qh0bp94lzoTYZkzyQv2-CVrB5l91FkG2vclXg24UFC0,2910
 konduktor/backends/jobset.py,sha256=UdhwAuZODLMbLY51Y2zOBsh6wg4Pb84oHVvUKzx3Z2w,8434
-konduktor/backends/jobset_utils.py,sha256=DSdWdCUPdTh9EzFG0z_GoHVeCW49kLwTujaKjC1ko3I,21430
+konduktor/backends/jobset_utils.py,sha256=zBo-DTYbOLppH7fYSEhYqOjeFJz_RYNoid-dImu_Uj0,22068
 konduktor/check.py,sha256=JennyWoaqSKhdyfUldd266KwVXTPJpcYQa4EED4a_BA,7569
-konduktor/cli.py,sha256=4eYDqSvwEn38rDNk6fKiSk4BwdmSna2XonUrI9-o7w0,33903
+konduktor/cli.py,sha256=i5vXN_p21Tj7etX_QIkq1HCJ3I2Pn_OpKdddBMqRR-g,33877
 konduktor/config.py,sha256=J50JxC6MsXMnlrJPXdDUMr38C89xvOO7mR8KJ6fyils,15520
 konduktor/constants.py,sha256=T3AeXXxuQHINW_bAWyztvDeS8r4g8kXBGIwIq13cys0,1814
 konduktor/controller/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -71,7 +71,7 @@ konduktor/manifests/pod_cleanup_controller.yaml,sha256=hziL1Ka1kCAEL9R7Tjvpb80iw
 konduktor/resource.py,sha256=Fg4kon7jQ9xDo9Iz8Q0J8doIRmTkSwIhYXLH6jbtRO8,19610
 konduktor/task.py,sha256=ofwd8WIhfD6C3ThLcv6X3GUzQHyZ6ddjUagE-umF4K0,35207
 konduktor/templates/jobset.yaml.j2,sha256=rdURknodtgLp4zoA2PX86Nn4wPpi3tr5l4IG55aWBRg,1059
-konduktor/templates/pod.yaml.j2,sha256=7KyUy4orakJ8nI7ee8AIGGY4HHfhs-6zwUlAzhekZHw,16112
+konduktor/templates/pod.yaml.j2,sha256=SlK6XKSwjuFJtBimlrUiFTcx7G_00XDtEopIKXBg5SI,16635
 konduktor/usage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 konduktor/usage/constants.py,sha256=gCL8afIHZhO0dcxbJGpESE9sCC1cBSbeRnQ8GwNOY4M,612
 konduktor/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -91,8 +91,8 @@ konduktor/utils/schemas.py,sha256=VGPERAso2G4sVAznsJ80qT2Q-I_EFxXw6Rfcw-vkYgQ,16
 konduktor/utils/subprocess_utils.py,sha256=WoFkoFhGecPR8-rF8WJxbIe-YtV94LXz9UG64SDhCY4,9448
 konduktor/utils/ux_utils.py,sha256=czCwiS1bDqgeKtzAJctczpLwFZzAse7WuozdvzEFYJ4,7437
 konduktor/utils/validator.py,sha256=tgBghVyedyzGx84-U2Qfoh_cJBE3oUk9gclMW90ORks,691
-konduktor_nightly-0.1.0.dev20250606105011.dist-info/LICENSE,sha256=MuuqTZbHvmqXR_aNKAXzggdV45ANd3wQ5YI7tnpZhm0,6586
-konduktor_nightly-0.1.0.dev20250606105011.dist-info/METADATA,sha256=yQ3_OsedDo68gju4ZF2vOa2OWe364dIgcep-BjqL_9c,4289
-konduktor_nightly-0.1.0.dev20250606105011.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-konduktor_nightly-0.1.0.dev20250606105011.dist-info/entry_points.txt,sha256=k3nG5wDFIJhNqsZWrHk4d0irIB2Ns9s47cjRWYsTCT8,48
-konduktor_nightly-0.1.0.dev20250606105011.dist-info/RECORD,,
+konduktor_nightly-0.1.0.dev20250608104639.dist-info/LICENSE,sha256=MuuqTZbHvmqXR_aNKAXzggdV45ANd3wQ5YI7tnpZhm0,6586
+konduktor_nightly-0.1.0.dev20250608104639.dist-info/METADATA,sha256=CQ9w6OfmAc9woXXE9jl18nTqOYgudGkRYX-dXUzZPh4,4289
+konduktor_nightly-0.1.0.dev20250608104639.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+konduktor_nightly-0.1.0.dev20250608104639.dist-info/entry_points.txt,sha256=k3nG5wDFIJhNqsZWrHk4d0irIB2Ns9s47cjRWYsTCT8,48
+konduktor_nightly-0.1.0.dev20250608104639.dist-info/RECORD,,