feat(memory): POV-aware context filtering (#5)

packages/core/src/agents/writer.ts

@@ -10,6 +10,7 @@ import { readGenreProfile, readBookRules } from "./rules-reader.js";
 import { validatePostWrite, type PostWriteViolation } from "./post-write-validator.js";
 import { analyzeAITells } from "./ai-tells.js";
 import { filterHooks, filterSummaries, filterSubplots, filterEmotionalArcs, filterCharacterMatrix } from "../utils/context-filter.js";
+import { extractPOVFromOutline, filterMatrixByPOV, filterHooksByPOV } from "../utils/pov-filter.js";
 import { parseCreativeOutput } from "./writer-parser.js";
 import { readFile, writeFile, mkdir, readdir } from "node:fs/promises";
 import { join } from "node:path";
@@ -118,20 +119,29 @@ export class WriterAgent extends BaseAgent {
     const filteredArcs = filterEmotionalArcs(emotionalArcs, chapterNumber);
     const filteredMatrix = filterCharacterMatrix(characterMatrix, volumeOutline, bookRules?.protagonist?.name);
 
+    // POV-aware filtering: limit context to what the POV character knows
+    const povCharacter = extractPOVFromOutline(volumeOutline, chapterNumber);
+    const povFilteredMatrix = povCharacter
+      ? filterMatrixByPOV(filteredMatrix, povCharacter)
+      : filteredMatrix;
+    const povFilteredHooks = povCharacter
+      ? filterHooksByPOV(filteredHooks, povCharacter, chapterSummaries)
+      : filteredHooks;
+
     const creativeUserPrompt = this.buildUserPrompt({
       chapterNumber,
       storyBible,
       volumeOutline,
       currentState,
       ledger: genreProfile.numericalSystem ? ledger : "",
-      hooks: filteredHooks,
+      hooks: povFilteredHooks,
       recentChapters,
       wordCount: input.wordCountOverride ?? book.chapterWordCount,
       externalContext: input.externalContext,
       chapterSummaries: filteredSummaries,
       subplotBoard: filteredSubplots,
       emotionalArcs: filteredArcs,
-      characterMatrix: filteredMatrix,
+      characterMatrix: povFilteredMatrix,
       dialogueFingerprints,
       relevantSummaries,
       parentCanon: hasParentCanon ? parentCanon : undefined,

packages/core/src/index.ts

@@ -38,6 +38,7 @@ export { buildFanficCanonSection, buildCharacterVoiceProfiles, buildFanficModeIn
 // Utils
 export { fetchUrl, searchWeb } from "./utils/web-search.js";
 export { filterHooks, filterSummaries, filterSubplots, filterEmotionalArcs, filterCharacterMatrix } from "./utils/context-filter.js";
+export { extractPOVFromOutline, filterMatrixByPOV, filterHooksByPOV } from "./utils/pov-filter.js";
 export { ConsolidatorAgent } from "./agents/consolidator.js";
 export { MemoryDB, type Fact, type StoredSummary } from "./state/memory-db.js";
 export { StateValidatorAgent } from "./agents/state-validator.js";

packages/core/src/utils/pov-filter.ts

@@ -0,0 +1,149 @@
+/**
+ * POV-aware context filtering.
+ *
+ * Filters truth file content based on the current POV character's
+ * information boundaries. Characters should only "see" information
+ * they've actually witnessed or been told about.
+ *
+ * Works with markdown-based truth files (no DB dependency).
+ * When MemoryDB is available, can do more precise queries.
+ */
+
+/**
+ * Extract the POV character from the volume outline for a given chapter.
+ * Looks for patterns like "POV: 角色名" or "视角: 角色名" or "POV: CharacterName"
+ * in the chapter's section of the outline.
+ */
+export function extractPOVFromOutline(volumeOutline: string, chapterNumber: number): string | null {
+  // Find the section for this chapter
+  const lines = volumeOutline.split("\n");
+
+  // Look for chapter reference near the chapter number
+  const chapterPatterns = [
+    new RegExp(`第${chapterNumber}章`),
+    new RegExp(`Chapter\\s+${chapterNumber}\\b`),
+    new RegExp(`\\b${chapterNumber}\\b.*章`),
+  ];
+
+  let inChapterSection = false;
+  for (const line of lines) {
+    // Check if we're in the right chapter section
+    if (chapterPatterns.some((p) => p.test(line))) {
+      inChapterSection = true;
+    } else if (inChapterSection && /^[#-]/.test(line) && !line.includes(String(chapterNumber))) {
+      // Left the chapter section
+      break;
+    }
+
+    if (inChapterSection) {
+      // Look for POV declaration
+      const povMatch = line.match(/(?:POV|视角|pov)[：:\s]+([^\s，,。.、]+)/i);
+      if (povMatch) return povMatch[1]!;
+    }
+  }
+
+  return null;
+}
+
+/**
+ * Filter character_matrix information boundaries for the POV character.
+ * Returns only what the POV character knows — strips other characters' "known info".
+ */
+export function filterMatrixByPOV(characterMatrix: string, povCharacter: string): string {
+  if (!characterMatrix || characterMatrix === "(文件尚未创建)") return characterMatrix;
+  if (!povCharacter) return characterMatrix;
+
+  // Find the 信息边界 / Information Boundaries section
+  const sections = characterMatrix.split(/(?=^###)/m);
+  const filtered = sections.map((section) => {
+    const isInfoBoundary = /信息边界|Information\s+Boundar/i.test(section);
+    if (!isInfoBoundary) return section;
+
+    // In the info boundary table, keep only the POV character's row
+    // and add a note about what other characters know
+    const lines = section.split("\n");
+    const headerLines = lines.filter((l) =>
+      l.startsWith("|") && (l.includes("---") || l.includes("角色") || l.includes("Character") || l.includes("已知") || l.includes("Known")),
+    );
+    const dataLines = lines.filter((l) =>
+      l.startsWith("|") && !l.includes("---") && !l.includes("角色") && !l.includes("Character") && !l.includes("已知") && !l.includes("Known"),
+    );
+
+    // Keep POV character's row + a summary note
+    const povRows = dataLines.filter((l) => l.includes(povCharacter));
+    const otherCharCount = dataLines.length - povRows.length;
+
+    const sectionHeader = lines.find((l) => l.startsWith("###"));
+    const result = [
+      sectionHeader ?? "### 信息边界",
+      `（当前视角：${povCharacter}，其他 ${otherCharCount} 个角色的信息边界已隐藏）`,
+      ...headerLines,
+      ...povRows,
+    ];
+
+    return result.join("\n");
+  });
+
+  return filtered.join("\n");
+}
+
+/**
+ * Filter pending_hooks by POV character's knowledge.
+ * Hooks planted in scenes where the POV character was NOT present are hidden.
+ *
+ * This is a heuristic: if the hook's chapter summary mentions the POV character,
+ * they likely know about it.
+ */
+export function filterHooksByPOV(
+  hooks: string,
+  povCharacter: string,
+  chapterSummaries: string,
+): string {
+  if (!hooks || hooks === "(文件尚未创建)") return hooks;
+  if (!povCharacter) return hooks;
+
+  const lines = hooks.split("\n");
+  const headerLines = lines.filter((l) =>
+    l.startsWith("|") && (l.includes("hook_id") || l.includes("---")),
+  );
+  const dataLines = lines.filter((l) =>
+    l.startsWith("|") && !l.includes("hook_id") && !l.includes("---"),
+  );
+
+  // Parse summary rows to find which chapters the POV character appeared in
+  const povChapters = new Set<number>();
+  if (chapterSummaries) {
+    for (const line of chapterSummaries.split("\n")) {
+      if (line.includes(povCharacter)) {
+        const match = line.match(/\|\s*(\d+)\s*\|/);
+        if (match) povChapters.add(parseInt(match[1]!, 10));
+      }
+    }
+  }
+
+  // Keep hooks where:
+  // 1. The POV character was present in the source chapter, OR
+  // 2. The hook mentions the POV character directly, OR
+  // 3. We can't determine (keep to be safe)
+  const filtered = dataLines.filter((row) => {
+    // If hook directly mentions POV character, keep it
+    if (row.includes(povCharacter)) return true;
+
+    // Extract source chapter from hook row
+    const chapterMatch = row.match(/\|\s*(\d+)\s*\|/);
+    if (!chapterMatch) return true; // can't determine, keep
+
+    const sourceChapter = parseInt(chapterMatch[1]!, 10);
+    // If POV was in that chapter, they know about the hook
+    if (povChapters.has(sourceChapter)) return true;
+
+    // POV wasn't in that chapter — hide this hook
+    return false;
+  });
+
+  // Fallback: if filtering removes everything, return original
+  if (filtered.length === 0 && dataLines.length > 0) return hooks;
+
+  const nonTableLines = lines.filter((l) => !l.startsWith("|"));
+  return [...nonTableLines, ...headerLines, ...filtered].join("\n");
+}