wiki-graph-core 0.3.1 → 0.4.1

package/dist/api/chapter/advance.js

@@ -0,0 +1,171 @@
+import { resolveExtractionPrompt } from "../../runtime/common/prompts.js";
+import { isStageBefore, selectChapterEntries } from "./details.js";
+import { generateChapterGraph, generateChapterSummary } from "./generate.js";
+async function advanceChapterStages(document, options) {
+  if (options.targetStage === "planned") {
+    return {
+      advanced: [],
+      pending: await selectChapterEntries(document, options.chapterId),
+      skipped: []
+    };
+  }
+  const advancedIds = /* @__PURE__ */ new Set();
+  const entries = await selectChapterEntries(document, options.chapterId);
+  const progressState = createAdvanceProgressState(
+    entries,
+    options.targetStage
+  );
+  await emitAdvanceProgress(options, {
+    state: progressState.snapshot(),
+    targetStage: options.targetStage,
+    totalChapters: entries.length,
+    type: "selected"
+  });
+  await emitPlannedSkips(entries, options);
+  if (isStageBefore("sourced", options.targetStage)) {
+    await advanceEntriesToGraphed(document, entries, options, {
+      advancedIds,
+      progressState
+    });
+  }
+  if (isStageBefore("graphed", options.targetStage)) {
+    const graphedEntries = await selectChapterEntries(
+      document,
+      options.chapterId
+    );
+    await advanceEntriesToSummarized(document, graphedEntries, options, {
+      advancedIds,
+      progressState
+    });
+  }
+  const nextEntries = await selectChapterEntries(document, options.chapterId);
+  return {
+    advanced: nextEntries.filter((entry) => advancedIds.has(entry.chapterId)),
+    pending: nextEntries.filter(
+      (entry) => isStageBefore(entry.stage, options.targetStage)
+    ),
+    skipped: nextEntries.filter((entry) => entry.stage === "planned")
+  };
+}
+async function advanceEntriesToGraphed(document, entries, options, state) {
+  for (const entry of entries) {
+    if (entry.stage !== "sourced") {
+      continue;
+    }
+    await emitAdvanceProgress(options, {
+      chapter: entry,
+      step: "graph",
+      targetStage: options.targetStage,
+      type: "started"
+    });
+    await generateChapterGraph(document, entry.chapterId, {
+      extractionPrompt: resolveExtractionPrompt(options.extractionPrompt),
+      llm: options.llm,
+      ...options.logDirPath === void 0 ? {} : { logDirPath: options.logDirPath },
+      ...options.userLanguage === void 0 ? {} : { userLanguage: options.userLanguage },
+      progressTracker: {
+        async advance(wordsCount) {
+          state.progressState.addGraphWords(wordsCount);
+          await emitAdvanceProgress(options, {
+            state: state.progressState.snapshot(),
+            targetStage: options.targetStage,
+            type: "progress"
+          });
+        },
+        async complete() {
+        }
+      }
+    });
+    await emitAdvanceProgress(options, {
+      chapter: entry,
+      step: "graph",
+      targetStage: options.targetStage,
+      type: "completed"
+    });
+    state.advancedIds.add(entry.chapterId);
+  }
+  return await selectChapterEntries(document, options.chapterId);
+}
+async function advanceEntriesToSummarized(document, entries, options, state) {
+  for (const entry of entries) {
+    if (entry.stage !== "graphed") {
+      continue;
+    }
+    await emitAdvanceProgress(options, {
+      chapter: entry,
+      step: "summary",
+      targetStage: options.targetStage,
+      type: "started"
+    });
+    await generateChapterSummary(document, entry.chapterId, {
+      llm: options.llm,
+      ...options.logDirPath === void 0 ? {} : { logDirPath: options.logDirPath },
+      ...options.userLanguage === void 0 ? {} : { userLanguage: options.userLanguage }
+    });
+    state.progressState.addSummaryWords(entry.words);
+    await emitAdvanceProgress(options, {
+      state: state.progressState.snapshot(),
+      targetStage: options.targetStage,
+      type: "progress"
+    });
+    await emitAdvanceProgress(options, {
+      chapter: entry,
+      step: "summary",
+      targetStage: options.targetStage,
+      type: "completed"
+    });
+    state.advancedIds.add(entry.chapterId);
+  }
+  return await selectChapterEntries(document, options.chapterId);
+}
+function createAdvanceProgressState(entries, targetStage) {
+  const totalGraphWords = isStageBefore("sourced", targetStage) ? entries.filter((entry) => entry.stage === "sourced").reduce((sum, entry) => sum + entry.words, 0) : 0;
+  const totalSummaryWords = isStageBefore("graphed", targetStage) ? entries.filter(
+    (entry) => entry.stage === "graphed" || entry.stage === "sourced"
+  ).reduce((sum, entry) => sum + entry.words, 0) : 0;
+  let graphWords = 0;
+  let summaryWords = 0;
+  return {
+    addGraphWords(words) {
+      graphWords = Math.min(totalGraphWords, graphWords + Math.max(0, words));
+    },
+    addSummaryWords(words) {
+      summaryWords = Math.min(
+        totalSummaryWords,
+        summaryWords + Math.max(0, words)
+      );
+    },
+    snapshot() {
+      return {
+        graphWords,
+        summaryWords,
+        totalGraphWords,
+        totalSummaryWords
+      };
+    }
+  };
+}
+async function emitPlannedSkips(entries, options) {
+  for (const entry of entries) {
+    if (entry.stage !== "planned") {
+      continue;
+    }
+    await emitAdvanceProgress(options, {
+      chapter: entry,
+      reason: "planned",
+      targetStage: options.targetStage,
+      type: "skipped"
+    });
+  }
+}
+async function emitAdvanceProgress(options, event) {
+  try {
+    await options.onProgress?.(event);
+  } catch {
+    return;
+  }
+}
+export {
+  advanceChapterStages
+};
+//# sourceMappingURL=advance.js.map

