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

kalavai_client/__init__.py

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

kalavai_client/assets/docker-compose-template.yaml

@@ -30,6 +30,9 @@ services:
     restart: unless-stopped
     command: >
       --command={{command}}
+    {% if user_id %}
+      --user_id={{user_id}}
+    {% endif %}
       --node_name="{{node_name}}"
       --node_ip="{{node_ip_address}}"
       {% if command == "server" %}

kalavai_client/env.py

@@ -4,6 +4,7 @@ import importlib.resources
 
 
 DEFAULT_KALAVAI_PATH = os.getenv("KALAVAI_PATH", "")
+KALAVAI_USER_ID = os.getenv("KALAVAI_USER_ID", None)
 
 def user_path(relative_path, create_path=False):
     """Transform a relative path into the user's cache folder path"""

kalavai_client/utils.py

@@ -22,10 +22,10 @@ from kalavai_client.env import (
     CONTAINER_HOST_PATH,
     USER_COMPOSE_FILE,
     USER_COOKIE,
+    KALAVAI_USER_ID,
     user_path
 )
 
-
 GITHUB_ORG = "kalavai-net"
 GITHUB_REPO = "kalavai-client"
 GITHUB_TEMPLATE_PATH = "templates"
@@ -121,7 +121,8 @@ def generate_compose_config(role, node_name, write_to_file=True, node_ip_address
         "k3s_path": f"{CONTAINER_HOST_PATH}/{rand_suffix}/k3s",
         "etc_path": f"{CONTAINER_HOST_PATH}/{rand_suffix}/etc",
         "node_labels": node_labels,
-        "flannel_iface": DEFAULT_FLANNEL_IFACE if vpn_token is not None else ""
+        "flannel_iface": DEFAULT_FLANNEL_IFACE if vpn_token is not None else "",
+        "user_id": load_user_id()
     }
     # generate local config files
     compose_yaml = load_template(
@@ -148,7 +149,6 @@ def is_watcher_alive(server_creds, user_cookie, timeout=30):
     return True
 
 
-
 def load_server_info(data_key, file):
     try:
         with open(file, "r") as f:
@@ -157,10 +157,18 @@ def load_server_info(data_key, file):
         return None
 
 def load_user_session():
-    return KALAVAI_AUTH.load_user_session()
+    if KALAVAI_USER_ID is None:
+        user_id = KALAVAI_AUTH.load_user_session()
+    else:
+        user_id = KALAVAI_USER_ID
+    return user_id
 
 def load_user_id():
-    return KALAVAI_AUTH.get_user_id()
+    if KALAVAI_USER_ID is None:
+        user_id = KALAVAI_AUTH.get_user_id()
+    else:
+        user_id = KALAVAI_USER_ID
+    return user_id
 
 def get_public_seeds(user_only, user_cookie):
     return []

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

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

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

@@ -1,10 +1,10 @@
-kalavai_client/__init__.py,sha256=ybIqssynIyB7zG5xVOgYdLdkjkSMc5Kwtdb49R-Wp2g,22
+kalavai_client/__init__.py,sha256=vHDA_1Wsc8bqniOeYrKHuU5fuCZ_w3-plFaYRacXd4Q,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_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=88Bi8mxynoK9lHi4d_yqaRVQCwbJfzkY8SUoYnEArRU,1589
+kalavai_client/assets/docker-compose-template.yaml,sha256=Nz_JzeBnQCzPCyWP5cEQHFeZzPwQqqBJ3C_xrToWlMA,1654
 kalavai_client/assets/nginx.conf,sha256=drVVCg8GHucz7hmt_BI6giAhK92OV71257NTs3LthwM,225
 kalavai_client/assets/pool_config_template.yaml,sha256=fFz4w2-fMKD5KvyzFdfcWD_jSneRlmnjLc8hCctweX0,576
 kalavai_client/assets/pool_config_values.yaml,sha256=VrM3XHQfQo6QLZ68qvagooUptaYgl1pszniY_JUtemk,233
@@ -16,10 +16,10 @@ kalavai_client/bridge_models.py,sha256=IGPLAYEZG4f5EUtBuoR2u_cYoyCCVkhx4H2rMeGuJ
 kalavai_client/cli.py,sha256=tBn3l12XhOJF17cWeX84xm10HPG7rSzalyOyugzcLhw,46900
 kalavai_client/cluster.py,sha256=ojUBXp2bR3hVyikIEkiGDbXvQfhBXBSk_mCqLxvyP0c,12943
 kalavai_client/core.py,sha256=bi8r7Twh2aQEDNn1shySlwx4704KG1pDmgMSSkzA1qk,32641
-kalavai_client/env.py,sha256=wAd-P8XDlDB8Oau76f0bG2aHTkNA793xb_V0frbNeJE,2770
-kalavai_client/utils.py,sha256=k2Bpvl16zDpWBXMIMsd_LbkinZtctPO6bygfB7DyiZ8,12018
-kalavai_client-0.6.8.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-kalavai_client-0.6.8.dist-info/METADATA,sha256=j06JF7kS9Mkuiz1tmhE2bVkAdLA5joRgq8rgXifbNyM,13353
-kalavai_client-0.6.8.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-kalavai_client-0.6.8.dist-info/entry_points.txt,sha256=9T6D45gxwzfVbglMm1r6XPdXuuZdHfy_7fCeu2jUphc,50
-kalavai_client-0.6.8.dist-info/RECORD,,
+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,,