feat: unify gateway heartbeat

src/auto-reply/heartbeat.test.ts

@@ -0,0 +1,55 @@
+import { describe, expect, it } from "vitest";
+
+import { stripHeartbeatToken } from "./heartbeat.js";
+import { HEARTBEAT_TOKEN } from "./tokens.js";
+
+describe("stripHeartbeatToken", () => {
+  it("skips empty or token-only replies", () => {
+    expect(stripHeartbeatToken(undefined)).toEqual({
+      shouldSkip: true,
+      text: "",
+    });
+    expect(stripHeartbeatToken("  ")).toEqual({
+      shouldSkip: true,
+      text: "",
+    });
+    expect(stripHeartbeatToken(HEARTBEAT_TOKEN)).toEqual({
+      shouldSkip: true,
+      text: "",
+    });
+  });
+
+  it("keeps content and removes token when mixed", () => {
+    expect(stripHeartbeatToken(`ALERT ${HEARTBEAT_TOKEN}`)).toEqual({
+      shouldSkip: false,
+      text: "ALERT",
+    });
+    expect(stripHeartbeatToken("hello")).toEqual({
+      shouldSkip: false,
+      text: "hello",
+    });
+  });
+
+  it("strips repeated OK tails after heartbeat token", () => {
+    expect(stripHeartbeatToken("HEARTBEAT_OK_OK_OK")).toEqual({
+      shouldSkip: true,
+      text: "",
+    });
+    expect(stripHeartbeatToken("HEARTBEAT_OK_OK")).toEqual({
+      shouldSkip: true,
+      text: "",
+    });
+    expect(stripHeartbeatToken("HEARTBEAT_OK _OK")).toEqual({
+      shouldSkip: true,
+      text: "",
+    });
+    expect(stripHeartbeatToken("HEARTBEAT_OK OK")).toEqual({
+      shouldSkip: true,
+      text: "",
+    });
+    expect(stripHeartbeatToken("ALERT HEARTBEAT_OK_OK")).toEqual({
+      shouldSkip: false,
+      text: "ALERT",
+    });
+  });
+});

src/auto-reply/heartbeat.ts

@@ -0,0 +1,21 @@
+import { HEARTBEAT_TOKEN } from "./tokens.js";
+
+export const HEARTBEAT_PROMPT = "HEARTBEAT";
+
+export function stripHeartbeatToken(raw?: string) {
+  if (!raw) return { shouldSkip: true, text: "" };
+  const trimmed = raw.trim();
+  if (!trimmed) return { shouldSkip: true, text: "" };
+  if (trimmed === HEARTBEAT_TOKEN) return { shouldSkip: true, text: "" };
+  const hadToken = trimmed.includes(HEARTBEAT_TOKEN);
+  let withoutToken = trimmed.replaceAll(HEARTBEAT_TOKEN, "").trim();
+  if (hadToken && withoutToken) {
+    // LLMs sometimes echo malformed HEARTBEAT_OK_OK... tails; strip trailing OK runs to avoid spam.
+    withoutToken = withoutToken.replace(/[\s_]*OK(?:[\s_]*OK)*$/gi, "").trim();
+  }
+  const shouldSkip = withoutToken.length === 0;
+  return {
+    shouldSkip,
+    text: shouldSkip ? "" : withoutToken || trimmed,
+  };
+}