kalavai-client 0.5.4__tar.gz → 0.5.5__tar.gz

{kalavai_client-0.5.4 → kalavai_client-0.5.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kalavai-client
-Version: 0.5.4
+Version: 0.5.5
 Summary: Client app for kalavai platform
 License: Apache-2.0
 Keywords: LLM,platform
@@ -26,7 +26,6 @@ Requires-Dist: Pillow (==10.3.0)
 Requires-Dist: anvil-uplink (==0.5.1)
 Requires-Dist: arguably (>=1.2.5)
 Requires-Dist: build ; extra == "dev"
-Requires-Dist: gputil (==1.4.0)
 Requires-Dist: importlib_resources (==6.5.2)
 Requires-Dist: jinja2 (==3.1.4)
 Requires-Dist: netifaces (==0.11.0)
@@ -36,7 +35,6 @@ Requires-Dist: pyinstaller (==6.5.0) ; extra == "dev"
 Requires-Dist: pyyaml (==6.0.2)
 Requires-Dist: requests (>=2.25)
 Requires-Dist: rich (==13.7.1)
-Requires-Dist: setuptools (<70.0.0)
 Requires-Dist: twine ; extra == "dev"
 Project-URL: Homepage, https://platform.kalavai.net
 Project-URL: Website, https://kalavai.net
@@ -145,6 +143,18 @@ From release **v0.5.0, you can now install `kalavai` client in non-worker comput
 
 - Python 3.10+
 
+If you see the following error:
+
+```bash
+fatal error: Python.h: No such file or directory | #include <Python.h>
+```
+
+Make sure you also install python3-dev package. For ubuntu distros:
+
+```bash
+sudo apt install python3-dev
+```
+
 
 ### Install the client
 

{kalavai_client-0.5.4 → kalavai_client-0.5.5}/README.md

@@ -101,6 +101,18 @@ From release **v0.5.0, you can now install `kalavai` client in non-worker comput
 
 - Python 3.10+
 
+If you see the following error:
+
+```bash
+fatal error: Python.h: No such file or directory | #include <Python.h>
+```
+
+Make sure you also install python3-dev package. For ubuntu distros:
+
+```bash
+sudo apt install python3-dev
+```
+
 
 ### Install the client
 

kalavai_client-0.5.5/kalavai_client/__init__.py

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

{kalavai_client-0.5.4 → kalavai_client-0.5.5}/kalavai_client/utils.py

@@ -2,7 +2,6 @@ import json, base64
 import os
 import requests
 from pathlib import Path
-from urllib.parse import urljoin
 import shutil
 import subprocess
 import re
@@ -12,10 +11,7 @@ from jinja2 import Template
 
 from rich.table import Table
 import yaml
-import platform
-import psutil
 
-import GPUtil
 
 from kalavai_client.auth import KalavaiAuthClient
 
@@ -387,41 +383,6 @@ def encode_dict(data: dict):
 def decode_dict(str_data: str):
     return json.loads(base64.b64decode(str_data.encode()))
 
-def get_gpus():
-    GPUs = GPUtil.getGPUs()
-    gpus = []
-    for gpu in GPUs:
-        name = "nvidia" if "nvidia" in gpu.name.lower() else None
-        if name is None:
-            continue
-        mem = int(gpu.memoryTotal / 1000) # in GBs
-        gpus.append(f"{name}-{mem}GB")
-    return ",".join(gpus)
-
-def system_uptick_request(username, node_name, backend_endpoint, backend_api_key, local_version=0):
-    gpus = get_gpus()
-    data = {
-        "username": username,
-        "system_info": {
-            "os": platform.system(),
-            "cpu_count": os.cpu_count(),
-            "cpu": platform.processor(),
-            "platform": platform.platform(),
-            "ram": round(psutil.virtual_memory().total / (1024.0 **3)),
-            "hostname": node_name,
-            "gpus": gpus
-        },
-        "version": local_version
-    }
-
-    response = requests.post(
-        url=urljoin(backend_endpoint, "/uptick"),
-        json=data,
-        headers={'X-API-KEY': backend_api_key}
-    )
-    response.raise_for_status()
-    return response.json()
-
 def resource_path(relative_path: str):
     """ Get absolute path to resource """
     try:

{kalavai_client-0.5.4 → kalavai_client-0.5.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name            = "kalavai-client"
-version         = "0.5.4"
+version         = "0.5.5"
 authors = [
   {name = "Carlos Fernandez Musoles", email = "carlos@kalavai.net"}
 ]
@@ -20,9 +20,7 @@ dependencies = [
     "pyyaml==6.0.2",
     "rich==13.7.1",
     "arguably>=1.2.5",
-    "gputil==1.4.0",
     "Pillow==10.3.0",
-    "setuptools<70.0.0",
     "netifaces==0.11.0",
     "anvil-uplink==0.5.1",
     "importlib_resources==6.5.2"

kalavai_client-0.5.4/kalavai_client/__init__.py

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