blitzwing 0.1.9 → 0.1.10

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "blitzwing",
-  "version": "0.1.9",
+  "version": "0.1.10",
   "description": "Interactive setup wizard to join a Blitzwing Petals mother swarm as a compute contributor",
   "bin": {
     "blitzwing": "bin/blitzwing.js"

package/src/heartbeat.js

@@ -0,0 +1,64 @@
+import { spawn } from "node:child_process";
+import fs from "node:fs";
+import path from "node:path";
+import { heartbeatHost } from "./api.js";
+import { HOME_DIR } from "./config.js";
+
+/** Must be well below mother HEARTBEAT_TTL_SECONDS (default 60). */
+export const DEFAULT_HEARTBEAT_INTERVAL_MS = 20_000;
+
+/**
+ * Start a detached background process that heartbeats the mother orchestrator.
+ * @param {{ motherUrl: string, hostId: string, intervalMs?: number }} opts
+ */
+export function startContributorHeartbeatDaemon({
+  motherUrl,
+  hostId,
+  intervalMs = DEFAULT_HEARTBEAT_INTERVAL_MS,
+}) {
+  const hbPath = path.join(HOME_DIR, "heartbeat.mjs");
+  fs.mkdirSync(HOME_DIR, { recursive: true });
+  fs.writeFileSync(
+    hbPath,
+    `const mother = ${JSON.stringify(motherUrl)};
+const hostId = ${JSON.stringify(hostId)};
+const intervalMs = ${intervalMs};
+async function beat() {
+  try {
+    await fetch(mother.replace(/\\/$/, "") + "/v1/hosts/heartbeat", {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        "ngrok-skip-browser-warning": "true",
+      },
+      body: JSON.stringify({ host_id: hostId }),
+    });
+  } catch {}
+}
+setInterval(beat, intervalMs);
+beat();
+`
+  );
+  const child = spawn(process.execPath, [hbPath], {
+    detached: true,
+    stdio: "ignore",
+  });
+  child.unref();
+}
+
+/** In-process heartbeat loop (for tests or long-running CLI). */
+export async function startContributorHeartbeatLoop({
+  motherUrl,
+  hostId,
+  intervalMs = DEFAULT_HEARTBEAT_INTERVAL_MS,
+}) {
+  async function beat() {
+    try {
+      await heartbeatHost(motherUrl, { host_id: hostId });
+    } catch {
+      /* mother may be briefly unavailable */
+    }
+  }
+  setInterval(beat, intervalMs);
+  await beat();
+}

package/src/wizard.js

@@ -18,6 +18,7 @@ import {
   venvPython,
 } from "./install.js";
 import { saveState, loadState, clearState, ensureHome } from "./state.js";
+import { startContributorHeartbeatDaemon } from "./heartbeat.js";
 
 function parseArgs(argv) {
   const out = { cmd: null, discoveryUrl: process.env.BLITZWING_DISCOVERY_URL || DEFAULT_DISCOVERY_URL };
@@ -322,7 +323,10 @@ async function wizard(args) {
     joined_at: new Date().toISOString(),
   };
   saveState(state);
-  startHeartbeatDaemon(state);
+  startContributorHeartbeatDaemon({
+    motherUrl: state.mother_url,
+    hostId: state.host_id,
+  });
 
   p.outro(
     `${color.green("You are online.")} Hosting ${color.cyan(String(state.layers_hosted))} layers of ${color.cyan(state.model)} at ${state.block_indices}\n` +
@@ -330,32 +334,6 @@ async function wizard(args) {
   );
 }
 
-function startHeartbeatDaemon(state) {
-  const hbPath = path.join(HOME_DIR, "heartbeat.mjs");
-  fs.writeFileSync(
-    hbPath,
-    `const mother = ${JSON.stringify(state.mother_url)};
-const hostId = ${JSON.stringify(state.host_id)};
-async function beat() {
-  try {
-    await fetch(mother.replace(/\\/$/, "") + "/v1/hosts/heartbeat", {
-      method: "POST",
-      headers: { "Content-Type": "application/json" },
-      body: JSON.stringify({ host_id: hostId }),
-    });
-  } catch {}
-}
-setInterval(beat, 60000);
-beat();
-`
-  );
-  const child = spawn(process.execPath, [hbPath], {
-    detached: true,
-    stdio: "ignore",
-  });
-  child.unref();
-}
-
 async function showStatus() {
   const state = loadState();
   if (!state) {