kalavai-client 0.6.1__py3-none-any.whl → 0.6.3__py3-none-any.whl

kalavai_client/__init__.py

@@ -1,2 +1,2 @@
 
-__version__ = "0.6.1"
+__version__ = "0.6.3"

kalavai_client/assets/docker-compose-gui.yaml

@@ -2,7 +2,6 @@ services:
   kalavai_gui:
     container_name: kalavai_gui
     image: bundenth/kalavai-gui:latest
-    #network_mode: host
     extra_hosts:
     - "host.docker.internal:host-gateway"
     networks:

kalavai_client/assets/docker-compose-template.yaml

@@ -7,34 +7,13 @@ services:
       - NET_ADMIN
       - SYS_MODULE
     network_mode: host
-    # networks:
-    # - custom-network
-    # ports:
-    # # https://docs.k3s.io/installation/requirements#inbound-rules-for-k3s-nodes
-    # - "6443:6443" # kube server
-    # - "10250:10250" # worker balancer
-    # - "8472:8472/udp" # flannel vxlan
-    # - "51820-51830:51820-51830" # flannel wireguard
-    # {% if command == "server" %}
-    # - "30000-30500:30000-30500"
-    # {% endif %}
     environment:
       - HOST_NAME={{node_name}}
       - TOKEN={{vpn_token}}
     volumes:
       - /dev/net/tun:/dev/net/tun
     restart: unless-stopped
-#   nginx:
-#     image: nginx:latest
-#     ports:
-#       - "{{redirect_source_port}}:{{redirect_source_port}}"
-#     restart: unless-stopped
-#     networks:
-#       - custom-network
-#     volumes:
-#       - {{nginx_path}}/nginx.conf:/etc/nginx/nginx.conf
 {% endif %}
-
 # run worker only if command is set
 {%if command %}
   {{service_name}}:
@@ -46,21 +25,6 @@ services:
     network_mode: "service:{{vpn_name}}"
   {% else %}
     network_mode: host
-    # hostname: {{node_name}}
-    # networks:
-    # - custom-network
-    # ports:
-    # - "6443:6443" # kube server
-    # - "2379-2380:2379-2380" # etcd server
-    # - "10259:10259" # kube scheduler
-    # - "10257:10257" # kube controller manager
-    # - "10250:10250" # worker balancer
-    # - "8285:8285" # flannel
-    # - "8472:8472" # flannel vxlan
-    # - "51820:51820" # flannel wireguard
-    # {% if command == "server" %}
-    # - "30000-32767:30000-32767"
-    # {% endif %}
   {% endif %}
     privileged: true
     restart: unless-stopped
@@ -100,8 +64,3 @@ services:
             capabilities: [gpu]
   {% endif %}
 {% endif %}
-
-# networks:
-#   custom-network:
-#     driver: bridge
-

kalavai_client/bridge_api.py

@@ -75,7 +75,8 @@ def pool_create(request: CreatePoolRequest, api_key: str = Depends(verify_api_ke
         only_registered_users=request.only_registered_users,
         location=request.location,
         description=request.description,
-        token_mode=request.token_mode
+        token_mode=request.token_mode,
+        frontend=request.frontend
     )
     return result
 
@@ -85,7 +86,8 @@ def pool_join(request: JoinPoolRequest, api_key: str = Depends(verify_api_key)):
         token=request.token,
         num_gpus=request.num_gpus,
         node_name=request.node_name,
-        ip_address=request.ip_address
+        ip_address=request.ip_address,
+        frontend=request.frontend
     )
     return result
 
@@ -93,7 +95,8 @@ def pool_join(request: JoinPoolRequest, api_key: str = Depends(verify_api_key)):
 def pool_attach(request: JoinPoolRequest, api_key: str = Depends(verify_api_key)):
     result = attach_to_pool(
         token=request.token,
-        node_name=request.node_name
+        node_name=request.node_name,
+        frontend=request.frontend
     )
     return result
 

