kalavai-client 0.5.24__py3-none-any.whl → 0.5.26__py3-none-any.whl

kalavai_client/__init__.py

@@ -1,2 +1,2 @@
 
-__version__ = "0.5.24"
+__version__ = "0.5.26"

kalavai_client/assets/apps.yaml

@@ -139,7 +139,7 @@ releases:
     - name: replicas
       value: 2
     - name: image_tag
-      value: "v2025.03.2"
+      value: "v2025.03.3"
     - name: deployment.in_cluster
       value: "True"
     - name: deployment.use_auth_key

kalavai_client/bridge_api.py

@@ -12,7 +12,7 @@ from kalavai_client.bridge_models import (
     DeployJobRequest,
     DeleteJobRequest,
     JobDetailsRequest,
-    DeleteNodesRequest
+    NodesActionRequest
 )
 from kalavai_client.core import (
     create_pool,
@@ -41,6 +41,8 @@ from kalavai_client.core import (
     get_ip_addresses,
     get_pool_token,
     delete_nodes,
+    cordon_nodes,
+    uncordon_nodes,
     TokenType
 )
 
@@ -85,12 +87,26 @@ def pool_stop(request: StopPoolRequest):
     return result
 
 @app.post("/delete_nodes")
-def device_delete(request: DeleteNodesRequest):
+def device_delete(request: NodesActionRequest):
     result = delete_nodes(
         nodes=request.nodes
     )
     return result
 
+@app.post("/cordon_nodes")
+def device_cordon(request: NodesActionRequest):
+    result = cordon_nodes(
+        nodes=request.nodes
+    )
+    return result
+
+@app.post("/uncordon_nodes")
+def device_uncordon(request: NodesActionRequest):
+    result = uncordon_nodes(
+        nodes=request.nodes
+    )
+    return result
+
 @app.get("/get_pool_token")
 def devices(mode: int):
 

kalavai_client/bridge_models.py

@@ -12,7 +12,7 @@ class CreatePoolRequest(BaseModel):
     only_registered_users: bool = False
     location: str = None
 
-class DeleteNodesRequest(BaseModel):
+class NodesActionRequest(BaseModel):
     nodes: list[str]
 
 class JoinPoolRequest(BaseModel):

kalavai_client/cli.py

@@ -54,6 +54,8 @@ from kalavai_client.core import (
     resume_agent,
     get_pool_token,
     delete_nodes,
+    cordon_nodes,
+    uncordon_nodes,
     TokenType
 )
 from kalavai_client.utils import (
@@ -127,27 +129,6 @@ def pre_join_check(node_name, server_url, server_key):
     except Exception as e:
         console.log(f"[red]Error when connecting to kalavai service: {str(e)}")
         return False
-    
-def set_schedulable(schedulable, node_name=load_server_info(data_key=NODE_NAME_KEY, file=USER_LOCAL_SERVER_FILE)):
-    """
-    Delete job in the cluster
-    """
-    # deploy template with kube-watcher
-    data = {
-        "schedulable": str(schedulable),
-        "node_names": [node_name]
-    }
-    try:
-        res = request_to_server(
-            method="post",
-            endpoint="/v1/set_node_schedulable",
-            data=data,
-            server_creds=USER_LOCAL_SERVER_FILE,
-            user_cookie=USER_COOKIE
-        )
-        console.log(f"{res}")
-    except Exception as e:
-        console.log(f"[red]Error when connecting to kalavai service: {str(e)}")
 
 def select_ip_address(subnet=None):
     ips = get_ip_addresses(subnet=subnet)
@@ -223,7 +204,7 @@ def input_gpus():
 ##################
 
 @arguably.command
-def gui__start(*others, backend_only=False, gui_frontend_port=3000, gui_backend_port=8000, bridge_port=8001):
+def gui__start(*others, backend_only=False, gui_frontend_port=3000, gui_backend_port=8000, bridge_port=8001, log_level="critical"):
     """Run GUI (docker) and kalavai core backend (api)"""
 
     if not backend_only:
@@ -241,7 +222,7 @@ def gui__start(*others, backend_only=False, gui_frontend_port=3000, gui_backend_
         run_cmd(f"docker compose --file {USER_GUI_COMPOSE_FILE} up -d")
 
         console.log(f"[green]Loading GUI, may take a few minutes. It will be available at http://localhost:{gui_frontend_port}")
-    run_api(port=bridge_port)
+    run_api(port=bridge_port, log_level=log_level)
     
     if not backend_only:
         run_cmd(f"docker compose --file {USER_GUI_COMPOSE_FILE} down")
@@ -958,8 +939,11 @@ def node__cordon(node_name, *others):
     except Exception as e:
         console.log(f"[red]Problems with your pool: {str(e)}")
         return
-    set_schedulable(schedulable=False, node_name=node_name)
-
+    result = cordon_nodes(nodes=[node_name])
+    if "error" in result:
+        console.log(f"[red]{result['error']}")
+    else:
+        console.log(result)
 
 @arguably.command
 def node__uncordon(node_name, *others):
@@ -971,8 +955,11 @@ def node__uncordon(node_name, *others):
     except Exception as e:
         console.log(f"[red]Problems with your pool: {str(e)}")
         return
-    set_schedulable(schedulable=True, node_name=node_name)
-
+    result = uncordon_nodes(nodes=[node_name])
+    if "error" in result:
+        console.log(f"[red]{result['error']}")
+    else:
+        console.log(result)
 
 @arguably.command
 def job__templates(*others):

kalavai_client/core.py

@@ -98,6 +98,30 @@ class TokenType(Enum):
     WORKER = 2
 
 
+def set_schedulable(schedulable, node_names):
+    """
+    Delete job in the cluster
+    """
+    # deploy template with kube-watcher
+    data = {
+        "schedulable": str(schedulable),
+        "node_names": node_names
+    }
+    try:
+        res = request_to_server(
+            method="post",
+            endpoint="/v1/set_node_schedulable",
+            data=data,
+            server_creds=USER_LOCAL_SERVER_FILE,
+            user_cookie=USER_COOKIE
+        )
+        if res is not None and "detail" in res:
+            return {"error": res["detail"]}
+        else:
+            return {"result": res}
+    except Exception as e:
+        return {"error": f"Error when connecting to kalavai service: {str(e)}"}
+
 def init_user_workspace(force_namespace=None):
     
     # load template config and populate with values
@@ -495,6 +519,12 @@ def delete_nodes(nodes):
     except Exception as e:
         return {"error": f"Error when removing nodes {nodes}: {str(e)}"}
 
+def cordon_nodes(nodes):
+    return set_schedulable(schedulable=False, node_names=nodes)
+
+def uncordon_nodes(nodes):
+    return set_schedulable(schedulable=True, node_names=nodes)
+
 def attach_to_pool(token, node_name=None):
     if node_name is None:
         node_name = f"{socket.gethostname()}-{uuid.uuid4().hex[:6]}"

{kalavai_client-0.5.24.dist-info → kalavai_client-0.5.26.dist-info}/METADATA

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

{kalavai_client-0.5.24.dist-info → kalavai_client-0.5.26.dist-info}/RECORD

@@ -1,7 +1,7 @@
-kalavai_client/__init__.py,sha256=mXfIOc2rN0BhNKx5tUIa95T0hrsXiddl_4uKwGp_7vU,23
+kalavai_client/__init__.py,sha256=yEQn5h7HJ9p7TIniys_VwtONGK1a_jUSpvxigYZe4PI,23
 kalavai_client/__main__.py,sha256=WQUfxvRsBJH5gsCJg8pLz95QnZIj7Ol8psTO77m0QE0,73
 kalavai_client/assets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kalavai_client/assets/apps.yaml,sha256=bMHyCrvdqmBvkoEm0uJDKZj7UziIopeqf7wqDBDSzao,5982
+kalavai_client/assets/apps.yaml,sha256=d13TzkWtqdwpWOxuoG7eG0Jp0UhVUsboS28496H8iH4,5982
 kalavai_client/assets/apps_values.yaml,sha256=CjKVelPQHd-hm-DTMEuya92feKiphU9mh3HrosLYYPE,1676
 kalavai_client/assets/docker-compose-gui.yaml,sha256=oE-neG3HC2PhdH-mIxrZdQlhqnycrSP_p8fRxjaxDFE,192
 kalavai_client/assets/docker-compose-template.yaml,sha256=ii24Nn-dM5cZk9lxFgrzxnmK7yv_6kIIw7KUlWhvYeI,2831
@@ -11,15 +11,15 @@ kalavai_client/assets/pool_config_values.yaml,sha256=VrM3XHQfQo6QLZ68qvagooUptaY
 kalavai_client/assets/user_workspace.yaml,sha256=wDvlMYknOPABAEo0dsQwU7bac8iubjAG9tdkFbJZ5Go,476
 kalavai_client/assets/user_workspace_values.yaml,sha256=G0HOzQUxrDMCwuW9kbWUZaKMzDDPVwDwzBHCL2Xi2ZM,542
 kalavai_client/auth.py,sha256=QsBh28L2LwjBBK6pTUE4Xu36lLDTyetyU1YfS1Hbb6g,1717
-kalavai_client/bridge_api.py,sha256=lFS1fvH7UQxZ24vrHGeea5z8bnGWhOZhgU_zKIrnlHI,4891
-kalavai_client/bridge_models.py,sha256=hWxEODqHwc8Sj21zkIZoZmU9W3PvNdK-CA1tH6yMQ4s,798
-kalavai_client/cli.py,sha256=Ev9SsnbgCU21WnxOOpSJbvqfQ1pMhIqSlk3tiQIuz7w,48099
+kalavai_client/bridge_api.py,sha256=vhkwgVOm0ZoKVz1dr-SGPQ7uMp-XPdhyzLJfKI2nsuc,5245
+kalavai_client/bridge_models.py,sha256=mEDdqUcK9BtijLpNo20HXuYdVccn4895hPzejSv666A,798
+kalavai_client/cli.py,sha256=U6Z4ESwV0L2jnKmpzEC4fKTcmrlYo4K0k9kSwUeiWL4,47700
 kalavai_client/cluster.py,sha256=gwjmdsd--YrffT0BmZDOEpbrdm3lPskUuN5jdgcrOR0,12947
-kalavai_client/core.py,sha256=c9UI_RJ7QmvWPfB9VQpc5QtNCbExOcPtGc-zed1ZgzI,30187
+kalavai_client/core.py,sha256=fhKRW5EAFwKUFdswgcuEzbSpyHAfIOrGkyghq04ckj8,31078
 kalavai_client/env.py,sha256=Zg2pP-xGJpQumo56KMBxBLgIsBmcNN0S9R-ZP2-s630,2604
 kalavai_client/utils.py,sha256=rz5W9PRZrTpgdmOs6yeqUi4f_q_L-3BJ5g1o7Asgnyo,13386
-kalavai_client-0.5.24.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-kalavai_client-0.5.24.dist-info/METADATA,sha256=ReywmarE5VApd0a8RuP50YoghGhswM7Lj1WyTozMsmk,14443
-kalavai_client-0.5.24.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-kalavai_client-0.5.24.dist-info/entry_points.txt,sha256=9T6D45gxwzfVbglMm1r6XPdXuuZdHfy_7fCeu2jUphc,50
-kalavai_client-0.5.24.dist-info/RECORD,,
+kalavai_client-0.5.26.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+kalavai_client-0.5.26.dist-info/METADATA,sha256=XlK_eljvXmKS6L2-XtvPzAuwCG6bVHZGPFEBifAOjlE,14443
+kalavai_client-0.5.26.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+kalavai_client-0.5.26.dist-info/entry_points.txt,sha256=9T6D45gxwzfVbglMm1r6XPdXuuZdHfy_7fCeu2jUphc,50
+kalavai_client-0.5.26.dist-info/RECORD,,