@lesscap/baton-cli 0.2.9 → 0.2.11

package/dist/baton.mjs

@@ -2446,16 +2446,30 @@ import { query as query2 } from "@anthropic-ai/claude-agent-sdk";
 // src/session/runner/title.ts
 var clip = (s, max) => s.length > max ? s.slice(0, max) : s;
 var isDeclined = (title) => title.replace(/[^a-z]/gi, "").toUpperCase() === "NONE";
-var sanitizeTitle = (raw) => raw.replace(/[\r\n]+/g, " ").replace(/[*_`~#>]/g, "").replace(/["']/g, "").replace(/^\s*(?:title|session|标题)\s*[:：-]\s*/i, "").replace(/^\s*(?:[-*•]|\d+[.)、])\s+/, "").replace(/\s+/g, " ").trim().slice(0, 30).replace(/[\s。．，,、；;：:!！?？.]+$/, "").trim();
+var sanitizeTitle = (raw) => raw.replace(/[\r\n]+/g, " ").replace(/[*_`~#>]/g, "").replace(/["']/g, "").replace(/^\s*(?:title|session|标题)\s*[:：-]\s*/i, "").replace(/^\s*(?:[-*•]|\d+[.)、])\s+/, "").replace(/\s+/g, " ").trim().replace(/(?:[。．！？，、；；]|[.!?;](?=\s|$)).*$/s, "").trim().slice(0, 30).replace(/[\s。．，,、；;：:!！?？.]+$/, "").trim();
 var TITLE_TIMEOUT_MS = 9e4;
 var buildPrompt2 = (userText, assistantText) => {
   const exchange = [
     `User: ${clip(userText, 800)}`,
     assistantText ? `Assistant: ${clip(assistantText, 800)}` : ""
   ].filter(Boolean).join("\n");
-  return `Reply with ONLY a short title for this session: a noun phrase like a filename \u2014 at most 6 words or ~12 Chinese characters. It must NOT be a sentence, no punctuation, no markdown, no quotes, and never an opener like "\u597D\u7684"/"\u6536\u5230"/"I'll". Only if the exchange below has no real content at all (e.g. just a greeting or a test ping), reply with exactly NONE.
-
-${exchange}`;
+  return [
+    "Name the TOPIC of this chat session \u2014 what it is ABOUT \u2014 as a short title:",
+    "a noun phrase like a filename, at most 6 words or ~12 Chinese characters.",
+    "Name the subject, do NOT restate the conclusion or the answer.",
+    "No sentence, no punctuation at all (including Chinese \uFF0C\u3002\u3001\uFF1B\uFF01\uFF1F), no markdown,",
+    `no quotes, and never an opener like "\u597D\u7684"/"\u6536\u5230"/"I'll".`,
+    "",
+    "Examples (exchange \u2192 title):",
+    "User asks whether the material-ui repo needs updating; assistant says it is",
+    "already up to date \u2192 material-ui \u4ED3\u5E93\u66F4\u65B0\u68C0\u67E5",
+    "User: curl the health endpoint and report status \u2192 \u5065\u5EB7\u68C0\u67E5\u63A5\u53E3\u6392\u67E5",
+    "",
+    "Only if the exchange below has no real content at all (just a greeting or a",
+    "test ping) reply with exactly NONE. Otherwise reply with ONLY the title.",
+    "",
+    exchange
+  ].join("\n");
 };
 var generateTitle = async (input) => {
   const { worktreePath, userText, assistantText, queryFn } = input;
@@ -2575,6 +2589,10 @@ var removeWorktree = (repo, worktreePath) => {
   }
 };
 
+// src/worker/watchdog.ts
+var stale = (okAt, now, thresholdMs) => now - okAt > thresholdMs;
+var streamWedged = (isOpen, okAt, now, thresholdMs) => !isOpen && stale(okAt, now, thresholdMs);
+
 // src/worker/daemon.ts
 var binPath = () => {
   const here = dirname2(fileURLToPath(import.meta.url));
@@ -2588,11 +2606,20 @@ var runWorkerDaemon = async (cfg, client, signal) => {
   const log = (m) => console.log(`[worker #${cfg.workerId} ${cfg.name}] ${m}`);
   const HEARTBEAT_MS = 3e4;
   const HEARTBEAT_TIMEOUT_MS = 15e3;
-  const MAX_HEARTBEAT_FAILURES = 5;
-  let heartbeatFailures = 0;
+  const HEARTBEAT_DEAD_MS = 15e4;
+  const STREAM_STALE_MS = 9e4;
+  const ES_OPEN = 1;
+  let heartbeatOkAt = Date.now();
+  let streamOkAt = Date.now();
   let fatal = false;
   let stop = () => {
   };
+  const trip = (reason) => {
+    if (fatal) return;
+    log(`${reason} \u2014 exiting so the supervisor restarts a clean worker`);
+    fatal = true;
+    stop();
+  };
   const ping = async () => {
     try {
       await Promise.race([
@@ -2601,20 +2628,11 @@ var runWorkerDaemon = async (cfg, client, signal) => {
           (_, reject) => setTimeout(() => reject(new Error("heartbeat timeout")), HEARTBEAT_TIMEOUT_MS)
         )
       ]);
-      heartbeatFailures = 0;
+      heartbeatOkAt = Date.now();
     } catch (e) {
-      heartbeatFailures += 1;
-      log(`heartbeat failed (${heartbeatFailures}/${MAX_HEARTBEAT_FAILURES}): ${String(e)}`);
-      if (heartbeatFailures >= MAX_HEARTBEAT_FAILURES) {
-        const downSec = MAX_HEARTBEAT_FAILURES * HEARTBEAT_MS / 1e3;
-        log(`heartbeat down ~${downSec}s \u2014 exiting so the supervisor restarts a clean worker`);
-        fatal = true;
-        stop();
-      }
+      log(`heartbeat failed: ${String(e)}`);
     }
   };
-  void ping();
-  const hb = setInterval(() => void ping(), HEARTBEAT_MS);
   const killChild = (child) => {
     if (child.pid === void 0) return;
     try {
@@ -2716,8 +2734,12 @@ var runWorkerDaemon = async (cfg, client, signal) => {
       }
     })
   });
-  es.onopen = () => void reconcile().catch((err) => log(`reconcile failed: ${String(err)}`));
+  es.onopen = () => {
+    streamOkAt = Date.now();
+    void reconcile().catch((err) => log(`reconcile failed: ${String(err)}`));
+  };
   es.onmessage = (e) => {
+    streamOkAt = Date.now();
     try {
       const cmd = JSON.parse(e.data);
       if (cmd.cmd === "session.start")
@@ -2733,6 +2755,18 @@ var runWorkerDaemon = async (cfg, client, signal) => {
   };
   es.onerror = () => log("command stream error (eventsource will retry)");
   log(`listening for commands (server ${cfg.server}, repo ${repo})`);
+  const watchdog = () => {
+    const now = Date.now();
+    if (stale(heartbeatOkAt, now, HEARTBEAT_DEAD_MS))
+      trip(`heartbeat down ~${HEARTBEAT_DEAD_MS / 1e3}s`);
+    else if (streamWedged(es.readyState === ES_OPEN, streamOkAt, now, STREAM_STALE_MS))
+      trip(`command stream wedged ~${STREAM_STALE_MS / 1e3}s`);
+  };
+  void ping();
+  const hb = setInterval(() => {
+    void ping();
+    watchdog();
+  }, HEARTBEAT_MS);
   await new Promise((resolve2) => {
     stop = resolve2;
     if (signal.aborted) return resolve2();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lesscap/baton-cli",
-  "version": "0.2.9",
+  "version": "0.2.11",
   "license": "MIT",
   "type": "module",
   "description": "baton CLI — register/run a worker, manage projects/requirements/tasks/sessions",
@@ -10,22 +10,23 @@
   "files": [
     "dist"
   ],
+  "scripts": {
+    "baton": "tsx src/index.ts",
+    "build": "esbuild src/index.ts --bundle --platform=node --format=esm --packages=bundle --external:@anthropic-ai/claude-agent-sdk --banner:js=\"#!/usr/bin/env node\" --outfile=dist/baton.mjs && chmod +x dist/baton.mjs",
+    "prepublishOnly": "pnpm build",
+    "typecheck": "tsc --noEmit -p tsconfig.json",
+    "test": "tsx --test 'src/**/*.test.ts'"
+  },
   "devDependencies": {
+    "@baton/shared": "workspace:*",
     "@types/node": "^25.9.1",
     "citty": "^0.2.2",
     "esbuild": "^0.28.0",
     "eventsource": "^4.1.0",
     "tsx": "^4.22.4",
-    "typescript": "^6.0.3",
-    "@baton/shared": "0.0.0"
+    "typescript": "^6.0.3"
   },
   "dependencies": {
     "@anthropic-ai/claude-agent-sdk": "^0.3.161"
-  },
-  "scripts": {
-    "baton": "tsx src/index.ts",
-    "build": "esbuild src/index.ts --bundle --platform=node --format=esm --packages=bundle --external:@anthropic-ai/claude-agent-sdk --banner:js=\"#!/usr/bin/env node\" --outfile=dist/baton.mjs && chmod +x dist/baton.mjs",
-    "typecheck": "tsc --noEmit -p tsconfig.json",
-    "test": "tsx --test 'src/**/*.test.ts'"
   }
-}
+}

package/LICENSE

@@ -1,21 +0,0 @@
-MIT License
-
-Copyright (c) 2026 bencode
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.