@11agents/cli 0.1.12 → 0.1.13

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@11agents/cli",
-  "version": "0.1.12",
+  "version": "0.1.13",
   "description": "11agents local runtime and telemetry CLI",
   "type": "module",
   "bin": {

package/src/commands/runtime.js

@@ -96,6 +96,41 @@ function createRetryState() {
   return { failures: 0 }
 }
 
+async function heartbeatRegisteredRuntime(registration, flags, deps) {
+  const config = configFromFlags(flags)
+  const machineKey = registration?.machine?.machine_key || machineOverride(flags) || ''
+  if (!machineKey) return null
+  return deps.requestJson('/api/runtime/machines/heartbeat', {
+    method: 'POST',
+    body: {
+      machine_key: machineKey,
+      runtime_providers: (registration?.runtimes || []).map(runtime => runtime.provider).filter(Boolean),
+      health: {
+        heartbeat_at: new Date().toISOString(),
+      },
+    },
+    config,
+  })
+}
+
+async function runWithRuntimeHeartbeat(operation, registration, flags, deps, heartbeatIntervalMs) {
+  const intervalMs = Math.max(10, Math.min(Number(heartbeatIntervalMs) || 15000, 60000))
+  const timer = setInterval(() => {
+    heartbeatRegisteredRuntime(registration, flags, deps).catch(error => {
+      deps.log(JSON.stringify({
+        warning: 'runtime heartbeat during task failed',
+        error: errorMessage(error),
+      }, null, 2))
+    })
+  }, intervalMs)
+  timer.unref?.()
+  try {
+    return await operation()
+  } finally {
+    clearInterval(timer)
+  }
+}
+
 async function runWithDaemonRetry(label, operation, deps, retryState) {
   while (true) {
     try {
@@ -906,7 +941,7 @@ function defaultTaskHandler(flags, deps) {
   }
 }
 
-async function claimAndRunRuntimeTasks(registration, flags, deps, handlerModule, retryState = createRetryState()) {
+async function claimAndRunRuntimeTasks(registration, flags, deps, handlerModule, retryState = createRetryState(), heartbeatIntervalMs = 15000) {
   if (!handlerModule) return 0
   const config = configFromFlags(flags)
   const machineKey = registration?.machine?.machine_key || machineOverride(flags) || ''
@@ -968,7 +1003,13 @@ async function claimAndRunRuntimeTasks(registration, flags, deps, handlerModule,
       executionContext = await prepareRuntimeTask(runtimeTask, flags, deps, config)
       runtimeTask.execution_context = executionContext
       try {
-        completion = await handlerModule.handleRuntimeTask(runtimeTask)
+        completion = await runWithRuntimeHeartbeat(
+          () => handlerModule.handleRuntimeTask(runtimeTask),
+          registration,
+          flags,
+          deps,
+          heartbeatIntervalMs
+        )
       } catch (error) {
         completion = {
           comment: error instanceof Error ? error.message : String(error),
@@ -1063,7 +1104,7 @@ export async function startRuntimeDaemon(flags = {}, deps = {}) {
   const retryState = createRetryState()
   let registration = await runWithDaemonRetry('register runtime', () => registerRuntime(flags, resolvedDeps), resolvedDeps, retryState)
   await syncRuntimeProjectMetadataBestEffort(flags, resolvedDeps)
-  await claimAndRunRuntimeTasks(registration, flags, resolvedDeps, handlerModule, retryState)
+  await claimAndRunRuntimeTasks(registration, flags, resolvedDeps, handlerModule, retryState, heartbeatIntervalMs)
   if (once) return
 
   let lastScan = Date.now()
@@ -1082,7 +1123,7 @@ export async function startRuntimeDaemon(flags = {}, deps = {}) {
       lastHeartbeat = now
     }
     if (now - lastTaskPoll >= taskIntervalMs) {
-      await claimAndRunRuntimeTasks(registration, flags, resolvedDeps, handlerModule, retryState)
+      await claimAndRunRuntimeTasks(registration, flags, resolvedDeps, handlerModule, retryState, heartbeatIntervalMs)
       lastTaskPoll = now
     }
     if (now - lastProjectRefresh >= projectRefreshIntervalMs) {