package/dist/api/chapter/advance.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/api/chapter/advance.ts"],"sourcesContent":["import type { Document } from \"../../document/index.js\";\nimport { resolveExtractionPrompt } from \"../../runtime/common/prompts.js\";\nimport { isStageBefore, selectChapterEntries } from \"./details.js\";\nimport { generateChapterGraph, generateChapterSummary } from \"./generate.js\";\nimport type {\n  AdvanceChapterStagesOptions,\n  AdvanceChapterStagesProgressEvent,\n  AdvanceChapterStagesResult,\n  ChapterEntry,\n  ChapterStage,\n  MutableAdvanceProgressState,\n} from \"./types.js\";\n\nexport async function advanceChapterStages(\n  document: Document,\n  options: AdvanceChapterStagesOptions,\n): Promise<AdvanceChapterStagesResult> {\n  if (options.targetStage === \"planned\") {\n    return {\n      advanced: [],\n      pending: await selectChapterEntries(document, options.chapterId),\n      skipped: [],\n    };\n  }\n\n  const advancedIds = new Set<number>();\n  const entries = await selectChapterEntries(document, options.chapterId);\n  const progressState = createAdvanceProgressState(\n    entries,\n    options.targetStage,\n  );\n  await emitAdvanceProgress(options, {\n    state: progressState.snapshot(),\n    targetStage: options.targetStage,\n    totalChapters: entries.length,\n    type: \"selected\",\n  });\n  await emitPlannedSkips(entries, options);\n\n  if (isStageBefore(\"sourced\", options.targetStage)) {\n    await advanceEntriesToGraphed(document, entries, options, {\n      advancedIds,\n      progressState,\n    });\n  }\n  if (isStageBefore(\"graphed\", options.targetStage)) {\n    const graphedEntries = await selectChapterEntries(\n      document,\n      options.chapterId,\n    );\n\n    await advanceEntriesToSummarized(document, graphedEntries, options, {\n      advancedIds,\n      progressState,\n    });\n  }\n\n  const nextEntries = await selectChapterEntries(document, options.chapterId);\n\n  return {\n    advanced: nextEntries.filter((entry) => advancedIds.has(entry.chapterId)),\n    pending: nextEntries.filter((entry) =>\n      isStageBefore(entry.stage, options.targetStage),\n    ),\n    skipped: nextEntries.filter((entry) => entry.stage === \"planned\"),\n  };\n}\n\nasync function advanceEntriesToGraphed(\n  document: Document,\n  entries: readonly ChapterEntry[],\n  options: AdvanceChapterStagesOptions,\n  state: {\n    readonly advancedIds: Set<number>;\n    readonly progressState: MutableAdvanceProgressState;\n  },\n): Promise<readonly ChapterEntry[]> {\n  for (const entry of entries) {\n    if (entry.stage !== \"sourced\") {\n      continue;\n    }\n\n    await emitAdvanceProgress(options, {\n      chapter: entry,\n      step: \"graph\",\n      targetStage: options.targetStage,\n      type: \"started\",\n    });\n    await generateChapterGraph(document, entry.chapterId, {\n      extractionPrompt: resolveExtractionPrompt(options.extractionPrompt),\n      llm: options.llm,\n      ...(options.logDirPath === undefined\n        ? {}\n        : { logDirPath: options.logDirPath }),\n      ...(options.userLanguage === undefined\n        ? {}\n        : { userLanguage: options.userLanguage }),\n      progressTracker: {\n        async advance(wordsCount) {\n          state.progressState.addGraphWords(wordsCount);\n          await emitAdvanceProgress(options, {\n            state: state.progressState.snapshot(),\n            targetStage: options.targetStage,\n            type: \"progress\",\n          });\n        },\n        async complete() {},\n      },\n    });\n    await emitAdvanceProgress(options, {\n      chapter: entry,\n      step: \"graph\",\n      targetStage: options.targetStage,\n      type: \"completed\",\n    });\n    state.advancedIds.add(entry.chapterId);\n  }\n\n  return await selectChapterEntries(document, options.chapterId);\n}\n\nasync function advanceEntriesToSummarized(\n  document: Document,\n  entries: readonly ChapterEntry[],\n  options: AdvanceChapterStagesOptions,\n  state: {\n    readonly advancedIds: Set<number>;\n    readonly progressState: MutableAdvanceProgressState;\n  },\n): Promise<readonly ChapterEntry[]> {\n  for (const entry of entries) {\n    if (entry.stage !== \"graphed\") {\n      continue;\n    }\n\n    await emitAdvanceProgress(options, {\n      chapter: entry,\n      step: \"summary\",\n      targetStage: options.targetStage,\n      type: \"started\",\n    });\n    await generateChapterSummary(document, entry.chapterId, {\n      llm: options.llm,\n      ...(options.logDirPath === undefined\n        ? {}\n        : { logDirPath: options.logDirPath }),\n      ...(options.userLanguage === undefined\n        ? {}\n        : { userLanguage: options.userLanguage }),\n    });\n    state.progressState.addSummaryWords(entry.words);\n    await emitAdvanceProgress(options, {\n      state: state.progressState.snapshot(),\n      targetStage: options.targetStage,\n      type: \"progress\",\n    });\n    await emitAdvanceProgress(options, {\n      chapter: entry,\n      step: \"summary\",\n      targetStage: options.targetStage,\n      type: \"completed\",\n    });\n    state.advancedIds.add(entry.chapterId);\n  }\n\n  return await selectChapterEntries(document, options.chapterId);\n}\n\nfunction createAdvanceProgressState(\n  entries: readonly ChapterEntry[],\n  targetStage: ChapterStage,\n): MutableAdvanceProgressState {\n  const totalGraphWords = isStageBefore(\"sourced\", targetStage)\n    ? entries\n        .filter((entry) => entry.stage === \"sourced\")\n        .reduce((sum, entry) => sum + entry.words, 0)\n    : 0;\n  const totalSummaryWords = isStageBefore(\"graphed\", targetStage)\n    ? entries\n        .filter(\n          (entry) => entry.stage === \"graphed\" || entry.stage === \"sourced\",\n        )\n        .reduce((sum, entry) => sum + entry.words, 0)\n    : 0;\n  let graphWords = 0;\n  let summaryWords = 0;\n\n  return {\n    addGraphWords(words) {\n      graphWords = Math.min(totalGraphWords, graphWords + Math.max(0, words));\n    },\n    addSummaryWords(words) {\n      summaryWords = Math.min(\n        totalSummaryWords,\n        summaryWords + Math.max(0, words),\n      );\n    },\n    snapshot() {\n      return {\n        graphWords,\n        summaryWords,\n        totalGraphWords,\n        totalSummaryWords,\n      };\n    },\n  };\n}\n\nasync function emitPlannedSkips(\n  entries: readonly ChapterEntry[],\n  options: AdvanceChapterStagesOptions,\n): Promise<void> {\n  for (const entry of entries) {\n    if (entry.stage !== \"planned\") {\n      continue;\n    }\n\n    await emitAdvanceProgress(options, {\n      chapter: entry,\n      reason: \"planned\",\n      targetStage: options.targetStage,\n      type: \"skipped\",\n    });\n  }\n}\n\nasync function emitAdvanceProgress(\n  options: AdvanceChapterStagesOptions,\n  event: AdvanceChapterStagesProgressEvent,\n): Promise<void> {\n  try {\n    await options.onProgress?.(event);\n  } catch {\n    return;\n  }\n}\n"],"mappings":"AACA,SAAS,+BAA+B;AACxC,SAAS,eAAe,4BAA4B;AACpD,SAAS,sBAAsB,8BAA8B;AAU7D,eAAsB,qBACpB,UACA,SACqC;AACrC,MAAI,QAAQ,gBAAgB,WAAW;AACrC,WAAO;AAAA,MACL,UAAU,CAAC;AAAA,MACX,SAAS,MAAM,qBAAqB,UAAU,QAAQ,SAAS;AAAA,MAC/D,SAAS,CAAC;AAAA,IACZ;AAAA,EACF;AAEA,QAAM,cAAc,oBAAI,IAAY;AACpC,QAAM,UAAU,MAAM,qBAAqB,UAAU,QAAQ,SAAS;AACtE,QAAM,gBAAgB;AAAA,IACpB;AAAA,IACA,QAAQ;AAAA,EACV;AACA,QAAM,oBAAoB,SAAS;AAAA,IACjC,OAAO,cAAc,SAAS;AAAA,IAC9B,aAAa,QAAQ;AAAA,IACrB,eAAe,QAAQ;AAAA,IACvB,MAAM;AAAA,EACR,CAAC;AACD,QAAM,iBAAiB,SAAS,OAAO;AAEvC,MAAI,cAAc,WAAW,QAAQ,WAAW,GAAG;AACjD,UAAM,wBAAwB,UAAU,SAAS,SAAS;AAAA,MACxD;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,MAAI,cAAc,WAAW,QAAQ,WAAW,GAAG;AACjD,UAAM,iBAAiB,MAAM;AAAA,MAC3B;AAAA,MACA,QAAQ;AAAA,IACV;AAEA,UAAM,2BAA2B,UAAU,gBAAgB,SAAS;AAAA,MAClE;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AAEA,QAAM,cAAc,MAAM,qBAAqB,UAAU,QAAQ,SAAS;AAE1E,SAAO;AAAA,IACL,UAAU,YAAY,OAAO,CAAC,UAAU,YAAY,IAAI,MAAM,SAAS,CAAC;AAAA,IACxE,SAAS,YAAY;AAAA,MAAO,CAAC,UAC3B,cAAc,MAAM,OAAO,QAAQ,WAAW;AAAA,IAChD;AAAA,IACA,SAAS,YAAY,OAAO,CAAC,UAAU,MAAM,UAAU,SAAS;AAAA,EAClE;AACF;AAEA,eAAe,wBACb,UACA,SACA,SACA,OAIkC;AAClC,aAAW,SAAS,SAAS;AAC3B,QAAI,MAAM,UAAU,WAAW;AAC7B;AAAA,IACF;AAEA,UAAM,oBAAoB,SAAS;AAAA,MACjC,SAAS;AAAA,MACT,MAAM;AAAA,MACN,aAAa,QAAQ;AAAA,MACrB,MAAM;AAAA,IACR,CAAC;AACD,UAAM,qBAAqB,UAAU,MAAM,WAAW;AAAA,MACpD,kBAAkB,wBAAwB,QAAQ,gBAAgB;AAAA,MAClE,KAAK,QAAQ;AAAA,MACb,GAAI,QAAQ,eAAe,SACvB,CAAC,IACD,EAAE,YAAY,QAAQ,WAAW;AAAA,MACrC,GAAI,QAAQ,iBAAiB,SACzB,CAAC,IACD,EAAE,cAAc,QAAQ,aAAa;AAAA,MACzC,iBAAiB;AAAA,QACf,MAAM,QAAQ,YAAY;AACxB,gBAAM,cAAc,cAAc,UAAU;AAC5C,gBAAM,oBAAoB,SAAS;AAAA,YACjC,OAAO,MAAM,cAAc,SAAS;AAAA,YACpC,aAAa,QAAQ;AAAA,YACrB,MAAM;AAAA,UACR,CAAC;AAAA,QACH;AAAA,QACA,MAAM,WAAW;AAAA,QAAC;AAAA,MACpB;AAAA,IACF,CAAC;AACD,UAAM,oBAAoB,SAAS;AAAA,MACjC,SAAS;AAAA,MACT,MAAM;AAAA,MACN,aAAa,QAAQ;AAAA,MACrB,MAAM;AAAA,IACR,CAAC;AACD,UAAM,YAAY,IAAI,MAAM,SAAS;AAAA,EACvC;AAEA,SAAO,MAAM,qBAAqB,UAAU,QAAQ,SAAS;AAC/D;AAEA,eAAe,2BACb,UACA,SACA,SACA,OAIkC;AAClC,aAAW,SAAS,SAAS;AAC3B,QAAI,MAAM,UAAU,WAAW;AAC7B;AAAA,IACF;AAEA,UAAM,oBAAoB,SAAS;AAAA,MACjC,SAAS;AAAA,MACT,MAAM;AAAA,MACN,aAAa,QAAQ;AAAA,MACrB,MAAM;AAAA,IACR,CAAC;AACD,UAAM,uBAAuB,UAAU,MAAM,WAAW;AAAA,MACtD,KAAK,QAAQ;AAAA,MACb,GAAI,QAAQ,eAAe,SACvB,CAAC,IACD,EAAE,YAAY,QAAQ,WAAW;AAAA,MACrC,GAAI,QAAQ,iBAAiB,SACzB,CAAC,IACD,EAAE,cAAc,QAAQ,aAAa;AAAA,IAC3C,CAAC;AACD,UAAM,cAAc,gBAAgB,MAAM,KAAK;AAC/C,UAAM,oBAAoB,SAAS;AAAA,MACjC,OAAO,MAAM,cAAc,SAAS;AAAA,MACpC,aAAa,QAAQ;AAAA,MACrB,MAAM;AAAA,IACR,CAAC;AACD,UAAM,oBAAoB,SAAS;AAAA,MACjC,SAAS;AAAA,MACT,MAAM;AAAA,MACN,aAAa,QAAQ;AAAA,MACrB,MAAM;AAAA,IACR,CAAC;AACD,UAAM,YAAY,IAAI,MAAM,SAAS;AAAA,EACvC;AAEA,SAAO,MAAM,qBAAqB,UAAU,QAAQ,SAAS;AAC/D;AAEA,SAAS,2BACP,SACA,aAC6B;AAC7B,QAAM,kBAAkB,cAAc,WAAW,WAAW,IACxD,QACG,OAAO,CAAC,UAAU,MAAM,UAAU,SAAS,EAC3C,OAAO,CAAC,KAAK,UAAU,MAAM,MAAM,OAAO,CAAC,IAC9C;AACJ,QAAM,oBAAoB,cAAc,WAAW,WAAW,IAC1D,QACG;AAAA,IACC,CAAC,UAAU,MAAM,UAAU,aAAa,MAAM,UAAU;AAAA,EAC1D,EACC,OAAO,CAAC,KAAK,UAAU,MAAM,MAAM,OAAO,CAAC,IAC9C;AACJ,MAAI,aAAa;AACjB,MAAI,eAAe;AAEnB,SAAO;AAAA,IACL,cAAc,OAAO;AACnB,mBAAa,KAAK,IAAI,iBAAiB,aAAa,KAAK,IAAI,GAAG,KAAK,CAAC;AAAA,IACxE;AAAA,IACA,gBAAgB,OAAO;AACrB,qBAAe,KAAK;AAAA,QAClB;AAAA,QACA,eAAe,KAAK,IAAI,GAAG,KAAK;AAAA,MAClC;AAAA,IACF;AAAA,IACA,WAAW;AACT,aAAO;AAAA,QACL;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;AAEA,eAAe,iBACb,SACA,SACe;AACf,aAAW,SAAS,SAAS;AAC3B,QAAI,MAAM,UAAU,WAAW;AAC7B;AAAA,IACF;AAEA,UAAM,oBAAoB,SAAS;AAAA,MACjC,SAAS;AAAA,MACT,QAAQ;AAAA,MACR,aAAa,QAAQ;AAAA,MACrB,MAAM;AAAA,IACR,CAAC;AAAA,EACH;AACF;AAEA,eAAe,oBACb,SACA,OACe;AACf,MAAI;AACF,UAAM,QAAQ,aAAa,KAAK;AAAA,EAClC,QAAQ;AACN;AAAA,EACF;AACF;","names":[]}

