kalavai-client 0.6.10__py3-none-any.whl → 0.6.12__py3-none-any.whl

kalavai_client/__init__.py

@@ -1,2 +1,2 @@
 
-__version__ = "0.6.10"
+__version__ = "0.6.12"

kalavai_client/assets/apps.yaml

@@ -152,7 +152,7 @@ releases:
     - name: replicas
       value: 1
     - name: image_tag
-      value: "v2025.04.2"
+      value: "v2025.05.2"
     - name: deployment.in_cluster
       value: "True"
     - name: deployment.kalavai_username_key

kalavai_client/bridge_api.py

@@ -14,7 +14,9 @@ from kalavai_client.bridge_models import (
     DeployJobRequest,
     DeleteJobRequest,
     JobDetailsRequest,
-    NodesActionRequest
+    NodesActionRequest,
+    NodeLabelsRequest,
+    GetNodeLabelsRequest
 )
 from kalavai_client.core import (
     create_pool,
@@ -46,6 +48,8 @@ from kalavai_client.core import (
     delete_nodes,
     cordon_nodes,
     uncordon_nodes,
+    add_node_labels,
+    get_node_labels,
     TokenType
 )
 from kalavai_client.utils import load_user_id
@@ -244,6 +248,20 @@ def pool_connected(user_only: bool=False, api_key: str = Depends(verify_api_key)
     result = list_available_pools(user_only=user_only)
     return result
 
+@app.post("/add_node_labels")
+def node_labels(request: NodeLabelsRequest, api_key: str = Depends(verify_api_key)):
+    result = add_node_labels(
+        node_name=request.node_name,
+        labels=request.labels
+    )
+    return result
+
+@app.post("/get_node_labels")
+def node_labels_get(request: GetNodeLabelsRequest, api_key: str = Depends(verify_api_key)):
+    result = get_node_labels(
+        node_names=request.node_names
+    )
+    return result
 
 def run_api(host="0.0.0.0", port=8001, log_level="critical"):
     uvicorn.run(

kalavai_client/bridge_models.py

@@ -1,4 +1,5 @@
 from pydantic import BaseModel
+from typing import List, Dict, Optional
 
 from kalavai_client.core import Job, TokenType
 
@@ -42,4 +43,11 @@ class DeployJobRequest(BaseModel):
 
 class DeleteJobRequest(BaseModel):
     name: str
-    force_namespace: str = None
+    force_namespace: str = None
+
+class NodeLabelsRequest(BaseModel):
+    node_name: str
+    labels: Dict[str, str]
+
+class GetNodeLabelsRequest(BaseModel):
+    node_names: List[str]

kalavai_client/core.py

@@ -550,7 +550,7 @@ def uncordon_nodes(nodes):
 
 def attach_to_pool(token, node_name=None):
     if node_name is None:
-        node_name = f"{socket.gethostname()}-{uuid.uuid4().hex[:6]}"
+        node_name = socket.gethostname()
     
     # check token
     valid = check_token(token=token)
@@ -651,7 +651,7 @@ def join_pool(token, num_gpus=None, node_name=None, ip_address=None):
         num_gpus = get_max_gpus()
 
     if node_name is None:
-        node_name = f"{socket.gethostname()}-{uuid.uuid4().hex[:6]}"
+        node_name = socket.gethostname()
     
     # check token
     valid = check_token(token=token)
@@ -743,7 +743,7 @@ def create_pool(
     if pool_config_values is None:
         pool_config_values = POOL_CONFIG_DEFAULT_VALUES
 
-    node_name = f"{socket.gethostname()}-{uuid.uuid4().hex[:6]}"
+    node_name = socket.gethostname()
     user_id = load_user_id()
     
     node_labels = {
@@ -1058,3 +1058,62 @@ def send_invites(invitees):
         user_cookie=USER_COOKIE
     )
     return result
+
+def add_node_labels(node_name: str, labels: dict):
+    """
+    Add labels to a node in the cluster.
+    
+    Args:
+        node_name (str): Name of the node to label
+        labels (dict): Dictionary of labels to add to the node
+        
+    Returns:
+        dict: Result of the operation with either success or error message
+    """
+    data = {
+        "node_name": node_name,
+        "labels": labels
+    }
+    try:
+        result = request_to_server(
+            method="post",
+            endpoint="/v1/add_labels_to_node",
+            data=data,
+            server_creds=USER_LOCAL_SERVER_FILE,
+            user_cookie=USER_COOKIE
+        )
+        if "error" in result:
+            return {"error": result["error"]}
+        else:
+            return {"success": result}
+
+    except Exception as e:
+        return {"error": f"Error when adding labels to node {node_name}: {str(e)}"}
+
+def get_node_labels(node_names: list[str]):
+    """
+    Get labels for specified nodes in the cluster.
+    
+    Args:
+        node_names (list[str]): List of node names to fetch labels from
+        
+    Returns:
+        dict: Result containing the labels for each node or error message
+    """
+    data = {
+        "node_names": node_names
+    }
+    try:
+        result = request_to_server(
+            method="post",
+            endpoint="/v1/get_node_labels",
+            data=data,
+            server_creds=USER_LOCAL_SERVER_FILE,
+            user_cookie=USER_COOKIE
+        )
+        if result is not None:
+            return {"labels": result}
+        else:
+            return {"error": "Failed to fetch node labels"}
+    except Exception as e:
+        return {"error": f"Error when fetching node labels: {str(e)}"}

{kalavai_client-0.6.10.dist-info → kalavai_client-0.6.12.dist-info}/METADATA

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

{kalavai_client-0.6.10.dist-info → kalavai_client-0.6.12.dist-info}/RECORD

@@ -1,7 +1,7 @@
-kalavai_client/__init__.py,sha256=vHDA_1Wsc8bqniOeYrKHuU5fuCZ_w3-plFaYRacXd4Q,23
+kalavai_client/__init__.py,sha256=RnWDr4VWcRjJDXe1q7REqBSUCBo44AvIa_8rGwJtT44,23
 kalavai_client/__main__.py,sha256=WQUfxvRsBJH5gsCJg8pLz95QnZIj7Ol8psTO77m0QE0,73
 kalavai_client/assets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kalavai_client/assets/apps.yaml,sha256=Wu2Pk3k4I_S3r_rW9bNDWMWIm5gzNpmz0K6agNH3YMg,6365
+kalavai_client/assets/apps.yaml,sha256=AhTA3VZI27y05xHoHJCA9nvGnk8sWMhFDruBI2is3LM,6365
 kalavai_client/assets/apps_values.yaml,sha256=WRew3bS1MztjzcJfphuJcKn0n2T1ICRupPpr_Csjt_s,1644
 kalavai_client/assets/docker-compose-gui.yaml,sha256=DGCyGYzz1kH6kkMbo62FJHe3F9vcAmA8DOHw-c_o0Kw,752
 kalavai_client/assets/docker-compose-template.yaml,sha256=Nz_JzeBnQCzPCyWP5cEQHFeZzPwQqqBJ3C_xrToWlMA,1654
@@ -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=EB3PMvKUn5_KAQkezkEHEt-OMZXyfkZguIQlUFkEHcA,3243
-kalavai_client/bridge_api.py,sha256=KecIZvsGRiZgUQaDSgvRBLaqPDBhxFKS6FBH_w5ArCc,7015
-kalavai_client/bridge_models.py,sha256=IGPLAYEZG4f5EUtBuoR2u_cYoyCCVkhx4H2rMeGuJ_Y,1030
+kalavai_client/bridge_api.py,sha256=ZLyFOOz_o4agm-7DrHzoSBFrH65y__hZUoEe4diBTOA,7557
+kalavai_client/bridge_models.py,sha256=GbIaqGFAVs-3ikVUQZldwTTc06SsxmP6iAifH0oVDro,1219
 kalavai_client/cli.py,sha256=tBn3l12XhOJF17cWeX84xm10HPG7rSzalyOyugzcLhw,46900
 kalavai_client/cluster.py,sha256=ojUBXp2bR3hVyikIEkiGDbXvQfhBXBSk_mCqLxvyP0c,12943
-kalavai_client/core.py,sha256=bi8r7Twh2aQEDNn1shySlwx4704KG1pDmgMSSkzA1qk,32641
+kalavai_client/core.py,sha256=QOzNSxNJZ8tXhUTUPZUEFI3PX8UlVVnONYqVA61NU6U,34268
 kalavai_client/env.py,sha256=YsfZj7LWf6ABquDsoIFFkXCFYwenpDk8zVnGsf7qv98,2823
 kalavai_client/utils.py,sha256=EmjWVLjTBILjw87_6ih_v5CbVqihsYHSKxfD-C_Z-es,12276
-kalavai_client-0.6.10.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-kalavai_client-0.6.10.dist-info/METADATA,sha256=sRmVhRkTkXpWaY0UUJHGclAOZPdjX97GG9VkXI1wXFg,13354
-kalavai_client-0.6.10.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-kalavai_client-0.6.10.dist-info/entry_points.txt,sha256=9T6D45gxwzfVbglMm1r6XPdXuuZdHfy_7fCeu2jUphc,50
-kalavai_client-0.6.10.dist-info/RECORD,,
+kalavai_client-0.6.12.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+kalavai_client-0.6.12.dist-info/METADATA,sha256=ShcxZWQofBxd_VVg6LbOGaPiPDrWSix_yLIbhn7MiJ4,13354
+kalavai_client-0.6.12.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+kalavai_client-0.6.12.dist-info/entry_points.txt,sha256=9T6D45gxwzfVbglMm1r6XPdXuuZdHfy_7fCeu2jUphc,50
+kalavai_client-0.6.12.dist-info/RECORD,,

{kalavai_client-0.6.10.dist-info → kalavai_client-0.6.12.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.2
+Generator: poetry-core 2.1.3
 Root-Is-Purelib: true
 Tag: py3-none-any