22 lines
833 B
TypeScript
22 lines
833 B
TypeScript
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,
|
|
};
|
|
}
|