package/dist/api/chapter/details.d.ts

@@ -0,0 +1,25 @@
+export { getChapterDetails, getChapterTree, isStageBefore, listChapters, requireChapterDetails, selectChapterEntries } from '../../document/chapter/query.js';
+import '../../document/directory/types.js';
+import '../../text/source/meta.js';
+import 'zod';
+import '../../text/source/toc.js';
+import '../../text/source/types.js';
+import '../../document/database.js';
+import 'sqlite3';
+import '../../document/types.js';
+import '../../document/text-streams/serial.js';
+import '../../document/text-streams/draft.js';
+import '../../document/text-streams/types.js';
+import '../../document/stores/chunk.js';
+import '../../document/stores/types.js';
+import '../../document/stores/fragment-group.js';
+import '../../document/stores/graph-build-parameter.js';
+import '../../document/stores/mention-link.js';
+import '../../document/stores/mention.js';
+import '../../document/stores/object-metadata.js';
+import '../../document/stores/reading-edge.js';
+import '../../document/stores/serial.js';
+import '../../document/stores/snake/chunk.js';
+import '../../document/stores/snake/core.js';
+import '../../document/stores/snake/edge.js';
+import '../../document/chapter/types.js';

package/dist/api/chapter/details.js

@@ -0,0 +1,17 @@
+import {
+  getChapterDetails,
+  getChapterTree,
+  isStageBefore,
+  listChapters,
+  requireChapterDetails,
+  selectChapterEntries
+} from "../../document/chapter/index.js";
+export {
+  getChapterDetails,
+  getChapterTree,
+  isStageBefore,
+  listChapters,
+  requireChapterDetails,
+  selectChapterEntries
+};
+//# sourceMappingURL=details.js.map

