dsh-repeat-guard 0.1.2 → 0.1.4

package/src/detect.ts

@@ -2,34 +2,84 @@
  * 复读退化判定：查表。
  *
  * 表里是已知会被模型复读的空话短句，**连标点一起写**，表本身可配置（见 config.ts）。
- * 判定是"思考里有没有独占一行的表项"：整行与表项全等，不拆解行的内部。
+ * 基本判据是"整行与表项全等"，不拆解行的内部。
  *
- * 阈值是"连续命中多少行才拦"：默认 1 表示发现即拦；调到 n 时，要连着 n 行都命中
- * （句子可以各不相同）才拦，中间夹一行不命中的就重新计数。
+ * 行内重复检测（可开关，默认关）打开时，"整行由多个表项首尾相接拼成"也算命中，
+ * 用来兜住把短句连成一串写在同一行里的模型（如"好。好。好。"）。
+ *
+ * 阈值是"连续命中多少次才拦"：默认 1 表示发现即拦。计数单位是表项——整行全等算一次，
+ * 开了行内检测后行内每一段各算一次；中间夹一行不命中的就重新计数。
+ */
+
+/**
+ * 在 line 的 at 处能匹配到的最长表项长度。
+ * @param line - 已小写化的行。
+ * @param at - 起始下标。
+ * @param fragments - 小写化的短句表。
+ * @returns 匹配到的长度，0 表示这个位置没有表项。
+ */
+function longestFragmentAt(line: string, at: number, fragments: ReadonlySet<string>): number {
+  let size = 0;
+  for (const fragment of fragments) {
+    if (fragment.length > size && line.startsWith(fragment, at)) {
+      size = fragment.length;
+    }
+  }
+  return size;
+}
+
+/**
+ * 这一行算几次命中。
+ * @param line - 已 trim 并小写化的行。
+ * @param fragments - 小写化的短句表。
+ * @param inlineRepeat - 是否把"整行由多个表项拼成"也算命中。
+ * @returns 0 表示不是命中行，其余为这一行贡献的次数。
  */
