kalavai-client 0.6.18__py3-none-any.whl → 0.6.19__py3-none-any.whl

kalavai_client/__init__.py

@@ -1,2 +1,2 @@
 
-__version__ = "0.6.18"
+__version__ = "0.6.19"

kalavai_client/assets/docker-compose-gui.yaml

@@ -1,7 +1,7 @@
 services:
   kalavai_gui:
     container_name: kalavai_gui
-    image: bundenth/kalavai-gui:latest
+    image: bundenth/kalavai-gui:{{gui_version}}
     platform: linux/amd64
     extra_hosts:
     - "host.docker.internal:host-gateway"

kalavai_client/cli.py

@@ -58,7 +58,8 @@ from kalavai_client.core import (
     stop_pool,
     uncordon_nodes,
     TokenType,
-    unregister_pool
+    unregister_pool,
+    update_pool
 )
 from kalavai_client.utils import (
     check_gpu_drivers,
@@ -208,7 +209,8 @@ def input_gpus(non_interactive=False):
 def gui__start(
     *others,
     log_level="critical",
-    backend_only=False
+    backend_only=False,
+    version="latest"
 ):
     """Run GUI (docker) and kalavai core backend (api)"""
     ports_needed = 1 if backend_only else 3
@@ -242,7 +244,8 @@ def gui__start(
             "gui_backend_port": ports[2],
             "bridge_port": ports[0],
             "path": user_path("", create_path=True),
-            "protected_access": user_key
+            "protected_access": user_key,
+            "gui_version": version
         }
         compose_yaml = load_template(
             template_path=DOCKER_COMPOSE_GUI,
@@ -679,23 +682,12 @@ def pool__update(*others):
     """
     Update kalavai pool
     """
-    try:
-        CLUSTER.validate_cluster()
-    except Exception as e:
-        console.log(f"[red]Problems with your pool: {str(e)}")
-        return
-    
-    if not CLUSTER.is_seed_node():
-        console.log("You can only update a pool from the seed node.")
-        return
-    
-    # update dependencies
-    try:
-        CLUSTER.update_dependencies(debug=True)
-        console.log("Pool updating. Expect some downtime on core services")
-    except Exception as e:
-        console.log(f"[red]Error when updating pool: {str(e)}")
-        return
+    result = update_pool(debug=True)
+
+    if "error" in result:
+        console.log(f"[red]{result['error']}")
+    else:
+        console.log(f"[green]{result}")
 
 
 @arguably.command

kalavai_client/core.py

@@ -39,6 +39,7 @@ from kalavai_client.utils import (
     CLUSTER_IP_KEY,
     CLUSTER_NAME_KEY,
     AUTH_KEY,
+    USER_ID_KEY,
     WATCHER_SERVICE_KEY,
     CLUSTER_TOKEN_KEY,
     READONLY_AUTH_KEY,
@@ -809,6 +810,7 @@ def create_pool(
     values = {
         CLUSTER_NAME_KEY: cluster_name,
         CLUSTER_IP_KEY: ip_address,
+        USER_ID_KEY: user_id if user_id is not None else "",
         AUTH_KEY: auth_key,
         READONLY_AUTH_KEY: readonly_auth_key,
         WRITE_AUTH_KEY: write_auth_key,
@@ -884,6 +886,22 @@ def create_pool(
     
     return {"success"}
 
+def update_pool(debug=True):
+    try:
+        CLUSTER.validate_cluster()
+    except Exception as e:
+        return {"error": f"Problems with your pool: {str(e)}"}
+    
+    if not CLUSTER.is_seed_node():
+        return {"error": "You can only update a pool from the seed node."}
+    
+    # update dependencies
+    try:
+        CLUSTER.update_dependencies(debug=debug)
+        return {"success": "Pool updating. Expect some downtime on core services"}
+    except Exception as e:
+        return {"error": f"[red]Error when updating pool: {str(e)}"}
+
 
 def get_pool_token(mode: TokenType):
 

kalavai_client/utils.py

@@ -36,6 +36,7 @@ NODE_NAME_KEY = "node_name"
 PUBLIC_LOCATION_KEY = "public_location"
 CLUSTER_NAME_KEY = "cluster_name"
 AUTH_KEY = "watcher_admin_key"
+USER_ID_KEY = "kalavai_user_id"
 WRITE_AUTH_KEY = "watcher_write_key"
 ALLOW_UNREGISTERED_USER_KEY = "watcher_allow_unregistered_user"
 NODE_ROLE_LABEL = "kalavai.node_role"

{kalavai_client-0.6.18.dist-info → kalavai_client-0.6.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kalavai-client
-Version: 0.6.18
+Version: 0.6.19
 Summary: Client app for kalavai platform
 License: Apache-2.0
 Keywords: LLM,platform

{kalavai_client-0.6.18.dist-info → kalavai_client-0.6.19.dist-info}/RECORD

@@ -1,9 +1,9 @@
-kalavai_client/__init__.py,sha256=ErdtY8HWYl_n6MmTR8hlH878NUP9glayQd1egl02vKY,23
+kalavai_client/__init__.py,sha256=29RPUdF-Jn8Tqqu5Mk5Ci4E0On_MZIVUBqNZv5nyu0s,23
 kalavai_client/__main__.py,sha256=WQUfxvRsBJH5gsCJg8pLz95QnZIj7Ol8psTO77m0QE0,73
 kalavai_client/assets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kalavai_client/assets/apps.yaml,sha256=17JuXSv-Qj5Az6ZTRyiEaQXVbI325uTrZzKk2irts2g,6410
 kalavai_client/assets/apps_values.yaml,sha256=LeSNd3PwkIx0wkTIlEk2KNz3Yy4sXSaHALQEkopdhKE,2165
-kalavai_client/assets/docker-compose-gui.yaml,sha256=shqN78YLw0QP7bqTKveI4ppz5E-5b1JowmsSB4OG3nA,778
+kalavai_client/assets/docker-compose-gui.yaml,sha256=OAVO0ohaCpDB9FGeih0yAbVNwUfDtaCzssZ25uiuJyA,787
 kalavai_client/assets/docker-compose-template.yaml,sha256=KHIwJ2WWX7Y7wQKiXRr82Jqd3IKRyls5zhTyl8mSmrc,1805
 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=Hd7whTX2TAiNYX1G237hv2rqtKUBGRJkzUoWOMZm44A,25562
 kalavai_client/bridge_models.py,sha256=3mHCqIHVysLLkQvGT-DKqKOrtAlQSfEOdrwSq2yTRRU,2439
-kalavai_client/cli.py,sha256=SzKG7_ZG0ehMQsECQRWSvqj2Fju2Gd5O7uBa60bFBAY,47830
+kalavai_client/cli.py,sha256=UC2aRUvNI53Xq-ubyoKDoVaRS1VDRmKtU9sVMf9C5Sg,47522
 kalavai_client/cluster.py,sha256=Z2PIXbZuSAv9xmw-MyZP1M41BpVMpirLzG51bqGA-zc,13548
-kalavai_client/core.py,sha256=weg54lc03gp2qGwEXl90XEnXGdwFFlaTqZjxyKsngj4,34765
+kalavai_client/core.py,sha256=haNLna0TWzxmGx9cEhJjnV3r9YSOS3Fhtr4dt70LnwQ,35390
 kalavai_client/env.py,sha256=YsfZj7LWf6ABquDsoIFFkXCFYwenpDk8zVnGsf7qv98,2823
-kalavai_client/utils.py,sha256=GeX1rKUdlQoOW_K2relER8jRQEN1M0UdhsLKOkv5D_g,13428
-kalavai_client-0.6.18.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-kalavai_client-0.6.18.dist-info/METADATA,sha256=nCvnC5f8QM1sHV4wk3HI9YjH0c_vkpOIkcVpMFIKEx0,12393
-kalavai_client-0.6.18.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-kalavai_client-0.6.18.dist-info/entry_points.txt,sha256=9T6D45gxwzfVbglMm1r6XPdXuuZdHfy_7fCeu2jUphc,50
-kalavai_client-0.6.18.dist-info/RECORD,,
+kalavai_client/utils.py,sha256=5cUpMVsADF3JdDW0wbu-f38MURkhQz9pPngg0WxssJw,13460
+kalavai_client-0.6.19.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+kalavai_client-0.6.19.dist-info/METADATA,sha256=CYVNusQKxd6KHa0UCx3QBDytEXGApHg2OrZd7O5LfIU,12393
+kalavai_client-0.6.19.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+kalavai_client-0.6.19.dist-info/entry_points.txt,sha256=9T6D45gxwzfVbglMm1r6XPdXuuZdHfy_7fCeu2jUphc,50
+kalavai_client-0.6.19.dist-info/RECORD,,