switchroom 0.15.45 → 0.16.4

package/dist/cli/self-improve-stop.mjs

@@ -0,0 +1,428 @@
+// src/cli/self-improve-stop.ts
+import { readFileSync as readFileSync2 } from "node:fs";
+import { createConnection } from "node:net";
+import { join as join2 } from "node:path";
+import { homedir } from "node:os";
+
+// src/self-improve/config.ts
+function intEnv(name, def) {
+  const raw = process.env[name];
+  if (raw === undefined)
+    return def;
+  const n = Number.parseInt(raw, 10);
+  return Number.isFinite(n) && n >= 0 ? n : def;
+}
+function resolveSelfImproveConfig() {
+  return {
+    repetitionThreshold: Math.max(2, intEnv("SWITCHROOM_SELF_IMPROVE_THRESHOLD", 2)),
+    t1MaxChangedLines: intEnv("SWITCHROOM_SELF_IMPROVE_T1_MAX_LINES", 30),
+    maxAutoAppliesPerDay: intEnv("SWITCHROOM_SELF_IMPROVE_MAX_AUTO_PER_DAY", 3),
+    maxOutstandingPending: intEnv("SWITCHROOM_SELF_IMPROVE_MAX_PENDING", 5)
+  };
+}
+function selfImproveEnabled() {
+  return process.env.SWITCHROOM_SELF_IMPROVE !== "0";
+}
+
+// src/self-improve/gate.ts
+var EVIDENCE_MAX = 160;
+function excerpt(s) {
+  const t = s.replace(/\s+/g, " ").trim();
+  return t.length > EVIDENCE_MAX ? t.slice(0, EVIDENCE_MAX - 1) + "\u2026" : t;
+}
+var CORRECTION_PATTERNS = [
+  /\bno,? that'?s (?:not right|wrong|incorrect|not what)\b/i,
+  /\bthat'?s (?:not right|wrong|incorrect)\b/i,
+  /\bthat'?s not what i (?:asked|wanted|meant|said)\b/i,
+  /\bi (?:already )?told you\b/i,
+  /\bi keep telling you\b/i,
+  /\b(?:stop|don'?t) (?:doing|do) (?:that|this)\b/i,
+  /\bwhy did you\b.*\b(?:again|still)\b/i,
+  /\byou (?:keep|always) (?:doing|getting)\b/i,
+  /\bnot again\b/i,
+  /\byou got it wrong\b/i
+];
+var DIRECTIVE_PATTERNS = [
+  /\b(?:always|never|every time|from now on|going forward|remember to|make sure (?:to|you))\b/i
+];
+var DIRECTIVE_STOPWORDS = new Set([
+  "always",
+  "never",
+  "every",
+  "time",
+  "from",
+  "now",
+  "on",
+  "going",
+  "forward",
+  "remember",
+  "to",
+  "make",
+  "sure",
+  "you",
+  "the",
+  "a",
+  "an",
+  "please",
+  "and",
+  "do",
+  "don't",
+  "dont",
+  "that",
+  "this",
+  "is",
+  "be",
+  "i",
+  "your",
+  "my"
+]);
+function directiveContentWords(text) {
+  return new Set(text.toLowerCase().replace(/[^a-z0-9\s'-]/g, " ").split(/\s+/).filter((w) => w.length > 2 && !DIRECTIVE_STOPWORDS.has(w)));
+}
+function jaccard(a, b) {
+  if (a.size === 0 || b.size === 0)
+    return 0;
+  let inter = 0;
+  for (const w of a)
+    if (b.has(w))
+      inter += 1;
+  return inter / (a.size + b.size - inter);
+}
+var DIRECTIVE_SIM_THRESHOLD = 0.5;
+function directiveStatements(text) {
+  const out = [];
+  for (const sentence of text.split(/(?<=[.!?\n])\s+/)) {
+    if (DIRECTIVE_PATTERNS.some((re) => re.test(sentence))) {
+      const words = directiveContentWords(sentence);
+      if (words.size > 0)
+        out.push({ words, sample: sentence.trim() });
+    }
+  }
+  return out;
+}
+var FIX_PATTERNS = [
+  /\bEdit\(([^)]+)\)/g,
+  /\bBash\(([^)]+)\)/g
+];
+function fixFingerprints(text) {
+  const out = [];
+  for (const re of FIX_PATTERNS) {
+    re.lastIndex = 0;
+    let m;
+    while ((m = re.exec(text)) !== null) {
+      const inner = (m[1] ?? "").replace(/\s+/g, " ").trim().toLowerCase();
+      if (inner.length >= 3)
+        out.push(`${m[0].split("(")[0]}::${inner}`);
+    }
+  }
+  return out;
+}
+function runGate(messages, cfg = resolveSelfImproveConfig()) {
+  const threshold = cfg.repetitionThreshold;
+  const signals = [];
+  if (!messages || messages.length === 0) {
+    return { tripped: false, signals };
+  }
+  let correctionCount = 0;
+  let lastCorrection = "";
+  const directiveClusters = [];
+  const fixCounts = new Map;
+  for (const m of messages) {
+    const text = typeof m.text === "string" ? m.text : "";
+    if (text.length === 0)
+      continue;
+    if (m.role === "user") {
+      if (CORRECTION_PATTERNS.some((re) => re.test(text))) {
+        correctionCount += 1;
+        lastCorrection = text;
+      }
+      for (const stmt of directiveStatements(text)) {
+        const hit = directiveClusters.find((c) => jaccard(c.words, stmt.words) >= DIRECTIVE_SIM_THRESHOLD);
+        if (hit) {
+          hit.count += 1;
+          if (stmt.sample.length > hit.sample.length)
+            hit.sample = stmt.sample;
+        } else {
+          directiveClusters.push({ words: stmt.words, count: 1, sample: stmt.sample });
+        }
+      }
+    } else if (m.role === "assistant") {
+      for (const fp of fixFingerprints(text)) {
+        fixCounts.set(fp, (fixCounts.get(fp) ?? 0) + 1);
+      }
+    }
+  }
+  if (correctionCount >= threshold) {
+    signals.push(buildSignal("operator-correction", `Operator pushed back ${correctionCount}\u00d7 this turn \u2014 the same correction recurred`, correctionCount, lastCorrection));
+  }
+  for (const { count, sample } of directiveClusters) {
+    if (count >= threshold) {
+      signals.push(buildSignal("directive-not-bound", `A standing rule was restated ${count}\u00d7 \u2014 it isn't binding where the action runs`, count, sample));
+    }
+  }
+  for (const [fp, count] of fixCounts) {
+    if (count >= threshold) {
+      signals.push(buildSignal("repeated-manual-fix", `The same manual fix was applied ${count}\u00d7 \u2014 codify it so it stops recurring`, count, fp));
+    }
+  }
+  return { tripped: signals.length > 0, signals };
+}
+function buildSignal(kind, reason, occurrences, evidence) {
+  return { kind, reason, occurrences, evidence: excerpt(evidence) };
+}
+
+// src/self-improve/review-prompt.ts
+var REVIEW_SOURCE = "self_improve_review";
+var REVIEW_BANNER = "[self-improvement review]";
+function isReviewTurn(text) {
+  if (!text)
+    return false;
+  if (text.startsWith(REVIEW_BANNER))
+    return true;
+  const env = /^<channel\b[^>]*>/i.exec(text);
+  if (env) {
+    if (env[0].includes(`source="${REVIEW_SOURCE}"`))
+      return true;
+    if (text.slice(env[0].length).replace(/^\s+/, "").startsWith(REVIEW_BANNER)) {
+      return true;
+    }
+  }
+  return false;
+}
+function buildReviewPrompt(signals) {
+  const cfg = resolveSelfImproveConfig();
+  const lines = [];
+  lines.push(`${REVIEW_BANNER} The turn-end gate detected a learning ` + "signal \u2014 a correction or pattern that should bind on future runs. " + "Run a focused, forked review. Use ONLY memory tools (recall/retain/" + "directives) and skill read/write tools (Read/Write/Edit under your " + "own .claude/skills/, skill_* / skill-personal). Do NOT touch " + "anything else, do NOT reply to the operator, and end the turn when " + "done.");
+  lines.push("");
+  lines.push("Signals detected this turn:");
+  for (const s of signals) {
+    lines.push(`  - [${s.kind}] ${s.reason}`);
+    if (s.evidence)
+      lines.push(`      evidence: ${s.evidence}`);
+  }
+  lines.push("");
+  lines.push("Diagnose and act, strictly within these rules:");
+  lines.push("  1. State the lesson in one line and the SMALLEST change that makes " + "it bind (prefer the strongest deterministic layer: a skill edit " + "over a Hindsight directive).");
+  lines.push(`  2. Classify the change's blast radius. T1 = a small (<= ${cfg.t1MaxChangedLines} ` + "changed lines, single file) reversible edit to a skill you ALREADY " + "own. Anything medium / shared / new-cron / new-skill / cross-agent " + "/ irreversible is T2 or T3.");
+  lines.push("  3. For a T1: BEFORE editing, confirm the skill has evals/evals.json " + "and that your edit passes them without regressing baseline (the " + "skill-creator grader + aggregate_benchmark.py). Land the edit via " + "the native skill-write path (Write/Edit) only if the evals pass. " + "If the skill has no evals.json, do NOT auto-apply \u2014 treat it as T2.");
+  lines.push("  4. For T2/T3: do NOT apply anything. Record a one-line pending " + "suggestion (the operator actions it later). Never auto-create a " + "cron or a new skill, never edit a shared/other-agent skill.");
+  lines.push("");
+  lines.push(`Rate limits in effect: <= ${cfg.maxAutoAppliesPerDay} auto-applies/day, ` + `<= ${cfg.maxOutstandingPending} outstanding pending suggestions. If a ` + "limit is reached, stop and leave a note rather than forcing the change.");
+  return lines.join(`
+`);
+}
+
+// src/self-improve/review-context.ts
+import {
+  existsSync,
+  mkdirSync,
+  readFileSync,
+  rmSync,
+  writeFileSync
+} from "node:fs";
+import { join } from "node:path";
+var REVIEW_CONTEXT_FILE = "self-improve-review-context.json";
+function contextPath(stateDir) {
+  return join(stateDir, REVIEW_CONTEXT_FILE);
+}
+function writeReviewContext(stateDir, ctx) {
+  if (!existsSync(stateDir)) {
+    mkdirSync(stateDir, { recursive: true, mode: 493 });
+  }
+  writeFileSync(contextPath(stateDir), JSON.stringify(ctx, null, 2), "utf-8");
+}
+function clearReviewContext(stateDir) {
+  try {
+    rmSync(contextPath(stateDir), { force: true });
+  } catch {}
+}
+
+// src/cli/self-improve-stop.ts
+var SCAN_WINDOW = 40;
+function readStdin() {
+  try {
+    return readFileSync2(0, "utf8");
+  } catch {
+    return "";
+  }
+}
+function shortHash(s) {
+  let h = 5381;
+  for (let i = 0;i < s.length; i++) {
+    h = (h << 5) + h + s.charCodeAt(i) | 0;
+  }
+  return (h >>> 0).toString(36);
+}
+function flatten(entry) {
+  if (!entry || typeof entry !== "object")
+    return null;
+  const e = entry;
+  let msg = null;
+  if ((e.type === "user" || e.type === "assistant") && e.message && typeof e.message === "object") {
+    msg = e.message;
+  } else if (typeof e.role === "string" && "content" in e) {
+    msg = e;
+  }
+  if (!msg)
+    return null;
+  const role = typeof msg.role === "string" ? msg.role : "";
+  if (role !== "user" && role !== "assistant")
+    return null;
+  const content = msg.content;
+  let text = "";
+  if (typeof content === "string") {
+    text = content;
+  } else if (Array.isArray(content)) {
+    const parts = [];
+    for (const part of content) {
+      if (!part || typeof part !== "object")
+        continue;
+      const p = part;
+      if (p.type === "text" && typeof p.text === "string") {
+        parts.push(p.text);
+      } else if (p.type === "tool_use" && typeof p.name === "string") {
+        const input = p.input ?? {};
+        const file = typeof input.file_path === "string" && input.file_path || typeof input.path === "string" && input.path || "";
+        let arg;
+        if (typeof input.command === "string") {
+          arg = input.command;
+        } else if (file) {
+          const editish = [
+            typeof input.old_string === "string" ? input.old_string : "",
+            typeof input.new_string === "string" ? input.new_string : "",
+            typeof input.content === "string" ? input.content : "",
+            input.edits !== undefined ? JSON.stringify(input.edits) : ""
+          ];
+          arg = editish.some((x) => x.length > 0) ? `${file} #${shortHash(editish.join(" "))}` : file;
+        } else {
+          arg = "";
+        }
+        parts.push(`${p.name}(${arg})`);
+      }
+    }
+    text = parts.join(" ");
+  }
+  return { role, text };
+}
+function readTranscript(path) {
+  if (!path)
+    return [];
+  let raw;
+  try {
+    raw = readFileSync2(path, "utf8");
+  } catch {
+    return [];
+  }
+  const out = [];
+  for (const line of raw.split(`
+`)) {
+    const t = line.trim();
+    if (!t)
+      continue;
+    try {
+      const m = flatten(JSON.parse(t));
+      if (m && m.text)
+        out.push(m);
+    } catch {}
+  }
+  return out.length > SCAN_WINDOW ? out.slice(-SCAN_WINDOW) : out;
+}
+function resolveStateDir() {
+  return process.env.TELEGRAM_STATE_DIR ?? join2(homedir(), ".claude", "channels", "telegram");
+}
+function resolveSocketPath() {
+  if (process.env.SWITCHROOM_GATEWAY_SOCKET) {
+    return process.env.SWITCHROOM_GATEWAY_SOCKET;
+  }
+  const stateDir = process.env.TELEGRAM_STATE_DIR ?? join2(homedir(), ".claude", "channels", "telegram");
+  return join2(stateDir, "gateway.sock");
+}
+function resolveChatId() {
+  return process.env.SWITCHROOM_SELF_IMPROVE_CHAT_ID ?? process.env.SWITCHROOM_DEFAULT_CHAT_ID ?? "self-improve";
+}
+function injectReview(agentName, text, sessionId) {
+  const socketPath = resolveSocketPath();
+  const chatId = resolveChatId();
+  const now = Date.now();
+  const envelope = {
+    type: "inject_inbound",
+    agentName,
+    inbound: {
+      type: "inbound",
+      chatId,
+      messageId: now,
+      user: "self-improve",
+      userId: 0,
+      ts: now,
+      text,
+      meta: {
+        source: REVIEW_SOURCE,
+        triggering_session: sessionId
+      }
+    }
+  };
+  let settled = false;
+  const sock = createConnection(socketPath);
+  const done = () => {
+    if (settled)
+      return;
+    settled = true;
+    try {
+      sock.end();
+    } catch {}
+    try {
+      sock.destroy();
+    } catch {}
+    process.exit(0);
+  };
+  const timer = setTimeout(done, 3000);
+  timer.unref?.();
+  sock.on("connect", () => {
+    try {
+      sock.write(JSON.stringify(envelope) + `
+`, () => done());
+    } catch {
+      done();
+    }
+  });
+  sock.on("error", () => done());
+}
+function main() {
+  if (!selfImproveEnabled())
+    process.exit(0);
+  const raw = readStdin().trim();
+  if (!raw)
+    process.exit(0);
+  let event;
+  try {
+    event = JSON.parse(raw);
+  } catch {
+    process.exit(0);
+  }
+  const transcriptPath = typeof event.transcript_path === "string" ? event.transcript_path : "";
+  const sessionId = typeof event.session_id === "string" ? event.session_id : "unknown";
+  const messages = readTranscript(transcriptPath);
+  const lastUser = [...messages].reverse().find((m) => m.role === "user");
+  if (lastUser && isReviewTurn(lastUser.text)) {
+    clearReviewContext(resolveStateDir());
+    process.exit(0);
+  }
+  const scanMessages = messages.filter((m) => !isReviewTurn(m.text));
+  const gate = runGate(scanMessages);
+  if (!gate.tripped)
+    process.exit(0);
+  const agentName = process.env.SWITCHROOM_AGENT_NAME ?? "";
+  if (!agentName)
+    process.exit(0);
+  try {
+    writeReviewContext(resolveStateDir(), {
+      created_at: new Date().toISOString(),
+      triggering_session: sessionId,
+      chat_id: resolveChatId(),
+      signals: gate.signals
+    });
+  } catch {}
+  const prompt = buildReviewPrompt(gate.signals);
+  injectReview(agentName, prompt, sessionId);
+}
+main();