package/dist/api/chapter/details.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/api/chapter/details.ts"],"sourcesContent":["export {\n  getChapterDetails,\n  getChapterTree,\n  isStageBefore,\n  listChapters,\n  requireChapterDetails,\n  selectChapterEntries,\n} from \"../../document/chapter/index.js\";\n"],"mappings":"AAAA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;","names":[]}

package/dist/api/chapter/entries.d.ts

@@ -0,0 +1,41 @@
+import { TocItem } from '../../text/source/toc.js';
+import { ReadonlyDocument, Document } from '../../document/directory/types.js';
+import { MutableTocFile } from './tree.js';
+import { ChapterEntry, ChapterStage } from '../../document/chapter/types.js';
+import 'zod';
+import '../../text/source/meta.js';
+import '../../text/source/types.js';
+import '../../document/database.js';
+import 'sqlite3';
+import '../../document/types.js';
+import '../../document/text-streams/serial.js';
+import '../../document/text-streams/draft.js';
+import '../../document/text-streams/types.js';
+import '../../document/stores/chunk.js';
+import '../../document/stores/types.js';
+import '../../document/stores/fragment-group.js';
+import '../../document/stores/graph-build-parameter.js';
+import '../../document/stores/mention-link.js';
+import '../../document/stores/mention.js';
+import '../../document/stores/object-metadata.js';
+import '../../document/stores/reading-edge.js';
+import '../../document/stores/serial.js';
+import '../../document/stores/snake/chunk.js';
+import '../../document/stores/snake/core.js';
+import '../../document/stores/snake/edge.js';
+import './types.js';
+import '../../runtime/common/language.js';
+import '../../runtime/common/llm-scope.js';
+import '../../external/llm/client/core.js';
+import '../../external/llm/context.js';
+import '../../external/llm/cache.js';
+import '../../external/llm/types.js';
+import 'ai';
+
+declare function normalizeChapterToc(document: Document): Promise<MutableTocFile>;
+declare function readChapterToc(document: ReadonlyDocument): Promise<MutableTocFile>;
+declare function findChapterEntry(document: ReadonlyDocument, items: readonly TocItem[], chapterId: number, ancestorTitles?: readonly string[], ancestorKeys?: readonly string[], depth?: number): Promise<ChapterEntry | undefined>;
+declare function collectChapterEntries(document: ReadonlyDocument, items: readonly TocItem[], ancestorTitles?: readonly string[], ancestorKeys?: readonly string[], depth?: number): Promise<ChapterEntry[]>;
+declare function resolveChapterStage(document: ReadonlyDocument, chapterId: number, fragmentCount: number): Promise<ChapterStage>;
+
+export { collectChapterEntries, findChapterEntry, normalizeChapterToc, readChapterToc, resolveChapterStage };

package/dist/api/chapter/entries.js

@@ -0,0 +1,216 @@
+import { formatChapterUri } from "../../document/chapter/path.js";
+import { ensureChapterKeys } from "../../document/chapter/toc.js";
+import { TOC_FILE_VERSION } from "../../text/source/index.js";
+import {
+  cloneTocItem,
+  normalizeTitle
+} from "./tree.js";
+async function normalizeChapterToc(document) {
+  const existingToc = await document.readToc();
+  const items = existingToc?.items.map(cloneTocItem) ?? [];
+  let changed = false;
+  const normalizeItems = async (items2) => {
+    for (const item of items2) {
+      if (item.serialId === void 0) {
+        item.serialId = await document.createSerial();
+        changed = true;
+      } else {
+        await document.serials.ensure(item.serialId);
+      }
+      await normalizeItems(item.children);
+    }
+  };
+  await normalizeItems(items);
+  const keysChanged = ensureChapterKeys(items);
+  changed ||= keysChanged;
+  const toc = {
+    items,
+    version: existingToc?.version ?? TOC_FILE_VERSION
+  };
+  if (existingToc === void 0 || changed) {
+    await document.replaceToc(toc);
+  }
+  return toc;
+}
+async function readChapterToc(document) {
+  const toc = await document.readToc();
+  const items = toc?.items.map(cloneTocItem) ?? [];
+  assertChapterKeys(items);
+  return toc === void 0 ? { items: [], version: TOC_FILE_VERSION } : {
+    items,
+    version: toc.version
+  };
+}
+function assertChapterKeys(items) {
+  const existingKeys = /* @__PURE__ */ new Set();
+  const visit = (nodes) => {
+    for (const item of nodes) {
+      if (item.key === void 0) {
+        throw new Error(
+          "Missing chapter key in TOC. Run a writable chapter operation or `wg maintenance upgrade` before using read-only chapter paths."
+        );
+      }
+      if (existingKeys.has(item.key)) {
+        throw new Error(`Duplicate chapter key: ${item.key}.`);
+      }
+      existingKeys.add(item.key);
+      visit(item.children);
+    }
+  };
+  visit(items);
+}
+async function findChapterEntry(document, items, chapterId, ancestorTitles = [], ancestorKeys = [], depth = 0) {
+  for (const item of items) {
+    const title = normalizeTitle(item.title) ?? null;
+    const key = item.key ?? `chapter-${item.serialId ?? "group"}`;
+    const tocPath = item.serialId === void 0 ? [...ancestorTitles, ...title === null ? [] : [title]] : [...ancestorTitles, title ?? `Chapter ${item.serialId}`];
+    const chapterPath = [...ancestorKeys, key];
+    if (item.serialId === chapterId) {
+      return await createChapterEntry(document, item, item.serialId, {
+        chapterPath,
+        depth,
+        title,
+        tocPath
+      });
+    }
+    const childEntry = await findChapterEntry(
+      document,
+      item.children,
+      chapterId,
+      tocPath,
+      chapterPath,
+      depth + 1
+    );
+    if (childEntry !== void 0) {
+      return childEntry;
+    }
+  }
+  return void 0;
+}
+async function collectChapterEntries(document, items, ancestorTitles = [], ancestorKeys = [], depth = 0) {
+  const entries = [];
+  for (const item of items) {
+    const title = normalizeTitle(item.title) ?? null;
+    const key = item.key ?? `chapter-${item.serialId ?? "group"}`;
+    const tocPath = item.serialId === void 0 ? [...ancestorTitles, ...title === null ? [] : [title]] : [...ancestorTitles, title ?? `Chapter ${item.serialId}`];
+    const chapterPath = [...ancestorKeys, key];
+    if (item.serialId === void 0) {
+      entries.push(
+        ...await collectChapterEntries(
+          document,
+          item.children,
+          tocPath,
+          chapterPath,
+          depth + 1
+        )
+      );
+      continue;
+    }
+    entries.push(
+      await createChapterEntry(document, item, item.serialId, {
+        chapterPath,
+        depth,
+        title,
+        tocPath
+      })
+    );
+    entries.push(
+      ...await collectChapterEntries(
+        document,
+        item.children,
+        tocPath,
+        chapterPath,
+        depth + 1
+      )
+    );
+  }
+  return entries;
+}
+async function createChapterEntry(document, item, serialId, input) {
+  const [serial, sourceSummary] = await Promise.all([
+    document.serials.getById(serialId),
+    summarizeSerialSource(document, serialId)
+  ]);
+  return {
+    chapterId: serialId,
+    childCount: item.children.length,
+    depth: input.depth,
+    documentOrder: serial?.documentOrder ?? serialId,
+    fragmentCount: sourceSummary.fragmentCount,
+    key: input.chapterPath.at(-1) ?? `chapter-${serialId}`,
+    path: input.chapterPath.join("/"),
+    stage: await resolveChapterStage(
+      document,
+      serialId,
+      sourceSummary.fragmentCount
+    ),
+    title: input.title,
+    tocPath: input.tocPath,
+    uri: formatChapterUri(input.chapterPath.join("/")),
+    words: sourceSummary.words
+  };
+}
+async function summarizeSerialSource(document, serialId) {
+  const sentenceWords = await document.readDatabase(
+    async (database) => await database.queryAll(
+      `
+        SELECT words_count
+        FROM text_sentence_records
+        WHERE kind = 1 AND chapter_id = ?
+        ORDER BY sentence_index
+      `,
+      [serialId],
+      (row) => Number(row.words_count)
+    )
+  );
+  let fragmentCount = 0;
+  let fragmentWords = 0;
+  let words = 0;
+  for (const sentenceWordCount of sentenceWords) {
+    if (fragmentWords > 0 && fragmentWords + sentenceWordCount > 600) {
+      fragmentCount += 1;
+      fragmentWords = 0;
+    }
+    fragmentWords += sentenceWordCount;
+    words += sentenceWordCount;
+  }
+  if (fragmentWords > 0) {
+    fragmentCount += 1;
+  }
+  return {
+    fragmentCount,
+    words
+  };
+}
+async function resolveChapterStage(document, chapterId, fragmentCount) {
+  const summarySentenceCount = await document.readDatabase(
+    async (database) => await database.queryOne(
+      `
+          SELECT COUNT(*) AS count
+          FROM text_sentence_records
+          WHERE kind = 2 AND chapter_id = ?
+        `,
+      [chapterId],
+      (row) => Number(row.count)
+    ) ?? 0
+  );
+  if (summarySentenceCount > 0) {
+    return "summarized";
+  }
+  const serial = await document.serials.getById(chapterId);
+  if (serial?.topologyReady === true) {
+    return "graphed";
+  }
+  if (fragmentCount > 0) {
+    return "sourced";
+  }
+  return "planned";
+}
+export {
+  collectChapterEntries,
+  findChapterEntry,
+  normalizeChapterToc,
+  readChapterToc,
+  resolveChapterStage
+};
+//# sourceMappingURL=entries.js.map

