kalavai-client 0.7.7__py3-none-any.whl → 0.7.9__py3-none-any.whl

kalavai_client/__init__.py

@@ -1,2 +1,2 @@
 
-__version__ = "0.7.7"
+__version__ = "0.7.9"

kalavai_client/assets/apps.yaml

@@ -45,6 +45,7 @@ releases:
     namespace: kalavai
     chart: rocm/gpu-operator-charts
     installed: {{deploy_rocm|default("false", true)}}
+    version: "v1.15.1"
   - name: lago
     namespace: kalavai
     chart: kalavai/lago
@@ -134,6 +135,7 @@ releases:
     namespace: kalavai
     chart: volcano-sh/volcano
     installed: {{deploy_volcano|default("false", true)}}
+    version: "1.13.0"
   - name: kuberay
     namespace: kuberay
     chart: kuberay/kuberay-operator
@@ -174,7 +176,7 @@ releases:
     - name: replicas
       value: 1
     - name: image_tag
-      value: "v2025.10.2" #"v2025.07.34"
+      value: "{{watcher_image_tag}}" #"v2025.07.34"
     - name: deployment.in_cluster
       value: "True"
     - name: deployment.kalavai_username_key
@@ -205,6 +207,7 @@ releases:
     namespace: kalavai
     chart: hami-charts/hami
     installed: {{deploy_hami|default("false", true)}}
+    version: "2.5.2"
     set:
     - name: resourceCores
       value: "nvidia.com/gpucores"

kalavai_client/assets/default_pool_config.yaml

@@ -3,14 +3,15 @@ server:
   ip_address: null
   location: null
   name: "kalavai_cluster"
-  mtu: 1280
+  mtu: ""
+  watcher_image_tag: "v2025.10.3"
 
 core:
   # Deploy systems
   deploy:
   - volcano
   - cert_manager
-  #- rocm
+  - rocm
   - watcher
   - hami
   #- lago

kalavai_client/assets/docker-compose-template.yaml

@@ -60,7 +60,9 @@ services:
       {% if node_labels %}
       --extra="{{node_labels}}"
       {% endif %}
+      {% if mtu != "" %}
       --mtu={{mtu}}
+      {% endif %}
 
     # volumes:
     # - {{k3s_path}}:/var/lib/rancher/k3s # Persist data

kalavai_client/cli.py