kalavai_client/bridge_models.py

@@ -16,6 +16,7 @@ class CreatePoolRequest(BaseModel):
     location: str = None
     token_mode: TokenType = TokenType.USER
     description: str = ""
+    frontend: bool = False
 
 class NodesActionRequest(BaseModel):
     nodes: list[str]
@@ -25,7 +26,7 @@ class JoinPoolRequest(BaseModel):
     ip_address: str = None
     node_name: str = None
     num_gpus: int = None
-
+    frontend: bool = False
 class JobDetailsRequest(BaseModel):
     jobs: list[Job]
 

kalavai_client/cli.py

@@ -205,33 +205,29 @@ def input_gpus(non_interactive=False):
 @arguably.command
 def gui__start(
     *others,
-    backend_only=False,
     gui_frontend_port=3000,
     gui_backend_port=8000,
     bridge_port=8001,
     log_level="critical",
-    protected_access=False
+    protected_access=False,
+    backend_only=False
 ):
     """Run GUI (docker) and kalavai core backend (api)"""
     if len(set([gui_frontend_port, gui_backend_port, bridge_port])) < 3:
         console.log("[red]Error: ports must be unique")
         return
     
-    user_key = None
-    if protected_access:
-        user_key = load_user_id()
-        if user_key is None:
-            console.log("[red]Error: user key not found (required for protected access)")
-            return
-    
+    user_key = load_user_id()
+    if user_key is not None:
+        console.log(f"[green]Using user key: {user_key}")
     if not backend_only:
         values = {
             "gui_frontend_port": gui_frontend_port,
             "gui_backend_port": gui_backend_port,
             "bridge_port": bridge_port,
-            "path": user_path(""),
+            "path": user_path("", create_path=True),
             "protected_access": user_key
-        }
+    }
         compose_yaml = load_template(
             template_path=DOCKER_COMPOSE_GUI,
             values=values)
@@ -245,11 +241,10 @@ def gui__start(
         "Deploying bridge API"
     )
     run_api(port=bridge_port, log_level=log_level)
-    
+
     if not backend_only:
         run_cmd(f"docker compose --file {USER_GUI_COMPOSE_FILE} down")
-    console.log("[green]Kalavai GUI has been stopped")
-
+        console.log("[green]Kalavai GUI has been stopped")
 
 @arguably.command
 def auth(user_key, *others):

kalavai_client/core.py

