kalavai-client 0.6.18__tar.gz → 0.6.19__tar.gz

{kalavai_client-0.6.18 → kalavai_client-0.6.19}/PKG-INFO

@@ -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.19/kalavai_client/__init__.py

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

{kalavai_client-0.6.18 → kalavai_client-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-0.6.18 → kalavai_client-0.6.19}/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-0.6.18 → kalavai_client-0.6.19}/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-0.6.18 → kalavai_client-0.6.19}/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 → kalavai_client-0.6.19}/pyproject.toml

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

kalavai_client-0.6.18/kalavai_client/__init__.py

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