package/dist/api/chapter/entries.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/api/chapter/entries.ts"],"sourcesContent":["import type { Document, ReadonlyDocument } from \"../../document/index.js\";\nimport { formatChapterUri } from \"../../document/chapter/path.js\";\nimport { ensureChapterKeys } from \"../../document/chapter/toc.js\";\nimport { TOC_FILE_VERSION, type TocItem } from \"../../text/source/index.js\";\n\nimport {\n  cloneTocItem,\n  normalizeTitle,\n  type MutableTocFile,\n  type MutableTocItem,\n} from \"./tree.js\";\nimport type { ChapterEntry, ChapterStage } from \"./types.js\";\n\nexport async function normalizeChapterToc(\n  document: Document,\n): Promise<MutableTocFile> {\n  const existingToc = await document.readToc();\n  const items = existingToc?.items.map(cloneTocItem) ?? [];\n  let changed = false;\n\n  const normalizeItems = async (items: MutableTocItem[]): Promise<void> => {\n    for (const item of items) {\n      if (item.serialId === undefined) {\n        item.serialId = await document.createSerial();\n        changed = true;\n      } else {\n        await document.serials.ensure(item.serialId);\n      }\n\n      await normalizeItems(item.children);\n    }\n  };\n\n  await normalizeItems(items);\n  const keysChanged = ensureChapterKeys(items);\n  changed ||= keysChanged;\n\n  const toc: MutableTocFile = {\n    items,\n    version: existingToc?.version ?? TOC_FILE_VERSION,\n  };\n\n  if (existingToc === undefined || changed) {\n    await document.replaceToc(toc);\n  }\n\n  return toc;\n}\n\nexport async function readChapterToc(\n  document: ReadonlyDocument,\n): Promise<MutableTocFile> {\n  const toc = await document.readToc();\n  const items = toc?.items.map(cloneTocItem) ?? [];\n  assertChapterKeys(items);\n\n  return toc === undefined\n    ? { items: [], version: TOC_FILE_VERSION }\n    : {\n        items,\n        version: toc.version,\n      };\n}\n\nfunction assertChapterKeys(items: MutableTocItem[]): void {\n  const existingKeys = new Set<string>();\n  const visit = (nodes: MutableTocItem[]): void => {\n    for (const item of nodes) {\n      if (item.key === undefined) {\n        throw new Error(\n          \"Missing chapter key in TOC. Run a writable chapter operation or `wg maintenance upgrade` before using read-only chapter paths.\",\n        );\n      }\n      if (existingKeys.has(item.key)) {\n        throw new Error(`Duplicate chapter key: ${item.key}.`);\n      }\n      existingKeys.add(item.key);\n      visit(item.children);\n    }\n  };\n\n  visit(items);\n}\n\nexport async function findChapterEntry(\n  document: ReadonlyDocument,\n  items: readonly TocItem[],\n  chapterId: number,\n  ancestorTitles: readonly string[] = [],\n  ancestorKeys: readonly string[] = [],\n  depth = 0,\n): Promise<ChapterEntry | undefined> {\n  for (const item of items) {\n    const title = normalizeTitle(item.title) ?? null;\n    const key = item.key ?? `chapter-${item.serialId ?? \"group\"}`;\n    const tocPath =\n      item.serialId === undefined\n        ? [...ancestorTitles, ...(title === null ? [] : [title])]\n        : [...ancestorTitles, title ?? `Chapter ${item.serialId}`];\n    const chapterPath = [...ancestorKeys, key];\n\n    if (item.serialId === chapterId) {\n      return await createChapterEntry(document, item, item.serialId, {\n        chapterPath,\n        depth,\n        title,\n        tocPath,\n      });\n    }\n\n    const childEntry = await findChapterEntry(\n      document,\n      item.children,\n      chapterId,\n      tocPath,\n      chapterPath,\n      depth + 1,\n    );\n\n    if (childEntry !== undefined) {\n      return childEntry;\n    }\n  }\n\n  return undefined;\n}\n\nexport async function collectChapterEntries(\n  document: ReadonlyDocument,\n  items: readonly TocItem[],\n  ancestorTitles: readonly string[] = [],\n  ancestorKeys: readonly string[] = [],\n  depth = 0,\n): Promise<ChapterEntry[]> {\n  const entries: ChapterEntry[] = [];\n\n  for (const item of items) {\n    const title = normalizeTitle(item.title) ?? null;\n    const key = item.key ?? `chapter-${item.serialId ?? \"group\"}`;\n    const tocPath =\n      item.serialId === undefined\n        ? [...ancestorTitles, ...(title === null ? [] : [title])]\n        : [...ancestorTitles, title ?? `Chapter ${item.serialId}`];\n    const chapterPath = [...ancestorKeys, key];\n\n    if (item.serialId === undefined) {\n      entries.push(\n        ...(await collectChapterEntries(\n          document,\n          item.children,\n          tocPath,\n          chapterPath,\n          depth + 1,\n        )),\n      );\n      continue;\n    }\n\n    entries.push(\n      await createChapterEntry(document, item, item.serialId, {\n        chapterPath,\n        depth,\n        title,\n        tocPath,\n      }),\n    );\n    entries.push(\n      ...(await collectChapterEntries(\n        document,\n        item.children,\n        tocPath,\n        chapterPath,\n        depth + 1,\n      )),\n    );\n  }\n\n  return entries;\n}\n\nasync function createChapterEntry(\n  document: ReadonlyDocument,\n  item: TocItem,\n  serialId: number,\n  input: {\n    readonly chapterPath: readonly string[];\n    readonly depth: number;\n    readonly title: string | null;\n    readonly tocPath: readonly string[];\n  },\n): Promise<ChapterEntry> {\n  const [serial, sourceSummary] = await Promise.all([\n    document.serials.getById(serialId),\n    summarizeSerialSource(document, serialId),\n  ]);\n\n  return {\n    chapterId: serialId,\n    childCount: item.children.length,\n    depth: input.depth,\n    documentOrder: serial?.documentOrder ?? serialId,\n    fragmentCount: sourceSummary.fragmentCount,\n    key: input.chapterPath.at(-1) ?? `chapter-${serialId}`,\n    path: input.chapterPath.join(\"/\"),\n    stage: await resolveChapterStage(\n      document,\n      serialId,\n      sourceSummary.fragmentCount,\n    ),\n    title: input.title,\n    tocPath: input.tocPath,\n    uri: formatChapterUri(input.chapterPath.join(\"/\")),\n    words: sourceSummary.words,\n  };\n}\n\nasync function summarizeSerialSource(\n  document: ReadonlyDocument,\n  serialId: number,\n): Promise<{ readonly fragmentCount: number; readonly words: number }> {\n  const sentenceWords = await document.readDatabase(\n    async (database) =>\n      await database.queryAll(\n        `\n        SELECT words_count\n        FROM text_sentence_records\n        WHERE kind = 1 AND chapter_id = ?\n        ORDER BY sentence_index\n      `,\n        [serialId],\n        (row) => Number(row.words_count),\n      ),\n  );\n  let fragmentCount = 0;\n  let fragmentWords = 0;\n  let words = 0;\n\n  for (const sentenceWordCount of sentenceWords) {\n    if (fragmentWords > 0 && fragmentWords + sentenceWordCount > 600) {\n      fragmentCount += 1;\n      fragmentWords = 0;\n    }\n\n    fragmentWords += sentenceWordCount;\n    words += sentenceWordCount;\n  }\n\n  if (fragmentWords > 0) {\n    fragmentCount += 1;\n  }\n\n  return {\n    fragmentCount,\n    words,\n  };\n}\n\nexport async function resolveChapterStage(\n  document: ReadonlyDocument,\n  chapterId: number,\n  fragmentCount: number,\n): Promise<ChapterStage> {\n  const summarySentenceCount = await document.readDatabase(\n    async (database) =>\n      (await database.queryOne(\n        `\n          SELECT COUNT(*) AS count\n          FROM text_sentence_records\n          WHERE kind = 2 AND chapter_id = ?\n        `,\n        [chapterId],\n        (row) => Number(row.count),\n      )) ?? 0,\n  );\n\n  if (summarySentenceCount > 0) {\n    return \"summarized\";\n  }\n\n  const serial = await document.serials.getById(chapterId);\n\n  if (serial?.topologyReady === true) {\n    return \"graphed\";\n  }\n\n  if (fragmentCount > 0) {\n    return \"sourced\";\n  }\n\n  return \"planned\";\n}\n"],"mappings":"AACA,SAAS,wBAAwB;AACjC,SAAS,yBAAyB;AAClC,SAAS,wBAAsC;AAE/C;AAAA,EACE;AAAA,EACA;AAAA,OAGK;AAGP,eAAsB,oBACpB,UACyB;AACzB,QAAM,cAAc,MAAM,SAAS,QAAQ;AAC3C,QAAM,QAAQ,aAAa,MAAM,IAAI,YAAY,KAAK,CAAC;AACvD,MAAI,UAAU;AAEd,QAAM,iBAAiB,OAAOA,WAA2C;AACvE,eAAW,QAAQA,QAAO;AACxB,UAAI,KAAK,aAAa,QAAW;AAC/B,aAAK,WAAW,MAAM,SAAS,aAAa;AAC5C,kBAAU;AAAA,MACZ,OAAO;AACL,cAAM,SAAS,QAAQ,OAAO,KAAK,QAAQ;AAAA,MAC7C;AAEA,YAAM,eAAe,KAAK,QAAQ;AAAA,IACpC;AAAA,EACF;AAEA,QAAM,eAAe,KAAK;AAC1B,QAAM,cAAc,kBAAkB,KAAK;AAC3C,cAAY;AAEZ,QAAM,MAAsB;AAAA,IAC1B;AAAA,IACA,SAAS,aAAa,WAAW;AAAA,EACnC;AAEA,MAAI,gBAAgB,UAAa,SAAS;AACxC,UAAM,SAAS,WAAW,GAAG;AAAA,EAC/B;AAEA,SAAO;AACT;AAEA,eAAsB,eACpB,UACyB;AACzB,QAAM,MAAM,MAAM,SAAS,QAAQ;AACnC,QAAM,QAAQ,KAAK,MAAM,IAAI,YAAY,KAAK,CAAC;AAC/C,oBAAkB,KAAK;AAEvB,SAAO,QAAQ,SACX,EAAE,OAAO,CAAC,GAAG,SAAS,iBAAiB,IACvC;AAAA,IACE;AAAA,IACA,SAAS,IAAI;AAAA,EACf;AACN;AAEA,SAAS,kBAAkB,OAA+B;AACxD,QAAM,eAAe,oBAAI,IAAY;AACrC,QAAM,QAAQ,CAAC,UAAkC;AAC/C,eAAW,QAAQ,OAAO;AACxB,UAAI,KAAK,QAAQ,QAAW;AAC1B,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAAA,MACF;AACA,UAAI,aAAa,IAAI,KAAK,GAAG,GAAG;AAC9B,cAAM,IAAI,MAAM,0BAA0B,KAAK,GAAG,GAAG;AAAA,MACvD;AACA,mBAAa,IAAI,KAAK,GAAG;AACzB,YAAM,KAAK,QAAQ;AAAA,IACrB;AAAA,EACF;AAEA,QAAM,KAAK;AACb;AAEA,eAAsB,iBACpB,UACA,OACA,WACA,iBAAoC,CAAC,GACrC,eAAkC,CAAC,GACnC,QAAQ,GAC2B;AACnC,aAAW,QAAQ,OAAO;AACxB,UAAM,QAAQ,eAAe,KAAK,KAAK,KAAK;AAC5C,UAAM,MAAM,KAAK,OAAO,WAAW,KAAK,YAAY,OAAO;AAC3D,UAAM,UACJ,KAAK,aAAa,SACd,CAAC,GAAG,gBAAgB,GAAI,UAAU,OAAO,CAAC,IAAI,CAAC,KAAK,CAAE,IACtD,CAAC,GAAG,gBAAgB,SAAS,WAAW,KAAK,QAAQ,EAAE;AAC7D,UAAM,cAAc,CAAC,GAAG,cAAc,GAAG;AAEzC,QAAI,KAAK,aAAa,WAAW;AAC/B,aAAO,MAAM,mBAAmB,UAAU,MAAM,KAAK,UAAU;AAAA,QAC7D;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH;AAEA,UAAM,aAAa,MAAM;AAAA,MACvB;AAAA,MACA,KAAK;AAAA,MACL;AAAA,MACA;AAAA,MACA;AAAA,MACA,QAAQ;AAAA,IACV;AAEA,QAAI,eAAe,QAAW;AAC5B,aAAO;AAAA,IACT;AAAA,EACF;AAEA,SAAO;AACT;AAEA,eAAsB,sBACpB,UACA,OACA,iBAAoC,CAAC,GACrC,eAAkC,CAAC,GACnC,QAAQ,GACiB;AACzB,QAAM,UAA0B,CAAC;AAEjC,aAAW,QAAQ,OAAO;AACxB,UAAM,QAAQ,eAAe,KAAK,KAAK,KAAK;AAC5C,UAAM,MAAM,KAAK,OAAO,WAAW,KAAK,YAAY,OAAO;AAC3D,UAAM,UACJ,KAAK,aAAa,SACd,CAAC,GAAG,gBAAgB,GAAI,UAAU,OAAO,CAAC,IAAI,CAAC,KAAK,CAAE,IACtD,CAAC,GAAG,gBAAgB,SAAS,WAAW,KAAK,QAAQ,EAAE;AAC7D,UAAM,cAAc,CAAC,GAAG,cAAc,GAAG;AAEzC,QAAI,KAAK,aAAa,QAAW;AAC/B,cAAQ;AAAA,QACN,GAAI,MAAM;AAAA,UACR;AAAA,UACA,KAAK;AAAA,UACL;AAAA,UACA;AAAA,UACA,QAAQ;AAAA,QACV;AAAA,MACF;AACA;AAAA,IACF;AAEA,YAAQ;AAAA,MACN,MAAM,mBAAmB,UAAU,MAAM,KAAK,UAAU;AAAA,QACtD;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH;AACA,YAAQ;AAAA,MACN,GAAI,MAAM;AAAA,QACR;AAAA,QACA,KAAK;AAAA,QACL;AAAA,QACA;AAAA,QACA,QAAQ;AAAA,MACV;AAAA,IACF;AAAA,EACF;AAEA,SAAO;AACT;AAEA,eAAe,mBACb,UACA,MACA,UACA,OAMuB;AACvB,QAAM,CAAC,QAAQ,aAAa,IAAI,MAAM,QAAQ,IAAI;AAAA,IAChD,SAAS,QAAQ,QAAQ,QAAQ;AAAA,IACjC,sBAAsB,UAAU,QAAQ;AAAA,EAC1C,CAAC;AAED,SAAO;AAAA,IACL,WAAW;AAAA,IACX,YAAY,KAAK,SAAS;AAAA,IAC1B,OAAO,MAAM;AAAA,IACb,eAAe,QAAQ,iBAAiB;AAAA,IACxC,eAAe,cAAc;AAAA,IAC7B,KAAK,MAAM,YAAY,GAAG,EAAE,KAAK,WAAW,QAAQ;AAAA,IACpD,MAAM,MAAM,YAAY,KAAK,GAAG;AAAA,IAChC,OAAO,MAAM;AAAA,MACX;AAAA,MACA;AAAA,MACA,cAAc;AAAA,IAChB;AAAA,IACA,OAAO,MAAM;AAAA,IACb,SAAS,MAAM;AAAA,IACf,KAAK,iBAAiB,MAAM,YAAY,KAAK,GAAG,CAAC;AAAA,IACjD,OAAO,cAAc;AAAA,EACvB;AACF;AAEA,eAAe,sBACb,UACA,UACqE;AACrE,QAAM,gBAAgB,MAAM,SAAS;AAAA,IACnC,OAAO,aACL,MAAM,SAAS;AAAA,MACb;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,MAMA,CAAC,QAAQ;AAAA,MACT,CAAC,QAAQ,OAAO,IAAI,WAAW;AAAA,IACjC;AAAA,EACJ;AACA,MAAI,gBAAgB;AACpB,MAAI,gBAAgB;AACpB,MAAI,QAAQ;AAEZ,aAAW,qBAAqB,eAAe;AAC7C,QAAI,gBAAgB,KAAK,gBAAgB,oBAAoB,KAAK;AAChE,uBAAiB;AACjB,sBAAgB;AAAA,IAClB;AAEA,qBAAiB;AACjB,aAAS;AAAA,EACX;AAEA,MAAI,gBAAgB,GAAG;AACrB,qBAAiB;AAAA,EACnB;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,EACF;AACF;AAEA,eAAsB,oBACpB,UACA,WACA,eACuB;AACvB,QAAM,uBAAuB,MAAM,SAAS;AAAA,IAC1C,OAAO,aACJ,MAAM,SAAS;AAAA,MACd;AAAA;AAAA;AAAA;AAAA;AAAA,MAKA,CAAC,SAAS;AAAA,MACV,CAAC,QAAQ,OAAO,IAAI,KAAK;AAAA,IAC3B,KAAM;AAAA,EACV;AAEA,MAAI,uBAAuB,GAAG;AAC5B,WAAO;AAAA,EACT;AAEA,QAAM,SAAS,MAAM,SAAS,QAAQ,QAAQ,SAAS;AAEvD,MAAI,QAAQ,kBAAkB,MAAM;AAClC,WAAO;AAAA,EACT;AAEA,MAAI,gBAAgB,GAAG;AACrB,WAAO;AAAA,EACT;AAEA,SAAO;AACT;","names":["items"]}