@@ -789,7 +789,6 @@ def create_pool(
     auth_key = user_id if user_id is not None else str(uuid.uuid4())
     write_auth_key = str(uuid.uuid4())
     readonly_auth_key = str(uuid.uuid4())
-    
     watcher_service = f"{ip_address}:{DEFAULT_WATCHER_PORT}"
     values = {
         CLUSTER_NAME_KEY: cluster_name,

kalavai_client/env.py

@@ -3,11 +3,11 @@ from pathlib import Path
 import importlib.resources
 
 
-DEFAULT_KALAVAI_PATH = os.getenv("KALAVAI_PATH", None)
+DEFAULT_KALAVAI_PATH = os.getenv("KALAVAI_PATH", "")
 
 def user_path(relative_path, create_path=False):
     """Transform a relative path into the user's cache folder path"""
-    if DEFAULT_KALAVAI_PATH is not None:
+    if len(DEFAULT_KALAVAI_PATH.strip()) > 0:
         kalavai_user_path = DEFAULT_KALAVAI_PATH
     else:
         base = os.path.expanduser("~")

{kalavai_client-0.6.1.dist-info → kalavai_client-0.6.3.dist-info}/METADATA

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

{kalavai_client-0.6.1.dist-info → kalavai_client-0.6.3.dist-info}/RECORD

@@ -1,25 +1,25 @@
-kalavai_client/__init__.py,sha256=RUEIQh8z_V_kGKsMyzH8WJLz-dW7GvzomrpPti1qfDw,22
+kalavai_client/__init__.py,sha256=YPRovmUtXL2teOic_KNYxHp09zJ6yS_T_hR6yiFy5hI,22
 kalavai_client/__main__.py,sha256=WQUfxvRsBJH5gsCJg8pLz95QnZIj7Ol8psTO77m0QE0,73
 kalavai_client/assets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kalavai_client/assets/apps.yaml,sha256=H2HZQ_oqdNz2qPDs5_NtHvJo4rblH1DRPugzQYq_t5Q,6366
 kalavai_client/assets/apps_values.yaml,sha256=WRew3bS1MztjzcJfphuJcKn0n2T1ICRupPpr_Csjt_s,1644
-kalavai_client/assets/docker-compose-gui.yaml,sha256=IHClIltiu8C4nHbgDgAscf3Ls6UFngbeQP07kj31kjM,776
-kalavai_client/assets/docker-compose-template.yaml,sha256=uct7xmXtO1XgNimdirSiQusb-RYsPbnL67TVkj2oWrU,2772
+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/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
 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=IkKq_CODA4p4UJkk6UqXKHkz5m2ESq2mL3tKqkQk6T0,6910
-kalavai_client/bridge_models.py,sha256=k4ILxa8jfAcfgIIBJqK1DunDzHh_oNUpuiEpmhAfTP0,977
-kalavai_client/cli.py,sha256=TUZJDmHdpBnF7Po-rnI97S1VYDPWxde1JqaHtb9xlIQ,46996
+kalavai_client/bridge_api.py,sha256=KecIZvsGRiZgUQaDSgvRBLaqPDBhxFKS6FBH_w5ArCc,7015
+kalavai_client/bridge_models.py,sha256=IGPLAYEZG4f5EUtBuoR2u_cYoyCCVkhx4H2rMeGuJ_Y,1030
+kalavai_client/cli.py,sha256=kvQXuIdSd2BYavnqLkd7phFUD3htxjVcSJaqnaEAGtI,46904
 kalavai_client/cluster.py,sha256=gwjmdsd--YrffT0BmZDOEpbrdm3lPskUuN5jdgcrOR0,12947
-kalavai_client/core.py,sha256=CWJQ38BZ2mLavfOCyR7wsDdy4fRTpjnYmggoXcR9Ji8,32531
-kalavai_client/env.py,sha256=J-9_YopIIVBCnwTFxoQ9jq8JWUDdKrB__S-Wig6TWSU,2767
+kalavai_client/core.py,sha256=1qT6d4LpZIHHQaCDZHVmk4crkcOShzVMCL4Pax8ShVI,32526
+kalavai_client/env.py,sha256=wAd-P8XDlDB8Oau76f0bG2aHTkNA793xb_V0frbNeJE,2770
 kalavai_client/utils.py,sha256=k2Bpvl16zDpWBXMIMsd_LbkinZtctPO6bygfB7DyiZ8,12018
-kalavai_client-0.6.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-kalavai_client-0.6.1.dist-info/METADATA,sha256=0VhOSaYS1GdoXSE0w_YNYjfjiXHZ57SiLe3IRVJ7Sz4,13353
-kalavai_client-0.6.1.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-kalavai_client-0.6.1.dist-info/entry_points.txt,sha256=9T6D45gxwzfVbglMm1r6XPdXuuZdHfy_7fCeu2jUphc,50
-kalavai_client-0.6.1.dist-info/RECORD,,
+kalavai_client-0.6.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+kalavai_client-0.6.3.dist-info/METADATA,sha256=8Q4iCcpjFF9gRDYEeQbaZHAegUn_v4yn8-w5Y7sAiM4,13353
+kalavai_client-0.6.3.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+kalavai_client-0.6.3.dist-info/entry_points.txt,sha256=9T6D45gxwzfVbglMm1r6XPdXuuZdHfy_7fCeu2jUphc,50
+kalavai_client-0.6.3.dist-info/RECORD,,