+function countHits(line: string, fragments: ReadonlySet<string>, inlineRepeat: boolean): number {
+  if (fragments.has(line)) {
+    return 1;
+  }
+  if (!inlineRepeat) {
+    return 0;
+  }
+  let at = 0;
+  let parts = 0;
+  while (at < line.length) {
+    const size = longestFragmentAt(line, at, fragments);
+    if (size === 0) {
+      return 0;
+    }
+    at += size;
+    parts += 1;
+  }
+  return parts > 1 ? parts : 0;
+}
 
 /**
  * 找出退化的碎片行。
  * @param text - 已生成的思考文本。
  * @param fragments - 小写化的短句表。
- * @param threshold - 连续命中多少行才判定为复读。
+ * @param threshold - 连续命中多少次才判定为复读。
+ * @param inlineRepeat - 是否启用行内重复检测。
  * @returns 计数成立时返回那一行的原文，否则返回 null。
  */
 export function findDegenerateLine(
   text: string,
   fragments: ReadonlySet<string>,
   threshold: number,
+  inlineRepeat: boolean,
 ): string | null {
   let run = 0;
   for (const raw of text.split('\n')) {
     const line = raw.trim();
-    if (line !== '' && fragments.has(line.toLowerCase())) {
-      run += 1;
-      if (run >= threshold) {
-        return line;
-      }
-    } else {
+    const hits = line === '' ? 0 : countHits(line.toLowerCase(), fragments, inlineRepeat);
+    if (hits === 0) {
       run = 0;
+      continue;
+    }
+    run += hits;
+    if (run >= threshold) {
+      return line;
     }
   }
   return null;

package/src/index.ts

@@ -22,7 +22,7 @@
 // 本文件只做装配，具体逻辑在各自的模块里。
 
 import type { Context } from '@deepseek-ai/cordis';
-import { createConfigSource } from './config.js';
+import { createRuntime } from './config.js';
 import { createStreamGuard } from './stream-guard.js';
 import { createTurnStoppingGuard } from './turn-stopping-guard.js';
 import type { GuardState } from './types.js';
@@ -35,7 +35,7 @@ export default function repeatGuard(ctx: Context): void {
   // 直接写 stdout：dsh 把插件的 stdout 收进 journal，便于确认插件确实被加载。
   console.log('[repeat-guard] 已加载，复读拦截生效');
   const state: GuardState = { pending: new Set() };
-  const readConfig = createConfigSource(ctx);
-  ctx.on('llm/stream', createStreamGuard(state, readConfig), { global: true });
-  ctx.on('agent/turn-stopping', createTurnStoppingGuard(state));
+  const runtime = createRuntime(ctx);
+  ctx.on('llm/stream', createStreamGuard(state, runtime.read, runtime.countHit), { global: true });
+  ctx.on('agent/turn-stopping', createTurnStoppingGuard(state, runtime.read));
 }

package/src/resume.ts

@@ -11,37 +11,22 @@
  *
  * 所以在监听器里调 `agent.steer(...)` 推入一条输入，本轮就会再跑一步，而不是停下来等
  * 用户输入。dsh 自己的 dsh-hooks-claude-code/lib/index.js:300 就是这么用的。
+ *
+ * 推什么内容由配置决定（见 config.ts 的 resumeText / resumeSummary）。
  */
 
 import type { Agent } from '@deepseek-ai/dsh-agent';
 import { createUserMessage } from '@deepseek-ai/dsh-llm';
+import type { RepeatGuardConfig } from './config.js';
 import type { GuardState } from './types.js';
 
-/** 续跑时推给模型的指令正文。 */
-const RESUME_TEXT = [
-  '[复读拦截]',
-  '你上一段思考退化成了碎片复读（反复输出"好。"一类的空话），已被系统截断。',
-  '请直接继续执行下一步，不要再输出任何确认语、寒暄或空话。',
-].join('\n');
-
-/**
- * 这条注入消息的来源标记。
- * `form: 'notice'` 要求同时给出 `summary`（dsh-llm 的 ContextFormed），
- * 客户端据此把它渲染成折叠的 context 行，而不是用户气泡。
- */
-const PLUGIN_SOURCE = {
-  kind: 'plugin',
-  plugin: 'repeat-guard',
-  form: 'notice',
-  summary: '复读已截断：请继续执行',
-} as const;
-
 /**
  * 本轮即将关闭时，若上一步刚被掐断过，就推一条输入让本轮继续。
  * @param state - 跨监听保留的拦截状态。
  * @param agent - 本轮所属的 agent 句柄。
+ * @param config - 当前配置，正文与摘要都从这里取。
  */
-export function reviveTurn(state: GuardState, agent: Agent): void {
+export function reviveTurn(state: GuardState, agent: Agent, config: RepeatGuardConfig): void {
   const sessionId = agent.session.id;
   if (!state.pending.has(sessionId)) {
     return;
@@ -50,8 +35,15 @@ export function reviveTurn(state: GuardState, agent: Agent): void {
   console.log(`[repeat-guard] turn-stopping：会话 ${sessionId} 续跑一步`);
   agent.steer(
     createUserMessage({
-      content: [{ type: 'text', text: RESUME_TEXT }],
-      source: PLUGIN_SOURCE,
+      content: [{ type: 'text', text: config.resumeText }],
+      // `form: 'notice'` 要求同时给出 `summary`（dsh-llm 的 ContextFormed），
+      // 客户端据此把它渲染成折叠的 context 行，而不是用户气泡。
+      source: {
+        kind: 'plugin',
+        plugin: 'repeat-guard',
+        form: 'notice',
+        summary: config.resumeSummary,
+      },
     }),
   );
 }

package/src/stream-guard.ts

@@ -29,6 +29,7 @@ type StreamListener = (
  * @param sessionId - 当前会话 id；有值才登记待续跑标记。
  * @param state - 跨监听保留的拦截状态。
  * @param readConfig - 取当前配置；每个 chunk 现取，改设置立即生效。
+ * @param countHit - 记一次拦截；只在真的掐断时调用。
  * @returns 包好的流。
  */
 async function* guardStream(
@@ -36,6 +37,7 @@ async function* guardStream(
   sessionId: GenerateOptions['sessionId'],
   state: GuardState,
   readConfig: ConfigSource,
+  countHit: () => void,
 ): AsyncGenerator<StreamChunk> {
   const iterator = downstream[Symbol.asyncIterator]();
   let accumulated = '';
@@ -49,10 +51,16 @@ async function* guardStream(
       if (chunk.type === 'reasoning-delta') {
         accumulated += chunk.text;
         const config = readConfig();
-        const hit = findDegenerateLine(accumulated, config.fragments, config.threshold);
+        const hit = findDegenerateLine(
+          accumulated,
+          config.fragments,
+          config.threshold,
+          config.inlineRepeat,
+        );
         if (hit !== null) {
           const probe = await iterator.next();
           if (!probe.done && probe.value.type === 'reasoning-delta') {
+            countHit();
             if (sessionId !== undefined) {
               state.pending.add(sessionId);
             }
@@ -92,14 +100,19 @@ async function* guardStream(
  * 造一个 `llm/stream` 监听器。
  * @param state - 跨监听保留的拦截状态。
  * @param readConfig - 取当前配置。
+ * @param countHit - 记一次拦截。
  * @returns 监听器；辅助调用直接透传，其余包一层复读检测。
  */
-export function createStreamGuard(state: GuardState, readConfig: ConfigSource): StreamListener {
+export function createStreamGuard(
+  state: GuardState,
+  readConfig: ConfigSource,
+  countHit: () => void,
+): StreamListener {
   return (options, next) => {
     // 辅助调用（上下文压缩、会话标题）不参与检测。
     if (options.purpose !== undefined) {
       return next();
     }
-    return guardStream(next(), options.sessionId, state, readConfig);
+    return guardStream(next(), options.sessionId, state, readConfig, countHit);
   };
 }

package/src/turn-stopping-guard.ts

@@ -6,16 +6,18 @@
  */
 
 import type { Agent } from '@deepseek-ai/dsh-agent';
+import type { ConfigSource } from './config.js';
 import { reviveTurn } from './resume.js';
 import type { GuardState } from './types.js';
 
 /**
  * 造一个 `agent/turn-stopping` 监听器。
  * @param state - 跨监听保留的拦截状态。
+ * @param readConfig - 取当前配置。
  * @returns 监听器；只在有待续跑标记时推一条输入。
  */
-export function createTurnStoppingGuard(state: GuardState) {
+export function createTurnStoppingGuard(state: GuardState, readConfig: ConfigSource) {
   return (payload: { agent: Agent }): void => {
-    reviveTurn(state, payload.agent);
+    reviveTurn(state, payload.agent, readConfig());
   };
 }