package/dist/api/chapter/generate.d.ts

@@ -0,0 +1,42 @@
+import { Document } from '../../document/directory/types.js';
+import { GenerateChapterGraphOptions, GenerateChapterSummaryOptions } from './options.js';
+import { ChapterDetails } from '../../document/chapter/types.js';
+import '../../text/source/meta.js';
+import 'zod';
+import '../../text/source/toc.js';
+import '../../text/source/types.js';
+import '../../document/database.js';
+import 'sqlite3';
+import '../../document/types.js';
+import '../../document/text-streams/serial.js';
+import '../../document/text-streams/draft.js';
+import '../../document/text-streams/types.js';
+import '../../document/stores/chunk.js';
+import '../../document/stores/types.js';
+import '../../document/stores/fragment-group.js';
+import '../../document/stores/graph-build-parameter.js';
+import '../../document/stores/mention-link.js';
+import '../../document/stores/mention.js';
+import '../../document/stores/object-metadata.js';
+import '../../document/stores/reading-edge.js';
+import '../../document/stores/serial.js';
+import '../../document/stores/snake/chunk.js';
+import '../../document/stores/snake/core.js';
+import '../../document/stores/snake/edge.js';
+import '../../runtime/common/language.js';
+import '../../runtime/common/llm-scope.js';
+import '../../external/llm/client/core.js';
+import '../../external/llm/context.js';
+import '../../external/llm/cache.js';
+import '../../external/llm/types.js';
+import 'ai';
+import '../../text/serial/options.js';
+import '../../text/reader/types.js';
+import '../../text/reader/attention/core.js';
+import '../../text/reader/chunk-batch/types.js';
+import '../../text/reader/segment/types.js';
+
+declare function generateChapterGraph(document: Document, chapterId: number, options: GenerateChapterGraphOptions): Promise<ChapterDetails>;
+declare function generateChapterSummary(document: Document, chapterId: number, options: GenerateChapterSummaryOptions): Promise<ChapterDetails>;
+
+export { generateChapterGraph, generateChapterSummary };

