tuneplane-node 0.3.15__py3-none-any.whl

tuneplane_node/wire.py

@@ -0,0 +1,85 @@
+"""Node wire protocol: JSON encoding of ContainerSpec / ContainerState.
+
+Its own module because the daemon (node side) and NodeExecutor (console side) must share one
+codec; two copies of it drift. The protocol carries an explicit version so that a console and a
+node daemon at different versions fail closed instead of guessing.
+"""
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    from tuneplane_node.runtime import ContainerSpec, ContainerState
+
+#: Wire protocol version. Bumped on a breaking change; both sides check it on every request.
+PROTOCOL = 1
+
+
+def spec_to_dict(spec: "ContainerSpec") -> dict[str, Any]:
+    return {
+        "name": spec.name,
+        "image": spec.image,
+        "command": list(spec.command),
+        "env": dict(spec.env),
+        "gpus": list(spec.gpus),
+        "mounts": [[host, cont, ro] for host, cont, ro in spec.mounts],
+        "labels": dict(spec.labels),
+        "workdir": spec.workdir,
+        "shm_size": spec.shm_size,
+        "cpu_limit": spec.cpu_limit,
+        "memory_limit": spec.memory_limit,
+        "network": spec.network,
+        "user": spec.user,
+    }
+
+
+def spec_from_dict(data: dict[str, Any]) -> "ContainerSpec":
+    from tuneplane_node.runtime import ContainerSpec
+
+    return ContainerSpec(
+        name=str(data["name"]),
+        image=str(data["image"]),
+        command=[str(x) for x in data["command"]],
+        env={str(k): str(v) for k, v in (data.get("env") or {}).items()},
+        gpus=[int(x) for x in (data.get("gpus") or [])],
+        mounts=[
+            (str(host), str(cont), bool(ro))
+            for host, cont, ro in (data.get("mounts") or [])
+        ],
+        labels={str(k): str(v) for k, v in (data.get("labels") or {}).items()},
+        workdir=str(data.get("workdir") or ""),
+        shm_size=str(data.get("shm_size") or ""),
+        cpu_limit=str(data.get("cpu_limit") or ""),
+        memory_limit=str(data.get("memory_limit") or ""),
+        network=str(data.get("network") or "host"),
+        user=str(data.get("user") or ""),
+    )
+
+
+def state_to_dict(state: "ContainerState") -> dict[str, Any]:
+    return {
+        "name": state.name,
+        "exists": state.exists,
+        "status": state.status,
+        "exit_code": state.exit_code,
+        "labels": dict(state.labels),
+        "started_at": state.started_at,
+        "finished_at": state.finished_at,
+        "oom_killed": state.oom_killed,
+    }
+
+
+def state_from_dict(data: dict[str, Any]) -> "ContainerState":
+    from tuneplane_node.runtime import ContainerState
+
+    exit_code = data.get("exit_code")
+    return ContainerState(
+        name=str(data["name"]),
+        exists=bool(data.get("exists")),
+        status=str(data.get("status") or ""),
+        exit_code=None if exit_code is None else int(exit_code),
+        labels={str(k): str(v) for k, v in (data.get("labels") or {}).items()},
+        started_at=str(data.get("started_at") or ""),
+        finished_at=str(data.get("finished_at") or ""),
+        oom_killed=bool(data.get("oom_killed")),
+    )

tuneplane_node-0.3.15.dist-info/METADATA

@@ -0,0 +1,12 @@
+Metadata-Version: 2.5
+Name: tuneplane-node
+Version: 0.3.15
+Summary: TunePlane node daemon: runs job containers on one machine and reports it to a console.
+License-Expression: AGPL-3.0-only
+Requires-Python: >=3.12
+Requires-Dist: fastapi>=0.110
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: pydantic-settings>=2.0
+Requires-Dist: tuneplane
+Requires-Dist: typer>=0.12
+Requires-Dist: uvicorn>=0.29

tuneplane_node-0.3.15.dist-info/RECORD

@@ -0,0 +1,13 @@
+tuneplane_node/__init__.py,sha256=Byymvzjv-JXH5kbRUcChRlpr1d9fDq69N0DLnqR2w_U,69
+tuneplane_node/allocator.py,sha256=JviCsvpyhEEmk4H6ahZ-dOjlchmgfzccJIb7AFsqh7M,28783
+tuneplane_node/cli.py,sha256=fCTrWz9XygzeBLXARtElPI7PziwmoJxp3Jda_-Btyhg,7272
+tuneplane_node/daemon.py,sha256=n3X0H4TGvGcxTgWLUpao_IAAzMDRJNHq5d67eomND78,25516
+tuneplane_node/inventory.py,sha256=PXcc8qTshbQX0_D-qOqmsmZ8tr7nII8aykP1WtJwyTc,6729
+tuneplane_node/join.py,sha256=Bu9CAvi9VQPso_3IFQ_AnK52_mx3-JC5-7QZU7IOo80,6834
+tuneplane_node/runtime.py,sha256=U4mi0gCvx7LVhU9mAlXKCRuzqhODodF4nNekd1ubRHs,26656
+tuneplane_node/settings.py,sha256=DJd49Ap8HpjRA5k7yl3_bd_MQERtl-pD61I5IwElBek,3541
+tuneplane_node/wire.py,sha256=z8uHfR9Y5e6sraxiNUiAVutwfYjtu-R6hTn1mv_j39k,3058
+tuneplane_node-0.3.15.dist-info/METADATA,sha256=rGCgjON6VXiUH-kI09AEUyNpbEEhXePnLTFsMeW4v9A,391
+tuneplane_node-0.3.15.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
+tuneplane_node-0.3.15.dist-info/entry_points.txt,sha256=Fk7kqMVHU8AzJQN8XUHrQs3cUs2GVd8Pc83jjL1ILRU,58
+tuneplane_node-0.3.15.dist-info/RECORD,,

tuneplane_node-0.3.15.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.32.0
+Root-Is-Purelib: true
+Tag: py3-none-any

tuneplane_node-0.3.15.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+tuneplane-node = tuneplane_node.cli:app