@@ -402,7 +402,8 @@ def pool__start(
     *others, 
     pool_config_file=None,
     apps: list=None,
-    mtu: str=None,
+    mtu: str="",
+    watcher_image_tag: str=None,
     platform="amd64",
     ip_address: str=None,
     location: str=None,
@@ -454,6 +455,7 @@ def pool__start(
         ip_address=ip_address,
         location=location,
         target_platform=platform,
+        watcher_image_tag=watcher_image_tag,
         pool_config_file=pool_config_file,
         apps=apps,
         num_gpus=input_gpus(non_interactive=non_interactive),
@@ -519,7 +521,7 @@ def pool__check_token(token, *others, public=False, verbose=False):
 def pool__join(
     token,
     *others,
-    mtu=None,
+    mtu="",
     platform="amd64",
     node_name=None,
     non_interactive=False,

kalavai_client/core.py

@@ -47,6 +47,7 @@ from kalavai_client.utils import (
     WRITE_AUTH_KEY,
     WATCHER_PORT_KEY,
     WATCHER_SERVICE_KEY,
+    WATCHER_IMAGE_TAG_KEY,
     USER_NODE_LABEL_KEY,
     ALLOW_UNREGISTERED_USER_KEY,
     KALAVAI_AUTH
@@ -671,7 +672,7 @@ def join_pool(
         node_name=None,
         ip_address=None,
         target_platform="amd64",
-        mtu="1420",
+        mtu="",
         node_labels={}
 ):
     compatibility = check_worker_compatibility()
@@ -683,9 +684,6 @@ def join_pool(
 
     if node_name is None:
         node_name = socket.gethostname()
-
-    if mtu is None:
-        mtu = "1420"
     
     # check token
     valid = check_token(token=token)
@@ -764,12 +762,13 @@ def create_pool(
         ip_address: str=None,
         location: str=None,
         target_platform: str="amd64",
+        watcher_image_tag: str=None,
         pool_config_file: str=None,
         description: str="",
         token_mode: TokenType=TokenType.USER,
         num_gpus: int=-1,
         node_name: str=None,
-        mtu: str=None,
+        mtu: str="",
         apps: list=[],
         node_labels: dict={}
     ):
@@ -799,11 +798,12 @@ def create_pool(
         config_values = yaml.safe_load(f)
     # use default values if not provided
     try:
+        watcher_image_tag = config_values["server"]["watcher_image_tag"] if watcher_image_tag is None else watcher_image_tag
         cluster_name = config_values["server"]["name"] if cluster_name is None else cluster_name
         ip_address = config_values["server"]["ip_address"] if ip_address is None else ip_address
         location = config_values["server"]["location"] if location is None else location
         target_platform = config_values["server"]["platform"] if target_platform is None else target_platform
-        mtu = config_values["server"]["mtu"] if mtu is None else mtu
+        mtu = config_values["server"]["mtu"] if mtu == "" or mtu is None else mtu
         app_values = config_values["core"]
         post_config_values = config_values["pool"]
         deploy_apps = {
@@ -856,7 +856,8 @@ def create_pool(
         WATCHER_PORT_KEY: DEFAULT_WATCHER_PORT,
         WATCHER_SERVICE_KEY: watcher_service,
         USER_NODE_LABEL_KEY: USER_NODE_LABEL,
-        ALLOW_UNREGISTERED_USER_KEY: True, # Change this if only registered users are allowed
+        WATCHER_IMAGE_TAG_KEY: watcher_image_tag,
+        ALLOW_UNREGISTERED_USER_KEY: True # Change this if only registered users are allowed
     }
 
     store_server_info(

kalavai_client/utils.py

@@ -44,6 +44,7 @@ USER_API_KEY = "user_api_key"
 READONLY_AUTH_KEY = "watcher_readonly_key"
 WATCHER_SERVICE_KEY = "watcher_service"
 WATCHER_PORT_KEY = "watcher_port"
+WATCHER_IMAGE_TAG_KEY = "watcher_image_tag"
 ENDPOINT_PORTS_KEY = "endpoint_ports"
 TEMPLATE_ID_FIELD = "id_field"
 TEMPLATE_ID_KEY = "deployment_id"
@@ -155,8 +156,19 @@ def extract_auth_token(headers):
         return {"error": str(e)}
 
 
-def generate_compose_config(role, node_name, mtu="1420", target_platform="amd64", write_to_file=True, node_ip_address="0.0.0.0", num_gpus=0, node_labels=None, pool_ip=None, vpn_token=None, pool_token=None):
-    
+def generate_compose_config(
+    role,
+    node_name,
+    mtu="",
+    target_platform="amd64",
+    write_to_file=True,
+    node_ip_address="0.0.0.0",
+    num_gpus=0,
+    node_labels=None,
+    pool_ip=None,
+    vpn_token=None,
+    pool_token=None
+):
     if node_labels is not None:
         node_labels = " ".join([f"--node-label {key}={value}" for key, value in node_labels.items()])
     rand_suffix = uuid.uuid4().hex[:8]

{kalavai_client-0.7.7.dist-info → kalavai_client-0.7.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kalavai-client
-Version: 0.7.7
+Version: 0.7.9
 Summary: Client app for kalavai platform
 License-Expression: Apache-2.0
 License-File: LICENSE

{kalavai_client-0.7.7.dist-info → kalavai_client-0.7.9.dist-info}/RECORD

@@ -1,10 +1,10 @@
-kalavai_client/__init__.py,sha256=_he-4_uHDJNaqLXyzYdBG_rgSv3K0P5DCywjUCh7v5c,22
+kalavai_client/__init__.py,sha256=J9AfUW8mWqmQTjAy5u_qrftQuhEY12KMdAOeZLcahBM,22
 kalavai_client/__main__.py,sha256=WQUfxvRsBJH5gsCJg8pLz95QnZIj7Ol8psTO77m0QE0,73
 kalavai_client/assets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kalavai_client/assets/apps.yaml,sha256=juO6kIl8LjykWI-BPsALSdADc9cDE73t8wP2W6ll2co,7087
-kalavai_client/assets/default_pool_config.yaml,sha256=FLfT0i-xPgEZhleTOZmfyFYrB9fAEMOmZ0YGLyJr0OA,1879
+kalavai_client/assets/apps.yaml,sha256=vvO_F7gA7tdcJK9_4HkZZaKNkac-1dMhQFDFVPdekfQ,7164
+kalavai_client/assets/default_pool_config.yaml,sha256=AqBmNr5GA0baLYv-c23Ru_JeezZpG0em3tA78w7YBBM,1910
 kalavai_client/assets/docker-compose-gui.yaml,sha256=OAVO0ohaCpDB9FGeih0yAbVNwUfDtaCzssZ25uiuJyA,787
-kalavai_client/assets/docker-compose-template.yaml,sha256=8giwNUVbVGxF_YppNMIsS7evWZ8qb4n2enK22u-x6Pk,1920
+kalavai_client/assets/docker-compose-template.yaml,sha256=3OZC3vDLCkG9RQ93UdrO5Aynd3hXQWPpiiXQn9Oi-2M,1963
 kalavai_client/assets/model_deployment_values.yaml,sha256=PKD9DGJnRRwYfk6pO4cmGN9gIgylHX3IFW2sDweEGdU,1698
 kalavai_client/assets/nginx.conf,sha256=drVVCg8GHucz7hmt_BI6giAhK92OV71257NTs3LthwM,225
 kalavai_client/assets/pool_config_template.yaml,sha256=MhBZQsEMKrBgbUVSKgIGmXWhybeGKG6l5XvJb38y5GI,577
@@ -13,13 +13,13 @@ kalavai_client/assets/user_workspace_values.yaml,sha256=G0HOzQUxrDMCwuW9kbWUZaKM
 kalavai_client/auth.py,sha256=EB3PMvKUn5_KAQkezkEHEt-OMZXyfkZguIQlUFkEHcA,3243
 kalavai_client/bridge_api.py,sha256=0TvAGgsyfMkbcVqoPKsjhXQLo06WetBH93mZH-pOM7U,26921
 kalavai_client/bridge_models.py,sha256=bq6vQNTI1py7e_1YgnBZhorFsAKoBqBVN7nRukCuQRE,2960
-kalavai_client/cli.py,sha256=yGCOS-eI01E_EdxWFG9IgUZnSQyjaEV8NqrX2M-YcZw,49010
+kalavai_client/cli.py,sha256=82bXvjvRZmyAzJAqNVsaG4y_IoVK-OFP23uVpdu0KCo,49084
 kalavai_client/cluster.py,sha256=Z2PIXbZuSAv9xmw-MyZP1M41BpVMpirLzG51bqGA-zc,13548
-kalavai_client/core.py,sha256=ElDeF0AjzoZaECUg1RbwolJItuX6b71i03Gr9gIXLfY,36529
+kalavai_client/core.py,sha256=OpFLnmJgXuk20l1c1jukatDSKm1Vc1xkj6cPpRy680U,36732
 kalavai_client/env.py,sha256=0L5gfEo5KY8gflrW-rSADx10ffDa-8gXmmrGWztKUd8,3099
-kalavai_client/utils.py,sha256=swMB-3elI20GihzZgqSFD5sQfQ-Nh2_TGbEnKNHx7EU,14230
-kalavai_client-0.7.7.dist-info/METADATA,sha256=o6ywphe6vy6cDQI7RuRhBFjcW5-KZOYV_xMCFrPmpMc,13175
-kalavai_client-0.7.7.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-kalavai_client-0.7.7.dist-info/entry_points.txt,sha256=9T6D45gxwzfVbglMm1r6XPdXuuZdHfy_7fCeu2jUphc,50
-kalavai_client-0.7.7.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-kalavai_client-0.7.7.dist-info/RECORD,,
+kalavai_client/utils.py,sha256=xbHsPgyvsA7X2vTl7OIMLe6MMFTYK1-iiSS_VmWJ-Rg,14311
+kalavai_client-0.7.9.dist-info/METADATA,sha256=-9WMQI9veuF8cb5ot91fTTNFvSOXjUBCkcjpeoVhKtg,13175
+kalavai_client-0.7.9.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+kalavai_client-0.7.9.dist-info/entry_points.txt,sha256=9T6D45gxwzfVbglMm1r6XPdXuuZdHfy_7fCeu2jUphc,50
+kalavai_client-0.7.9.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+kalavai_client-0.7.9.dist-info/RECORD,,