tuneplane-node 0.3.18__tar.gz → 0.3.19__tar.gz

{tuneplane_node-0.3.18 → tuneplane_node-0.3.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: tuneplane-node
-Version: 0.3.18
+Version: 0.3.19
 Summary: TunePlane node daemon: runs job containers on one machine and reports it to a console.
 License-Expression: AGPL-3.0-only
 License-File: LICENSE

{tuneplane_node-0.3.18 → tuneplane_node-0.3.19}/pyproject.toml

@@ -10,7 +10,7 @@
 # the local backend runs the same container runtime and allocator on its own host.
 [project]
 name = "tuneplane-node"
-version = "0.3.18"
+version = "0.3.19"
 description = "TunePlane node daemon: runs job containers on one machine and reports it to a console."
 requires-python = ">=3.12"
 # Same licence as the control plane: this is infrastructure an organization runs,

{tuneplane_node-0.3.18 → tuneplane_node-0.3.19}/src/tuneplane_node/allocator.py

@@ -2,7 +2,7 @@
 
 Who decides "which cards this job gets" when there is no Kubernetes
 ──────────────────────────────────────────────────────────────────────────────
-On the kuberay backend kube-scheduler and the NVIDIA device plugin do it: the Pod
+On the kubernetes backend kube-scheduler and the NVIDIA device plugin do it: the Pod
 declares `nvidia.com/gpu: 4`, the scheduler guarantees no over-subscription, and the
 device plugin injects the actual cards into the container.
 

{tuneplane_node-0.3.18 → tuneplane_node-0.3.19}/src/tuneplane_node/join.py

@@ -123,7 +123,10 @@ async def join(
         "inventory": inventory,
     }
     async with httpx.AsyncClient(timeout=timeout, transport=transport) as client:
-        response = await client.post(f"{base}/api/fleets/{fleet_id}/nodes", json=payload)
+        response = await client.post(
+            f"{base}/api/fleets/{fleet_id}/nodes", json=payload,
+            headers={"User-Agent": _user_agent(node_version)},
+        )
     if response.status_code != 201:
         raise JoinError(_detail(response))
     body = response.json()
@@ -155,7 +158,11 @@ async def beat(
     }
     async with httpx.AsyncClient(timeout=timeout, transport=transport) as client:
         response = await client.post(
-            url, json=payload, headers={"Authorization": f"Bearer {identity.credential}"}
+            url, json=payload,
+            headers={
+                "Authorization": f"Bearer {identity.credential}",
+                "User-Agent": _user_agent(node_version),
+            },
         )
     if response.status_code != 200:
         raise JoinError(_detail(response))
@@ -190,6 +197,12 @@ async def heartbeat_forever(
         await asyncio.sleep(interval_s)
 
 
+def _user_agent(node_version: str) -> str:
+    """How the daemon names itself, so a join reads as one in the console's audit trail
+    rather than as `python-httpx`."""
+    return f"tuneplane-node/{node_version}"
+
+
 def _detail(response: httpx.Response) -> str:
     try:
         body = response.json()

{tuneplane_node-0.3.18 → tuneplane_node-0.3.19}/src/tuneplane_node/runtime.py

@@ -312,14 +312,14 @@ class CliRuntime:
         # `vllm` command -- died at once with
         #     vllm: error: unrecognized arguments: -lc python3 -m vllm...
         # which is this `bash -lc` arriving where a subcommand was expected. The
-        # same launch works on kuberay, because a Kubernetes container's
+        # same launch works on kubernetes, because a Kubernetes container's
         # `command` overrides the entrypoint by definition -- so the two backends
         # disagreed about the one field that says what to run, and the process
         # runtime (which execs the argv directly) agreed with Kubernetes.
         #
         # What this gives up is an image's own entrypoint wrapper, such as
         # nvidia_entrypoint.sh on the NGC bases. That wrapper prints a banner and
-        # `exec "$@"`, and the kuberay path has been skipping it since it was
+        # `exec "$@"`, and the kubernetes path has been skipping it since it was
         # written, which is the evidence that nothing depends on it.
         if spec.command:
             args += ["--entrypoint", spec.command[0]]