@actalk/inkos-core 1.6.3 → 1.7.0

package/dist/pipeline/runner.js

@@ -1,3 +1,4 @@
+import { AsyncLocalStorage } from "node:async_hooks";
 import { chatCompletion, createLLMClient } from "../llm/provider.js";
 import { ArchitectAgent } from "../agents/architect.js";
 import { FoundationReviewerAgent } from "../agents/foundation-reviewer.js";
@@ -174,15 +175,35 @@ export function buildImportFoundationSource(chapters, language, options = {}) {
         anchorText,
     ].join("\n");
 }
+/** Human-readable description of each manual-revision gate, surfaced in revisionDiagnostics. */
+const REVISION_GATE_STANDARDS = {
+    strict: "A revision is applied only when blocking, critical, and AI-tell counts do not worsen, and at least blocking or AI-tell issues improve.",
+    lenient: "A revision is applied whenever blocking, critical, and AI-tell counts do not worsen; no improvement is required (lenient gate).",
+    always: "Manual revisions are always applied; audit counts are recorded for reference only (always gate).",
+};
 export class PipelineRunner {
     state;
     config;
     agentClients = new Map();
+    operationContext = new AsyncLocalStorage();
     memoryIndexFallbackWarned = false;
     constructor(config) {
         this.config = config;
         this.state = new StateManager(config.projectRoot);
     }
+    async runWithAbortSignal(signal, task) {
+        signal?.throwIfAborted();
+        return this.operationContext.run({ signal }, async () => {
+            signal?.throwIfAborted();
+            return task();
+        });
+    }
+    currentAbortSignal() {
+        return this.operationContext.getStore()?.signal;
+    }
+    throwIfOperationAborted() {
+        this.currentAbortSignal()?.throwIfAborted();
+    }
     localize(language, messages) {
         return language === "en" ? messages.en : messages.zh;
     }
@@ -361,6 +382,7 @@ export class PipelineRunner {
             bookId,
             logger: this.config.logger?.child(agent),
             onStreamProgress: this.config.onStreamProgress,
+            signal: this.currentAbortSignal(),
         };
     }
     createAgentContext(agent, bookId) {
@@ -991,10 +1013,13 @@ export class PipelineRunner {
             const blockingDidNotWorsen = effectivePostRevision.blockingCount <= preRevision.blockingCount;
             const criticalDidNotWorsen = effectivePostRevision.criticalCount <= preRevision.criticalCount;
             const aiDidNotWorsen = effectivePostRevision.aiTellCount <= preRevision.aiTellCount;
-            const shouldApplyRevision = blockingDidNotWorsen
-                && criticalDidNotWorsen
-                && aiDidNotWorsen
-                && (improvedBlocking || improvedAITells);
+            const didNotWorsen = blockingDidNotWorsen && criticalDidNotWorsen && aiDidNotWorsen;
+            const revisionGate = this.config.revisionGate ?? "strict";
+            const shouldApplyRevision = revisionGate === "always"
+                ? true
+                : revisionGate === "lenient"
+                    ? didNotWorsen
+                    : didNotWorsen && (improvedBlocking || improvedAITells);
             if (!shouldApplyRevision) {
                 const remainingIssues = effectivePostRevision.revisionBlockingIssues
                     .filter((issue) => issue.severity === "warning" || issue.severity === "critical")
@@ -1013,7 +1038,7 @@ export class PipelineRunner {
                     status: "unchanged",
                     skippedReason: `Manual revision kept original chapter: before blocking=${preRevision.blockingCount}, critical=${preRevision.criticalCount}, aiTell=${preRevision.aiTellCount}; after blocking=${effectivePostRevision.blockingCount}, critical=${effectivePostRevision.criticalCount}, aiTell=${effectivePostRevision.aiTellCount}.`,
                     revisionDiagnostics: {
-                        standard: "A revision is applied only when blocking, critical, and AI-tell counts do not worsen, and at least blocking or AI-tell issues improve.",
+                        standard: REVISION_GATE_STANDARDS[revisionGate],
                         before: {
                             blockingCount: preRevision.blockingCount,
                             criticalCount: preRevision.criticalCount,
@@ -1157,6 +1182,7 @@ export class PipelineRunner {
     // Full pipeline (convenience — runs draft + audit + revise in one shot)
     // ---------------------------------------------------------------------------
     async writeNextChapter(bookId, wordCount, temperatureOverride) {
+        this.throwIfOperationAborted();
         const releaseLock = await this.state.acquireBookLock(bookId);
         try {
             return await this._writeNextChapterLocked(bookId, wordCount, temperatureOverride, this.config.externalContext);
@@ -1184,6 +1210,7 @@ export class PipelineRunner {
         }
     }
     async _writeNextChapterLocked(bookId, wordCount, temperatureOverride, externalContext) {
+        this.throwIfOperationAborted();
         await this.state.ensureControlDocuments(bookId);
         const book = await this.state.loadBookConfig(bookId);
         const bookDir = this.state.bookDir(bookId);
@@ -1220,6 +1247,7 @@ export class PipelineRunner {
             ...(wordCount ? { wordCountOverride: wordCount } : {}),
             ...(temperatureOverride ? { temperatureOverride } : {}),
         });
+        this.throwIfOperationAborted();
         const writerCount = countChapterLength(output.content, lengthSpec.countingMode);
         // Token usage accumulator
         let totalUsage = output.tokenUsage ?? { promptTokens: 0, completionTokens: 0, totalTokens: 0 };
@@ -1303,6 +1331,7 @@ export class PipelineRunner {
             normalizeApplied = reviewResult.normalizeApplied;
             preAuditNormalizedWordCount = reviewResult.preAuditNormalizedWordCount;
         }
+        this.throwIfOperationAborted();
         // 3b. Lightweight per-chapter promotion pass — check if any hooks should
         // be promoted based on advanced_count derived from chapter_summaries.
         // Runs BEFORE persistence so the reviewer of the NEXT chapter sees the
@@ -1326,6 +1355,7 @@ export class PipelineRunner {
                 }
             }
         }
+        this.throwIfOperationAborted();
         // 4. Save the final chapter and truth files from a single persistence source
         this.logStage(stageLanguage, { zh: "落盘最终章节", en: "persisting final chapter" });
         this.logStage(stageLanguage, { zh: "生成最终真相文件", en: "rebuilding final truth files" });
@@ -1848,7 +1878,7 @@ Base the analysis on the text's actual features, not generalities. Support each
                 const response = await chatCompletion(this.config.client, this.config.model, [
                     { role: "system", content: styleSystemPrompt },
                     { role: "user", content: styleUserPrompt },
-                ], { temperature: 0.3 });
+                ], { temperature: 0.3, signal: this.currentAbortSignal() });
                 qualitativeGuide = response.content.trim()
                     ? response.content
                     : this.buildDeterministicStyleGuide(profile, {
@@ -2036,7 +2066,7 @@ ${emotions}
 ## 正传角色矩阵
 ${matrix}`,
             },
-        ], { temperature: 0.3 });
+        ], { temperature: 0.3, signal: this.currentAbortSignal() });
         // Append deterministic meta block (LLM may hallucinate timestamps)
         const metaBlock = [
             "",
@@ -2089,6 +2119,7 @@ ${matrix}`,
      * Step 2: Sequentially replay each chapter through ChapterAnalyzer to build truth files.
      */
     async importChapters(input) {
+        this.throwIfOperationAborted();
         const releaseLock = await this.state.acquireBookLock(input.bookId);
         try {
             const book = await this.state.loadBookConfig(input.bookId);
@@ -2116,6 +2147,7 @@ ${matrix}`,
                         targetChapters: book.targetChapters,
                     })
                     : await architect.generateFoundationFromImport(book, foundationSource);
+                this.throwIfOperationAborted();
                 await architect.writeFoundationFiles(bookDir, foundation, gp.numericalSystem, resolvedLanguage);
                 await this.resetImportReplayTruthFiles(bookDir, resolvedLanguage);
                 await this.state.saveChapterIndex(input.bookId, [], { allowEmptyWithChapterFiles: true });
@@ -2144,6 +2176,7 @@ ${matrix}`,
             let totalWords = 0;
             let importedCount = 0;
             for (let i = startFrom - 1; i < input.chapters.length; i++) {
+                this.throwIfOperationAborted();
                 const ch = input.chapters[i];
                 const chapterNumber = i + 1;
                 const governedInput = await this.prepareWriteInput(book, bookDir, chapterNumber);
@@ -2162,6 +2195,7 @@ ${matrix}`,
                     contextPackage: governedInput.contextPackage,
                     ruleStack: governedInput.ruleStack,
                 });
+                this.throwIfOperationAborted();
                 const chapterWordCount = countChapterLength(ch.content, countingMode);
                 const persistedOutput = {
                     ...output,