package/dist/api/chapter/generate.js

@@ -0,0 +1,61 @@
+import { normalizeLanguageCode } from "../../runtime/common/language.js";
+import { resolveExtractionPrompt } from "../../runtime/common/prompts.js";
+import { SerialGeneration } from "../../serial.js";
+import { getChapterDetails, requireChapterDetails } from "./details.js";
+import { createTopologyOptions } from "./options.js";
+async function generateChapterGraph(document, chapterId, options) {
+  return await document.openSession(async (openedDocument) => {
+    const details = await requireChapterDetails(openedDocument, chapterId);
+    if (details.stage !== "sourced") {
+      throw new Error(
+        `Chapter ${chapterId} is ${details.stage}. Generate a graph only for sourced chapters.`
+      );
+    }
+    const generation = new SerialGeneration({
+      document: openedDocument,
+      llm: options.llm,
+      ...options.logDirPath === void 0 ? {} : { logDirPath: options.logDirPath }
+    });
+    await openedDocument.clearSerialGraph(chapterId);
+    await generation.buildTopologyInto(
+      chapterId,
+      createTopologyOptions(options),
+      options.progressTracker
+    );
+    const language = normalizeLanguageCode(options.userLanguage);
+    const parameter = await openedDocument.graphBuildParameters.save({
+      prompt: resolveExtractionPrompt(options.extractionPrompt),
+      ...language === void 0 ? {} : { language }
+    });
+    await openedDocument.serials.setTopologyReady(
+      chapterId,
+      true,
+      parameter.hash
+    );
+    return await getChapterDetails(openedDocument, chapterId);
+  });
+}
+async function generateChapterSummary(document, chapterId, options) {
+  return await document.openSession(async (openedDocument) => {
+    const details = await requireChapterDetails(openedDocument, chapterId);
+    if (details.stage !== "graphed") {
+      throw new Error(
+        `Chapter ${chapterId} is ${details.stage}. Generate a summary only for graphed chapters.`
+      );
+    }
+    const generation = new SerialGeneration({
+      document: openedDocument,
+      llm: options.llm,
+      ...options.logDirPath === void 0 ? {} : { logDirPath: options.logDirPath }
+    });
+    await generation.buildSummary(chapterId, {
+      ...options.userLanguage === void 0 ? {} : { userLanguage: options.userLanguage }
+    });
+    return await getChapterDetails(openedDocument, chapterId);
+  });
+}
+export {
+  generateChapterGraph,
+  generateChapterSummary
+};
+//# sourceMappingURL=generate.js.map

