kalavai-client 0.7.7__tar.gz → 0.7.8__tar.gz

{kalavai_client-0.7.7 → kalavai_client-0.7.8}/PKG-INFO

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

kalavai_client-0.7.8/kalavai_client/__init__.py

@@ -0,0 +1,2 @@
+
+__version__ = "0.7.8"

{kalavai_client-0.7.7 → kalavai_client-0.7.8}/kalavai_client/assets/apps.yaml

@@ -174,7 +174,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

{kalavai_client-0.7.7 → kalavai_client-0.7.8}/kalavai_client/assets/default_pool_config.yaml

@@ -4,6 +4,7 @@ server:
   location: null
   name: "kalavai_cluster"
   mtu: 1280
+  watcher_image_tag: "v2025.10.3"
 
 core:
   # Deploy systems

{kalavai_client-0.7.7 → kalavai_client-0.7.8}/kalavai_client/cli.py

@@ -403,6 +403,7 @@ def pool__start(
     pool_config_file=None,
     apps: list=None,
     mtu: str=None,
+    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),

{kalavai_client-0.7.7 → kalavai_client-0.7.8}/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
@@ -764,6 +765,7 @@ 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,
@@ -799,6 +801,7 @@ 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
@@ -856,7 +859,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-0.7.7 → kalavai_client-0.7.8}/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="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
+):
     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 → kalavai_client-0.7.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name            = "kalavai-client"
-version         = "0.7.7"
+version         = "0.7.8"
 authors = [
   {name = "Carlos Fernandez Musoles", email = "carlos@kalavai.net"}
 ]

kalavai_client-0.7.7/kalavai_client/__init__.py

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