package/dist/api/chapter/generate.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/api/chapter/generate.ts"],"sourcesContent":["import { normalizeLanguageCode } from \"../../runtime/common/language.js\";\nimport { resolveExtractionPrompt } from \"../../runtime/common/prompts.js\";\nimport type { Document } from \"../../document/index.js\";\nimport { SerialGeneration } from \"../../serial.js\";\nimport { getChapterDetails, requireChapterDetails } from \"./details.js\";\nimport { createTopologyOptions } from \"./options.js\";\nimport type {\n  GenerateChapterGraphOptions,\n  GenerateChapterSummaryOptions,\n} from \"./options.js\";\nimport type { ChapterDetails } from \"./types.js\";\n\nexport async function generateChapterGraph(\n  document: Document,\n  chapterId: number,\n  options: GenerateChapterGraphOptions,\n): Promise<ChapterDetails> {\n  return await document.openSession(async (openedDocument) => {\n    const details = await requireChapterDetails(openedDocument, chapterId);\n\n    if (details.stage !== \"sourced\") {\n      throw new Error(\n        `Chapter ${chapterId} is ${details.stage}. Generate a graph only for sourced chapters.`,\n      );\n    }\n\n    const generation = new SerialGeneration({\n      document: openedDocument,\n      llm: options.llm,\n      ...(options.logDirPath === undefined\n        ? {}\n        : { logDirPath: options.logDirPath }),\n    });\n    await openedDocument.clearSerialGraph(chapterId);\n\n    await generation.buildTopologyInto(\n      chapterId,\n      createTopologyOptions(options),\n      options.progressTracker,\n    );\n    const language = normalizeLanguageCode(options.userLanguage);\n    const parameter = await openedDocument.graphBuildParameters.save({\n      prompt: resolveExtractionPrompt(options.extractionPrompt),\n      ...(language === undefined ? {} : { language }),\n    });\n    await openedDocument.serials.setTopologyReady(\n      chapterId,\n      true,\n      parameter.hash,\n    );\n    return await getChapterDetails(openedDocument, chapterId);\n  });\n}\n\nexport async function generateChapterSummary(\n  document: Document,\n  chapterId: number,\n  options: GenerateChapterSummaryOptions,\n): Promise<ChapterDetails> {\n  return await document.openSession(async (openedDocument) => {\n    const details = await requireChapterDetails(openedDocument, chapterId);\n\n    if (details.stage !== \"graphed\") {\n      throw new Error(\n        `Chapter ${chapterId} is ${details.stage}. Generate a summary only for graphed chapters.`,\n      );\n    }\n\n    const generation = new SerialGeneration({\n      document: openedDocument,\n      llm: options.llm,\n      ...(options.logDirPath === undefined\n        ? {}\n        : { logDirPath: options.logDirPath }),\n    });\n\n    await generation.buildSummary(chapterId, {\n      ...(options.userLanguage === undefined\n        ? {}\n        : { userLanguage: options.userLanguage }),\n    });\n    return await getChapterDetails(openedDocument, chapterId);\n  });\n}\n"],"mappings":"AAAA,SAAS,6BAA6B;AACtC,SAAS,+BAA+B;AAExC,SAAS,wBAAwB;AACjC,SAAS,mBAAmB,6BAA6B;AACzD,SAAS,6BAA6B;AAOtC,eAAsB,qBACpB,UACA,WACA,SACyB;AACzB,SAAO,MAAM,SAAS,YAAY,OAAO,mBAAmB;AAC1D,UAAM,UAAU,MAAM,sBAAsB,gBAAgB,SAAS;AAErE,QAAI,QAAQ,UAAU,WAAW;AAC/B,YAAM,IAAI;AAAA,QACR,WAAW,SAAS,OAAO,QAAQ,KAAK;AAAA,MAC1C;AAAA,IACF;AAEA,UAAM,aAAa,IAAI,iBAAiB;AAAA,MACtC,UAAU;AAAA,MACV,KAAK,QAAQ;AAAA,MACb,GAAI,QAAQ,eAAe,SACvB,CAAC,IACD,EAAE,YAAY,QAAQ,WAAW;AAAA,IACvC,CAAC;AACD,UAAM,eAAe,iBAAiB,SAAS;AAE/C,UAAM,WAAW;AAAA,MACf;AAAA,MACA,sBAAsB,OAAO;AAAA,MAC7B,QAAQ;AAAA,IACV;AACA,UAAM,WAAW,sBAAsB,QAAQ,YAAY;AAC3D,UAAM,YAAY,MAAM,eAAe,qBAAqB,KAAK;AAAA,MAC/D,QAAQ,wBAAwB,QAAQ,gBAAgB;AAAA,MACxD,GAAI,aAAa,SAAY,CAAC,IAAI,EAAE,SAAS;AAAA,IAC/C,CAAC;AACD,UAAM,eAAe,QAAQ;AAAA,MAC3B;AAAA,MACA;AAAA,MACA,UAAU;AAAA,IACZ;AACA,WAAO,MAAM,kBAAkB,gBAAgB,SAAS;AAAA,EAC1D,CAAC;AACH;AAEA,eAAsB,uBACpB,UACA,WACA,SACyB;AACzB,SAAO,MAAM,SAAS,YAAY,OAAO,mBAAmB;AAC1D,UAAM,UAAU,MAAM,sBAAsB,gBAAgB,SAAS;AAErE,QAAI,QAAQ,UAAU,WAAW;AAC/B,YAAM,IAAI;AAAA,QACR,WAAW,SAAS,OAAO,QAAQ,KAAK;AAAA,MAC1C;AAAA,IACF;AAEA,UAAM,aAAa,IAAI,iBAAiB;AAAA,MACtC,UAAU;AAAA,MACV,KAAK,QAAQ;AAAA,MACb,GAAI,QAAQ,eAAe,SACvB,CAAC,IACD,EAAE,YAAY,QAAQ,WAAW;AAAA,IACvC,CAAC;AAED,UAAM,WAAW,aAAa,WAAW;AAAA,MACvC,GAAI,QAAQ,iBAAiB,SACzB,CAAC,IACD,EAAE,cAAc,QAAQ,aAAa;AAAA,IAC3C,CAAC;AACD,WAAO,MAAM,kBAAkB,gBAAgB,SAAS;AAAA,EAC1D,CAAC;AACH;","names":[]}