wiki-graph-sdk 0.7.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (98) hide show
  1. package/LICENSE +201 -0
  2. package/README.md +32 -0
  3. package/dist/archives.cjs +2799 -0
  4. package/dist/archives.cjs.map +1 -0
  5. package/dist/archives.d.ts +249 -0
  6. package/dist/archives.js +2814 -0
  7. package/dist/archives.js.map +1 -0
  8. package/dist/conversions.cjs +1332 -0
  9. package/dist/conversions.cjs.map +1 -0
  10. package/dist/conversions.d.ts +37 -0
  11. package/dist/conversions.js +1308 -0
  12. package/dist/conversions.js.map +1 -0
  13. package/dist/data/editor/clue_reviewer.jinja +410 -0
  14. package/dist/data/editor/clue_reviewer_generator.jinja +428 -0
  15. package/dist/data/editor/revision_feedback.jinja +5 -0
  16. package/dist/data/editor/text_compressor.jinja +131 -0
  17. package/dist/data/guaranteed/response_intent_classifier.jinja +8 -0
  18. package/dist/data/output/epub/cover.xhtml.jinja +13 -0
  19. package/dist/data/output/epub/nav.xhtml.jinja +12 -0
  20. package/dist/data/output/epub/package.opf.xml.jinja +39 -0
  21. package/dist/data/output/epub/section.xhtml.jinja +12 -0
  22. package/dist/data/topologization/book_coherence_extraction.jinja +223 -0
  23. package/dist/data/topologization/chunk_extraction.jinja +233 -0
  24. package/dist/data/topologization/evidence_choice.jinja +38 -0
  25. package/dist/data/topologization/translate_chunks.jinja +138 -0
  26. package/dist/data/topologization/user_focused_extraction.jinja +378 -0
  27. package/dist/default-worker.cjs +1709 -0
  28. package/dist/default-worker.cjs.map +1 -0
  29. package/dist/default-worker.d.ts +10 -0
  30. package/dist/default-worker.js +1705 -0
  31. package/dist/default-worker.js.map +1 -0
  32. package/dist/embedding.cjs +1091 -0
  33. package/dist/embedding.cjs.map +1 -0
  34. package/dist/embedding.d.ts +26 -0
  35. package/dist/embedding.js +1062 -0
  36. package/dist/embedding.js.map +1 -0
  37. package/dist/entry-context.cjs +91 -0
  38. package/dist/entry-context.cjs.map +1 -0
  39. package/dist/entry-context.d.ts +20 -0
  40. package/dist/entry-context.js +68 -0
  41. package/dist/entry-context.js.map +1 -0
  42. package/dist/gc.cjs +731 -0
  43. package/dist/gc.cjs.map +1 -0
  44. package/dist/gc.d.ts +14 -0
  45. package/dist/gc.js +707 -0
  46. package/dist/gc.js.map +1 -0
  47. package/dist/index.cjs +4760 -0
  48. package/dist/index.cjs.map +1 -0
  49. package/dist/index.d.ts +213 -0
  50. package/dist/index.js +4808 -0
  51. package/dist/index.js.map +1 -0
  52. package/dist/jobs-Dd5PMSId.d.ts +98 -0
  53. package/dist/llm.cjs +101 -0
  54. package/dist/llm.cjs.map +1 -0
  55. package/dist/llm.d.ts +7 -0
  56. package/dist/llm.js +79 -0
  57. package/dist/llm.js.map +1 -0
  58. package/dist/local-config.cjs +1024 -0
  59. package/dist/local-config.cjs.map +1 -0
  60. package/dist/local-config.d.ts +14 -0
  61. package/dist/local-config.js +989 -0
  62. package/dist/local-config.js.map +1 -0
  63. package/dist/maintenance.cjs +800 -0
  64. package/dist/maintenance.cjs.map +1 -0
  65. package/dist/maintenance.d.ts +24 -0
  66. package/dist/maintenance.js +778 -0
  67. package/dist/maintenance.js.map +1 -0
  68. package/dist/node-platform.cjs +758 -0
  69. package/dist/node-platform.cjs.map +1 -0
  70. package/dist/node-platform.d.ts +49 -0
  71. package/dist/node-platform.js +723 -0
  72. package/dist/node-platform.js.map +1 -0
  73. package/dist/planning.cjs +164 -0
  74. package/dist/planning.cjs.map +1 -0
  75. package/dist/planning.d.ts +43 -0
  76. package/dist/planning.js +137 -0
  77. package/dist/planning.js.map +1 -0
  78. package/dist/runtime-config.cjs +1155 -0
  79. package/dist/runtime-config.cjs.map +1 -0
  80. package/dist/runtime-config.d.ts +50 -0
  81. package/dist/runtime-config.js +1127 -0
  82. package/dist/runtime-config.js.map +1 -0
  83. package/dist/stage.cjs +1265 -0
  84. package/dist/stage.cjs.map +1 -0
  85. package/dist/stage.d.ts +16 -0
  86. package/dist/stage.js +1239 -0
  87. package/dist/stage.js.map +1 -0
  88. package/dist/wikispine.cjs +85 -0
  89. package/dist/wikispine.cjs.map +1 -0
  90. package/dist/wikispine.d.ts +6 -0
  91. package/dist/wikispine.js +63 -0
  92. package/dist/wikispine.js.map +1 -0
  93. package/dist/worker.cjs +738 -0
  94. package/dist/worker.cjs.map +1 -0
  95. package/dist/worker.d.ts +157 -0
  96. package/dist/worker.js +715 -0
  97. package/dist/worker.js.map +1 -0
  98. package/package.json +151 -0
package/dist/index.cjs ADDED
@@ -0,0 +1,4760 @@
1
+ globalThis.__WIKIGRAPH_DATA_DIR__ ??= require("path").resolve(__dirname, "data");
2
+ "use strict";
3
+ var __create = Object.create;
4
+ var __defProp = Object.defineProperty;
5
+ var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
6
+ var __getOwnPropNames = Object.getOwnPropertyNames;
7
+ var __getProtoOf = Object.getPrototypeOf;
8
+ var __hasOwnProp = Object.prototype.hasOwnProperty;
9
+ var __export = (target, all) => {
10
+ for (var name in all)
11
+ __defProp(target, name, { get: all[name], enumerable: true });
12
+ };
13
+ var __copyProps = (to, from, except, desc) => {
14
+ if (from && typeof from === "object" || typeof from === "function") {
15
+ for (let key of __getOwnPropNames(from))
16
+ if (!__hasOwnProp.call(to, key) && key !== except)
17
+ __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
18
+ }
19
+ return to;
20
+ };
21
+ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
22
+ // If the importer is in node compatibility mode or this is not an ESM
23
+ // file that has been converted to a CommonJS file using a Babel-
24
+ // compatible transform (i.e. "__esModule" has not been set), then set
25
+ // "default" to the CommonJS "module.exports" for node compatibility.
26
+ isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
27
+ mod
28
+ ));
29
+ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
30
+
31
+ // src/index.ts
32
+ var src_exports = {};
33
+ __export(src_exports, {
34
+ DANGEROUS_RUNTIME_ENV_NAMES: () => DANGEROUS_RUNTIME_ENV_NAMES,
35
+ DEFAULT_EXTRACTION_PROMPT: () => import_wiki_graph_core8.DEFAULT_EXTRACTION_PROMPT,
36
+ DEFAULT_GENERATION_JOB_CONCURRENCY: () => DEFAULT_GENERATION_JOB_CONCURRENCY,
37
+ DEFAULT_GENERATION_REQUEST_CONCURRENCY: () => DEFAULT_GENERATION_REQUEST_CONCURRENCY,
38
+ DEFAULT_KNOWLEDGE_GRAPH_RECALL_PROMPT: () => import_wiki_graph_core8.DEFAULT_KNOWLEDGE_GRAPH_RECALL_PROMPT,
39
+ LLMPaymentRequiredError: () => import_wiki_graph_core16.LLMPaymentRequiredError,
40
+ LOCAL_CONFIG_SECTIONS: () => LOCAL_CONFIG_SECTIONS,
41
+ NodeDirectory: () => NodeDirectory,
42
+ NodeFile: () => NodeFile,
43
+ WIKI_GRAPH_URI_PREFIX: () => import_wiki_graph_core16.WIKI_GRAPH_URI_PREFIX,
44
+ WikiGraphArchiveExistsError: () => WikiGraphArchiveExistsError,
45
+ WikiGraphArchiveHandle: () => WikiGraphArchiveHandle,
46
+ WikiGraphArchiveManager: () => WikiGraphArchiveManager,
47
+ WikiGraphConfigManager: () => WikiGraphConfigManager,
48
+ WikiGraphContinuationManager: () => WikiGraphContinuationManager,
49
+ WikiGraphConversionManager: () => WikiGraphConversionManager,
50
+ WikiGraphError: () => import_wiki_graph_core16.WikiGraphError,
51
+ WikiGraphJob: () => WikiGraphJob,
52
+ WikiGraphJobManager: () => WikiGraphJobManager,
53
+ WikiGraphLibrary: () => WikiGraphLibrary,
54
+ WikiGraphLibraryManager: () => WikiGraphLibraryManager,
55
+ WikiGraphMaintenanceManager: () => WikiGraphMaintenanceManager,
56
+ WikiGraphSDK: () => WikiGraphSDK,
57
+ buildSearchIndexEmbeddingProvider: () => buildSearchIndexEmbeddingProvider,
58
+ buildWikiGraphLLMOptions: () => buildWikiGraphLLMOptions,
59
+ clearLocalConfigSection: () => clearLocalConfigSection,
60
+ createEntryRuntimeContext: () => createEntryRuntimeContext,
61
+ createGenerationPerformanceHints: () => createGenerationPerformanceHints,
62
+ createNodeTemplateEnvironment: () => createNodeTemplateEnvironment,
63
+ createNodeWikiGraphStorage: () => createNodeWikiGraphStorage,
64
+ createStageLLM: () => createStageLLM,
65
+ createWikiGraphArchiveInspectReport: () => createWikiGraphArchiveInspectReport,
66
+ createWikiGraphSDK: () => createWikiGraphSDK,
67
+ deleteLocalConfigValue: () => deleteLocalConfigValue,
68
+ embedQueryText: () => embedQueryText,
69
+ embedQueryTextWithLocalConfig: () => embedQueryTextWithLocalConfig,
70
+ ensureNodeWikiGraphPlatform: () => ensureNodeWikiGraphPlatform,
71
+ formatError: () => import_wiki_graph_core16.formatError,
72
+ formatGenerationPlanningDuration: () => formatGenerationPlanningDuration,
73
+ formatGenerationPlanningModel: () => formatGenerationPlanningModel,
74
+ formatLocatedChapterResourceUri: () => import_wiki_graph_core16.formatLocatedChapterResourceUri,
75
+ formatLocatedChapterSourceCollectionUri: () => import_wiki_graph_core16.formatLocatedChapterSourceCollectionUri,
76
+ formatLocatedChapterUri: () => import_wiki_graph_core16.formatLocatedChapterUri,
77
+ formatLocatedWikiGraphUri: () => import_wiki_graph_core16.formatLocatedWikiGraphUri,
78
+ formatWikiGraphCommandUri: () => import_wiki_graph_core16.formatWikiGraphCommandUri,
79
+ formatWikiGraphLibraryUri: () => import_wiki_graph_core16.formatWikiGraphLibraryUri,
80
+ getNodeResourcePath: () => getNodeResourcePath,
81
+ getWikiGraphSDKRuntimeContext: () => getWikiGraphSDKRuntimeContext,
82
+ getWikiGraphSDKStateDir: () => getWikiGraphSDKStateDir,
83
+ installNodeWikiGraphPlatform: () => installNodeWikiGraphPlatform,
84
+ isSourceLocatorScopeUri: () => import_wiki_graph_core16.isSourceLocatorScopeUri,
85
+ isWikiGraphLibraryUri: () => import_wiki_graph_core16.isWikiGraphLibraryUri,
86
+ loadRequiredStageConfig: () => loadRequiredStageConfig,
87
+ loadWikiGraphRuntimeConfig: () => loadWikiGraphRuntimeConfig,
88
+ maskLocalConfigSection: () => maskLocalConfigSection,
89
+ nodeWikiGraphPlatform: () => nodeWikiGraphPlatform,
90
+ nodeWikispineCommandRunner: () => nodeWikispineCommandRunner,
91
+ normalizeLocalConfigKey: () => normalizeLocalConfigKey,
92
+ parseChapterPath: () => import_wiki_graph_core16.parseChapterPath,
93
+ parseChapterTreeInput: () => import_wiki_graph_core16.parseChapterTreeInput,
94
+ parseChapterUriPath: () => import_wiki_graph_core16.parseChapterUriPath,
95
+ parseLocalConfigSection: () => parseLocalConfigSection,
96
+ parseLocatedWikiGraphUri: () => import_wiki_graph_core16.parseLocatedWikiGraphUri,
97
+ parseSourceTextJsonl: () => import_wiki_graph_core16.parseSourceTextJsonl,
98
+ parseWikiGraphLibraryUri: () => import_wiki_graph_core16.parseWikiGraphLibraryUri,
99
+ parseWikiGraphUriSyntax: () => import_wiki_graph_core16.parseWikiGraphUriSyntax,
100
+ planGenerationTask: () => planGenerationTask,
101
+ putLocalConfigValue: () => putLocalConfigValue,
102
+ readLocalConfigSection: () => readLocalConfigSection,
103
+ readWikiGraphEmbeddingConfig: () => readWikiGraphEmbeddingConfig,
104
+ replaceLocalConfigSection: () => replaceLocalConfigSection,
105
+ requireKnowledgeGraphWikispineConfig: () => requireKnowledgeGraphWikispineConfig,
106
+ resolveExtractionPrompt: () => import_wiki_graph_core8.resolveExtractionPrompt,
107
+ resolveHostedProviderEndpoint: () => resolveHostedProviderEndpoint,
108
+ resolveKnowledgeGraphRecallPrompt: () => import_wiki_graph_core8.resolveKnowledgeGraphRecallPrompt,
109
+ resolveWikiGraphArchiveLocation: () => resolveWikiGraphArchiveLocation,
110
+ resolveWikispineConfig: () => resolveWikispineConfig,
111
+ runWikiGraphQueueWorker: () => runWikiGraphQueueWorker,
112
+ validateLocalConfigSection: () => validateLocalConfigSection,
113
+ withNodeWikiGraphStorage: () => withNodeWikiGraphStorage,
114
+ withWikiGraphSDKRuntimeContext: () => withWikiGraphSDKRuntimeContext,
115
+ writeNodeFileHandle: () => writeNodeFileHandle
116
+ });
117
+ module.exports = __toCommonJS(src_exports);
118
+
119
+ // src/archive/archive.ts
120
+ var import_crypto = require("crypto");
121
+ var import_promises3 = require("fs/promises");
122
+ var import_os3 = require("os");
123
+ var import_path3 = require("path");
124
+ var import_stream = require("stream");
125
+ var import_wiki_graph_core5 = require("wiki-graph-core");
126
+
127
+ // src/conversions.ts
128
+ var import_promises2 = require("fs/promises");
129
+ var import_wiki_graph_core2 = require("wiki-graph-core");
130
+
131
+ // src/llm.ts
132
+ var import_anthropic = require("@ai-sdk/anthropic");
133
+ var import_google = require("@ai-sdk/google");
134
+ var import_openai = require("@ai-sdk/openai");
135
+ var import_openai_compatible = require("@ai-sdk/openai-compatible");
136
+ function buildWikiGraphLLMOptions(config) {
137
+ const llm = config.llm;
138
+ if (llm?.provider === void 0 || llm.model === void 0) {
139
+ throw new Error(
140
+ "Missing LLM configuration. Set --llm for one run, or configure `wikg://local/config/llm` with provider and model."
141
+ );
142
+ }
143
+ return {
144
+ model: createLanguageModel(llm.provider, llm.model, {
145
+ apiKey: llm.apiKey,
146
+ baseURL: llm.baseURL,
147
+ name: llm.name
148
+ }),
149
+ ...config.concurrent?.request === void 0 ? {} : { concurrent: config.concurrent.request }
150
+ };
151
+ }
152
+ function createLanguageModel(provider, model, options) {
153
+ switch (provider) {
154
+ case "anthropic": {
155
+ const anthropic = (0, import_anthropic.createAnthropic)({
156
+ ...options.apiKey === void 0 ? {} : { apiKey: options.apiKey },
157
+ ...options.name === void 0 ? {} : { name: options.name }
158
+ });
159
+ return anthropic(model);
160
+ }
161
+ case "google": {
162
+ const google = (0, import_google.createGoogleGenerativeAI)({
163
+ ...options.apiKey === void 0 ? {} : { apiKey: options.apiKey },
164
+ ...options.name === void 0 ? {} : { name: options.name }
165
+ });
166
+ return google(model);
167
+ }
168
+ case "openai": {
169
+ if (options.baseURL !== void 0) {
170
+ throw new Error(
171
+ "openai does not accept llm.baseURL or baseURL in --llm JSON. Use openai-compatible for third-party OpenAI-style APIs."
172
+ );
173
+ }
174
+ const openai = (0, import_openai.createOpenAI)({
175
+ ...options.apiKey === void 0 ? {} : { apiKey: options.apiKey },
176
+ ...options.name === void 0 ? {} : { name: options.name }
177
+ });
178
+ return openai(model);
179
+ }
180
+ case "openai-compatible": {
181
+ if (options.baseURL === void 0) {
182
+ throw new Error(
183
+ "openai-compatible requires llm.baseURL or baseURL in --llm JSON."
184
+ );
185
+ }
186
+ const openaiCompatible = (0, import_openai_compatible.createOpenAICompatible)({
187
+ baseURL: options.baseURL,
188
+ name: options.name ?? createOpenAICompatibleName(options.baseURL),
189
+ ...options.apiKey === void 0 ? {} : { apiKey: options.apiKey }
190
+ });
191
+ return openaiCompatible(model);
192
+ }
193
+ }
194
+ }
195
+ function createOpenAICompatibleName(baseURL) {
196
+ try {
197
+ return new URL(baseURL).hostname;
198
+ } catch {
199
+ return "openai-compatible";
200
+ }
201
+ }
202
+
203
+ // src/node-platform.ts
204
+ var asyncHooks = __toESM(require("async_hooks"), 1);
205
+ var crypto = __toESM(require("crypto"), 1);
206
+ var fs = __toESM(require("fs"), 1);
207
+ var fsPromises = __toESM(require("fs/promises"), 1);
208
+ var os = __toESM(require("os"), 1);
209
+ var pathModule = __toESM(require("path"), 1);
210
+ var process2 = __toESM(require("process"), 1);
211
+ var import_node_stream = require("stream");
212
+ var import_promises = require("stream/promises");
213
+ var sqlite3 = __toESM(require("sqlite3"), 1);
214
+ var yauzl = __toESM(require("yauzl"), 1);
215
+ var yazl = __toESM(require("yazl"), 1);
216
+ var import_nunjucks = require("nunjucks");
217
+ var import_platform = require("wiki-graph-core/platform");
218
+ var nodeProcess = process2.default ?? process2;
219
+ var nodeSqlite3 = sqlite3.default ?? sqlite3;
220
+ var NODE_RESOURCE_KIND = /* @__PURE__ */ Symbol.for("wiki-graph-sdk.node-resource.v1");
221
+ var NodeFile = class {
222
+ constructor(path, name = pathModule.basename(path)) {
223
+ this.path = path;
224
+ this.identity = encodeNodeResourceIdentity("file", path);
225
+ this.name = name;
226
+ }
227
+ path;
228
+ kind = "file";
229
+ [NODE_RESOURCE_KIND] = "file";
230
+ identity;
231
+ name;
232
+ async openReader() {
233
+ const handle = await fsPromises.open(this.path, "r");
234
+ let closed = false;
235
+ try {
236
+ const stats = await handle.stat();
237
+ if (!stats.isFile())
238
+ throw new Error("File reader requires a regular file");
239
+ const { size } = stats;
240
+ return {
241
+ size,
242
+ close: async () => {
243
+ if (!closed) await handle.close();
244
+ closed = true;
245
+ },
246
+ read: async (offset, length) => {
247
+ if (closed) throw new Error("Cannot read from a closed FileReader");
248
+ assertFileRange(offset, length, size);
249
+ const buffer = Buffer.allocUnsafe(length);
250
+ let bytesRead = 0;
251
+ while (bytesRead < length) {
252
+ const result = await handle.read(
253
+ buffer,
254
+ bytesRead,
255
+ length - bytesRead,
256
+ offset + bytesRead
257
+ );
258
+ if (result.bytesRead === 0) {
259
+ throw new Error("File changed while reading its byte range");
260
+ }
261
+ bytesRead += result.bytesRead;
262
+ }
263
+ return new Uint8Array(buffer.buffer, buffer.byteOffset, bytesRead);
264
+ }
265
+ };
266
+ } catch (error) {
267
+ await handle.close();
268
+ throw error;
269
+ }
270
+ }
271
+ async getLastModified() {
272
+ try {
273
+ return (await fsPromises.stat(this.path)).mtimeMs;
274
+ } catch (error) {
275
+ if (typeof error === "object" && error !== null && "code" in error && error.code === "ENOENT") {
276
+ return void 0;
277
+ }
278
+ throw error;
279
+ }
280
+ }
281
+ async openWriter() {
282
+ await fsPromises.mkdir(pathModule.dirname(this.path), { recursive: true });
283
+ const temporary = `${this.path}.tmp-${crypto.randomUUID()}`;
284
+ const handle = await fsPromises.open(temporary, "wx");
285
+ let closed = false;
286
+ return {
287
+ write: async (data) => {
288
+ if (closed) throw new Error("Cannot write to a closed FileWriter");
289
+ await writeNodeFileHandle(handle, data);
290
+ },
291
+ writeAt: async (offset, data) => {
292
+ if (closed) throw new Error("Cannot write to a closed FileWriter");
293
+ assertFileOffset(offset);
294
+ await writeNodeFileHandle(handle, data, offset);
295
+ },
296
+ commit: async () => {
297
+ if (!closed) {
298
+ await handle.close();
299
+ await fsPromises.rename(temporary, this.path);
300
+ }
301
+ closed = true;
302
+ },
303
+ abort: async () => {
304
+ if (!closed) {
305
+ await handle.close();
306
+ await fsPromises.rm(temporary, { force: true });
307
+ }
308
+ closed = true;
309
+ }
310
+ };
311
+ }
312
+ };
313
+ async function writeNodeFileHandle(handle, data, position) {
314
+ const buffer = Buffer.from(data);
315
+ let written = 0;
316
+ while (written < buffer.byteLength) {
317
+ const result = await handle.write(
318
+ buffer,
319
+ written,
320
+ buffer.byteLength - written,
321
+ position === void 0 ? null : position + written
322
+ );
323
+ if (result.bytesWritten === 0) {
324
+ throw new Error("File writer made no progress");
325
+ }
326
+ written += result.bytesWritten;
327
+ }
328
+ }
329
+ function assertFileRange(offset, length, size) {
330
+ assertFileOffset(offset);
331
+ if (!Number.isSafeInteger(length) || length < 0) {
332
+ throw new RangeError(
333
+ "File read length must be a non-negative safe integer"
334
+ );
335
+ }
336
+ if (offset > size || length > size - offset) {
337
+ throw new RangeError("File read range exceeds the file size");
338
+ }
339
+ }
340
+ function assertFileOffset(offset) {
341
+ if (!Number.isSafeInteger(offset) || offset < 0) {
342
+ throw new RangeError("File offset must be a non-negative safe integer");
343
+ }
344
+ }
345
+ var NodeDirectory = class _NodeDirectory {
346
+ constructor(path) {
347
+ this.path = path;
348
+ this.identity = encodeNodeResourceIdentity("directory", path);
349
+ this.name = pathModule.basename(path);
350
+ }
351
+ path;
352
+ kind = "directory";
353
+ [NODE_RESOURCE_KIND] = "directory";
354
+ identity;
355
+ name;
356
+ async getFile(name) {
357
+ assertChildName(name);
358
+ const file = new NodeFile(pathModule.join(this.path, name));
359
+ try {
360
+ return (await fsPromises.stat(file.path)).isFile() ? file : void 0;
361
+ } catch {
362
+ return void 0;
363
+ }
364
+ }
365
+ async getLastModified() {
366
+ try {
367
+ return (await fsPromises.stat(this.path)).mtimeMs;
368
+ } catch (error) {
369
+ if (typeof error === "object" && error !== null && "code" in error && error.code === "ENOENT") {
370
+ return void 0;
371
+ }
372
+ throw error;
373
+ }
374
+ }
375
+ async getDirectory(name) {
376
+ assertChildName(name);
377
+ const directory = new _NodeDirectory(pathModule.join(this.path, name));
378
+ try {
379
+ return (await fsPromises.stat(directory.path)).isDirectory() ? directory : void 0;
380
+ } catch {
381
+ return void 0;
382
+ }
383
+ }
384
+ async list() {
385
+ const entries = await fsPromises.readdir(this.path, {
386
+ withFileTypes: true
387
+ });
388
+ return entries.map(
389
+ (entry) => entry.isDirectory() ? new _NodeDirectory(pathModule.join(this.path, entry.name)) : new NodeFile(pathModule.join(this.path, entry.name), entry.name)
390
+ );
391
+ }
392
+ async createFile(name) {
393
+ assertChildName(name);
394
+ await fsPromises.mkdir(this.path, { recursive: true });
395
+ const file = new NodeFile(pathModule.join(this.path, name));
396
+ await fsPromises.writeFile(file.path, "", { flag: "wx" });
397
+ return file;
398
+ }
399
+ async createDirectory(name) {
400
+ assertChildName(name);
401
+ await fsPromises.mkdir(this.path, { recursive: true });
402
+ const directory = new _NodeDirectory(pathModule.join(this.path, name));
403
+ await fsPromises.mkdir(directory.path);
404
+ return directory;
405
+ }
406
+ async remove(name, options) {
407
+ assertChildName(name);
408
+ await fsPromises.rm(pathModule.join(this.path, name), {
409
+ force: true,
410
+ recursive: options?.recursive === true
411
+ });
412
+ }
413
+ };
414
+ function encodeNodeResourceIdentity(kind, path) {
415
+ return `node-${kind}:${Buffer.from(pathModule.resolve(path), "utf8").toString("base64url")}`;
416
+ }
417
+ function decodeNodeResourceIdentity(identity, kind) {
418
+ const prefix = `node-${kind}:`;
419
+ if (!identity.startsWith(prefix)) {
420
+ return pathModule.isAbsolute(identity) ? identity : void 0;
421
+ }
422
+ try {
423
+ return Buffer.from(identity.slice(prefix.length), "base64url").toString(
424
+ "utf8"
425
+ );
426
+ } catch {
427
+ return void 0;
428
+ }
429
+ }
430
+ function getNodeResourcePath(resource) {
431
+ const brandedPath = getBrandedNodeResourcePath(resource, resource.kind);
432
+ if (brandedPath !== void 0) return brandedPath;
433
+ if (typeof resource === "object" && resource !== null && "path" in resource && typeof resource.path === "string") {
434
+ return resource.path;
435
+ }
436
+ throw new TypeError("Expected a Node File or Directory resource");
437
+ }
438
+ function assertChildName(name) {
439
+ if (name.length === 0 || name === "." || name === ".." || pathModule.isAbsolute(name) || name.includes("/") || name.includes("\\")) {
440
+ throw new TypeError(`Directory child must be a relative name: ${name}`);
441
+ }
442
+ }
443
+ var NodeDatabaseConnection = class {
444
+ constructor(database) {
445
+ this.database = database;
446
+ }
447
+ database;
448
+ async close() {
449
+ await new Promise((resolve3, reject) => {
450
+ this.database.close((error) => {
451
+ if (error) reject(error);
452
+ else resolve3();
453
+ });
454
+ });
455
+ }
456
+ async execute(sql) {
457
+ await new Promise((resolve3, reject) => {
458
+ this.database.exec(sql, (error) => {
459
+ if (error) reject(error);
460
+ else resolve3();
461
+ });
462
+ });
463
+ }
464
+ async queryAll(sql, params = []) {
465
+ return await new Promise((resolve3, reject) => {
466
+ this.database.all(sql, [...params], (error, rows) => {
467
+ if (error) reject(error);
468
+ else resolve3(rows);
469
+ });
470
+ });
471
+ }
472
+ async queryOne(sql, params = []) {
473
+ return await new Promise((resolve3, reject) => {
474
+ this.database.get(sql, [...params], (error, row) => {
475
+ if (error) reject(error);
476
+ else resolve3(row);
477
+ });
478
+ });
479
+ }
480
+ async run(sql, params = []) {
481
+ await new Promise((resolve3, reject) => {
482
+ this.database.run(sql, [...params], (error) => {
483
+ if (error) reject(error);
484
+ else resolve3();
485
+ });
486
+ });
487
+ }
488
+ };
489
+ async function openNodeDatabase(file, options) {
490
+ const flags = (options.mode === "readonly" ? nodeSqlite3.OPEN_READONLY : nodeSqlite3.OPEN_READWRITE | (options.create ? nodeSqlite3.OPEN_CREATE : 0)) | nodeSqlite3.OPEN_FULLMUTEX;
491
+ return new NodeDatabaseConnection(await openNativeNodeDatabase(file, flags));
492
+ }
493
+ async function openNativeNodeDatabase(file, flags) {
494
+ const path = getBrandedNodeResourcePath(file, "file");
495
+ if (path === void 0) {
496
+ throw new TypeError("The Node database adapter requires a NodeFile");
497
+ }
498
+ return await new Promise((resolve3, reject) => {
499
+ const database = new nodeSqlite3.Database(path, flags, (error) => {
500
+ if (error) reject(error);
501
+ else resolve3(database);
502
+ });
503
+ });
504
+ }
505
+ async function openNodeZip(file) {
506
+ const path = getBrandedNodeResourcePath(file, "file");
507
+ if (path === void 0) {
508
+ throw new TypeError("The Node ZIP adapter requires a NodeFile");
509
+ }
510
+ const zipFile = await new Promise((resolve3, reject) => {
511
+ yauzl.open(
512
+ path,
513
+ { autoClose: false, lazyEntries: true },
514
+ (error, opened) => {
515
+ if (error || opened === void 0)
516
+ reject(error ?? new Error("Cannot open ZIP"));
517
+ else resolve3(opened);
518
+ }
519
+ );
520
+ });
521
+ const entries = await new Promise(
522
+ (resolve3, reject) => {
523
+ const discovered = /* @__PURE__ */ new Map();
524
+ const rejectAndClose = (error) => {
525
+ zipFile.close();
526
+ reject(
527
+ error instanceof Error ? error : new Error("Cannot scan ZIP central directory")
528
+ );
529
+ };
530
+ const resolveEntries = () => {
531
+ zipFile.off("error", rejectAndClose);
532
+ resolve3(discovered);
533
+ };
534
+ zipFile.on("entry", (entry) => {
535
+ if (!entry.fileName.endsWith("/")) {
536
+ discovered.set(entry.fileName, entry);
537
+ }
538
+ zipFile.readEntry();
539
+ });
540
+ zipFile.once("error", rejectAndClose);
541
+ zipFile.once("end", resolveEntries);
542
+ zipFile.readEntry();
543
+ }
544
+ );
545
+ let closed = false;
546
+ return {
547
+ close: () => {
548
+ if (!closed) zipFile.close();
549
+ closed = true;
550
+ return Promise.resolve();
551
+ },
552
+ copyEntry: async (name, target) => {
553
+ if (closed) throw new Error("Cannot read a closed ZIP archive");
554
+ const entry = entries.get(name);
555
+ if (entry === void 0) return false;
556
+ await copyNodeZipEntry(zipFile, entry, target);
557
+ return true;
558
+ },
559
+ getEntrySize: (name) => {
560
+ if (closed) throw new Error("Cannot read a closed ZIP archive");
561
+ return Promise.resolve(entries.get(name)?.uncompressedSize);
562
+ },
563
+ listEntries: () => Promise.resolve([...entries.keys()]),
564
+ readEntryRange: async (name, offset, length) => {
565
+ if (closed) throw new Error("Cannot read a closed ZIP archive");
566
+ const entry = entries.get(name);
567
+ if (entry === void 0) return void 0;
568
+ return await readNodeZipEntryRange(zipFile, entry, offset, length);
569
+ },
570
+ readEntry: async (name) => {
571
+ if (closed) throw new Error("Cannot read a closed ZIP archive");
572
+ const entry = entries.get(name);
573
+ if (entry === void 0) return void 0;
574
+ return await readNodeZipEntry(zipFile, entry);
575
+ }
576
+ };
577
+ }
578
+ async function copyNodeZipEntry(zipFile, entry, target) {
579
+ const writer = await target.openWriter();
580
+ try {
581
+ const stream = await openNodeZipEntryStream(zipFile, entry);
582
+ await writeNodeStream(stream, writer);
583
+ await writer.commit();
584
+ } catch (error) {
585
+ await writer.abort().catch(() => void 0);
586
+ throw error;
587
+ }
588
+ }
589
+ async function readNodeZipEntry(zipFile, entry) {
590
+ return await collectNodeStream(await openNodeZipEntryStream(zipFile, entry));
591
+ }
592
+ async function readNodeZipEntryRange(zipFile, entry, offset, length) {
593
+ assertFileRange(offset, length, entry.uncompressedSize);
594
+ if (length === 0) return new Uint8Array();
595
+ if (entry.compressionMethod !== 0) {
596
+ const content2 = await collectDecodedNodeStreamRange(
597
+ await openNodeZipEntryStream(zipFile, entry),
598
+ offset,
599
+ length
600
+ );
601
+ return new Uint8Array(
602
+ content2.buffer,
603
+ content2.byteOffset,
604
+ content2.byteLength
605
+ );
606
+ }
607
+ const content = await collectNodeStream(
608
+ await openNodeZipEntryStream(zipFile, entry, {
609
+ decompress: null,
610
+ decrypt: null,
611
+ end: offset + length,
612
+ start: offset
613
+ })
614
+ );
615
+ return new Uint8Array(content.buffer, content.byteOffset, content.byteLength);
616
+ }
617
+ async function collectDecodedNodeStreamRange(input, offset, length) {
618
+ const chunks = [];
619
+ const end = offset + length;
620
+ let position = 0;
621
+ let collected = 0;
622
+ for await (const value of input) {
623
+ const chunk = Buffer.isBuffer(value) ? value : Buffer.from(value);
624
+ const chunkEnd = position + chunk.byteLength;
625
+ if (chunkEnd > offset && position < end) {
626
+ const selected = chunk.subarray(
627
+ Math.max(0, offset - position),
628
+ Math.min(chunk.byteLength, end - position)
629
+ );
630
+ chunks.push(selected);
631
+ collected += selected.byteLength;
632
+ }
633
+ position = chunkEnd;
634
+ if (position >= end) break;
635
+ }
636
+ if (collected !== length) {
637
+ throw new Error("ZIP entry changed while reading its byte range");
638
+ }
639
+ return Buffer.concat(chunks, collected);
640
+ }
641
+ async function openNodeZipEntryStream(zipFile, entry, options) {
642
+ return await new Promise((resolve3, reject) => {
643
+ const callback = (error, opened) => {
644
+ if (error || opened === void 0) {
645
+ reject(error ?? new Error(`Cannot read ZIP entry: ${entry.fileName}`));
646
+ } else {
647
+ resolve3(opened);
648
+ }
649
+ };
650
+ if (options === void 0) zipFile.openReadStream(entry, callback);
651
+ else zipFile.openReadStream(entry, options, callback);
652
+ });
653
+ }
654
+ async function writeNodeZip(file, entries) {
655
+ const zipFile = new yazl.ZipFile();
656
+ const writer = await file.openWriter();
657
+ const outputStream = zipFile.outputStream;
658
+ const outputDone = writeNodeStream(outputStream, writer);
659
+ const entryState = { cancelled: false, complete: false };
660
+ let iterator;
661
+ try {
662
+ iterator = getNodeZipEntryIterator(entries);
663
+ const entriesDone = addNodeZipEntries(zipFile, iterator, entryState);
664
+ await Promise.race([entriesDone, outputDone]);
665
+ if (!entryState.complete) {
666
+ throw new Error("ZIP output ended before entry production completed");
667
+ }
668
+ zipFile.end();
669
+ await outputDone;
670
+ await writer.commit();
671
+ } catch (error) {
672
+ entryState.cancelled = true;
673
+ if (!entryState.complete && iterator !== void 0) {
674
+ closeNodeZipEntryIterator(iterator);
675
+ }
676
+ if (!outputStream.destroyed) {
677
+ outputStream.destroy(
678
+ error instanceof Error ? error : new Error("Cannot write ZIP archive")
679
+ );
680
+ }
681
+ await outputDone.catch(() => void 0);
682
+ await writer.abort().catch(() => void 0);
683
+ throw error;
684
+ }
685
+ }
686
+ async function addNodeZipEntries(zipFile, iterator, state) {
687
+ while (!state.cancelled) {
688
+ const result = await iterator.next();
689
+ if (state.cancelled) return;
690
+ if (result.done) {
691
+ state.complete = true;
692
+ return;
693
+ }
694
+ if ("file" in result.value) {
695
+ zipFile.addReadStream(
696
+ createNodeFileReadStream(result.value.file),
697
+ result.value.name,
698
+ { compress: false }
699
+ );
700
+ } else if ("data" in result.value) {
701
+ zipFile.addBuffer(Buffer.from(result.value.data), result.value.name, {
702
+ compress: false
703
+ });
704
+ } else {
705
+ zipFile.addReadStream(
706
+ createNodeRangeReadStream(result.value),
707
+ result.value.name,
708
+ { compress: false, size: result.value.size }
709
+ );
710
+ }
711
+ }
712
+ }
713
+ function createNodeFileReadStream(file) {
714
+ return import_node_stream.Readable.from(
715
+ (async function* () {
716
+ const reader = await file.openReader();
717
+ try {
718
+ for (let offset = 0; offset < reader.size; ) {
719
+ const chunk = await reader.read(
720
+ offset,
721
+ Math.min(64 * 1024, reader.size - offset)
722
+ );
723
+ offset += chunk.byteLength;
724
+ yield chunk;
725
+ }
726
+ } finally {
727
+ await reader.close();
728
+ }
729
+ })()
730
+ );
731
+ }
732
+ function getBrandedNodeResourcePath(resource, kind) {
733
+ if (typeof resource !== "object" || resource === null) return void 0;
734
+ const candidate = resource;
735
+ return candidate[NODE_RESOURCE_KIND] === kind && typeof candidate.path === "string" ? candidate.path : void 0;
736
+ }
737
+ function createNodeRangeReadStream(entry) {
738
+ return import_node_stream.Readable.from(
739
+ (async function* () {
740
+ for (let offset = 0; offset < entry.size; ) {
741
+ const chunk = await entry.read(
742
+ offset,
743
+ Math.min(64 * 1024, entry.size - offset)
744
+ );
745
+ offset += chunk.byteLength;
746
+ yield chunk;
747
+ }
748
+ })()
749
+ );
750
+ }
751
+ function getNodeZipEntryIterator(entries) {
752
+ const asyncIterator = entries[Symbol.asyncIterator];
753
+ return asyncIterator === void 0 ? entries[Symbol.iterator]() : asyncIterator.call(entries);
754
+ }
755
+ function closeNodeZipEntryIterator(iterator) {
756
+ if (iterator.return === void 0) return;
757
+ try {
758
+ void Promise.resolve(iterator.return()).catch(() => void 0);
759
+ } catch {
760
+ }
761
+ }
762
+ async function writeNodeStream(input, writer) {
763
+ await (0, import_promises.pipeline)(
764
+ input,
765
+ new import_node_stream.Writable({
766
+ write: (chunk, _encoding, callback) => {
767
+ void writer.write(chunk).then(
768
+ () => callback(),
769
+ (error) => callback(
770
+ error instanceof Error ? error : new Error("Cannot stream ZIP archive")
771
+ )
772
+ );
773
+ }
774
+ })
775
+ );
776
+ }
777
+ async function collectNodeStream(input) {
778
+ return await new Promise((resolve3, reject) => {
779
+ const chunks = [];
780
+ input.on("data", (chunk) => chunks.push(chunk));
781
+ input.once("error", reject);
782
+ input.once("end", () => resolve3(Buffer.concat(chunks)));
783
+ });
784
+ }
785
+ var nodeWikiGraphPlatform = {
786
+ asyncContext: {
787
+ create: () => new asyncHooks.AsyncLocalStorage()
788
+ },
789
+ database: { open: openNodeDatabase },
790
+ lifecycle: {
791
+ instanceId: `node-process:${nodeProcess.pid}`,
792
+ isInstanceAlive: (instanceId) => {
793
+ const match = /^node-process:(\d+)$/u.exec(instanceId);
794
+ if (match === null) return Promise.resolve(void 0);
795
+ try {
796
+ nodeProcess.kill(Number(match[1]), 0);
797
+ return Promise.resolve(true);
798
+ } catch (error) {
799
+ if (typeof error === "object" && error !== null && "code" in error && error.code === "ESRCH") {
800
+ return Promise.resolve(false);
801
+ }
802
+ return Promise.resolve(true);
803
+ }
804
+ }
805
+ },
806
+ resources: {
807
+ getDirectory: (identity) => {
808
+ const path = decodeNodeResourceIdentity(identity, "directory");
809
+ return Promise.resolve(
810
+ path === void 0 ? void 0 : new NodeDirectory(path)
811
+ );
812
+ },
813
+ getFile: (identity) => {
814
+ const path = decodeNodeResourceIdentity(identity, "file");
815
+ return Promise.resolve(
816
+ path === void 0 ? void 0 : new NodeFile(path)
817
+ );
818
+ },
819
+ resolveLegacyDirectory: (reference) => {
820
+ const path = decodeNodeResourceIdentity(reference, "directory");
821
+ return Promise.resolve(
822
+ path === void 0 ? void 0 : new NodeDirectory(path)
823
+ );
824
+ }
825
+ },
826
+ templates: {
827
+ createEnvironment: (options) => new import_nunjucks.Environment(new NodeTemplateLoader(), options)
828
+ },
829
+ zip: {
830
+ open: openNodeZip,
831
+ write: writeNodeZip
832
+ }
833
+ };
834
+ var NodeTemplateLoader = class extends import_nunjucks.Loader {
835
+ #dataDirectory;
836
+ constructor(dataDirectory) {
837
+ super();
838
+ this.#dataDirectory = dataDirectory;
839
+ }
840
+ getSource(templateName) {
841
+ const name = (0, import_platform.normalizeTemplateName)(templateName);
842
+ const filePath = pathModule.join(
843
+ this.#dataDirectory ?? resolveNodeDataDirectory(),
844
+ name
845
+ );
846
+ return {
847
+ noCache: false,
848
+ path: name,
849
+ src: fs.readFileSync(filePath, "utf8")
850
+ };
851
+ }
852
+ };
853
+ function createNodeTemplateEnvironment(dataDirectory, options = {}) {
854
+ return new import_nunjucks.Environment(new NodeTemplateLoader(dataDirectory), {
855
+ autoescape: options.autoescape ?? true,
856
+ trimBlocks: options.trimBlocks ?? true
857
+ });
858
+ }
859
+ function resolveNodeDataDirectory() {
860
+ const injected = globalThis.__WIKIGRAPH_DATA_DIR__;
861
+ if (typeof injected === "string" && injected !== "") return injected;
862
+ let directory = nodeProcess.cwd();
863
+ while (true) {
864
+ for (const candidate of [
865
+ pathModule.join(directory, "data"),
866
+ pathModule.join(directory, "packages", "core", "data")
867
+ ]) {
868
+ try {
869
+ if (fs.statSync(candidate).isDirectory()) return candidate;
870
+ } catch {
871
+ }
872
+ }
873
+ const parent = pathModule.dirname(directory);
874
+ if (parent === directory)
875
+ throw new Error("Could not locate data directory");
876
+ directory = parent;
877
+ }
878
+ }
879
+ function createNodeWikiGraphStorage(stateRoot = pathModule.join(os.homedir(), ".wikigraph")) {
880
+ return {
881
+ library: new NodeDirectory(stateRoot),
882
+ documentStore: new NodeDirectory(pathModule.join(stateRoot, "documents"))
883
+ };
884
+ }
885
+ function installNodeWikiGraphPlatform(stateRoot) {
886
+ (0, import_platform.installWikiGraphPlatform)(nodeWikiGraphPlatform);
887
+ (0, import_platform.installWikiGraphStorage)(createNodeWikiGraphStorage(stateRoot));
888
+ }
889
+ function ensureNodeWikiGraphPlatform() {
890
+ (0, import_platform.installWikiGraphPlatform)(nodeWikiGraphPlatform);
891
+ try {
892
+ (0, import_platform.getWikiGraphStorage)();
893
+ } catch {
894
+ (0, import_platform.installWikiGraphStorage)(createNodeWikiGraphStorage());
895
+ }
896
+ }
897
+ async function withNodeWikiGraphStorage(stateRoot, operation) {
898
+ return await (0, import_platform.withWikiGraphStorage)(
899
+ stateRoot === void 0 ? void 0 : createNodeWikiGraphStorage(stateRoot),
900
+ operation
901
+ );
902
+ }
903
+
904
+ // src/runtime-config.ts
905
+ var import_zod = require("zod");
906
+
907
+ // src/local-config.ts
908
+ var import_wiki_graph_core = require("wiki-graph-core");
909
+
910
+ // src/runtime-context.ts
911
+ var import_async_hooks = require("async_hooks");
912
+ var RUNTIME_CONTEXT_KEY = /* @__PURE__ */ Symbol.for("wiki-graph-sdk.runtime-context.v1");
913
+ var runtimeContext = getRuntimeContextStorage();
914
+ function getRuntimeContextStorage() {
915
+ const runtimeGlobal = globalThis;
916
+ const existing = runtimeGlobal[RUNTIME_CONTEXT_KEY];
917
+ if (existing !== void 0) {
918
+ return existing;
919
+ }
920
+ const storage = new import_async_hooks.AsyncLocalStorage();
921
+ Object.defineProperty(runtimeGlobal, RUNTIME_CONTEXT_KEY, {
922
+ configurable: false,
923
+ enumerable: false,
924
+ value: storage,
925
+ writable: false
926
+ });
927
+ return storage;
928
+ }
929
+ async function withWikiGraphSDKRuntimeContext(context, operation) {
930
+ return await runtimeContext.run(context, operation);
931
+ }
932
+ function getWikiGraphSDKRuntimeContext() {
933
+ return runtimeContext.getStore() ?? {
934
+ cwd: process.cwd(),
935
+ env: process.env,
936
+ envPolicy: "production"
937
+ };
938
+ }
939
+ function getWikiGraphSDKStateDir() {
940
+ return getWikiGraphSDKRuntimeContext().stateDir;
941
+ }
942
+
943
+ // src/local-config.ts
944
+ var import_path = require("path");
945
+ var import_os = require("os");
946
+ var LOCAL_CONFIG_SECTIONS = [
947
+ "concurrent",
948
+ "embeddings",
949
+ "llm",
950
+ "wikimedia",
951
+ "wikispine"
952
+ ];
953
+ var LOCAL_CONFIG_SCHEMA_SQL = `
954
+ CREATE TABLE IF NOT EXISTS config_sections (
955
+ section TEXT PRIMARY KEY,
956
+ value_json TEXT NOT NULL,
957
+ updated_at INTEGER NOT NULL
958
+ );
959
+ `;
960
+ async function readLocalConfigSection(section) {
961
+ return await withLocalConfigDatabase(async (database) => {
962
+ return await readLocalConfigSectionInDatabase(database, section);
963
+ });
964
+ }
965
+ async function replaceLocalConfigSection(section, value) {
966
+ const normalized = validateLocalConfigSection(section, value);
967
+ return await withLocalConfigDatabase(async (database) => {
968
+ await writeLocalConfigSectionInDatabase(database, section, normalized);
969
+ return normalized;
970
+ });
971
+ }
972
+ async function putLocalConfigValue(section, key, value) {
973
+ return await withLocalConfigDatabase(async (database) => {
974
+ const current = await readLocalConfigSectionInDatabase(database, section);
975
+ const next = validateLocalConfigSection(section, {
976
+ ...current,
977
+ [normalizeLocalConfigKey(section, key)]: value
978
+ });
979
+ await writeLocalConfigSectionInDatabase(database, section, next);
980
+ return next;
981
+ });
982
+ }
983
+ async function deleteLocalConfigValue(section, key) {
984
+ return await withLocalConfigDatabase(async (database) => {
985
+ const current = await readLocalConfigSectionInDatabase(database, section);
986
+ const normalizedKey = normalizeLocalConfigKey(section, key);
987
+ const { [normalizedKey]: _removed, ...nextInput } = current;
988
+ const next = validateLocalConfigSection(section, nextInput);
989
+ await writeLocalConfigSectionInDatabase(database, section, next);
990
+ return next;
991
+ });
992
+ }
993
+ async function clearLocalConfigSection(section) {
994
+ return await replaceLocalConfigSection(section, {});
995
+ }
996
+ function parseLocalConfigSection(value) {
997
+ if (value === "embedding") {
998
+ return "embeddings";
999
+ }
1000
+ return LOCAL_CONFIG_SECTIONS.find((section) => section === value);
1001
+ }
1002
+ function maskLocalConfigSection(section, value) {
1003
+ if (section !== "llm" && section !== "embeddings" && section !== "wikimedia" && section !== "wikispine" || value.apiKey === void 0 && value.token === void 0) {
1004
+ return value;
1005
+ }
1006
+ return {
1007
+ ...value,
1008
+ ...value.apiKey === void 0 ? {} : { apiKey: "****" },
1009
+ ...value.token === void 0 ? {} : { token: "****" }
1010
+ };
1011
+ }
1012
+ function normalizeLocalConfigKey(section, key) {
1013
+ const normalized = key.trim();
1014
+ if (normalized === "") {
1015
+ throw new Error("Config key cannot be empty.");
1016
+ }
1017
+ if (section === "llm" || section === "embeddings") {
1018
+ switch (normalized) {
1019
+ case "api-key":
1020
+ return "apiKey";
1021
+ case "base-url":
1022
+ return "baseURL";
1023
+ default:
1024
+ return normalized;
1025
+ }
1026
+ }
1027
+ return normalized;
1028
+ }
1029
+ function validateLocalConfigSection(section, value) {
1030
+ switch (section) {
1031
+ case "embeddings":
1032
+ return validateEmbeddingsConfig(value);
1033
+ case "llm":
1034
+ return validateLLMConfig(value);
1035
+ case "concurrent":
1036
+ return validateConcurrentConfig(value);
1037
+ case "wikimedia":
1038
+ return validateWikimediaConfig(value);
1039
+ case "wikispine":
1040
+ return validateWikispineConfig(value);
1041
+ }
1042
+ }
1043
+ async function withLocalConfigDatabase(operation) {
1044
+ const database = await (0, import_wiki_graph_core.openSharedStateDatabase)(
1045
+ new NodeFile(
1046
+ (0, import_path.join)(
1047
+ getWikiGraphSDKStateDir() ?? (0, import_path.join)((0, import_os.homedir)(), ".wikigraph"),
1048
+ "core.sqlite"
1049
+ )
1050
+ ),
1051
+ LOCAL_CONFIG_SCHEMA_SQL
1052
+ );
1053
+ try {
1054
+ return await operation(database);
1055
+ } finally {
1056
+ await database.close();
1057
+ }
1058
+ }
1059
+ async function readLocalConfigSectionInDatabase(database, section) {
1060
+ const value = await database.queryOne(
1061
+ "SELECT value_json FROM config_sections WHERE section = ?",
1062
+ [section],
1063
+ (row) => String(row.value_json)
1064
+ );
1065
+ return value === void 0 ? {} : parseSectionJSON(section, value);
1066
+ }
1067
+ async function writeLocalConfigSectionInDatabase(database, section, value) {
1068
+ await database.run(
1069
+ `
1070
+ INSERT INTO config_sections (section, value_json, updated_at)
1071
+ VALUES (?, ?, ?)
1072
+ ON CONFLICT(section) DO UPDATE SET
1073
+ value_json = excluded.value_json,
1074
+ updated_at = excluded.updated_at
1075
+ `,
1076
+ [section, JSON.stringify(value), Date.now()]
1077
+ );
1078
+ }
1079
+ function parseSectionJSON(section, value) {
1080
+ let parsed;
1081
+ try {
1082
+ parsed = JSON.parse(value);
1083
+ } catch (error) {
1084
+ throw new Error(
1085
+ `Invalid local config JSON for ${section}: ${formatError(error)}`
1086
+ );
1087
+ }
1088
+ if (parsed === null || typeof parsed !== "object" || Array.isArray(parsed)) {
1089
+ throw new Error(
1090
+ `Invalid local config section ${section}: expected object.`
1091
+ );
1092
+ }
1093
+ return validateLocalConfigSection(section, parsed);
1094
+ }
1095
+ function validateEmbeddingsConfig(value) {
1096
+ const allowedKeys = /* @__PURE__ */ new Set([
1097
+ "apiKey",
1098
+ "baseURL",
1099
+ "dimensions",
1100
+ "model",
1101
+ "name",
1102
+ "provider"
1103
+ ]);
1104
+ const allowedProviders = /* @__PURE__ */ new Set(["openai", "openai-compatible"]);
1105
+ const next = {};
1106
+ for (const [key, entry] of Object.entries(value)) {
1107
+ if (!allowedKeys.has(key)) {
1108
+ throw new Error(`Unknown embeddings config key: ${key}`);
1109
+ }
1110
+ if (key === "dimensions") {
1111
+ const parsed = typeof entry === "number" ? entry : typeof entry === "string" ? Number(entry) : NaN;
1112
+ if (!Number.isInteger(parsed) || parsed <= 0) {
1113
+ throw new Error("embeddings.dimensions must be a positive integer.");
1114
+ }
1115
+ next[key] = parsed;
1116
+ continue;
1117
+ }
1118
+ if (typeof entry !== "string" || entry.trim() === "") {
1119
+ throw new Error(`embeddings.${key} must be a non-empty string.`);
1120
+ }
1121
+ const normalized = entry.trim();
1122
+ if (key === "provider" && !allowedProviders.has(normalized)) {
1123
+ throw new Error(`Unknown embeddings.provider: ${normalized}`);
1124
+ }
1125
+ next[key] = normalized;
1126
+ }
1127
+ return next;
1128
+ }
1129
+ function validateLLMConfig(value) {
1130
+ const allowedKeys = /* @__PURE__ */ new Set([
1131
+ "apiKey",
1132
+ "baseURL",
1133
+ "model",
1134
+ "name",
1135
+ "provider"
1136
+ ]);
1137
+ const allowedProviders = /* @__PURE__ */ new Set([
1138
+ "anthropic",
1139
+ "google",
1140
+ "openai",
1141
+ "openai-compatible"
1142
+ ]);
1143
+ const next = {};
1144
+ for (const [key, entry] of Object.entries(value)) {
1145
+ if (!allowedKeys.has(key)) {
1146
+ throw new Error(`Unknown llm config key: ${key}`);
1147
+ }
1148
+ if (typeof entry !== "string" || entry.trim() === "") {
1149
+ throw new Error(`llm.${key} must be a non-empty string.`);
1150
+ }
1151
+ const normalized = entry.trim();
1152
+ if (key === "provider" && !allowedProviders.has(normalized)) {
1153
+ throw new Error(`Unknown llm.provider: ${normalized}`);
1154
+ }
1155
+ next[key] = normalized;
1156
+ }
1157
+ return next;
1158
+ }
1159
+ function validateConcurrentConfig(value) {
1160
+ const allowedKeys = /* @__PURE__ */ new Set(["job", "request"]);
1161
+ const next = {};
1162
+ for (const [key, entry] of Object.entries(value)) {
1163
+ if (!allowedKeys.has(key)) {
1164
+ throw new Error(`Unknown concurrent config key: ${key}`);
1165
+ }
1166
+ const parsed = typeof entry === "number" ? entry : typeof entry === "string" ? Number(entry) : NaN;
1167
+ if (!Number.isInteger(parsed) || parsed <= 0) {
1168
+ throw new Error(`concurrent.${key} must be a positive integer.`);
1169
+ }
1170
+ next[key] = parsed;
1171
+ }
1172
+ return next;
1173
+ }
1174
+ function validateWikispineConfig(value) {
1175
+ const allowedKeys = /* @__PURE__ */ new Set(["endpoint", "provider", "token"]);
1176
+ const allowedProviders = /* @__PURE__ */ new Set(["cli", "fetch"]);
1177
+ const next = {};
1178
+ for (const [key, entry] of Object.entries(value)) {
1179
+ if (!allowedKeys.has(key)) {
1180
+ throw new Error(`Unknown wikispine config key: ${key}`);
1181
+ }
1182
+ if (typeof entry !== "string" || entry.trim() === "") {
1183
+ throw new Error(`wikispine.${key} must be a non-empty string.`);
1184
+ }
1185
+ const normalized = entry.trim();
1186
+ if (key === "provider" && !allowedProviders.has(normalized)) {
1187
+ throw new Error(`Unknown wikispine.provider: ${normalized}`);
1188
+ }
1189
+ next[key] = normalized;
1190
+ }
1191
+ return next;
1192
+ }
1193
+ function validateWikimediaConfig(value) {
1194
+ return validateEndpointConfig("wikimedia", value);
1195
+ }
1196
+ function validateEndpointConfig(section, value) {
1197
+ const allowedKeys = /* @__PURE__ */ new Set(["endpoint", "token"]);
1198
+ const next = {};
1199
+ for (const [key, entry] of Object.entries(value)) {
1200
+ if (!allowedKeys.has(key)) {
1201
+ throw new Error(`Unknown ${section} config key: ${key}`);
1202
+ }
1203
+ if (typeof entry !== "string" || entry.trim() === "") {
1204
+ throw new Error(`${section}.${key} must be a non-empty string.`);
1205
+ }
1206
+ next[key] = entry.trim();
1207
+ }
1208
+ return next;
1209
+ }
1210
+ function formatError(error) {
1211
+ return error instanceof Error ? error.message : String(error);
1212
+ }
1213
+
1214
+ // src/runtime-config.ts
1215
+ var WIKI_GRAPH_PROVIDER_VALUES = [
1216
+ "anthropic",
1217
+ "google",
1218
+ "openai",
1219
+ "openai-compatible"
1220
+ ];
1221
+ var wikiGraphProviderSchema = import_zod.z.enum(WIKI_GRAPH_PROVIDER_VALUES);
1222
+ var inlineLLMConfigSchema = import_zod.z.object({
1223
+ apiKey: import_zod.z.string().min(1).optional(),
1224
+ baseURL: import_zod.z.string().min(1).optional(),
1225
+ baseUrl: import_zod.z.string().min(1).optional(),
1226
+ chatCompletionsUrl: import_zod.z.string().min(1).optional(),
1227
+ llm: import_zod.z.object({
1228
+ apiKey: import_zod.z.string().min(1).optional(),
1229
+ baseURL: import_zod.z.string().min(1).optional(),
1230
+ baseUrl: import_zod.z.string().min(1).optional(),
1231
+ chatCompletionsUrl: import_zod.z.string().min(1).optional(),
1232
+ model: import_zod.z.string().min(1).optional(),
1233
+ name: import_zod.z.string().min(1).optional(),
1234
+ provider: wikiGraphProviderSchema.optional()
1235
+ }).optional(),
1236
+ model: import_zod.z.string().min(1).optional(),
1237
+ name: import_zod.z.string().min(1).optional(),
1238
+ provider: wikiGraphProviderSchema.optional()
1239
+ });
1240
+ async function loadWikiGraphRuntimeConfig(options) {
1241
+ const [embedding, localLLM, concurrent, wikimedia, wikispine] = await Promise.all([
1242
+ readLocalConfigSection("embeddings"),
1243
+ readLocalConfigSection("llm"),
1244
+ readLocalConfigSection("concurrent"),
1245
+ readLocalConfigSection("wikimedia"),
1246
+ readLocalConfigSection("wikispine")
1247
+ ]);
1248
+ const inlineLLMConfig = options?.llmJSON === void 0 ? void 0 : parseInlineLLMConfig(options.llmJSON);
1249
+ const llm = createLLMConfig({
1250
+ apiKey: firstDefined(inlineLLMConfig?.apiKey, readString(localLLM.apiKey)),
1251
+ baseURL: firstDefined(
1252
+ inlineLLMConfig?.baseURL,
1253
+ readString(localLLM.baseURL)
1254
+ ),
1255
+ model: firstDefined(inlineLLMConfig?.model, readString(localLLM.model)),
1256
+ name: firstDefined(inlineLLMConfig?.name, readString(localLLM.name)),
1257
+ provider: firstDefined(
1258
+ inlineLLMConfig?.provider,
1259
+ readProvider(localLLM.provider)
1260
+ )
1261
+ });
1262
+ const requestConcurrent = readPositiveInteger(concurrent.request);
1263
+ const jobConcurrent = readPositiveInteger(concurrent.job);
1264
+ const envPolicy = getWikiGraphSDKRuntimeContext().envPolicy;
1265
+ const wikispineConfig = resolveWikispineConfig(wikispine, envPolicy);
1266
+ const wikimediaConfig = createWikimediaConfig(wikimedia, envPolicy);
1267
+ const embeddingConfig = createEmbeddingConfig(embedding);
1268
+ return {
1269
+ ...jobConcurrent === void 0 && requestConcurrent === void 0 ? {} : {
1270
+ concurrent: {
1271
+ ...jobConcurrent === void 0 ? {} : { job: jobConcurrent },
1272
+ ...requestConcurrent === void 0 ? {} : { request: requestConcurrent }
1273
+ }
1274
+ },
1275
+ ...embeddingConfig === void 0 ? {} : { embedding: embeddingConfig },
1276
+ ...llm === void 0 ? {} : { llm },
1277
+ ...wikimediaConfig === void 0 ? {} : { wikimedia: wikimediaConfig },
1278
+ ...wikispineConfig === void 0 ? {} : { wikispine: wikispineConfig }
1279
+ };
1280
+ }
1281
+ function createWikimediaConfig(input, envPolicy) {
1282
+ const endpoint = readString(input?.endpoint);
1283
+ const token = readString(input?.token);
1284
+ if (endpoint === void 0 && token === void 0) return void 0;
1285
+ if (token === void 0) {
1286
+ throw new Error(
1287
+ "Remote Wikimedia access requires an API key. Configure `wikg://local/config/wikimedia` token with `put token --secret`."
1288
+ );
1289
+ }
1290
+ return {
1291
+ endpoint: endpoint ?? resolveHostedProviderEndpoint("wg-wikimedia", envPolicy),
1292
+ token
1293
+ };
1294
+ }
1295
+ function createEmbeddingConfig(input) {
1296
+ const value = input ?? {};
1297
+ const provider = readEmbeddingProvider(value.provider);
1298
+ const apiKey = readString(value.apiKey);
1299
+ const baseURL = readString(value.baseURL);
1300
+ const dimensions = typeof value.dimensions === "number" && Number.isInteger(value.dimensions) && value.dimensions > 0 ? value.dimensions : void 0;
1301
+ const model = readString(value.model);
1302
+ const name = readString(value.name);
1303
+ if (provider === void 0 || model === void 0 || provider === "openai-compatible" && baseURL === void 0 || provider === "openai" && baseURL !== void 0) {
1304
+ return void 0;
1305
+ }
1306
+ return {
1307
+ ...apiKey === void 0 ? {} : { apiKey },
1308
+ ...baseURL === void 0 ? {} : { baseURL },
1309
+ ...dimensions === void 0 ? {} : { dimensions },
1310
+ ...model === void 0 ? {} : { model },
1311
+ ...name === void 0 ? {} : { name },
1312
+ ...provider === void 0 ? {} : { provider }
1313
+ };
1314
+ }
1315
+ function parseInlineLLMConfig(value) {
1316
+ const normalized = normalizeString(value);
1317
+ if (normalized === void 0) {
1318
+ throw new Error("--llm must be a non-empty JSON object.");
1319
+ }
1320
+ let parsedJson;
1321
+ try {
1322
+ parsedJson = JSON.parse(normalized);
1323
+ } catch (error) {
1324
+ throw new Error(`Invalid --llm JSON: ${formatError2(error)}`);
1325
+ }
1326
+ const parsed = inlineLLMConfigSchema.safeParse(parsedJson);
1327
+ if (!parsed.success) {
1328
+ throw new Error(
1329
+ `Invalid --llm config: ${parsed.error.issues.map((issue) => `${issue.path.join(".") || "<root>"}: ${issue.message}`).join("; ")}`
1330
+ );
1331
+ }
1332
+ const input = parsed.data.llm ?? parsed.data;
1333
+ const provider = input.provider ?? inferInlineProvider(input);
1334
+ const baseURL = resolveInlineBaseURL(input);
1335
+ const config = createLLMConfig({
1336
+ apiKey: input.apiKey,
1337
+ baseURL,
1338
+ model: input.model,
1339
+ name: input.name,
1340
+ provider
1341
+ });
1342
+ if (config === void 0) {
1343
+ throw new Error("--llm must contain at least one supported LLM field.");
1344
+ }
1345
+ return config;
1346
+ }
1347
+ function inferInlineProvider(input) {
1348
+ if (input.provider === void 0 && (input.baseURL !== void 0 || input.baseUrl !== void 0 || input.chatCompletionsUrl !== void 0)) {
1349
+ return "openai-compatible";
1350
+ }
1351
+ return input.provider;
1352
+ }
1353
+ function inferBaseURLFromChatCompletionsURL(input) {
1354
+ if (input.chatCompletionsUrl === void 0) {
1355
+ return void 0;
1356
+ }
1357
+ const suffix = "/chat/completions";
1358
+ if (!input.chatCompletionsUrl.endsWith(suffix)) {
1359
+ throw new Error(
1360
+ "--llm chatCompletionsUrl must end with /chat/completions when baseURL is not provided."
1361
+ );
1362
+ }
1363
+ return input.chatCompletionsUrl.slice(0, -suffix.length);
1364
+ }
1365
+ function resolveInlineBaseURL(input) {
1366
+ if (input.baseURL !== void 0) {
1367
+ return input.baseURL;
1368
+ }
1369
+ if (input.baseUrl !== void 0) {
1370
+ return input.baseUrl;
1371
+ }
1372
+ return inferBaseURLFromChatCompletionsURL(input);
1373
+ }
1374
+ function createLLMConfig(input) {
1375
+ if (input.apiKey === void 0 && input.baseURL === void 0 && input.model === void 0 && input.name === void 0 && input.provider === void 0) {
1376
+ return void 0;
1377
+ }
1378
+ return {
1379
+ ...input.apiKey === void 0 ? {} : { apiKey: input.apiKey },
1380
+ ...input.baseURL === void 0 ? {} : { baseURL: input.baseURL },
1381
+ ...input.model === void 0 ? {} : { model: input.model },
1382
+ ...input.name === void 0 ? {} : { name: input.name },
1383
+ ...input.provider === void 0 ? {} : { provider: input.provider }
1384
+ };
1385
+ }
1386
+ function readProvider(value) {
1387
+ if (typeof value !== "string") {
1388
+ return void 0;
1389
+ }
1390
+ const parsed = wikiGraphProviderSchema.safeParse(value);
1391
+ return parsed.success ? parsed.data : void 0;
1392
+ }
1393
+ function resolveWikispineConfig(value, envPolicy = getWikiGraphSDKRuntimeContext().envPolicy) {
1394
+ const provider = readWikispineProvider(value.provider);
1395
+ const endpoint = readString(value.endpoint);
1396
+ const token = readString(value.token);
1397
+ if (provider === void 0) {
1398
+ return void 0;
1399
+ }
1400
+ if (provider === "cli") return { provider };
1401
+ if (token === void 0) {
1402
+ throw new Error(
1403
+ "Remote WikiSpine access requires an API key. Configure `wikg://local/config/wikispine` token with `put token --secret`."
1404
+ );
1405
+ }
1406
+ return {
1407
+ endpoint: endpoint ?? resolveHostedProviderEndpoint("wikispine", envPolicy),
1408
+ provider,
1409
+ token
1410
+ };
1411
+ }
1412
+ function resolveHostedProviderEndpoint(scope, envPolicy = getWikiGraphSDKRuntimeContext().envPolicy) {
1413
+ const origin = envPolicy === "development" ? "https://pdf-craft-api.oomol.dev" : "https://api.pdfcraft.ai";
1414
+ return `${origin}/v1/${scope}`;
1415
+ }
1416
+ function readWikispineProvider(value) {
1417
+ return value === "cli" || value === "fetch" ? value : void 0;
1418
+ }
1419
+ function readEmbeddingProvider(value) {
1420
+ return value === "openai" || value === "openai-compatible" ? value : void 0;
1421
+ }
1422
+ function readString(value) {
1423
+ return typeof value === "string" ? value : void 0;
1424
+ }
1425
+ function readPositiveInteger(value) {
1426
+ return typeof value === "number" && Number.isInteger(value) && value > 0 ? value : void 0;
1427
+ }
1428
+ function firstDefined(first, second) {
1429
+ return first ?? second;
1430
+ }
1431
+ function normalizeString(value) {
1432
+ const normalized = value?.trim();
1433
+ return normalized === void 0 || normalized === "" ? void 0 : normalized;
1434
+ }
1435
+ function formatError2(error) {
1436
+ return error instanceof Error ? error.message : String(error);
1437
+ }
1438
+
1439
+ // src/runtime-path.ts
1440
+ var import_os2 = require("os");
1441
+ var import_path2 = require("path");
1442
+ function resolveWikiGraphRuntimePath(path) {
1443
+ const context = getWikiGraphSDKRuntimeContext();
1444
+ const environmentHome = context.env.HOME?.trim();
1445
+ const home = environmentHome === void 0 || environmentHome === "" ? (0, import_os2.homedir)() : environmentHome;
1446
+ if (path === "~") return home;
1447
+ if (path.startsWith("~/")) return (0, import_path2.resolve)(home, path.slice(2));
1448
+ return (0, import_path2.isAbsolute)(path) ? path : (0, import_path2.resolve)(context.cwd, path);
1449
+ }
1450
+
1451
+ // src/conversions.ts
1452
+ var WikiGraphConversionManager = class {
1453
+ #runtime;
1454
+ constructor(runtime) {
1455
+ this.#runtime = runtime;
1456
+ }
1457
+ async convert(options) {
1458
+ return await this.#runtime.run(async () => await this.#convert(options));
1459
+ }
1460
+ async #convert(options) {
1461
+ const targetStage = options.targetStage ?? "summarized";
1462
+ if (options.targetStage !== void 0 && options.output.format !== "wikg") {
1463
+ throw new Error("targetStage is only supported for wikg output.");
1464
+ }
1465
+ if (options.targetStage !== void 0 && options.input.format === "wikg") {
1466
+ throw new Error(
1467
+ "targetStage is only supported when creating a wikg archive."
1468
+ );
1469
+ }
1470
+ const requiresDigest = options.input.format !== "wikg";
1471
+ const requiresLLM = requiresDigest && targetStage !== "planned" && targetStage !== "sourced";
1472
+ const config = await loadWikiGraphRuntimeConfig({
1473
+ ...options.llmJSON === void 0 ? {} : { llmJSON: options.llmJSON }
1474
+ });
1475
+ const app = new import_wiki_graph_core2.WikiGraph({
1476
+ ...options.verbose === true ? { verbose: true } : {},
1477
+ ...requiresLLM ? { llm: buildWikiGraphLLMOptions(config) } : {}
1478
+ });
1479
+ const documentDirectory = await prepareDigestDirectory(
1480
+ options.digestDirectory,
1481
+ requiresDigest
1482
+ );
1483
+ const write = async (archive) => {
1484
+ await writeArchive(archive, options.output.path, options.output.format);
1485
+ };
1486
+ if (options.input.format === "wikg") {
1487
+ if (!("path" in options.input)) {
1488
+ throw new Error("wikg input requires a file path.");
1489
+ }
1490
+ await app.openSession(
1491
+ new NodeFile(resolveWikiGraphRuntimePath(options.input.path)),
1492
+ write
1493
+ );
1494
+ } else if ("stream" in options.input) {
1495
+ await app.digestTextStreamSession(
1496
+ {
1497
+ ...documentDirectory === void 0 ? {} : { documentDirectory },
1498
+ ...options.onProgress === void 0 ? {} : { onProgress: options.onProgress },
1499
+ sourceFormat: options.input.format,
1500
+ stream: options.input.stream,
1501
+ ...(options.prompt ?? config.prompt) === void 0 ? {} : { extractionPrompt: options.prompt ?? config.prompt },
1502
+ targetStage
1503
+ },
1504
+ write
1505
+ );
1506
+ } else {
1507
+ const digestOptions = {
1508
+ ...documentDirectory === void 0 ? {} : { documentDirectory },
1509
+ file: new NodeFile(resolveWikiGraphRuntimePath(options.input.path)),
1510
+ ...options.onProgress === void 0 ? {} : { onProgress: options.onProgress },
1511
+ ...(options.prompt ?? config.prompt) === void 0 ? {} : { extractionPrompt: options.prompt ?? config.prompt },
1512
+ targetStage
1513
+ };
1514
+ if (options.input.format === "epub") {
1515
+ await app.digestEpubSession(digestOptions, write);
1516
+ } else if (options.input.format === "markdown") {
1517
+ await app.digestMarkdownSession(digestOptions, write);
1518
+ } else {
1519
+ await app.digestTxtSession(digestOptions, write);
1520
+ }
1521
+ }
1522
+ return {
1523
+ inputFormat: options.input.format,
1524
+ outputFormat: options.output.format,
1525
+ outputPath: options.output.path
1526
+ };
1527
+ }
1528
+ };
1529
+ async function prepareDigestDirectory(path, required) {
1530
+ const normalized = path?.trim();
1531
+ if (!required || normalized === void 0 || normalized === "")
1532
+ return void 0;
1533
+ const resolved = resolveWikiGraphRuntimePath(normalized);
1534
+ await (0, import_promises2.rm)(resolved, { force: true, recursive: true });
1535
+ return new NodeDirectory(resolved);
1536
+ }
1537
+ async function writeArchive(archive, path, format) {
1538
+ const file = new NodeFile(resolveWikiGraphRuntimePath(path));
1539
+ if (format === "epub") await archive.exportEpub(file);
1540
+ else if (format === "wikg") await archive.saveAs(file);
1541
+ else await archive.exportText(file);
1542
+ }
1543
+
1544
+ // src/query-runtime.ts
1545
+ var import_wiki_graph_core3 = require("wiki-graph-core");
1546
+
1547
+ // src/embedding.ts
1548
+ var import_ai = require("ai");
1549
+ var import_openai2 = require("@ai-sdk/openai");
1550
+ var import_openai_compatible2 = require("@ai-sdk/openai-compatible");
1551
+ var EMBEDDING_BATCH_SIZE = 10;
1552
+ async function readWikiGraphEmbeddingConfig() {
1553
+ const config = await readLocalConfigSection("embeddings");
1554
+ return {
1555
+ ...typeof config.apiKey === "string" ? { apiKey: config.apiKey } : {},
1556
+ ...typeof config.baseURL === "string" ? { baseURL: config.baseURL } : {},
1557
+ ...typeof config.dimensions === "number" ? { dimensions: config.dimensions } : {},
1558
+ ...typeof config.model === "string" ? { model: config.model } : {},
1559
+ ...typeof config.name === "string" ? { name: config.name } : {},
1560
+ ...typeof config.provider === "string" ? { provider: parseEmbeddingProvider(config.provider) } : {}
1561
+ };
1562
+ }
1563
+ async function embedQueryText(value, config = {}) {
1564
+ const normalized = value.trim();
1565
+ if (normalized === "") {
1566
+ throw new Error("Query text cannot be empty.");
1567
+ }
1568
+ const provider = requireEmbeddingProvider(config.provider);
1569
+ const model = requireEmbeddingModel(config.model);
1570
+ const embeddingModel = createEmbeddingModel(provider, model, config);
1571
+ const providerOptions = createEmbeddingProviderOptions(provider, config);
1572
+ const result = await (0, import_ai.embed)({
1573
+ model: embeddingModel,
1574
+ ...providerOptions === void 0 ? {} : { providerOptions },
1575
+ value: normalized
1576
+ });
1577
+ return {
1578
+ dimensions: result.embedding.length,
1579
+ embedding: result.embedding,
1580
+ model,
1581
+ provider,
1582
+ ...result.usage.tokens === void 0 ? {} : { usage: { tokens: result.usage.tokens } }
1583
+ };
1584
+ }
1585
+ function buildSearchIndexEmbeddingProvider(config) {
1586
+ const provider = requireEmbeddingProvider(config.provider);
1587
+ const model = requireEmbeddingModel(config.model);
1588
+ const embeddingModel = createEmbeddingModel(provider, model, config);
1589
+ const providerOptions = createEmbeddingProviderOptions(provider, config);
1590
+ return {
1591
+ ...config.dimensions === void 0 ? {} : { dimensions: config.dimensions },
1592
+ identity: createEmbeddingIdentity(provider, model, config),
1593
+ model,
1594
+ embedTexts: async (texts, options) => {
1595
+ const embeddings = [];
1596
+ let tokens = 0;
1597
+ for (const batch of chunkTexts(texts, EMBEDDING_BATCH_SIZE)) {
1598
+ options?.signal?.throwIfAborted();
1599
+ const result = await (0, import_ai.embedMany)({
1600
+ ...options?.signal === void 0 ? {} : { abortSignal: options.signal },
1601
+ model: embeddingModel,
1602
+ ...providerOptions === void 0 ? {} : { providerOptions },
1603
+ values: [...batch]
1604
+ });
1605
+ embeddings.push(...result.embeddings);
1606
+ tokens += result.usage.tokens ?? 0;
1607
+ }
1608
+ return {
1609
+ embeddings,
1610
+ ...tokens === 0 ? {} : { tokens }
1611
+ };
1612
+ }
1613
+ };
1614
+ }
1615
+ function createEmbeddingIdentity(provider, model, config) {
1616
+ return JSON.stringify({
1617
+ ...config.baseURL === void 0 ? {} : { baseURL: config.baseURL },
1618
+ ...config.dimensions === void 0 ? {} : { dimensions: config.dimensions },
1619
+ model,
1620
+ provider
1621
+ });
1622
+ }
1623
+ function* chunkTexts(texts, size) {
1624
+ for (let index = 0; index < texts.length; index += size) {
1625
+ yield texts.slice(index, index + size);
1626
+ }
1627
+ }
1628
+ async function embedQueryTextWithLocalConfig(value) {
1629
+ return await embedQueryText(value, await readWikiGraphEmbeddingConfig());
1630
+ }
1631
+ function createEmbeddingModel(provider, model, config) {
1632
+ switch (provider) {
1633
+ case "openai": {
1634
+ if (config.baseURL !== void 0) {
1635
+ throw new Error(
1636
+ "openai does not accept embeddings.baseURL. Use openai-compatible for third-party OpenAI-style embeddings APIs."
1637
+ );
1638
+ }
1639
+ return (0, import_openai2.createOpenAI)({
1640
+ ...config.apiKey === void 0 ? {} : { apiKey: config.apiKey },
1641
+ ...config.name === void 0 ? {} : { name: config.name }
1642
+ }).embeddingModel(model);
1643
+ }
1644
+ case "openai-compatible": {
1645
+ if (config.baseURL === void 0) {
1646
+ throw new Error("openai-compatible requires embeddings.baseURL.");
1647
+ }
1648
+ return (0, import_openai_compatible2.createOpenAICompatible)({
1649
+ baseURL: config.baseURL,
1650
+ name: config.name ?? createOpenAICompatibleName2(config.baseURL),
1651
+ ...config.apiKey === void 0 ? {} : { apiKey: config.apiKey }
1652
+ }).embeddingModel(model);
1653
+ }
1654
+ }
1655
+ }
1656
+ function createEmbeddingProviderOptions(provider, config) {
1657
+ if (config.dimensions === void 0) {
1658
+ return void 0;
1659
+ }
1660
+ return {
1661
+ [getEmbeddingProviderOptionsName(provider, config)]: {
1662
+ dimensions: config.dimensions
1663
+ }
1664
+ };
1665
+ }
1666
+ function getEmbeddingProviderOptionsName(provider, _config) {
1667
+ if (provider === "openai") {
1668
+ return "openai";
1669
+ }
1670
+ return "openaiCompatible";
1671
+ }
1672
+ function requireEmbeddingProvider(provider) {
1673
+ if (provider !== void 0) {
1674
+ return provider;
1675
+ }
1676
+ throw new Error(
1677
+ "Missing embeddings configuration. Configure `wikg://local/config/embeddings` with provider and model."
1678
+ );
1679
+ }
1680
+ function requireEmbeddingModel(model) {
1681
+ if (model !== void 0) {
1682
+ return model;
1683
+ }
1684
+ throw new Error(
1685
+ "Missing embeddings.model. Configure `wikg://local/config/embeddings` before using Dense search."
1686
+ );
1687
+ }
1688
+ function parseEmbeddingProvider(value) {
1689
+ switch (value) {
1690
+ case "openai":
1691
+ case "openai-compatible":
1692
+ return value;
1693
+ default:
1694
+ throw new Error(
1695
+ `Invalid embeddings.provider: ${value}. Expected openai or openai-compatible.`
1696
+ );
1697
+ }
1698
+ }
1699
+ function createOpenAICompatibleName2(baseURL) {
1700
+ try {
1701
+ return new URL(baseURL).hostname;
1702
+ } catch {
1703
+ return "openai-compatible";
1704
+ }
1705
+ }
1706
+
1707
+ // src/query-runtime.ts
1708
+ async function createConfiguredEmbeddingProvider() {
1709
+ const config = await readWikiGraphEmbeddingConfig();
1710
+ if (config.provider === void 0 || config.model === void 0 || config.provider === "openai-compatible" && config.baseURL === void 0 || config.provider === "openai" && config.baseURL !== void 0) {
1711
+ return void 0;
1712
+ }
1713
+ return buildSearchIndexEmbeddingProvider(config);
1714
+ }
1715
+ async function withConfiguredWikimediaResolver(objectUri, operation) {
1716
+ if (!objectUri.endsWith("/wikipage")) return await operation({});
1717
+ const config = await loadWikiGraphRuntimeConfig();
1718
+ const resolver = await (0, import_wiki_graph_core3.openWikimediaResolver)(
1719
+ config.wikimedia === void 0 ? { kind: "local" } : {
1720
+ endpoint: config.wikimedia.endpoint,
1721
+ kind: "remote",
1722
+ token: config.wikimedia.token
1723
+ }
1724
+ );
1725
+ try {
1726
+ return await operation({ wikimediaResolver: resolver });
1727
+ } finally {
1728
+ await resolver.close();
1729
+ }
1730
+ }
1731
+
1732
+ // src/archive/inspect.ts
1733
+ var import_wiki_graph_core4 = require("wiki-graph-core");
1734
+
1735
+ // src/planning.ts
1736
+ var DEFAULT_GENERATION_JOB_CONCURRENCY = 3;
1737
+ var DEFAULT_GENERATION_REQUEST_CONCURRENCY = 6;
1738
+ function planGenerationTask(task, words, chapters, concurrent, model) {
1739
+ const profile = getGenerationPlanningProfile(task);
1740
+ const calls = Math.max(chapters, Math.ceil(words / profile.wordsPerCall));
1741
+ const effectiveConcurrency = Math.max(
1742
+ 1,
1743
+ task === "reading-graph" ? 1 : concurrent.request
1744
+ );
1745
+ const callBatches = Math.ceil(calls / effectiveConcurrency);
1746
+ const wordSeconds = words * profile.secondsPerWord;
1747
+ return {
1748
+ model,
1749
+ timeSeconds: {
1750
+ max: Math.ceil(callBatches * profile.maxSecondsPerCall + wordSeconds),
1751
+ min: Math.ceil(callBatches * profile.minSecondsPerCall + wordSeconds)
1752
+ },
1753
+ tokens: {
1754
+ cacheableInput: Math.ceil(words * profile.cacheableInputTokenPerWord),
1755
+ input: Math.ceil(words * profile.inputTokenPerWord),
1756
+ output: Math.ceil(words * profile.outputTokenPerWord)
1757
+ }
1758
+ };
1759
+ }
1760
+ function createGenerationPerformanceHints(input) {
1761
+ if (!input.hasJobWork) {
1762
+ return [];
1763
+ }
1764
+ const hints = [];
1765
+ const recommendedRequest = input.concurrent.request < 4 ? DEFAULT_GENERATION_REQUEST_CONCURRENCY : input.concurrent.request < 8 ? 8 : void 0;
1766
+ if (input.hasRequestWork && recommendedRequest !== void 0) {
1767
+ hints.push({
1768
+ configKey: "request",
1769
+ configValue: recommendedRequest,
1770
+ current: input.concurrent.request,
1771
+ kind: "request",
1772
+ message: "LLM request concurrency can often be higher. Use at least 4; 6-8 is usually faster when the provider allows it.",
1773
+ recommended: recommendedRequest
1774
+ });
1775
+ }
1776
+ if (input.concurrent.job < 4 && input.chapters > 1) {
1777
+ hints.push({
1778
+ configKey: "job",
1779
+ configValue: 4,
1780
+ current: input.concurrent.job,
1781
+ kind: "job",
1782
+ message: "Multiple chapters need work and job concurrency is below 4. Raising it lets chapter jobs run in parallel.",
1783
+ recommended: 4
1784
+ });
1785
+ }
1786
+ return hints;
1787
+ }
1788
+ function formatGenerationPlanningModel(llm) {
1789
+ if (llm?.model === void 0) {
1790
+ return "not configured";
1791
+ }
1792
+ return llm.provider === void 0 ? llm.model : `${llm.provider}/${llm.model}`;
1793
+ }
1794
+ function formatGenerationPlanningDuration(seconds) {
1795
+ if (seconds < 60) {
1796
+ return `${seconds}s`;
1797
+ }
1798
+ const minutes = Math.round(seconds / 60);
1799
+ if (minutes < 60) {
1800
+ return `${minutes}m`;
1801
+ }
1802
+ return `${Math.round(minutes / 60)}h`;
1803
+ }
1804
+ function getGenerationPlanningProfile(task) {
1805
+ switch (task) {
1806
+ case "index-fts":
1807
+ return {
1808
+ cacheableInputTokenPerWord: 0,
1809
+ inputTokenPerWord: 0,
1810
+ maxSecondsPerCall: 1,
1811
+ minSecondsPerCall: 0,
1812
+ outputTokenPerWord: 0,
1813
+ secondsPerWord: 5e-4,
1814
+ wordsPerCall: 1e5
1815
+ };
1816
+ case "index-embedding-source":
1817
+ case "index-embedding-summary":
1818
+ return {
1819
+ cacheableInputTokenPerWord: 0,
1820
+ inputTokenPerWord: 0,
1821
+ maxSecondsPerCall: 10,
1822
+ minSecondsPerCall: 1,
1823
+ outputTokenPerWord: 0,
1824
+ secondsPerWord: 2e-3,
1825
+ wordsPerCall: 1e4
1826
+ };
1827
+ case "knowledge-graph":
1828
+ return {
1829
+ cacheableInputTokenPerWord: 60,
1830
+ inputTokenPerWord: 80,
1831
+ maxSecondsPerCall: 35,
1832
+ minSecondsPerCall: 15,
1833
+ outputTokenPerWord: 25,
1834
+ secondsPerWord: 0.02,
1835
+ wordsPerCall: 35
1836
+ };
1837
+ case "reading-graph":
1838
+ return {
1839
+ cacheableInputTokenPerWord: 20,
1840
+ inputTokenPerWord: 25,
1841
+ maxSecondsPerCall: 45,
1842
+ minSecondsPerCall: 15,
1843
+ outputTokenPerWord: 4,
1844
+ secondsPerWord: 0.01,
1845
+ wordsPerCall: 160
1846
+ };
1847
+ case "reading-summary":
1848
+ return {
1849
+ cacheableInputTokenPerWord: 45,
1850
+ inputTokenPerWord: 55,
1851
+ maxSecondsPerCall: 25,
1852
+ minSecondsPerCall: 10,
1853
+ outputTokenPerWord: 4,
1854
+ secondsPerWord: 0.01,
1855
+ wordsPerCall: 110
1856
+ };
1857
+ }
1858
+ }
1859
+
1860
+ // src/archive/inspect.ts
1861
+ async function createWikiGraphArchiveInspectReport(document, options) {
1862
+ const { archiveUri } = options;
1863
+ const [
1864
+ chapters,
1865
+ summaryWords,
1866
+ indexCacheCurrent,
1867
+ config,
1868
+ ftsArtifactCoverage,
1869
+ sourceEmbeddingArtifactCoverage,
1870
+ summaryEmbeddingArtifactCoverage
1871
+ ] = await Promise.all([
1872
+ readInspectChapters(document, options.chapterId),
1873
+ readSummaryWords(document, options.chapterId),
1874
+ (0, import_wiki_graph_core4.isArchiveSearchIndexCurrent)(document),
1875
+ loadWikiGraphRuntimeConfig(),
1876
+ readIndexArtifactCoverage(document, "fts", options.chapterId),
1877
+ readIndexArtifactCoverage(document, "embedding-source", options.chapterId),
1878
+ readIndexArtifactCoverage(document, "embedding-summary", options.chapterId)
1879
+ ]);
1880
+ const selectedChapter = options.chapterId === void 0 ? void 0 : chapters[0];
1881
+ const scopeUri = selectedChapter === void 0 ? archiveUri : formatInspectLocatedChapterUri(archiveUri, selectedChapter);
1882
+ const concurrent = {
1883
+ job: config.concurrent?.job ?? DEFAULT_GENERATION_JOB_CONCURRENCY,
1884
+ request: config.concurrent?.request ?? DEFAULT_GENERATION_REQUEST_CONCURRENCY
1885
+ };
1886
+ const planningModel = formatGenerationPlanningModel(config.llm);
1887
+ const contentChapters = chapters.filter(
1888
+ (chapter) => chapter.stage !== "planned"
1889
+ );
1890
+ const sourceWords = sumWords(contentChapters);
1891
+ const readingGraphCovered = contentChapters.filter(
1892
+ (chapter) => chapter.readingGraphReady
1893
+ );
1894
+ const knowledgeGraphCovered = contentChapters.filter(
1895
+ (chapter) => chapter.knowledgeGraphReady
1896
+ );
1897
+ const summaryCovered = contentChapters.filter(
1898
+ (chapter) => chapter.summaryReady
1899
+ );
1900
+ const ftsArtifactCovered = filterChaptersByCurrentIndexArtifact(
1901
+ contentChapters,
1902
+ ftsArtifactCoverage
1903
+ );
1904
+ const sourceEmbeddingArtifactCovered = filterChaptersByCurrentIndexArtifact(
1905
+ contentChapters,
1906
+ sourceEmbeddingArtifactCoverage
1907
+ );
1908
+ const summaryEmbeddingArtifactCovered = filterChaptersByCurrentIndexArtifact(
1909
+ contentChapters,
1910
+ summaryEmbeddingArtifactCoverage
1911
+ );
1912
+ const queryBlockedContentChapters = contentChapters.filter(
1913
+ (chapter) => !ftsArtifactCovered.includes(chapter) && !sourceEmbeddingArtifactCovered.includes(chapter)
1914
+ );
1915
+ const queryBlockedChapters = queryBlockedContentChapters.map(
1916
+ (chapter) => createInspectChapterReference(archiveUri, chapter)
1917
+ );
1918
+ const queryReady = queryBlockedChapters.length === 0;
1919
+ const improvements = createInspectImprovements({
1920
+ concurrent,
1921
+ contentChapters,
1922
+ indexCacheCurrent,
1923
+ planningModel,
1924
+ queryBlockedChapters: queryBlockedContentChapters,
1925
+ summaryCovered,
1926
+ knowledgeGraphCovered,
1927
+ readingGraphCovered
1928
+ });
1929
+ const performanceHints = createGenerationPerformanceHints({
1930
+ chapters: Math.max(
1931
+ 0,
1932
+ ...improvements.map((improvement) => improvement.missingChapters ?? 0)
1933
+ ),
1934
+ concurrent,
1935
+ hasJobWork: improvements.some(
1936
+ (improvement) => improvement.planning !== void 0
1937
+ ),
1938
+ hasRequestWork: improvements.some(
1939
+ (improvement) => improvement.planning !== void 0
1940
+ )
1941
+ });
1942
+ return {
1943
+ archiveUri,
1944
+ uri: scopeUri,
1945
+ scope: options.chapterId === void 0 ? { type: "archive" } : { chapterId: options.chapterId, type: "chapter" },
1946
+ content: {
1947
+ chapters: {
1948
+ content: contentChapters.length,
1949
+ planned: chapters.length - contentChapters.length,
1950
+ total: chapters.length
1951
+ },
1952
+ sourceWords,
1953
+ summaryWords
1954
+ },
1955
+ localIndexCache: {
1956
+ current: indexCacheCurrent,
1957
+ ...!queryReady ? {
1958
+ blockedBy: "missing-search-artifacts",
1959
+ impact: "The cache cannot be built until every content chapter has a current FTS or source embedding artifact."
1960
+ } : indexCacheCurrent ? {} : {
1961
+ impact: "The next archive query may need to sync index cache first.",
1962
+ resource: "local CPU/disk time only; no provider calls."
1963
+ },
1964
+ status: !queryReady ? "blocked" : indexCacheCurrent ? "current" : "missing-or-outdated",
1965
+ syncAvailable: queryReady && !indexCacheCurrent
1966
+ },
1967
+ coverage: {
1968
+ ftsIndexArtifact: createInspectCoverage(
1969
+ ftsArtifactCovered,
1970
+ contentChapters
1971
+ ),
1972
+ knowledgeGraph: createInspectCoverage(
1973
+ knowledgeGraphCovered,
1974
+ contentChapters
1975
+ ),
1976
+ readingGraph: createInspectCoverage(readingGraphCovered, contentChapters),
1977
+ summary: createInspectCoverage(summaryCovered, contentChapters),
1978
+ sourceEmbeddingIndexArtifact: createInspectCoverage(
1979
+ sourceEmbeddingArtifactCovered,
1980
+ contentChapters
1981
+ ),
1982
+ summaryEmbeddingIndexArtifact: createInspectCoverage(
1983
+ summaryEmbeddingArtifactCovered,
1984
+ contentChapters
1985
+ )
1986
+ },
1987
+ query: {
1988
+ ready: queryReady
1989
+ },
1990
+ searchArtifacts: {
1991
+ blockedChapters: queryBlockedChapters,
1992
+ ...!queryReady ? {
1993
+ resource: "local CPU/disk time only; no provider calls."
1994
+ } : {},
1995
+ ready: queryReady,
1996
+ status: queryReady ? "ready" : "incomplete"
1997
+ },
1998
+ retrievalGuidance: formatRetrievalGuidance({
1999
+ indexCacheCurrent,
2000
+ knowledgeGraphCovered,
2001
+ readingGraphCovered,
2002
+ contentChapters,
2003
+ queryReady,
2004
+ sourceWords
2005
+ }),
2006
+ improvements,
2007
+ performanceHints,
2008
+ summaryCoverageNote: "Summary coverage means a current summary artifact exists for the chapter; it does not guarantee that every source topic is represented in the summary text."
2009
+ };
2010
+ }
2011
+ function createInspectChapterReference(archiveUri, chapter) {
2012
+ return {
2013
+ chapterId: chapter.chapterId,
2014
+ locatedUri: formatInspectLocatedChapterUri(archiveUri, chapter),
2015
+ title: chapter.title,
2016
+ uri: chapter.uri
2017
+ };
2018
+ }
2019
+ function formatInspectLocatedChapterUri(archiveUri, chapter) {
2020
+ return `${archiveUri.replace(/\/+$/u, "")}/${chapter.uri.replace(/^wikg:\/\//u, "")}`;
2021
+ }
2022
+ async function readInspectChapters(document, chapterId) {
2023
+ const chapters = chapterId === void 0 ? await (0, import_wiki_graph_core4.listChapters)(document) : (await (0, import_wiki_graph_core4.listChapters)(document)).filter(
2024
+ (chapter) => chapter.chapterId === chapterId
2025
+ );
2026
+ if (chapterId !== void 0 && chapters.length === 0) {
2027
+ throw new Error(`Chapter ${chapterId} does not exist.`);
2028
+ }
2029
+ return await Promise.all(
2030
+ chapters.map(async (chapter) => {
2031
+ const serial = await document.serials.getById(chapter.chapterId);
2032
+ return {
2033
+ ...chapter,
2034
+ knowledgeGraphReady: serial?.knowledgeGraphReady === true,
2035
+ readingGraphReady: serial?.topologyReady === true,
2036
+ summaryReady: chapter.stage === "summarized"
2037
+ };
2038
+ })
2039
+ );
2040
+ }
2041
+ async function readSummaryWords(document, chapterId) {
2042
+ return await document.readDatabase(
2043
+ async (database) => await database.queryOne(
2044
+ `
2045
+ SELECT COALESCE(SUM(words_count), 0) AS words
2046
+ FROM text_sentence_records
2047
+ WHERE kind = 2
2048
+ ${chapterId === void 0 ? "" : "AND chapter_id = ?"}
2049
+ `,
2050
+ chapterId === void 0 ? void 0 : [chapterId],
2051
+ (row) => Number(row.words)
2052
+ ) ?? 0
2053
+ );
2054
+ }
2055
+ async function readIndexArtifactCoverage(document, kind, chapterId) {
2056
+ const coverage = await document.indexArtifacts.listCoverage(kind);
2057
+ return chapterId === void 0 ? coverage : coverage.filter((item) => item.serialId === chapterId);
2058
+ }
2059
+ function filterChaptersByCurrentIndexArtifact(chapters, coverage) {
2060
+ const currentChapterIds = new Set(
2061
+ coverage.filter((item) => item.current).map((item) => item.serialId)
2062
+ );
2063
+ return chapters.filter((chapter) => currentChapterIds.has(chapter.chapterId));
2064
+ }
2065
+ function createInspectCoverage(covered, total) {
2066
+ const coveredWords = sumWords(covered);
2067
+ const totalWords = sumWords(total);
2068
+ return {
2069
+ coveredChapters: covered.length,
2070
+ coveredWords,
2071
+ percent: formatPercent(coveredWords, totalWords),
2072
+ totalChapters: total.length,
2073
+ totalWords
2074
+ };
2075
+ }
2076
+ function formatRetrievalGuidance(input) {
2077
+ if (input.sourceWords === 0 || input.contentChapters.length === 0) {
2078
+ return [
2079
+ "Source content: empty.",
2080
+ "Add source text before using query coverage or graph-based retrieval."
2081
+ ];
2082
+ }
2083
+ const lines = [
2084
+ input.queryReady ? `Query: available; local index cache is ${input.indexCacheCurrent ? "current" : "missing or outdated and can be synced from artifacts"}.` : "Query: unavailable until every content chapter has a current FTS or source embedding artifact."
2085
+ ];
2086
+ lines.push(
2087
+ formatObjectSearchGuidance(
2088
+ "Reading Graph object retrieval",
2089
+ input.readingGraphCovered,
2090
+ input.contentChapters
2091
+ )
2092
+ );
2093
+ lines.push(
2094
+ formatObjectSearchGuidance(
2095
+ "Entity/triple retrieval",
2096
+ input.knowledgeGraphCovered,
2097
+ input.contentChapters
2098
+ )
2099
+ );
2100
+ return lines;
2101
+ }
2102
+ function formatObjectSearchGuidance(label, covered, total) {
2103
+ const coveredWords = sumWords(covered);
2104
+ const totalWords = sumWords(total);
2105
+ const ratio = totalWords === 0 ? 0 : coveredWords / totalWords;
2106
+ const coverage = formatPercent(coveredWords, totalWords);
2107
+ if (ratio >= 1) {
2108
+ return `${label}: covers all source content; it can represent the full scope for object retrieval.`;
2109
+ }
2110
+ if (ratio >= 0.9) {
2111
+ return `${label}: covers ${coverage}; use it as the main path, but source --query is needed for uncovered content.`;
2112
+ }
2113
+ if (ratio >= 0.5) {
2114
+ return `${label}: covers ${coverage}; use it as leads, not as a full-scope substitute for source --query.`;
2115
+ }
2116
+ return `${label}: covers ${coverage}; build missing graph coverage before relying on object retrieval.`;
2117
+ }
2118
+ function createInspectImprovements(input) {
2119
+ const improvements = [];
2120
+ if (input.contentChapters.length === 0) {
2121
+ improvements.push({
2122
+ kind: "add-source",
2123
+ recommendation: "No source content is available yet; add or import source text before graph or summary generation.",
2124
+ title: "Add source content"
2125
+ });
2126
+ return improvements;
2127
+ }
2128
+ if (input.queryBlockedChapters.length > 0) {
2129
+ improvements.push({
2130
+ kind: "build",
2131
+ missingChapters: input.queryBlockedChapters.length,
2132
+ missingWords: sumWords(input.queryBlockedChapters),
2133
+ recommendation: "Build local FTS artifacts to make this scope queryable. This does not call a provider.",
2134
+ task: "index-fts",
2135
+ title: "Enable query with local FTS"
2136
+ });
2137
+ } else if (!input.indexCacheCurrent) {
2138
+ improvements.push({
2139
+ kind: "sync-index",
2140
+ recommendation: "Sync the index cache from chapter index artifacts so query starts without a lazy cache rebuild.",
2141
+ title: "Sync index cache"
2142
+ });
2143
+ }
2144
+ improvements.push(
2145
+ ...createGraphImprovement({
2146
+ concurrent: input.concurrent,
2147
+ covered: input.readingGraphCovered,
2148
+ planningModel: input.planningModel,
2149
+ task: "reading-graph",
2150
+ title: "Complete Reading Graph coverage",
2151
+ total: input.contentChapters
2152
+ })
2153
+ );
2154
+ improvements.push(
2155
+ ...createGraphImprovement({
2156
+ concurrent: input.concurrent,
2157
+ covered: input.knowledgeGraphCovered,
2158
+ planningModel: input.planningModel,
2159
+ task: "knowledge-graph",
2160
+ title: "Complete Knowledge Graph coverage",
2161
+ total: input.contentChapters
2162
+ })
2163
+ );
2164
+ improvements.push(
2165
+ ...createGraphImprovement({
2166
+ concurrent: input.concurrent,
2167
+ covered: input.summaryCovered,
2168
+ planningModel: input.planningModel,
2169
+ task: "reading-summary",
2170
+ title: "Complete Summary coverage",
2171
+ total: input.contentChapters
2172
+ })
2173
+ );
2174
+ return improvements;
2175
+ }
2176
+ function createGraphImprovement(input) {
2177
+ const coveredIds = new Set(input.covered.map((chapter) => chapter.chapterId));
2178
+ const missing = input.total.filter(
2179
+ (chapter) => !coveredIds.has(chapter.chapterId)
2180
+ );
2181
+ if (missing.length === 0) {
2182
+ return [];
2183
+ }
2184
+ const missingWords = sumWords(missing);
2185
+ return [
2186
+ {
2187
+ kind: "build",
2188
+ missingChapters: missing.length,
2189
+ missingWords,
2190
+ planning: planGenerationTask(
2191
+ input.task,
2192
+ missingWords,
2193
+ missing.length,
2194
+ input.concurrent,
2195
+ input.planningModel
2196
+ ),
2197
+ recommendation: formatImprovementRecommendation(
2198
+ input.covered,
2199
+ input.total,
2200
+ missingWords
2201
+ ),
2202
+ task: input.task,
2203
+ title: input.title
2204
+ }
2205
+ ];
2206
+ }
2207
+ function formatImprovementRecommendation(covered, total, missingWords) {
2208
+ const totalWords = sumWords(total);
2209
+ const coveredWords = sumWords(covered);
2210
+ if (totalWords === 0) {
2211
+ return "No source content is available.";
2212
+ }
2213
+ if (total.length === 1 || coveredWords / totalWords >= 0.9 || missingWords <= 1e3) {
2214
+ return "Queue the full scope to finish the remaining gap.";
2215
+ }
2216
+ return "Queue selected chapters first if only part of the scope matters.";
2217
+ }
2218
+ function sumWords(chapters) {
2219
+ return chapters.reduce((total, chapter) => total + chapter.words, 0);
2220
+ }
2221
+ function formatPercent(numerator, denominator) {
2222
+ if (denominator === 0) {
2223
+ return "n/a";
2224
+ }
2225
+ const percent = numerator / denominator * 100;
2226
+ return `${Number.isInteger(percent) ? percent.toFixed(0) : percent.toFixed(1)}%`;
2227
+ }
2228
+
2229
+ // src/archive/archive.ts
2230
+ var WikiGraphArchiveExistsError = class extends Error {
2231
+ code = "WIKI_GRAPH_ARCHIVE_EXISTS";
2232
+ path;
2233
+ constructor(path) {
2234
+ super(`Archive already exists: ${path}`);
2235
+ this.name = "WikiGraphArchiveExistsError";
2236
+ this.path = path;
2237
+ }
2238
+ };
2239
+ var WikiGraphArchiveManager = class {
2240
+ #runtime;
2241
+ constructor(runtime) {
2242
+ this.#runtime = runtime;
2243
+ }
2244
+ async create(options) {
2245
+ return await this.#runtime.run(async () => {
2246
+ const path = resolveWikiGraphRuntimePath(options.path);
2247
+ if (options.replace !== true && await nodePathExists(path)) {
2248
+ throw new WikiGraphArchiveExistsError(path);
2249
+ }
2250
+ const outputPath = (0, import_path3.join)(
2251
+ (0, import_path3.dirname)(path),
2252
+ `.${(0, import_path3.basename)(path)}.${(0, import_crypto.randomUUID)()}.tmp.wikg`
2253
+ );
2254
+ try {
2255
+ if (options.importPath === void 0) {
2256
+ await createEmptyArchiveFile(outputPath);
2257
+ } else {
2258
+ await new WikiGraphConversionManager(this.#runtime).convert({
2259
+ input: { format: "epub", path: options.importPath },
2260
+ ...options.onProgress === void 0 ? {} : { onProgress: options.onProgress },
2261
+ output: { format: "wikg", path: outputPath },
2262
+ targetStage: "sourced"
2263
+ });
2264
+ }
2265
+ if (options.replace === true) {
2266
+ await (0, import_promises3.rename)(outputPath, path);
2267
+ } else {
2268
+ try {
2269
+ await (0, import_promises3.link)(outputPath, path);
2270
+ } catch (error) {
2271
+ if (isNodeEEXISTError(error)) {
2272
+ throw new WikiGraphArchiveExistsError(path);
2273
+ }
2274
+ throw error;
2275
+ }
2276
+ }
2277
+ return { locatedUri: (0, import_wiki_graph_core5.formatLocatedWikiGraphUri)(path), path };
2278
+ } finally {
2279
+ await (0, import_promises3.rm)(outputPath, { force: true, recursive: true });
2280
+ }
2281
+ }, options.signal);
2282
+ }
2283
+ async open(uriOrPath, options = {}) {
2284
+ const location = await this.#runtime.run(
2285
+ async () => await resolveWikiGraphArchiveLocation(uriOrPath),
2286
+ options.signal
2287
+ );
2288
+ return new WikiGraphArchiveHandle(this.#runtime, location);
2289
+ }
2290
+ };
2291
+ var WikiGraphArchiveHandle = class {
2292
+ #runtime;
2293
+ #location;
2294
+ constructor(runtime, location) {
2295
+ this.#runtime = runtime;
2296
+ this.#location = location;
2297
+ }
2298
+ get archiveKey() {
2299
+ return this.#location.archiveKey;
2300
+ }
2301
+ get indexScope() {
2302
+ return this.#location.indexScope;
2303
+ }
2304
+ get locatedUri() {
2305
+ return this.#location.locatedUri;
2306
+ }
2307
+ get objectUri() {
2308
+ return (0, import_wiki_graph_core5.parseLocatedWikiGraphUri)(this.#location.locatedUri).objectUri ?? "wikg://";
2309
+ }
2310
+ get path() {
2311
+ return this.#location.archivePath;
2312
+ }
2313
+ async inspect(options = {}) {
2314
+ return await this.#readDocument(
2315
+ async (document) => await createWikiGraphArchiveInspectReport(document, {
2316
+ archiveUri: this.#location.publicArchiveUri ?? this.locatedUri,
2317
+ ...options.chapterId === void 0 ? {} : { chapterId: options.chapterId }
2318
+ }),
2319
+ options
2320
+ );
2321
+ }
2322
+ async search(query, options = {}) {
2323
+ const embeddingProvider = await this.#runtime.run(
2324
+ createConfiguredEmbeddingProvider,
2325
+ options.signal
2326
+ );
2327
+ return await this.#writeDocument(
2328
+ async (document) => {
2329
+ const chapters = await this.#resolveQueryChapters(document, options);
2330
+ await ensureArchiveSearchIndex(document, {
2331
+ ...chapters === void 0 ? {} : { chapters },
2332
+ ...embeddingProvider === void 0 ? {} : { embeddingProvider }
2333
+ });
2334
+ return await (0, import_wiki_graph_core5.findArchiveObjects)(document, query, {
2335
+ ...withoutOperationAndScope(options),
2336
+ archiveKey: this.archiveKey,
2337
+ ...chapters === void 0 ? {} : { chapters },
2338
+ ...embeddingProvider === void 0 ? {} : { embeddingProvider }
2339
+ });
2340
+ },
2341
+ {
2342
+ searchIndexWritebackPolicy: "cache",
2343
+ ...options.signal === void 0 ? {} : { signal: options.signal }
2344
+ }
2345
+ );
2346
+ }
2347
+ async list(options = {}) {
2348
+ return await this.#readDocument(async (document) => {
2349
+ if ((0, import_wiki_graph_core5.isSourceLocatorScopeUri)(this.objectUri)) {
2350
+ return await (0, import_wiki_graph_core5.listArchiveSourceLocators)(document, this.objectUri, {
2351
+ ...options.cursor === void 0 ? {} : { cursor: options.cursor },
2352
+ ...options.limit === void 0 ? {} : { limit: options.limit }
2353
+ });
2354
+ }
2355
+ const chapters = await this.#resolveScope(document, options);
2356
+ return await (0, import_wiki_graph_core5.listArchiveCollection)(document, {
2357
+ ...withoutOperationAndScope(options),
2358
+ ...chapters === void 0 ? {} : { chapters }
2359
+ });
2360
+ }, options);
2361
+ }
2362
+ async ensureSearchIndex(options = {}) {
2363
+ const embeddingProvider = await this.#runtime.run(
2364
+ createConfiguredEmbeddingProvider,
2365
+ options.signal
2366
+ );
2367
+ await this.#writeDocument(
2368
+ async (document) => {
2369
+ const chapters = await this.#resolveScope(document, options);
2370
+ await ensureArchiveSearchIndex(document, {
2371
+ ...chapters === void 0 ? {} : { chapters },
2372
+ ...embeddingProvider === void 0 ? {} : { embeddingProvider }
2373
+ });
2374
+ },
2375
+ {
2376
+ searchIndexWritebackPolicy: "cache",
2377
+ ...options.signal === void 0 ? {} : { signal: options.signal }
2378
+ }
2379
+ );
2380
+ }
2381
+ async getSearchIndexStatus(options = {}) {
2382
+ return await this.#readDocument(
2383
+ async (document) => ({
2384
+ capabilities: await (0, import_wiki_graph_core5.readSearchIndexCapabilityStatus)(document),
2385
+ current: await (0, import_wiki_graph_core5.isArchiveSearchIndexCurrent)(document)
2386
+ }),
2387
+ options
2388
+ );
2389
+ }
2390
+ async syncSearchIndex(options = {}) {
2391
+ const rebuilt = await this.#writeDocument(
2392
+ async (document) => {
2393
+ const scope = options.skipUnindexed === true ? { chapters: await (0, import_wiki_graph_core5.listArchiveQueryableChapterIds)(document) } : {};
2394
+ if (scope.chapters?.length === 0) {
2395
+ throw new Error(
2396
+ "Wiki Graph index cache is not ready. No chapters have a current FTS artifact or source embedding artifact."
2397
+ );
2398
+ }
2399
+ if (await (0, import_wiki_graph_core5.isArchiveSearchIndexCurrent)(document, scope)) return false;
2400
+ await (0, import_wiki_graph_core5.rebuildArchiveSearchIndex)(document, options.onProgress, scope);
2401
+ return true;
2402
+ },
2403
+ {
2404
+ searchIndexWritebackPolicy: "cache",
2405
+ ...options.signal === void 0 ? {} : { signal: options.signal }
2406
+ }
2407
+ );
2408
+ await this.#runtime.run(
2409
+ async () => await (0, import_wiki_graph_core5.deleteArchiveSearchSessions)(this.path),
2410
+ options.signal
2411
+ );
2412
+ return { rebuilt };
2413
+ }
2414
+ async cleanSearchIndex(options = {}) {
2415
+ const status = await this.#writeDocument(
2416
+ async (document) => {
2417
+ await document.deleteSearchIndexDatabase();
2418
+ return {
2419
+ capabilities: {
2420
+ dense: { current: false },
2421
+ indexes: "missing"
2422
+ },
2423
+ current: false
2424
+ };
2425
+ },
2426
+ {
2427
+ searchIndexWritebackPolicy: "cache",
2428
+ ...options.signal === void 0 ? {} : { signal: options.signal }
2429
+ }
2430
+ );
2431
+ await this.#runtime.run(
2432
+ async () => await (0, import_wiki_graph_core5.deleteArchiveSearchSessions)(this.path),
2433
+ options.signal
2434
+ );
2435
+ return status;
2436
+ }
2437
+ async listChapters(options = {}) {
2438
+ return await this.#readDocument(
2439
+ async (document) => await (0, import_wiki_graph_core5.listChapters)(document),
2440
+ options
2441
+ );
2442
+ }
2443
+ async getChapter(path, options = {}) {
2444
+ return await this.#readDocument(async (document) => {
2445
+ const chapterId = await (0, import_wiki_graph_core5.resolveChapterPathReadonly)(document, path);
2446
+ return requireChapter(await (0, import_wiki_graph_core5.listChapters)(document), chapterId);
2447
+ }, options);
2448
+ }
2449
+ async addChapter(options) {
2450
+ return await this.#writeDocument(async (document) => {
2451
+ const parentChapterId = await resolveOptionalChapterPath(
2452
+ document,
2453
+ options.parentPath
2454
+ );
2455
+ await (0, import_wiki_graph_core5.assertNoActiveBuildJobConflicts)({
2456
+ archive: this.#location.archiveFile,
2457
+ operation: "Adding chapter",
2458
+ scope: { kind: "archive" }
2459
+ });
2460
+ let details = await (0, import_wiki_graph_core5.addChapter)(document, {
2461
+ ...parentChapterId === void 0 ? {} : { parentChapterId },
2462
+ ...options.title === void 0 ? {} : { title: options.title }
2463
+ });
2464
+ if (options.source !== void 0) {
2465
+ details = await (0, import_wiki_graph_core5.setChapterSource)(
2466
+ document,
2467
+ details.chapterId,
2468
+ asTextStream(options.source)
2469
+ );
2470
+ }
2471
+ return details;
2472
+ });
2473
+ }
2474
+ async getChapterArtifact(path, kind) {
2475
+ return await this.#readDocument(async (document) => {
2476
+ const chapterId = await (0, import_wiki_graph_core5.resolveChapterPathReadonly)(document, path);
2477
+ const [artifact, revision] = await Promise.all([
2478
+ document.indexArtifacts.get(chapterId, kind),
2479
+ document.serials.getRevision(chapterId)
2480
+ ]);
2481
+ return {
2482
+ ...artifact === void 0 ? {} : {
2483
+ artifact: {
2484
+ createdAt: artifact.createdAt,
2485
+ metadata: artifact.metadata,
2486
+ sourceRevision: artifact.sourceRevision
2487
+ }
2488
+ },
2489
+ chapterId,
2490
+ current: artifact?.sourceRevision === revision,
2491
+ kind,
2492
+ missing: artifact === void 0,
2493
+ revision
2494
+ };
2495
+ });
2496
+ }
2497
+ async deleteChapterArtifact(path, kind) {
2498
+ return await this.#writeDocument(async (document) => {
2499
+ const chapterId = await (0, import_wiki_graph_core5.resolveChapterPathReadonly)(document, path);
2500
+ await document.indexArtifacts.delete(chapterId, kind);
2501
+ return { chapterId, deleted: true, kind };
2502
+ });
2503
+ }
2504
+ async moveChapter(path, options) {
2505
+ return await this.#writeDocument(async (document) => {
2506
+ const chapterId = await (0, import_wiki_graph_core5.resolveChapterPathReadonly)(document, path);
2507
+ const [afterChapterId, beforeChapterId, parentChapterId] = await Promise.all([
2508
+ resolveOptionalChapterPath(document, options.afterPath),
2509
+ resolveOptionalChapterPath(document, options.beforePath),
2510
+ resolveOptionalChapterPath(document, options.parentPath)
2511
+ ]);
2512
+ await (0, import_wiki_graph_core5.assertNoActiveBuildJobConflicts)({
2513
+ archive: this.#location.archiveFile,
2514
+ operation: "Moving chapter",
2515
+ scope: { kind: "archive" }
2516
+ });
2517
+ return await (0, import_wiki_graph_core5.moveChapter)(document, chapterId, {
2518
+ ...afterChapterId === void 0 ? {} : { afterChapterId },
2519
+ ...beforeChapterId === void 0 ? {} : { beforeChapterId },
2520
+ ...parentChapterId === void 0 ? {} : { parentChapterId },
2521
+ ...options.first === void 0 ? {} : { first: options.first },
2522
+ ...options.last === void 0 ? {} : { last: options.last },
2523
+ ...options.root === void 0 ? {} : { root: options.root }
2524
+ });
2525
+ });
2526
+ }
2527
+ async removeChapter(path, recursive = false) {
2528
+ await this.#writeDocument(async (document) => {
2529
+ const chapterId = await (0, import_wiki_graph_core5.resolveChapterPathReadonly)(document, path);
2530
+ await (0, import_wiki_graph_core5.assertNoActiveBuildJobConflicts)({
2531
+ archive: this.#location.archiveFile,
2532
+ operation: "Removing chapter",
2533
+ scope: { kind: "archive" }
2534
+ });
2535
+ await (0, import_wiki_graph_core5.removeChapter)(document, chapterId, { recursive });
2536
+ });
2537
+ }
2538
+ async resetChapter(path, stage) {
2539
+ return await this.#writeDocument(async (document) => {
2540
+ const chapterId = await (0, import_wiki_graph_core5.resolveChapterPathReadonly)(document, path);
2541
+ await assertChapterResetAllowed(
2542
+ this.#location.archiveFile,
2543
+ chapterId,
2544
+ stage
2545
+ );
2546
+ return await (0, import_wiki_graph_core5.resetChapter)(document, chapterId, stage);
2547
+ });
2548
+ }
2549
+ async setChapterSource(path, source, options = {}) {
2550
+ return await this.#writeDocument(async (document) => {
2551
+ const chapterId = await (0, import_wiki_graph_core5.resolveChapterPathReadonly)(document, path);
2552
+ await (0, import_wiki_graph_core5.assertNoActiveBuildJobs)({
2553
+ archive: this.#location.archiveFile,
2554
+ chapterIds: [chapterId],
2555
+ operation: "Setting chapter source"
2556
+ });
2557
+ return await (0, import_wiki_graph_core5.setChapterSource)(
2558
+ document,
2559
+ chapterId,
2560
+ asTextStream(source),
2561
+ options
2562
+ );
2563
+ });
2564
+ }
2565
+ async setChapterSummary(path, summary) {
2566
+ return await this.#writeDocument(async (document) => {
2567
+ const chapterId = await (0, import_wiki_graph_core5.resolveChapterPathReadonly)(document, path);
2568
+ await (0, import_wiki_graph_core5.assertNoActiveBuildJobs)({
2569
+ archive: this.#location.archiveFile,
2570
+ chapterIds: [chapterId],
2571
+ operation: "Setting chapter summary",
2572
+ requiresTarget: "reading-summary"
2573
+ });
2574
+ return await (0, import_wiki_graph_core5.setChapterSummary)(document, chapterId, summary);
2575
+ });
2576
+ }
2577
+ async setChapterTitle(path, title) {
2578
+ return await this.#writeDocument(async (document) => {
2579
+ const chapterId = await (0, import_wiki_graph_core5.resolveChapterPathReadonly)(document, path);
2580
+ await (0, import_wiki_graph_core5.assertNoActiveBuildJobs)({
2581
+ archive: this.#location.archiveFile,
2582
+ chapterIds: [chapterId],
2583
+ operation: "Setting chapter title"
2584
+ });
2585
+ return await (0, import_wiki_graph_core5.setChapterTitle)(document, chapterId, title);
2586
+ });
2587
+ }
2588
+ async getChapterTree() {
2589
+ return await this.#readDocument(
2590
+ async (document) => await (0, import_wiki_graph_core5.getChapterTree)(document)
2591
+ );
2592
+ }
2593
+ async applyChapterTree(tree, options = {}) {
2594
+ return await this.#writeDocument(
2595
+ async (document) => {
2596
+ if (options.dryRun !== true) {
2597
+ await (0, import_wiki_graph_core5.assertNoActiveBuildJobConflicts)({
2598
+ archive: this.#location.archiveFile,
2599
+ operation: "Changing chapter tree",
2600
+ scope: { kind: "archive" }
2601
+ });
2602
+ }
2603
+ return await (0, import_wiki_graph_core5.applyChapterTree)(document, tree, {
2604
+ dryRun: options.dryRun ?? false
2605
+ });
2606
+ },
2607
+ { refreshLibraryIndex: options.dryRun !== true }
2608
+ );
2609
+ }
2610
+ async page(objectUri = this.objectUri, options = {}) {
2611
+ return await this.#runtime.run(
2612
+ async () => await withConfiguredWikimediaResolver(
2613
+ objectUri,
2614
+ async (wikimediaOptions) => await this.#readDocument(
2615
+ async (document) => await (0, import_wiki_graph_core5.readArchivePage)(document, objectUri, {
2616
+ ...withoutOperation(options),
2617
+ ...wikimediaOptions
2618
+ }),
2619
+ options
2620
+ )
2621
+ ),
2622
+ options.signal
2623
+ );
2624
+ }
2625
+ async related(objectUri = this.objectUri, options = {}) {
2626
+ if (options.query === void 0) {
2627
+ return await this.#readDocument(async (document) => {
2628
+ const chapters = await this.#resolveScope(document, {});
2629
+ return await (0, import_wiki_graph_core5.listRelatedArchiveObjects)(document, objectUri, {
2630
+ ...withoutOperation(options),
2631
+ ...chapters === void 0 ? {} : { chapters }
2632
+ });
2633
+ }, options);
2634
+ }
2635
+ const embeddingProvider = await this.#runtime.run(
2636
+ createConfiguredEmbeddingProvider,
2637
+ options.signal
2638
+ );
2639
+ return await this.#writeDocument(
2640
+ async (document) => {
2641
+ const chapters = await this.#resolveQueryChapters(document, options);
2642
+ await ensureArchiveSearchIndex(document, {
2643
+ ...chapters === void 0 ? {} : { chapters },
2644
+ ...embeddingProvider === void 0 ? {} : { embeddingProvider }
2645
+ });
2646
+ return await (0, import_wiki_graph_core5.listRelatedArchiveObjects)(document, objectUri, {
2647
+ ...withoutOperation(options),
2648
+ ...chapters === void 0 ? {} : { chapters }
2649
+ });
2650
+ },
2651
+ {
2652
+ searchIndexWritebackPolicy: "cache",
2653
+ ...options.signal === void 0 ? {} : { signal: options.signal }
2654
+ }
2655
+ );
2656
+ }
2657
+ async evidence(objectUri = this.objectUri, options = {}) {
2658
+ if (options.query === void 0) {
2659
+ return await this.#readDocument(async (document) => {
2660
+ const chapters = await this.#resolveScope(document, {});
2661
+ return await (0, import_wiki_graph_core5.listArchiveEvidence)(document, objectUri, {
2662
+ ...withoutOperation(options),
2663
+ ...chapters === void 0 ? {} : { chapters }
2664
+ });
2665
+ }, options);
2666
+ }
2667
+ const embeddingProvider = await this.#runtime.run(
2668
+ createConfiguredEmbeddingProvider,
2669
+ options.signal
2670
+ );
2671
+ return await this.#writeDocument(
2672
+ async (document) => {
2673
+ const chapters = await this.#resolveQueryChapters(document, options);
2674
+ await ensureArchiveSearchIndex(document, {
2675
+ ...chapters === void 0 ? {} : { chapters },
2676
+ ...embeddingProvider === void 0 ? {} : { embeddingProvider }
2677
+ });
2678
+ return await (0, import_wiki_graph_core5.listArchiveEvidence)(document, objectUri, {
2679
+ ...withoutOperation(options),
2680
+ ...chapters === void 0 ? {} : { chapters }
2681
+ });
2682
+ },
2683
+ {
2684
+ searchIndexWritebackPolicy: "cache",
2685
+ ...options.signal === void 0 ? {} : { signal: options.signal }
2686
+ }
2687
+ );
2688
+ }
2689
+ async pack(objectUri = this.objectUri, budget = 5e3, options = {}) {
2690
+ return await this.#readDocument(
2691
+ async (document) => await (0, import_wiki_graph_core5.packArchiveContext)(document, objectUri, budget),
2692
+ options
2693
+ );
2694
+ }
2695
+ async getMetadata(objectPath, options = {}) {
2696
+ return await this.#readDocument(
2697
+ async (document) => await document.metadata.getMap(objectPath),
2698
+ options
2699
+ );
2700
+ }
2701
+ async readBookMeta(options = {}) {
2702
+ return await this.#readDocument(
2703
+ async (document) => await document.readBookMeta(),
2704
+ options
2705
+ );
2706
+ }
2707
+ async replaceBookMeta(meta, options = {}) {
2708
+ return await this.#writeDocument(async (document) => {
2709
+ await document.replaceBookMeta(meta);
2710
+ return meta;
2711
+ }, options);
2712
+ }
2713
+ async readCover(options = {}) {
2714
+ const app = new import_wiki_graph_core5.WikiGraph({});
2715
+ return await this.#runtime.run(
2716
+ async () => await app.openSession(
2717
+ this.#location.archiveFile,
2718
+ async (archive) => await archive.readCover()
2719
+ ),
2720
+ options.signal
2721
+ );
2722
+ }
2723
+ async replaceMetadata(objectPath, value, options = {}) {
2724
+ return await this.#writeDocument(async (document) => {
2725
+ await document.metadata.replaceMap(
2726
+ parseObjectMetadataTarget(objectPath),
2727
+ value
2728
+ );
2729
+ return await document.metadata.getMap(objectPath);
2730
+ }, options);
2731
+ }
2732
+ async putMetadata(objectPath, key, value, options = {}) {
2733
+ return await this.#writeDocument(async (document) => {
2734
+ await document.metadata.put(
2735
+ parseObjectMetadataTarget(objectPath),
2736
+ key,
2737
+ value
2738
+ );
2739
+ return await document.metadata.getMap(objectPath);
2740
+ }, options);
2741
+ }
2742
+ async deleteMetadata(objectPath, key, options = {}) {
2743
+ return await this.#writeDocument(async (document) => {
2744
+ await document.metadata.deleteKey(objectPath, key);
2745
+ return await document.metadata.getMap(objectPath);
2746
+ }, options);
2747
+ }
2748
+ async clearMetadata(objectPath, options = {}) {
2749
+ return await this.#writeDocument(async (document) => {
2750
+ await document.metadata.clear(objectPath);
2751
+ return await document.metadata.getMap(objectPath);
2752
+ }, options);
2753
+ }
2754
+ async resolveScope(options = {}) {
2755
+ return await this.#readDocument(
2756
+ async (document) => await resolveArchiveScope(document, this.objectUri, options.depth),
2757
+ options
2758
+ );
2759
+ }
2760
+ async #readDocument(operation, options = {}) {
2761
+ return await this.#runtime.run(
2762
+ async () => await new import_wiki_graph_core5.WikiGraphArchiveFile(this.#location.archiveFile).readDocument(
2763
+ operation
2764
+ ),
2765
+ options.signal
2766
+ );
2767
+ }
2768
+ async #writeDocument(operation, options = {}) {
2769
+ return await this.#runtime.run(async () => {
2770
+ const result = await new import_wiki_graph_core5.WikiGraphArchiveFile(
2771
+ this.#location.archiveFile
2772
+ ).write(operation, {
2773
+ ...options.searchIndexWritebackPolicy === void 0 ? {} : {
2774
+ searchIndexWritebackPolicy: options.searchIndexWritebackPolicy
2775
+ }
2776
+ });
2777
+ if (options.refreshLibraryIndex !== false) {
2778
+ await this.#refreshLibraryIndex(options.onIndexSyncError);
2779
+ }
2780
+ return result;
2781
+ }, options.signal);
2782
+ }
2783
+ async #resolveScope(document, options) {
2784
+ if (options.chapters !== void 0) return options.chapters;
2785
+ return (await resolveArchiveScope(document, this.objectUri, options.depth))?.chapterIds;
2786
+ }
2787
+ async #resolveQueryChapters(document, options) {
2788
+ const chapters = await this.#resolveScope(document, options);
2789
+ if (options.skipUnindexed !== true) return chapters;
2790
+ const queryable = await (0, import_wiki_graph_core5.listArchiveQueryableChapterIds)(document, {
2791
+ ...chapters === void 0 ? {} : { chapters }
2792
+ });
2793
+ if (queryable.length === 0) {
2794
+ throw new Error(
2795
+ "Wiki Graph query is not ready. No chapters in this scope have a current FTS artifact or source embedding artifact."
2796
+ );
2797
+ }
2798
+ return queryable;
2799
+ }
2800
+ async #refreshLibraryIndex(onError) {
2801
+ if (this.#location.libraryDirtyTarget === void 0) return;
2802
+ if (this.#location.libraryArchiveTarget !== void 0) {
2803
+ await (0, import_wiki_graph_core5.finalizeWikiGraphLibraryArchiveWrite)({
2804
+ target: this.#location.libraryArchiveTarget
2805
+ });
2806
+ }
2807
+ try {
2808
+ const state = await (0, import_wiki_graph_core5.readWikiGraphLibraryIndexState)(
2809
+ this.#location.libraryDirtyTarget
2810
+ );
2811
+ if (state.status !== "missing") {
2812
+ await (0, import_wiki_graph_core5.rebuildWikiGraphLibraryIndex)(this.#location.libraryDirtyTarget);
2813
+ }
2814
+ } catch (error) {
2815
+ onError?.(error);
2816
+ }
2817
+ }
2818
+ };
2819
+ async function resolveWikiGraphArchiveLocation(uriOrPath) {
2820
+ if (!uriOrPath.startsWith("wikg://")) {
2821
+ return createPathLocation(resolveWikiGraphRuntimePath(uriOrPath));
2822
+ }
2823
+ const parsed = (0, import_wiki_graph_core5.parseLocatedWikiGraphUri)(uriOrPath);
2824
+ const portableArchiveLocator = parsed.archivePath ?? uriOrPath;
2825
+ const archiveLocator = portableArchiveLocator.startsWith("wikg://lib/") ? portableArchiveLocator : resolveWikiGraphRuntimePath(portableArchiveLocator);
2826
+ const libraryArchiveTarget = archiveLocator.startsWith("wikg://lib/") ? (0, import_wiki_graph_core5.parseWikiGraphLibraryUri)(archiveLocator) : void 0;
2827
+ const archiveFile = libraryArchiveTarget?.kind === "archive" ? await (0, import_wiki_graph_core5.resolveWikiGraphLibraryArchiveFile)(archiveLocator) : new NodeFile(archiveLocator);
2828
+ const archivePath = getNodeResourcePath(archiveFile);
2829
+ return {
2830
+ archiveFile,
2831
+ archiveKey: archivePath,
2832
+ archivePath,
2833
+ indexScope: { archiveKey: archivePath, archivePath, kind: "archive-index" },
2834
+ ...libraryArchiveTarget?.kind === "archive" ? {
2835
+ libraryArchiveTarget,
2836
+ libraryDirtyTarget: {
2837
+ isDefault: libraryArchiveTarget.isDefault,
2838
+ kind: "scope",
2839
+ ...libraryArchiveTarget.publicId === void 0 ? {} : { publicId: libraryArchiveTarget.publicId }
2840
+ },
2841
+ publicArchiveUri: portableArchiveLocator
2842
+ } : {},
2843
+ locatedUri: (0, import_wiki_graph_core5.formatLocatedWikiGraphUri)(archivePath, parsed.objectUri)
2844
+ };
2845
+ }
2846
+ async function ensureArchiveSearchIndex(document, options = {}) {
2847
+ const scope = options.chapters === void 0 ? {} : { chapters: options.chapters };
2848
+ if (await (0, import_wiki_graph_core5.isArchiveSearchIndexCurrent)(document, scope)) return;
2849
+ await (0, import_wiki_graph_core5.rebuildArchiveSearchIndex)(document, void 0, {
2850
+ ...scope,
2851
+ ...options.embeddingProvider === void 0 ? {} : { embeddingProvider: options.embeddingProvider }
2852
+ });
2853
+ }
2854
+ async function resolveArchiveScope(document, objectUri, depth) {
2855
+ const parsed = parseChapterScopePath(objectUri);
2856
+ if (parsed === void 0 || parsed.kind === "collection" && depth === void 0) {
2857
+ return void 0;
2858
+ }
2859
+ const chapters = await (0, import_wiki_graph_core5.listChapters)(document);
2860
+ const entries = parsed.kind === "collection" ? chapters.filter((chapter) => chapter.depth <= depth) : selectChapterSubtree(
2861
+ chapters,
2862
+ await (0, import_wiki_graph_core5.resolveChapterPathReadonly)(document, parsed.chapterPath),
2863
+ depth
2864
+ );
2865
+ return { chapterIds: entries.map((entry) => entry.chapterId), entries };
2866
+ }
2867
+ function parseChapterScopePath(objectUri) {
2868
+ if (objectUri === "wikg://chapter" || objectUri === "wikg://chapter/") {
2869
+ return { kind: "collection" };
2870
+ }
2871
+ const match = /^wikg:\/\/chapter\/([^/].*?)(?:\/)?$/u.exec(objectUri);
2872
+ return match?.[1] === void 0 ? void 0 : { chapterPath: decodeURIComponent(match[1]), kind: "chapter" };
2873
+ }
2874
+ function selectChapterSubtree(chapters, rootChapterId, depth) {
2875
+ const root = chapters.find((chapter) => chapter.chapterId === rootChapterId);
2876
+ if (root === void 0) {
2877
+ throw new Error(`Chapter ${rootChapterId} does not exist.`);
2878
+ }
2879
+ const prefix = `${root.path}/`;
2880
+ return chapters.filter(
2881
+ (chapter) => chapter.chapterId === root.chapterId || chapter.path.startsWith(prefix) && (depth === void 0 || chapter.depth - root.depth <= depth)
2882
+ );
2883
+ }
2884
+ function createPathLocation(path) {
2885
+ return {
2886
+ archiveFile: new NodeFile(path),
2887
+ archiveKey: path,
2888
+ archivePath: path,
2889
+ indexScope: { archiveKey: path, archivePath: path, kind: "archive-index" },
2890
+ locatedUri: (0, import_wiki_graph_core5.formatLocatedWikiGraphUri)(path)
2891
+ };
2892
+ }
2893
+ async function createEmptyArchiveFile(path) {
2894
+ const directoryPath = await (0, import_promises3.mkdtemp)(
2895
+ (0, import_path3.join)((0, import_os3.tmpdir)(), "wikigraph-archive-write-")
2896
+ );
2897
+ try {
2898
+ const directory = new NodeDirectory(directoryPath);
2899
+ const document = await import_wiki_graph_core5.DirectoryDocument.open(directory);
2900
+ try {
2901
+ await document.openSession(async (openedDocument) => {
2902
+ await openedDocument.writeToc({ items: [], version: import_wiki_graph_core5.TOC_FILE_VERSION });
2903
+ });
2904
+ } finally {
2905
+ await document.release();
2906
+ }
2907
+ await (0, import_wiki_graph_core5.writeWikgArchive)(directory, new NodeFile(path));
2908
+ } finally {
2909
+ await (0, import_promises3.rm)(directoryPath, { force: true, recursive: true });
2910
+ }
2911
+ }
2912
+ function isNodeEEXISTError(error) {
2913
+ return error instanceof Error && "code" in error && error.code === "EEXIST";
2914
+ }
2915
+ async function nodePathExists(path) {
2916
+ try {
2917
+ await (0, import_promises3.stat)(path);
2918
+ return true;
2919
+ } catch (error) {
2920
+ if (isNodeENOENTError(error)) return false;
2921
+ throw error;
2922
+ }
2923
+ }
2924
+ function isNodeENOENTError(error) {
2925
+ return error instanceof Error && "code" in error && error.code === "ENOENT";
2926
+ }
2927
+ function parseObjectMetadataTarget(objectPath) {
2928
+ if (objectPath === "") {
2929
+ return { kind: import_wiki_graph_core5.ObjectMetadataKind.Archive, objectPath };
2930
+ }
2931
+ const chapter = /^chapter\/([1-9][0-9]*)(?:\/.*)?$/u.exec(objectPath);
2932
+ if (chapter?.[1] !== void 0) {
2933
+ return {
2934
+ chapterId: Number(chapter[1]),
2935
+ kind: import_wiki_graph_core5.ObjectMetadataKind.Chapter,
2936
+ objectPath
2937
+ };
2938
+ }
2939
+ const chunk = /^chunk\/([1-9][0-9]*)$/u.exec(objectPath);
2940
+ if (chunk?.[1] !== void 0) {
2941
+ return {
2942
+ chunkId: Number(chunk[1]),
2943
+ kind: import_wiki_graph_core5.ObjectMetadataKind.Chunk,
2944
+ objectPath
2945
+ };
2946
+ }
2947
+ const entity = /^entity\/(Q[1-9][0-9]*)$/u.exec(objectPath);
2948
+ if (entity?.[1] !== void 0) {
2949
+ return {
2950
+ entityQid: entity[1],
2951
+ kind: import_wiki_graph_core5.ObjectMetadataKind.Entity,
2952
+ objectPath
2953
+ };
2954
+ }
2955
+ const triple = /^triple\/(Q[1-9][0-9]*)\/([^/]+)\/(Q[1-9][0-9]*)$/u.exec(
2956
+ objectPath
2957
+ );
2958
+ if (triple?.[1] !== void 0) {
2959
+ return {
2960
+ kind: import_wiki_graph_core5.ObjectMetadataKind.Triple,
2961
+ objectPath,
2962
+ tripleObjectQid: triple[3],
2963
+ triplePredicate: decodeURIComponent(triple[2]),
2964
+ tripleSubjectQid: triple[1]
2965
+ };
2966
+ }
2967
+ return { kind: import_wiki_graph_core5.ObjectMetadataKind.Object, objectPath };
2968
+ }
2969
+ function withoutOperation(options) {
2970
+ const { signal: _signal, ...rest } = options;
2971
+ return rest;
2972
+ }
2973
+ function withoutOperationAndScope(options) {
2974
+ const {
2975
+ chapters: _chapters,
2976
+ depth: _depth,
2977
+ signal: _signal,
2978
+ ...rest
2979
+ } = options;
2980
+ return rest;
2981
+ }
2982
+ async function resolveOptionalChapterPath(document, path) {
2983
+ return path === void 0 ? void 0 : await (0, import_wiki_graph_core5.resolveChapterPathReadonly)(document, path);
2984
+ }
2985
+ function requireChapter(chapters, chapterId) {
2986
+ const chapter = chapters.find((entry) => entry.chapterId === chapterId);
2987
+ if (chapter === void 0) {
2988
+ throw new Error(`Chapter internal id ${chapterId} does not exist.`);
2989
+ }
2990
+ return chapter;
2991
+ }
2992
+ function asTextStream(text) {
2993
+ return import_stream.Readable.from([text]);
2994
+ }
2995
+ async function assertChapterResetAllowed(archive, chapterId, stage) {
2996
+ if (stage === "planned") {
2997
+ await (0, import_wiki_graph_core5.assertNoActiveBuildJobs)({
2998
+ archive,
2999
+ chapterIds: [chapterId],
3000
+ operation: "Resetting chapter to planned"
3001
+ });
3002
+ return;
3003
+ }
3004
+ if (stage === "sourced") {
3005
+ await (0, import_wiki_graph_core5.assertNoActiveBuildJobs)({
3006
+ archive,
3007
+ chapterIds: [chapterId],
3008
+ operation: "Resetting chapter graph",
3009
+ requiresTarget: "reading-graph"
3010
+ });
3011
+ }
3012
+ await (0, import_wiki_graph_core5.assertNoActiveBuildJobs)({
3013
+ archive,
3014
+ chapterIds: [chapterId],
3015
+ operation: "Resetting chapter summary",
3016
+ requiresTarget: "reading-summary"
3017
+ });
3018
+ }
3019
+
3020
+ // src/wikispine.ts
3021
+ var import_child_process = require("child_process");
3022
+ var nodeWikispineCommandRunner = {
3023
+ run: async ({ args, command, input, onStdout, signal }) => await new Promise((resolve3, reject) => {
3024
+ signal?.throwIfAborted();
3025
+ const child = (0, import_child_process.spawn)(command, [...args], {
3026
+ stdio: ["pipe", "pipe", "pipe"]
3027
+ });
3028
+ const stderr = [];
3029
+ let settled = false;
3030
+ const settle = (operation) => {
3031
+ if (settled) return;
3032
+ settled = true;
3033
+ signal?.removeEventListener("abort", abort);
3034
+ operation();
3035
+ };
3036
+ const abort = () => {
3037
+ child.kill();
3038
+ settle(() => reject(abortError(signal)));
3039
+ };
3040
+ signal?.addEventListener("abort", abort, { once: true });
3041
+ child.stdout.setEncoding("utf8");
3042
+ child.stdout.on("data", (chunk) => {
3043
+ if (settled) return;
3044
+ try {
3045
+ onStdout(chunk);
3046
+ } catch (error) {
3047
+ child.kill();
3048
+ settle(
3049
+ () => reject(error instanceof Error ? error : new Error(String(error)))
3050
+ );
3051
+ }
3052
+ });
3053
+ child.stderr.on("data", (chunk) => stderr.push(chunk));
3054
+ child.on("error", (error) => {
3055
+ settle(
3056
+ () => reject(
3057
+ new Error(`Failed to start wikispine command: ${error.message}`)
3058
+ )
3059
+ );
3060
+ });
3061
+ child.on("close", (exitCode) => {
3062
+ settle(
3063
+ () => resolve3({
3064
+ exitCode,
3065
+ stderr: Buffer.concat(stderr).toString("utf8")
3066
+ })
3067
+ );
3068
+ });
3069
+ child.stdin.end(input);
3070
+ })
3071
+ };
3072
+ function abortError(signal) {
3073
+ return signal?.reason instanceof Error ? signal.reason : new Error("WikiSpine command aborted.");
3074
+ }
3075
+
3076
+ // src/config.ts
3077
+ var import_ai2 = require("ai");
3078
+ var import_wiki_graph_core6 = require("wiki-graph-core");
3079
+ var WikiGraphConfigManager = class {
3080
+ #runtime;
3081
+ constructor(runtime) {
3082
+ this.#runtime = runtime;
3083
+ }
3084
+ async get(section, options = {}) {
3085
+ const value = await this.#runtime.run(
3086
+ async () => await readLocalConfigSection(section)
3087
+ );
3088
+ return options.maskSecrets === true ? maskLocalConfigSection(section, value) : value;
3089
+ }
3090
+ async replace(section, value) {
3091
+ return await this.#runtime.run(
3092
+ async () => await replaceLocalConfigSection(section, value)
3093
+ );
3094
+ }
3095
+ async put(section, key, value) {
3096
+ return await this.#runtime.run(
3097
+ async () => await putLocalConfigValue(section, key, value)
3098
+ );
3099
+ }
3100
+ async delete(section, key) {
3101
+ return await this.#runtime.run(
3102
+ async () => await deleteLocalConfigValue(section, key)
3103
+ );
3104
+ }
3105
+ async clear(section) {
3106
+ return await this.#runtime.run(
3107
+ async () => await clearLocalConfigSection(section)
3108
+ );
3109
+ }
3110
+ async testLLM() {
3111
+ return await this.#runtime.run(async () => {
3112
+ const startedAt = Date.now();
3113
+ const llm = await readLocalConfigSection("llm");
3114
+ const config = {
3115
+ ...typeof llm.apiKey === "string" ? { apiKey: llm.apiKey } : {},
3116
+ ...typeof llm.baseURL === "string" ? { baseURL: llm.baseURL } : {},
3117
+ ...typeof llm.model === "string" ? { model: llm.model } : {},
3118
+ ...typeof llm.name === "string" ? { name: llm.name } : {},
3119
+ ...typeof llm.provider === "string" ? { provider: parseProvider(llm.provider) } : {}
3120
+ };
3121
+ const options = buildWikiGraphLLMOptions({ llm: config });
3122
+ const result = (0, import_ai2.streamText)({
3123
+ maxRetries: 0,
3124
+ messages: [{ content: "Reply with exactly: ok", role: "user" }],
3125
+ model: options.model,
3126
+ temperature: 0
3127
+ });
3128
+ const chunks = [];
3129
+ for await (const chunk of result.textStream) chunks.push(chunk);
3130
+ return {
3131
+ durationMs: Date.now() - startedAt,
3132
+ ...config.model === void 0 ? {} : { model: config.model },
3133
+ ...config.provider === void 0 ? {} : { provider: config.provider },
3134
+ response: chunks.join("").trim()
3135
+ };
3136
+ });
3137
+ }
3138
+ async testEmbedding() {
3139
+ return await this.#runtime.run(async () => {
3140
+ const startedAt = Date.now();
3141
+ const result = await embedQueryText(
3142
+ "Wiki Graph embedding connectivity test.",
3143
+ await readWikiGraphEmbeddingConfig()
3144
+ );
3145
+ return {
3146
+ dimensions: result.dimensions,
3147
+ durationMs: Date.now() - startedAt,
3148
+ model: result.model,
3149
+ provider: result.provider,
3150
+ ...result.usage?.tokens === void 0 ? {} : { tokens: result.usage.tokens }
3151
+ };
3152
+ });
3153
+ }
3154
+ async testWikispine() {
3155
+ return await this.#runtime.run(async () => {
3156
+ const value = await readLocalConfigSection("wikispine");
3157
+ const resolved = resolveWikispineConfig(value);
3158
+ if (resolved === void 0) {
3159
+ throw new Error("WikiSpine provider is not configured.");
3160
+ }
3161
+ const result = await (0, import_wiki_graph_core6.testWikispineRuntime)({
3162
+ ...resolved,
3163
+ ...resolved.provider === "cli" ? { commandRunner: nodeWikispineCommandRunner } : {}
3164
+ });
3165
+ return {
3166
+ durationMs: result.durationMs,
3167
+ ...resolved.provider === "fetch" ? { endpoint: resolved.endpoint } : {},
3168
+ ...result.metadata === void 0 ? {} : { metadata: result.metadata },
3169
+ provider: resolved.provider
3170
+ };
3171
+ });
3172
+ }
3173
+ };
3174
+ function parseProvider(value) {
3175
+ if (value === "anthropic" || value === "google" || value === "openai" || value === "openai-compatible") {
3176
+ return value;
3177
+ }
3178
+ throw new Error(
3179
+ `Invalid llm.provider: ${value}. Expected anthropic, google, openai, or openai-compatible.`
3180
+ );
3181
+ }
3182
+
3183
+ // src/continuations.ts
3184
+ var import_wiki_graph_core7 = require("wiki-graph-core");
3185
+ var WikiGraphContinuationManager = class {
3186
+ #runtime;
3187
+ constructor(runtime) {
3188
+ this.#runtime = runtime;
3189
+ }
3190
+ async next(options) {
3191
+ return await this.#runtime.run(
3192
+ async () => await this.#next(options),
3193
+ options.signal
3194
+ );
3195
+ }
3196
+ async create(context, cursor) {
3197
+ if (cursor === null || cursor === void 0) return null;
3198
+ return await this.#runtime.run(
3199
+ async () => await (0, import_wiki_graph_core7.createContinuationCursor)(
3200
+ createContinuationPayload(context, cursor)
3201
+ )
3202
+ );
3203
+ }
3204
+ async #next(options) {
3205
+ const cursor = await (0, import_wiki_graph_core7.readContinuationCursor)(options.cursor);
3206
+ const continuationCursor = options.format === void 0 ? cursor : { ...cursor, format: options.format };
3207
+ const limit = options.limit ?? 20;
3208
+ if (options.archive !== void 0) {
3209
+ const explicit = await resolveWikiGraphArchiveLocation(options.archive);
3210
+ const cursorArchivePath = getCursorArchivePath(cursor);
3211
+ if (explicit.archivePath !== cursorArchivePath) {
3212
+ throw new Error(
3213
+ `Continuation cursor ${options.cursor} belongs to ${cursorArchivePath}, not ${explicit.archivePath}.`
3214
+ );
3215
+ }
3216
+ }
3217
+ const result = cursor.indexScope.kind === "library-index" ? await continueLibraryCursor(cursor, limit) : await continueArchiveCursor(cursor, limit);
3218
+ const durableResult = await persistResultCursor(continuationCursor, result);
3219
+ return {
3220
+ cursor: continuationCursor,
3221
+ format: continuationCursor.format,
3222
+ kind: cursor.kind,
3223
+ limit,
3224
+ result: durableResult
3225
+ };
3226
+ }
3227
+ };
3228
+ async function continueArchiveCursor(cursor, limit) {
3229
+ const location = await resolveWikiGraphArchiveLocation(
3230
+ getCursorArchivePath(cursor)
3231
+ );
3232
+ return await new import_wiki_graph_core7.WikiGraphArchiveFile(location.archiveFile).readDocument(
3233
+ async (document) => {
3234
+ switch (cursor.kind) {
3235
+ case "source-locators":
3236
+ return await (0, import_wiki_graph_core7.listArchiveSourceLocators)(document, cursor.targetUri, {
3237
+ cursor: cursor.cursor,
3238
+ limit
3239
+ });
3240
+ case "collection":
3241
+ return await (0, import_wiki_graph_core7.listArchiveCollection)(
3242
+ document,
3243
+ createCollectionOptions(cursor, limit)
3244
+ );
3245
+ case "search":
3246
+ return await (0, import_wiki_graph_core7.findArchiveObjects)(
3247
+ document,
3248
+ cursor.query ?? "",
3249
+ await createFindOptions(cursor, limit)
3250
+ );
3251
+ case "evidence":
3252
+ return await (0, import_wiki_graph_core7.listArchiveEvidence)(document, cursor.targetUri, {
3253
+ ...cursor.chapters === null ? {} : { chapters: cursor.chapters },
3254
+ cursor: cursor.cursor,
3255
+ limit,
3256
+ order: cursor.order,
3257
+ ...cursor.query === void 0 ? {} : { query: cursor.query },
3258
+ ...cursor.skipUnindexed === void 0 ? {} : { skipUnindexed: cursor.skipUnindexed },
3259
+ ...cursor.sourceContext === void 0 ? {} : { sourceContext: cursor.sourceContext }
3260
+ });
3261
+ case "related":
3262
+ return await (0, import_wiki_graph_core7.listRelatedArchiveObjects)(document, cursor.targetUri, {
3263
+ ...cursor.chapters === null ? {} : { chapters: cursor.chapters },
3264
+ cursor: cursor.cursor,
3265
+ ...cursor.evidenceLimit === void 0 ? {} : { evidenceLimit: cursor.evidenceLimit },
3266
+ limit,
3267
+ order: cursor.order,
3268
+ ...cursor.query === void 0 ? {} : { query: cursor.query },
3269
+ ...cursor.role === void 0 ? {} : { role: cursor.role },
3270
+ ...cursor.skipUnindexed === void 0 ? {} : { skipUnindexed: cursor.skipUnindexed },
3271
+ ...cursor.sourceContext === void 0 ? {} : { sourceContext: cursor.sourceContext }
3272
+ });
3273
+ }
3274
+ }
3275
+ );
3276
+ }
3277
+ async function continueLibraryCursor(cursor, limit) {
3278
+ if (cursor.kind === "source-locators") {
3279
+ throw new Error(
3280
+ "Source locator cursors do not use the aggregate library index."
3281
+ );
3282
+ }
3283
+ if (cursor.indexScope.kind !== "library-index") {
3284
+ throw new Error("Internal error: expected a library index cursor.");
3285
+ }
3286
+ const target = await (0, import_wiki_graph_core7.resolveWikiGraphLibraryQueryTargetById)(
3287
+ cursor.indexScope.libraryId
3288
+ );
3289
+ switch (cursor.kind) {
3290
+ case "collection":
3291
+ return await (requireLibraryQuery(cursor) === "archive-members" ? (0, import_wiki_graph_core7.listWikiGraphLibraryArchiveMembers)(
3292
+ target,
3293
+ createCollectionOptions(cursor, limit)
3294
+ ) : (0, import_wiki_graph_core7.listWikiGraphLibraryObjects)(
3295
+ target,
3296
+ createCollectionOptions(cursor, limit)
3297
+ ));
3298
+ case "search":
3299
+ return await (requireLibraryQuery(cursor) === "archive-members" ? (0, import_wiki_graph_core7.findWikiGraphLibraryArchiveMembers)(
3300
+ target,
3301
+ cursor.query ?? "",
3302
+ await createFindOptions(cursor, limit)
3303
+ ) : (0, import_wiki_graph_core7.findWikiGraphLibraryObjects)(
3304
+ target,
3305
+ cursor.query ?? "",
3306
+ await createFindOptions(cursor, limit)
3307
+ ));
3308
+ case "evidence":
3309
+ return await (0, import_wiki_graph_core7.listWikiGraphLibraryEvidence)(target, cursor.targetUri, {
3310
+ cursor: cursor.cursor,
3311
+ limit,
3312
+ order: cursor.order,
3313
+ ...cursor.query === void 0 ? {} : { query: cursor.query },
3314
+ ...cursor.skipUnindexed === void 0 ? {} : { skipUnindexed: cursor.skipUnindexed },
3315
+ ...cursor.sourceContext === void 0 ? {} : { sourceContext: cursor.sourceContext }
3316
+ });
3317
+ case "related":
3318
+ return await (0, import_wiki_graph_core7.listRelatedWikiGraphLibraryObjects)(
3319
+ target,
3320
+ cursor.targetUri,
3321
+ {
3322
+ cursor: cursor.cursor,
3323
+ ...cursor.evidenceLimit === void 0 ? {} : { evidenceLimit: cursor.evidenceLimit },
3324
+ limit,
3325
+ order: cursor.order,
3326
+ ...cursor.query === void 0 ? {} : { query: cursor.query },
3327
+ ...cursor.role === void 0 ? {} : { role: cursor.role },
3328
+ ...cursor.skipUnindexed === void 0 ? {} : { skipUnindexed: cursor.skipUnindexed },
3329
+ ...cursor.sourceContext === void 0 ? {} : { sourceContext: cursor.sourceContext }
3330
+ }
3331
+ );
3332
+ }
3333
+ }
3334
+ function createCollectionOptions(cursor, limit) {
3335
+ return {
3336
+ ...cursor.backlinks === void 0 ? {} : { backlinks: cursor.backlinks },
3337
+ ...cursor.chapters === null ? {} : { chapters: cursor.chapters },
3338
+ cursor: cursor.cursor,
3339
+ ...cursor.evidenceLimit === void 0 ? {} : { evidenceLimit: cursor.evidenceLimit },
3340
+ ...cursor.ids === null ? {} : { ids: cursor.ids },
3341
+ limit,
3342
+ order: cursor.order,
3343
+ ...cursor.sourceContext === void 0 ? {} : { sourceContext: cursor.sourceContext },
3344
+ ...cursor.triplePattern === void 0 ? {} : { triplePattern: cursor.triplePattern },
3345
+ ...cursor.types === null ? {} : {
3346
+ types: cursor.types
3347
+ }
3348
+ };
3349
+ }
3350
+ async function createFindOptions(cursor, limit) {
3351
+ const embeddingProvider = await createConfiguredEmbeddingProvider();
3352
+ return {
3353
+ archiveKey: cursor.archiveKey,
3354
+ ...cursor.backlinks === void 0 ? {} : { backlinks: cursor.backlinks },
3355
+ ...cursor.chapters == null ? {} : { chapters: cursor.chapters },
3356
+ cursor: cursor.cursor,
3357
+ ...cursor.evidenceLimit === void 0 ? {} : { evidenceLimit: cursor.evidenceLimit },
3358
+ limit,
3359
+ ...cursor.skipUnindexed === void 0 ? {} : { skipUnindexed: cursor.skipUnindexed },
3360
+ ...cursor.sourceContext === void 0 ? {} : { sourceContext: cursor.sourceContext },
3361
+ ...cursor.triplePattern === void 0 ? {} : { triplePattern: cursor.triplePattern },
3362
+ ...cursor.types === null ? {} : { types: cursor.types },
3363
+ ...embeddingProvider === void 0 ? {} : { embeddingProvider }
3364
+ };
3365
+ }
3366
+ function requireLibraryQuery(cursor) {
3367
+ if ((cursor.kind === "collection" || cursor.kind === "search") && cursor.libraryQuery !== void 0) {
3368
+ return cursor.libraryQuery;
3369
+ }
3370
+ throw new Error(
3371
+ "Library continuation cursor is missing its typed query variant."
3372
+ );
3373
+ }
3374
+ function getCursorArchivePath(cursor) {
3375
+ return cursor.indexScope.kind === "archive-index" ? cursor.indexScope.archivePath : cursor.archivePath;
3376
+ }
3377
+ async function persistResultCursor(cursor, result) {
3378
+ if (result.nextCursor === null) return result;
3379
+ const nextCursor = await (0, import_wiki_graph_core7.createContinuationCursor)({
3380
+ ...cursor,
3381
+ cursor: result.nextCursor
3382
+ });
3383
+ return { ...result, nextCursor };
3384
+ }
3385
+ function createContinuationPayload(context, cursor) {
3386
+ switch (context.continuationKind ?? "search") {
3387
+ case "source-locators":
3388
+ return {
3389
+ archiveKey: context.archiveKey,
3390
+ archivePath: context.archivePath,
3391
+ cursor,
3392
+ format: context.format,
3393
+ indexScope: context.indexScope,
3394
+ kind: "source-locators",
3395
+ targetUri: requireTargetUri(context, "Source locator")
3396
+ };
3397
+ case "evidence":
3398
+ return {
3399
+ archiveKey: context.archiveKey,
3400
+ archivePath: context.archivePath,
3401
+ chapters: context.chapters ?? null,
3402
+ cursor,
3403
+ format: context.format,
3404
+ indexScope: context.indexScope,
3405
+ kind: "evidence",
3406
+ order: context.order ?? "doc-asc",
3407
+ ...context.query === void 0 ? {} : { query: context.query },
3408
+ ...context.skipUnindexed === void 0 ? {} : { skipUnindexed: context.skipUnindexed },
3409
+ ...context.sourceContext === void 0 ? {} : { sourceContext: context.sourceContext },
3410
+ targetUri: requireTargetUri(context, "Evidence")
3411
+ };
3412
+ case "related":
3413
+ return {
3414
+ archiveKey: context.archiveKey,
3415
+ archivePath: context.archivePath,
3416
+ chapters: context.chapters ?? null,
3417
+ cursor,
3418
+ ...context.evidenceLimit === void 0 ? {} : { evidenceLimit: context.evidenceLimit },
3419
+ format: context.format,
3420
+ indexScope: context.indexScope,
3421
+ kind: "related",
3422
+ order: context.order ?? "doc-asc",
3423
+ ...context.query === void 0 ? {} : { query: context.query },
3424
+ ...context.role === void 0 ? {} : { role: context.role },
3425
+ ...context.skipUnindexed === void 0 ? {} : { skipUnindexed: context.skipUnindexed },
3426
+ ...context.sourceContext === void 0 ? {} : { sourceContext: context.sourceContext },
3427
+ targetUri: requireTargetUri(context, "Related")
3428
+ };
3429
+ case "collection":
3430
+ return {
3431
+ archiveKey: context.archiveKey,
3432
+ archivePath: context.archivePath,
3433
+ ...context.backlinks === void 0 ? {} : { backlinks: context.backlinks },
3434
+ chapters: context.chapters ?? null,
3435
+ cursor,
3436
+ ...context.evidenceLimit === void 0 ? {} : { evidenceLimit: context.evidenceLimit },
3437
+ format: context.format,
3438
+ ids: context.ids ?? null,
3439
+ indexScope: context.indexScope,
3440
+ kind: "collection",
3441
+ ...createLibraryQueryPayload(context),
3442
+ order: context.order ?? "doc-asc",
3443
+ ...context.sourceContext === void 0 ? {} : { sourceContext: context.sourceContext },
3444
+ ...context.triplePattern === void 0 ? {} : { triplePattern: context.triplePattern },
3445
+ types: context.types
3446
+ };
3447
+ case "search":
3448
+ return {
3449
+ archiveKey: context.archiveKey,
3450
+ archivePath: context.archivePath,
3451
+ ...context.backlinks === void 0 ? {} : { backlinks: context.backlinks },
3452
+ ...context.chapters === void 0 ? {} : { chapters: context.chapters },
3453
+ cursor,
3454
+ ...context.evidenceLimit === void 0 ? {} : { evidenceLimit: context.evidenceLimit },
3455
+ format: context.format,
3456
+ indexScope: context.indexScope,
3457
+ kind: "search",
3458
+ ...createLibraryQueryPayload(context),
3459
+ ...context.query === void 0 ? {} : { query: context.query },
3460
+ ...context.skipUnindexed === void 0 ? {} : { skipUnindexed: context.skipUnindexed },
3461
+ ...context.sourceContext === void 0 ? {} : { sourceContext: context.sourceContext },
3462
+ ...context.triplePattern === void 0 ? {} : { triplePattern: context.triplePattern },
3463
+ types: context.types
3464
+ };
3465
+ }
3466
+ }
3467
+ function requireTargetUri(context, label) {
3468
+ if (context.targetUri !== void 0) return context.targetUri;
3469
+ throw new Error(`${label} continuation cursors require a target URI.`);
3470
+ }
3471
+ function createLibraryQueryPayload(context) {
3472
+ if (context.indexScope.kind !== "library-index") return {};
3473
+ if (context.libraryQuery !== void 0) {
3474
+ return { libraryQuery: context.libraryQuery };
3475
+ }
3476
+ throw new Error(
3477
+ "Library collection and search continuations require an explicit query variant."
3478
+ );
3479
+ }
3480
+
3481
+ // src/default-worker.ts
3482
+ var import_wiki_graph_core11 = require("wiki-graph-core");
3483
+ var import_worker3 = require("wiki-graph-core/worker");
3484
+
3485
+ // src/stage.ts
3486
+ var import_wiki_graph_core8 = require("wiki-graph-core");
3487
+ var import_wiki_graph_core9 = require("wiki-graph-core");
3488
+ var import_wiki_graph_core10 = require("wiki-graph-core");
3489
+ function createStageLLM(config, options) {
3490
+ const llmOptions = buildWikiGraphLLMOptions(config);
3491
+ return new import_wiki_graph_core10.LLM({
3492
+ sampling: (0, import_wiki_graph_core9.createDefaultWikiGraphSampling)({
3493
+ ...llmOptions.temperature === void 0 ? {} : { temperature: llmOptions.temperature },
3494
+ ...llmOptions.topP === void 0 ? {} : { topP: llmOptions.topP }
3495
+ }),
3496
+ ...llmOptions,
3497
+ ...options?.cacheDirectory === void 0 ? {} : { cacheDirectory: options.cacheDirectory },
3498
+ ...options?.logDirectory === void 0 ? {} : { logDirectory: options.logDirectory },
3499
+ ...options?.onStreamProgress === void 0 ? {} : { onStreamProgress: options.onStreamProgress },
3500
+ ...options?.onTokenUsage === void 0 ? {} : { onTokenUsage: options.onTokenUsage }
3501
+ });
3502
+ }
3503
+ async function loadRequiredStageConfig(options) {
3504
+ const config = await loadWikiGraphRuntimeConfig({
3505
+ ...options.llmJSON === void 0 ? {} : { llmJSON: options.llmJSON }
3506
+ });
3507
+ if (config.llm?.provider === void 0 || config.llm.model === void 0) {
3508
+ throw new Error(
3509
+ "Missing LLM configuration. Set --llm for one run, or configure `wikg://local/config/llm` with provider and model."
3510
+ );
3511
+ }
3512
+ return config;
3513
+ }
3514
+
3515
+ // src/worker.ts
3516
+ var import_worker = require("wiki-graph-core/worker");
3517
+ var import_worker2 = require("wiki-graph-core/worker");
3518
+ async function runBuildJobWorker(options) {
3519
+ installNodeWikiGraphPlatform();
3520
+ const { stateDir, ...coreOptions } = options;
3521
+ await withNodeWikiGraphStorage(
3522
+ stateDir,
3523
+ async () => await (0, import_worker.runBuildJobWorker)(coreOptions)
3524
+ );
3525
+ }
3526
+
3527
+ // src/default-worker.ts
3528
+ async function runWikiGraphQueueWorker(options = {}) {
3529
+ const config = await loadWikiGraphRuntimeConfig();
3530
+ await runBuildJobWorker({
3531
+ concurrency: config.concurrent?.job ?? DEFAULT_GENERATION_JOB_CONCURRENCY,
3532
+ executeJob: async (job, reporter, context) => {
3533
+ await (0, import_wiki_graph_core11.withLoggingContext)(
3534
+ { logDirectory: job.log, operation: "build-job" },
3535
+ async () => await executeBuildJob(job, reporter, context)
3536
+ );
3537
+ },
3538
+ ...options.signal === void 0 ? {} : { signal: options.signal }
3539
+ });
3540
+ }
3541
+ async function executeBuildJob(job, reporter, context) {
3542
+ const config = await loadWikiGraphRuntimeConfig({
3543
+ ...job.llmJSON === void 0 ? {} : { llmJSON: job.llmJSON }
3544
+ });
3545
+ const execution = await openJobExecutor(
3546
+ job,
3547
+ config,
3548
+ context.signal,
3549
+ reporter
3550
+ );
3551
+ try {
3552
+ if (job.target === "reading-summary") {
3553
+ let stage = await readChapterStage(job);
3554
+ if (stage === "sourced") {
3555
+ await executeStep(job, "reading-graph", execution, reporter, context, {
3556
+ ...execution.extractionPrompt === void 0 ? {} : { extractionPrompt: execution.extractionPrompt }
3557
+ });
3558
+ stage = await readChapterStage(job);
3559
+ }
3560
+ if (stage === "summarized") return;
3561
+ if (stage !== "graphed") {
3562
+ throw new Error(
3563
+ `Chapter ${job.chapterId} is ${stage}. Cannot generate summary.`
3564
+ );
3565
+ }
3566
+ await executeStep(job, "reading-summary", execution, reporter, context, {
3567
+ ...job.prompt === void 0 ? {} : { prompt: job.prompt }
3568
+ });
3569
+ return;
3570
+ }
3571
+ if (job.target === "reading-graph") {
3572
+ const stage = await readChapterStage(job);
3573
+ if (stage === "graphed" || stage === "summarized") return;
3574
+ if (stage !== "sourced") {
3575
+ throw new Error(
3576
+ `Chapter ${job.chapterId} is ${stage}. Cannot generate Reading Graph.`
3577
+ );
3578
+ }
3579
+ }
3580
+ if (job.target === "knowledge-graph") {
3581
+ const stage = await readChapterStage(job);
3582
+ if (stage === "planned") {
3583
+ throw new Error(
3584
+ `Chapter ${job.chapterId} is planned. Set source before generating Knowledge Graph.`
3585
+ );
3586
+ }
3587
+ }
3588
+ await executeStep(job, job.target, execution, reporter, context, {
3589
+ ...job.target === "reading-graph" ? { extractionPrompt: execution.extractionPrompt } : {},
3590
+ ...job.target === "knowledge-graph" ? { policyPrompt: execution.knowledgeGraphPrompt } : {}
3591
+ });
3592
+ } finally {
3593
+ await execution.close();
3594
+ }
3595
+ }
3596
+ async function executeStep(job, kind, execution, reporter, context, inputOptions) {
3597
+ await reporter.throwIfStopped();
3598
+ await reporter.stepStarted(kind);
3599
+ if (kind === "reading-graph" || kind === "knowledge-graph") {
3600
+ await assertCurrentFtsArtifact(job);
3601
+ }
3602
+ const stepWorkspace = await resetStepWorkspace(job.workspace, kind);
3603
+ const inputFile = await stepWorkspace.createFile("input.jsonl");
3604
+ const outputWorkspace = await stepWorkspace.createDirectory("output");
3605
+ const revision = await new import_wiki_graph_core11.WikiGraphArchiveFile(job.archive).readDocument(
3606
+ async (document) => {
3607
+ await (0, import_wiki_graph_core11.getChapterDetails)(document, job.chapterId);
3608
+ return await (0, import_worker3.writeChapterJobInputFile)(
3609
+ document,
3610
+ job.chapterId,
3611
+ kind,
3612
+ inputFile,
3613
+ inputOptions
3614
+ );
3615
+ }
3616
+ );
3617
+ await (0, import_wiki_graph_core11.recordBuildJobInputRevision)({
3618
+ currentRevision: revision,
3619
+ jobId: job.jobId,
3620
+ ownerId: requireRunningJobOwnerId(job)
3621
+ });
3622
+ await reporter.updatePhase({
3623
+ done: 0,
3624
+ phase: kind.startsWith("index-") ? "indexing" : "grounding",
3625
+ total: 1,
3626
+ unit: "item"
3627
+ });
3628
+ const result = await execution.executor({
3629
+ inputFile,
3630
+ kind,
3631
+ progress: reporter,
3632
+ revision,
3633
+ signal: context.signal,
3634
+ workspace: outputWorkspace
3635
+ });
3636
+ if (result.revision !== revision) {
3637
+ throw new Error(
3638
+ `Chapter job returned revision ${result.revision}; expected ${revision}.`
3639
+ );
3640
+ }
3641
+ await reporter.updatePhase({
3642
+ done: 0,
3643
+ phase: "committing",
3644
+ total: 1,
3645
+ unit: "item"
3646
+ });
3647
+ await new import_wiki_graph_core11.WikiGraphArchiveFile(job.archive).write(async (document) => {
3648
+ assertJobStillRunning(await (0, import_wiki_graph_core11.getBuildJob)(job.jobId));
3649
+ await (0, import_worker3.applyChapterJobArtifactFile)(
3650
+ document,
3651
+ job.chapterId,
3652
+ kind,
3653
+ revision,
3654
+ result.artifactFile
3655
+ );
3656
+ });
3657
+ await reporter.updatePhase({
3658
+ done: 1,
3659
+ phase: "committing",
3660
+ total: 1,
3661
+ unit: "item"
3662
+ });
3663
+ await reporter.stepCompleted(kind);
3664
+ assertJobStillRunning(await (0, import_wiki_graph_core11.getBuildJob)(job.jobId));
3665
+ }
3666
+ async function openJobExecutor(job, config, signal, reporter) {
3667
+ const generation = !job.target.startsWith("index-");
3668
+ const stageConfig = generation ? await loadRequiredStageConfig({
3669
+ ...job.llmJSON === void 0 ? {} : { llmJSON: job.llmJSON }
3670
+ }) : void 0;
3671
+ const llm = stageConfig === void 0 ? void 0 : createStageLLM(stageConfig, {
3672
+ cacheDirectory: job.cache,
3673
+ logDirectory: job.log,
3674
+ onTokenUsage: async (usage) => {
3675
+ await reporter.addTokenUsage(usage);
3676
+ }
3677
+ });
3678
+ const promptSource = job.prompt ?? stageConfig?.prompt;
3679
+ const wikispine = job.target === "knowledge-graph" ? requireKnowledgeGraphWikispineConfig(config) : void 0;
3680
+ const request = llm === void 0 ? void 0 : async (messages, index, maxRetries) => await llm.request(messages, {
3681
+ retryIndex: index,
3682
+ retryMax: maxRetries,
3683
+ scope: import_wiki_graph_core11.WikiGraphScope.ReaderExtraction,
3684
+ signal
3685
+ });
3686
+ if (request !== void 0 && llm !== void 0) {
3687
+ request.lazy = async (operation) => await llm.request(async () => await operation(request));
3688
+ }
3689
+ const wikimediaResolver = job.target !== "knowledge-graph" ? void 0 : await (0, import_wiki_graph_core11.openWikimediaResolver)(
3690
+ config.wikimedia === void 0 ? {
3691
+ kind: "local",
3692
+ llmRequest: requireValue(
3693
+ request,
3694
+ "Local Wikimedia resolution requires an LLM."
3695
+ )
3696
+ } : {
3697
+ endpoint: config.wikimedia.endpoint,
3698
+ kind: "remote",
3699
+ token: config.wikimedia.token
3700
+ }
3701
+ );
3702
+ const embeddingProvider = job.target === "index-embedding-source" || job.target === "index-embedding-summary" ? buildSearchIndexEmbeddingProvider(requireEmbeddingConfig(config)) : void 0;
3703
+ return {
3704
+ async close() {
3705
+ await wikimediaResolver?.close();
3706
+ },
3707
+ executor: (0, import_worker3.createLocalChapterJobFileExecutor)({
3708
+ ...embeddingProvider === void 0 ? {} : { embeddingProvider },
3709
+ ...llm === void 0 ? {} : { llm },
3710
+ ...wikimediaResolver === void 0 ? {} : { wikimediaResolver },
3711
+ ...wikispine === void 0 ? {} : {
3712
+ wikispine: {
3713
+ ...wikispine,
3714
+ ...wikispine.provider === "cli" ? { commandRunner: nodeWikispineCommandRunner } : {}
3715
+ }
3716
+ }
3717
+ }),
3718
+ ...promptSource === void 0 ? {} : { extractionPrompt: (0, import_wiki_graph_core8.resolveExtractionPrompt)(promptSource) },
3719
+ ...promptSource === void 0 ? {} : {
3720
+ knowledgeGraphPrompt: (0, import_wiki_graph_core8.resolveKnowledgeGraphRecallPrompt)(promptSource)
3721
+ }
3722
+ };
3723
+ }
3724
+ async function resetStepWorkspace(workspace, kind) {
3725
+ const name = `chapter-job-${kind}`;
3726
+ if (await workspace.getDirectory(name) !== void 0) {
3727
+ await workspace.remove(name, { recursive: true });
3728
+ }
3729
+ return await workspace.createDirectory(name);
3730
+ }
3731
+ async function readChapterStage(job) {
3732
+ return await new import_wiki_graph_core11.WikiGraphArchiveFile(job.archive).readDocument(
3733
+ async (document) => (await (0, import_wiki_graph_core11.getChapterDetails)(document, job.chapterId)).stage
3734
+ );
3735
+ }
3736
+ async function assertCurrentFtsArtifact(job) {
3737
+ await new import_wiki_graph_core11.WikiGraphArchiveFile(job.archive).readDocument(async (document) => {
3738
+ const revision = await document.serials.getRevision(job.chapterId);
3739
+ const artifact = await document.indexArtifacts.get(job.chapterId, "fts");
3740
+ if (artifact?.sourceRevision !== revision) {
3741
+ throw new Error(
3742
+ `Chapter ${job.chapterId} needs a current FTS index artifact before running ${job.target}.`
3743
+ );
3744
+ }
3745
+ });
3746
+ }
3747
+ function requireEmbeddingConfig(config) {
3748
+ if (config.embedding !== void 0) return config.embedding;
3749
+ throw new Error(
3750
+ "Missing embeddings configuration. Configure `wikg://local/config/embeddings` before building embedding index artifacts."
3751
+ );
3752
+ }
3753
+ function requireKnowledgeGraphWikispineConfig(config) {
3754
+ if (config.wikispine?.provider !== void 0) return config.wikispine;
3755
+ throw new Error(
3756
+ "Knowledge Graph requires WikiSpine. Configure `wikg://local/config/wikispine` with provider `cli` or `fetch` and test the provider before running the worker."
3757
+ );
3758
+ }
3759
+ function requireRunningJobOwnerId(job) {
3760
+ if (job.ownerId === void 0) {
3761
+ throw new Error(`Build job ${job.jobId} has no running owner.`);
3762
+ }
3763
+ return job.ownerId;
3764
+ }
3765
+ function assertJobStillRunning(job) {
3766
+ if (job.state !== "running") {
3767
+ throw new Error(`Job ${job.jobId} is ${job.state}. Stop before flushing.`);
3768
+ }
3769
+ }
3770
+ function requireValue(value, message) {
3771
+ if (value === void 0) throw new Error(message);
3772
+ return value;
3773
+ }
3774
+
3775
+ // src/entry-context.ts
3776
+ var DANGEROUS_RUNTIME_ENV_NAMES = [
3777
+ "WIKIGRAPH_DEV",
3778
+ "WIKIGRAPH_ENV_POLICY",
3779
+ "WIKIGRAPH_QUEUE_DISABLE_AUTOSTART",
3780
+ "WIKIGRAPH_STATE_DIR"
3781
+ ];
3782
+ function createEntryRuntimeContext(options) {
3783
+ const environment = createMergedEnvironment(options.env);
3784
+ if (options.envPolicy === "production") {
3785
+ const blocked = DANGEROUS_RUNTIME_ENV_NAMES.filter(
3786
+ (name) => environment[name] !== void 0
3787
+ );
3788
+ if (blocked.length > 0) {
3789
+ throw new Error(
3790
+ `Wiki Graph production entries do not support runtime env overrides: ${blocked.join(", ")}. Use the development entry or explicit SDK options instead.`
3791
+ );
3792
+ }
3793
+ return {
3794
+ env: environment,
3795
+ envPolicy: options.envPolicy,
3796
+ stateDir: options.stateDir
3797
+ };
3798
+ }
3799
+ const envStateDir = environment.WIKIGRAPH_DEV ?? environment.WIKIGRAPH_STATE_DIR;
3800
+ const stateDir = normalizeOptionalPath(options.stateDir ?? envStateDir);
3801
+ if (stateDir === void 0) {
3802
+ throw new Error(
3803
+ "Wiki Graph development entries require an explicit runtime state directory."
3804
+ );
3805
+ }
3806
+ const devProjectRoot = normalizeOptionalPath(options.devProjectRoot);
3807
+ if (devProjectRoot === void 0) {
3808
+ throw new Error(
3809
+ "Wiki Graph development entries require an explicit project root."
3810
+ );
3811
+ }
3812
+ return {
3813
+ devProjectRoot,
3814
+ env: environment,
3815
+ envPolicy: options.envPolicy,
3816
+ stateDir
3817
+ };
3818
+ }
3819
+ function createMergedEnvironment(environment) {
3820
+ if (environment === void 0) {
3821
+ return { ...process.env };
3822
+ }
3823
+ return {
3824
+ ...process.env,
3825
+ ...environment
3826
+ };
3827
+ }
3828
+ function normalizeOptionalPath(path) {
3829
+ if (path === void 0 || path.trim() === "") {
3830
+ return void 0;
3831
+ }
3832
+ return path;
3833
+ }
3834
+
3835
+ // src/jobs.ts
3836
+ var import_wiki_graph_core12 = require("wiki-graph-core");
3837
+ var TERMINAL_STATES = /* @__PURE__ */ new Set(["canceled", "failed", "succeeded"]);
3838
+ var TERMINAL_EVENT_GRACE_READS = 3;
3839
+ var CORE_BUILD_JOB_BACKEND = {
3840
+ add: async (options) => await (0, import_wiki_graph_core12.addBuildJob)(options),
3841
+ boost: async (jobId) => await (0, import_wiki_graph_core12.boostBuildJob)(jobId),
3842
+ cancel: async (jobId) => await (0, import_wiki_graph_core12.cancelBuildJob)(jobId),
3843
+ get: async (jobId) => await (0, import_wiki_graph_core12.getBuildJob)(jobId),
3844
+ list: async (options) => await (0, import_wiki_graph_core12.listBuildJobs)(options),
3845
+ pause: async (jobId) => await (0, import_wiki_graph_core12.pauseBuildJob)(jobId),
3846
+ readEventChunk: async (job, cursor) => await readIncrementalBuildJobEvents(job, cursor),
3847
+ readEvents: async (job) => await (0, import_wiki_graph_core12.readBuildJobEvents)(job),
3848
+ resume: async (jobId) => await (0, import_wiki_graph_core12.resumeBuildJob)(jobId),
3849
+ setTarget: async (jobId, target) => await (0, import_wiki_graph_core12.updateBuildJobTarget)(jobId, target)
3850
+ };
3851
+ async function readIncrementalBuildJobEvents(job, cursor) {
3852
+ let reader;
3853
+ try {
3854
+ reader = await job.events.openReader();
3855
+ } catch {
3856
+ return { cursor, events: await (0, import_wiki_graph_core12.readBuildJobEvents)(job) };
3857
+ }
3858
+ try {
3859
+ const offset = cursor <= reader.size ? cursor : 0;
3860
+ const content = new TextDecoder().decode(
3861
+ await reader.read(offset, reader.size - offset)
3862
+ );
3863
+ return {
3864
+ cursor: reader.size,
3865
+ events: content.split("\n").filter((line) => line.trim() !== "").map((line) => JSON.parse(line))
3866
+ };
3867
+ } finally {
3868
+ await reader.close();
3869
+ }
3870
+ }
3871
+ var WikiGraphJobManager = class {
3872
+ #runtime;
3873
+ #backend;
3874
+ #subscriptions = /* @__PURE__ */ new Set();
3875
+ constructor(runtime, backend = CORE_BUILD_JOB_BACKEND) {
3876
+ this.#runtime = runtime;
3877
+ this.#backend = backend;
3878
+ }
3879
+ async create(options) {
3880
+ const job = await this.#runtime.run(
3881
+ async () => await this.#backend.add({
3882
+ ...options,
3883
+ archive: typeof options.archive === "string" ? new NodeFile(resolveWikiGraphRuntimePath(options.archive)) : options.archive
3884
+ })
3885
+ );
3886
+ return this.#createHandle(job);
3887
+ }
3888
+ async clean() {
3889
+ return await this.#runtime.run(async () => await (0, import_wiki_graph_core12.cleanBuildJobs)());
3890
+ }
3891
+ async get(jobId) {
3892
+ return this.#createHandle(
3893
+ await this.#runtime.run(async () => await this.#backend.get(jobId))
3894
+ );
3895
+ }
3896
+ async resolveId(reference) {
3897
+ return await this.#runtime.run(
3898
+ async () => await (0, import_wiki_graph_core12.resolveBuildJobId)(reference)
3899
+ );
3900
+ }
3901
+ async list(options = {}) {
3902
+ const jobs = await this.#runtime.run(async () => {
3903
+ const normalized = {
3904
+ ...options.activeOnly === void 0 ? {} : { activeOnly: options.activeOnly },
3905
+ ...options.all === void 0 ? {} : { all: options.all },
3906
+ ...options.archive === void 0 ? {} : {
3907
+ archive: typeof options.archive === "string" ? new NodeFile(resolveWikiGraphRuntimePath(options.archive)) : options.archive
3908
+ }
3909
+ };
3910
+ return await this.#backend.list(normalized);
3911
+ });
3912
+ return jobs.map((job) => this.#createHandle(job));
3913
+ }
3914
+ async resolveChapters(jobs) {
3915
+ return await this.#runtime.run(async () => {
3916
+ const jobsByArchive = /* @__PURE__ */ new Map();
3917
+ for (const job of jobs) {
3918
+ const archivePath = getBuildJobArchivePath(job);
3919
+ const grouped = jobsByArchive.get(archivePath) ?? [];
3920
+ grouped.push(job);
3921
+ jobsByArchive.set(archivePath, grouped);
3922
+ }
3923
+ const entries = (await Promise.all(
3924
+ [...jobsByArchive].map(async ([archivePath, archiveJobs]) => {
3925
+ try {
3926
+ const chapters = await new import_wiki_graph_core12.WikiGraphArchiveFile(
3927
+ new NodeFile(archivePath)
3928
+ ).readDocument(async (document) => await (0, import_wiki_graph_core12.listChapters)(document));
3929
+ const chaptersById = new Map(
3930
+ chapters.map((chapter) => [chapter.chapterId, chapter])
3931
+ );
3932
+ return archiveJobs.flatMap((job) => {
3933
+ const chapter = chaptersById.get(job.chapterId);
3934
+ return chapter === void 0 ? [] : [
3935
+ [
3936
+ job.jobId,
3937
+ {
3938
+ locatedUri: (0, import_wiki_graph_core12.formatLocatedChapterUri)(
3939
+ archivePath,
3940
+ chapter.path
3941
+ ),
3942
+ title: chapter.title,
3943
+ uri: chapter.uri
3944
+ }
3945
+ ]
3946
+ ];
3947
+ });
3948
+ } catch {
3949
+ return [];
3950
+ }
3951
+ })
3952
+ )).flat();
3953
+ return new Map(entries);
3954
+ });
3955
+ }
3956
+ async resolveChapter(job) {
3957
+ return (await this.resolveChapters([job])).get(job.jobId);
3958
+ }
3959
+ async enqueue(options) {
3960
+ return await this.#runtime.run(async () => {
3961
+ const location = await resolveWikiGraphArchiveLocation(options.archive);
3962
+ const target = options.target ?? "reading-summary";
3963
+ await validateQueueTargetConfig(target, options.llmJSON);
3964
+ const selectedExplicitly = options.chapterId !== void 0 || options.chapterIds !== void 0 || options.chapterPath !== void 0 || options.depth !== void 0;
3965
+ const candidates = await new import_wiki_graph_core12.WikiGraphArchiveFile(
3966
+ location.archiveFile
3967
+ ).readDocument(async (document) => {
3968
+ const chapters = await (0, import_wiki_graph_core12.listChapters)(document);
3969
+ const selected = await selectQueueChapters(document, chapters, options);
3970
+ const checked = await Promise.all(
3971
+ selected.map(async (chapter) => ({
3972
+ chapter,
3973
+ reason: await readQueueReadinessReason(document, chapter, target)
3974
+ }))
3975
+ );
3976
+ return checked;
3977
+ });
3978
+ const created = [];
3979
+ const skipped = [];
3980
+ for (const candidate of candidates) {
3981
+ if (candidate.reason !== void 0) {
3982
+ if (selectedExplicitly && candidates.length === 1) {
3983
+ throw new Error(
3984
+ formatQueueReadinessError(
3985
+ candidate.chapter,
3986
+ target,
3987
+ candidate.reason
3988
+ )
3989
+ );
3990
+ }
3991
+ skipped.push({
3992
+ chapter: candidate.chapter,
3993
+ reason: candidate.reason
3994
+ });
3995
+ continue;
3996
+ }
3997
+ try {
3998
+ created.push({
3999
+ chapter: candidate.chapter,
4000
+ job: await this.create({
4001
+ archive: location.archiveFile,
4002
+ boost: options.boost ?? false,
4003
+ chapterId: candidate.chapter.chapterId,
4004
+ ...options.llmJSON === void 0 ? {} : { llmJSON: options.llmJSON },
4005
+ ...options.prompt === void 0 ? {} : { prompt: options.prompt },
4006
+ target
4007
+ })
4008
+ });
4009
+ } catch (error) {
4010
+ if (selectedExplicitly && candidates.length === 1) throw error;
4011
+ skipped.push({
4012
+ chapter: candidate.chapter,
4013
+ reason: error instanceof Error ? error.message : String(error)
4014
+ });
4015
+ }
4016
+ }
4017
+ return { created, skipped };
4018
+ });
4019
+ }
4020
+ async planEnqueue(options) {
4021
+ return await this.#runtime.run(async () => {
4022
+ const location = await resolveWikiGraphArchiveLocation(options.archive);
4023
+ const target = options.target ?? "reading-summary";
4024
+ await validateQueueTargetConfig(target, options.llmJSON);
4025
+ const selectedExplicitly = options.chapterId !== void 0 || options.chapterIds !== void 0 || options.chapterPath !== void 0 || options.depth !== void 0;
4026
+ const candidates = await new import_wiki_graph_core12.WikiGraphArchiveFile(
4027
+ location.archiveFile
4028
+ ).readDocument(async (document) => {
4029
+ const chapters = await (0, import_wiki_graph_core12.listChapters)(document);
4030
+ const selected = await selectQueueChapters(document, chapters, options);
4031
+ return await Promise.all(
4032
+ selected.map(async (chapter) => ({
4033
+ chapter,
4034
+ reason: await readQueueReadinessReason(document, chapter, target)
4035
+ }))
4036
+ );
4037
+ });
4038
+ if (selectedExplicitly && candidates.length === 1 && candidates[0]?.reason !== void 0) {
4039
+ throw new Error(
4040
+ formatQueueReadinessError(
4041
+ candidates[0].chapter,
4042
+ target,
4043
+ candidates[0].reason
4044
+ )
4045
+ );
4046
+ }
4047
+ return {
4048
+ ready: candidates.filter((item) => item.reason === void 0).map((item) => item.chapter),
4049
+ skipped: candidates.flatMap(
4050
+ (item) => item.reason === void 0 ? [] : [{ chapter: item.chapter, reason: item.reason }]
4051
+ )
4052
+ };
4053
+ });
4054
+ }
4055
+ close() {
4056
+ for (const controller of this.#subscriptions) controller.abort();
4057
+ this.#subscriptions.clear();
4058
+ }
4059
+ trackSubscription(controller) {
4060
+ this.#subscriptions.add(controller);
4061
+ return () => this.#subscriptions.delete(controller);
4062
+ }
4063
+ #createHandle(snapshot) {
4064
+ return new WikiGraphJob(this, this.#runtime, this.#backend, snapshot);
4065
+ }
4066
+ };
4067
+ function getBuildJobArchivePath(job) {
4068
+ const record = job;
4069
+ if (record.archive !== void 0) {
4070
+ return getNodeResourcePath(record.archive);
4071
+ }
4072
+ if (typeof record.archivePath === "string") return record.archivePath;
4073
+ throw new TypeError("Build job is missing archive");
4074
+ }
4075
+ var WikiGraphJob = class {
4076
+ #manager;
4077
+ #runtime;
4078
+ #backend;
4079
+ #snapshot;
4080
+ constructor(manager, runtime, backend, snapshot) {
4081
+ this.#manager = manager;
4082
+ this.#runtime = runtime;
4083
+ this.#backend = backend;
4084
+ this.#snapshot = snapshot;
4085
+ }
4086
+ get id() {
4087
+ return this.#snapshot.jobId;
4088
+ }
4089
+ get snapshot() {
4090
+ return this.#snapshot;
4091
+ }
4092
+ async status() {
4093
+ return await this.#update(async () => await this.#backend.get(this.id));
4094
+ }
4095
+ async pause() {
4096
+ return await this.#update(async () => await this.#backend.pause(this.id));
4097
+ }
4098
+ async resume() {
4099
+ return await this.#update(async () => await this.#backend.resume(this.id));
4100
+ }
4101
+ async cancel() {
4102
+ return await this.#update(async () => await this.#backend.cancel(this.id));
4103
+ }
4104
+ async boost() {
4105
+ return await this.#update(async () => await this.#backend.boost(this.id));
4106
+ }
4107
+ async setTarget(target) {
4108
+ return await this.#update(
4109
+ async () => await this.#backend.setTarget(this.id, target)
4110
+ );
4111
+ }
4112
+ async *events(options = {}) {
4113
+ let seenSeq = 0;
4114
+ let cursor = 0;
4115
+ let job = this.#snapshot;
4116
+ if (options.from === "now") {
4117
+ job = await this.status();
4118
+ const chunk = await this.#readEventChunkSafely(job, cursor);
4119
+ const events = chunk.events;
4120
+ cursor = chunk.cursor;
4121
+ seenSeq = events.at(-1)?.seq ?? 0;
4122
+ if (isTerminalState(job.state)) return;
4123
+ }
4124
+ let terminalReadsWithoutEvent = 0;
4125
+ while (options.signal?.aborted !== true) {
4126
+ const chunk = await this.#readEventChunkSafely(job, cursor);
4127
+ const events = chunk.events;
4128
+ cursor = chunk.cursor;
4129
+ let sawTerminalEvent = false;
4130
+ for (const event of events) {
4131
+ if (event.seq <= seenSeq) continue;
4132
+ seenSeq = event.seq;
4133
+ sawTerminalEvent ||= isTerminalEvent(event);
4134
+ yield event;
4135
+ }
4136
+ if (sawTerminalEvent) return;
4137
+ job = await this.status();
4138
+ if (isTerminalState(job.state)) {
4139
+ terminalReadsWithoutEvent += 1;
4140
+ if (terminalReadsWithoutEvent >= TERMINAL_EVENT_GRACE_READS) return;
4141
+ } else {
4142
+ terminalReadsWithoutEvent = 0;
4143
+ }
4144
+ await wait(options.pollIntervalMs ?? 1e3, options.signal);
4145
+ }
4146
+ }
4147
+ subscribe(listener, options = {}) {
4148
+ const controller = new AbortController();
4149
+ const untrack = this.#manager.trackSubscription(controller);
4150
+ const stopForwarding = forwardAbort(options.signal, controller);
4151
+ void (async () => {
4152
+ try {
4153
+ for await (const event of this.events({
4154
+ ...options,
4155
+ signal: controller.signal
4156
+ })) {
4157
+ try {
4158
+ await listener(event);
4159
+ } catch (error) {
4160
+ options.onError?.(error);
4161
+ }
4162
+ }
4163
+ } catch (error) {
4164
+ if (!controller.signal.aborted) options.onError?.(error);
4165
+ } finally {
4166
+ stopForwarding();
4167
+ untrack();
4168
+ }
4169
+ })();
4170
+ return () => controller.abort();
4171
+ }
4172
+ async #update(operation) {
4173
+ this.#snapshot = await this.#runtime.run(operation);
4174
+ return this.#snapshot;
4175
+ }
4176
+ async #readEventChunk(job, cursor) {
4177
+ return await this.#runtime.run(async () => {
4178
+ if (this.#backend.readEventChunk !== void 0) {
4179
+ return await this.#backend.readEventChunk(job, cursor);
4180
+ }
4181
+ const events = await this.#backend.readEvents?.(job);
4182
+ return { cursor: 0, events: events ?? [] };
4183
+ });
4184
+ }
4185
+ async #readEventChunkSafely(job, cursor) {
4186
+ try {
4187
+ return await this.#readEventChunk(job, cursor);
4188
+ } catch {
4189
+ return { cursor, events: [] };
4190
+ }
4191
+ }
4192
+ };
4193
+ async function selectQueueChapters(document, chapters, options) {
4194
+ const ids = options.chapterIds ?? (options.chapterId === void 0 ? void 0 : [options.chapterId]);
4195
+ if (ids !== void 0) {
4196
+ const selected = new Set(ids);
4197
+ for (const id of selected) requireQueueChapter(chapters, id);
4198
+ return chapters.filter((chapter) => selected.has(chapter.chapterId));
4199
+ }
4200
+ if (options.chapterPath === void 0) {
4201
+ return options.depth === void 0 ? chapters : chapters.filter((chapter) => chapter.depth <= options.depth);
4202
+ }
4203
+ const rootId = await (0, import_wiki_graph_core12.resolveChapterPathReadonly)(
4204
+ document,
4205
+ options.chapterPath
4206
+ );
4207
+ const root = requireQueueChapter(chapters, rootId);
4208
+ const prefix = `${root.path}/`;
4209
+ return chapters.filter(
4210
+ (chapter) => chapter.chapterId === rootId || chapter.path.startsWith(prefix) && (options.depth === void 0 || chapter.depth - root.depth <= options.depth)
4211
+ );
4212
+ }
4213
+ async function validateQueueTargetConfig(target, llmJSON) {
4214
+ const config = target === "knowledge-graph" || target === "reading-graph" || target === "reading-summary" ? await loadRequiredStageConfig({
4215
+ ...llmJSON === void 0 ? {} : { llmJSON }
4216
+ }) : await loadWikiGraphRuntimeConfig({
4217
+ ...llmJSON === void 0 ? {} : { llmJSON }
4218
+ });
4219
+ if ((target === "index-embedding-source" || target === "index-embedding-summary") && config.embedding === void 0) {
4220
+ throw new Error(
4221
+ "Missing embeddings configuration. Configure `wikg://local/config/embeddings` before queueing embedding index artifact jobs."
4222
+ );
4223
+ }
4224
+ if (target === "knowledge-graph")
4225
+ requireKnowledgeGraphWikispineConfig(config);
4226
+ }
4227
+ function requireQueueChapter(chapters, chapterId) {
4228
+ const chapter = chapters.find((entry) => entry.chapterId === chapterId);
4229
+ if (chapter === void 0)
4230
+ throw new Error(`Chapter ${chapterId} does not exist.`);
4231
+ return chapter;
4232
+ }
4233
+ async function readQueueReadinessReason(document, chapter, target) {
4234
+ if (chapter.stage === "planned") {
4235
+ return "planned";
4236
+ }
4237
+ if (target === "index-embedding-summary") {
4238
+ const summary = await document.readSummary(chapter.chapterId);
4239
+ if (summary === void 0 || summary.trim() === "") {
4240
+ return "missing summary";
4241
+ }
4242
+ }
4243
+ if (target === "knowledge-graph" || target === "reading-graph") {
4244
+ const [artifact, revision] = await Promise.all([
4245
+ document.indexArtifacts.get(chapter.chapterId, "fts"),
4246
+ document.serials.getRevision(chapter.chapterId)
4247
+ ]);
4248
+ if (artifact?.sourceRevision !== revision) {
4249
+ return "missing current FTS index artifact";
4250
+ }
4251
+ }
4252
+ return void 0;
4253
+ }
4254
+ function formatQueueReadinessError(chapter, target, reason) {
4255
+ if (reason === "planned") {
4256
+ return `Chapter ${chapter.uri} is planned. Set source before queueing a build job.`;
4257
+ }
4258
+ if (reason === "missing summary") {
4259
+ return `Chapter ${chapter.uri} has no summary. Build a reading summary before queueing a summary embedding index artifact job.`;
4260
+ }
4261
+ if (reason === "missing current FTS index artifact") {
4262
+ return `Chapter ${chapter.uri} needs a current FTS index artifact before queueing ${target}.`;
4263
+ }
4264
+ return reason;
4265
+ }
4266
+ async function wait(milliseconds, signal) {
4267
+ if (signal?.aborted === true) return;
4268
+ await new Promise((resolve3) => {
4269
+ let settled = false;
4270
+ const complete = () => {
4271
+ if (settled) return;
4272
+ settled = true;
4273
+ clearTimeout(timeout);
4274
+ signal?.removeEventListener("abort", complete);
4275
+ resolve3();
4276
+ };
4277
+ const timeout = setTimeout(complete, milliseconds);
4278
+ signal?.addEventListener("abort", complete, { once: true });
4279
+ if (signal?.aborted === true) complete();
4280
+ });
4281
+ }
4282
+ function isTerminalState(state) {
4283
+ return TERMINAL_STATES.has(state);
4284
+ }
4285
+ function isTerminalEvent(event) {
4286
+ return event.type === "canceled" || event.type === "failed" || event.type === "succeeded";
4287
+ }
4288
+ function forwardAbort(source, target) {
4289
+ if (source === void 0) return () => void 0;
4290
+ const abort = () => target.abort(source.reason);
4291
+ if (source.aborted) abort();
4292
+ else source.addEventListener("abort", abort, { once: true });
4293
+ return () => source.removeEventListener("abort", abort);
4294
+ }
4295
+
4296
+ // src/libraries.ts
4297
+ var import_wiki_graph_core13 = require("wiki-graph-core");
4298
+ var import_promises4 = require("fs/promises");
4299
+ var WikiGraphLibraryManager = class {
4300
+ #runtime;
4301
+ constructor(runtime) {
4302
+ this.#runtime = runtime;
4303
+ }
4304
+ async create(folder) {
4305
+ const record = await this.#runtime.run(async () => {
4306
+ const path = resolveWikiGraphRuntimePath(folder);
4307
+ await (0, import_promises4.mkdir)(path);
4308
+ return await (0, import_wiki_graph_core13.createWikiGraphLibrary)({ folder: new NodeDirectory(path) });
4309
+ });
4310
+ return new WikiGraphLibrary(this.#runtime, record);
4311
+ }
4312
+ async get(uri = (0, import_wiki_graph_core13.formatWikiGraphLibraryUri)()) {
4313
+ const target = requireLibraryTarget(uri);
4314
+ const record = await this.#runtime.run(
4315
+ async () => await (0, import_wiki_graph_core13.resolveWikiGraphLibrary)(target)
4316
+ );
4317
+ return new WikiGraphLibrary(this.#runtime, record);
4318
+ }
4319
+ async addArchive(options) {
4320
+ return await this.#runtime.run(
4321
+ async () => await (0, import_wiki_graph_core13.addWikiGraphLibraryArchive)({
4322
+ inputFile: new NodeFile(
4323
+ resolveWikiGraphRuntimePath(options.inputPath)
4324
+ ),
4325
+ target: requireLibraryTarget(options.target),
4326
+ ...options.to === void 0 ? {} : { to: options.to }
4327
+ })
4328
+ );
4329
+ }
4330
+ async assertCurrent(target) {
4331
+ await this.#runtime.run(
4332
+ async () => await (0, import_wiki_graph_core13.assertWikiGraphLibrarySchemaCurrent)(requireLibraryTarget(target))
4333
+ );
4334
+ }
4335
+ async rebind(target, folder) {
4336
+ return await this.#runtime.run(
4337
+ async () => await (0, import_wiki_graph_core13.rebindWikiGraphLibrary)({
4338
+ folder: new NodeDirectory(resolveWikiGraphRuntimePath(folder)),
4339
+ target: requireLibraryTarget(target)
4340
+ })
4341
+ );
4342
+ }
4343
+ async remove(target) {
4344
+ return await this.#runtime.run(
4345
+ async () => await (0, import_wiki_graph_core13.removeWikiGraphLibrary)(requireLibraryTarget(target))
4346
+ );
4347
+ }
4348
+ async scan(target) {
4349
+ return await this.#runtime.run(
4350
+ async () => await (0, import_wiki_graph_core13.scanWikiGraphLibrary)(requireLibraryTarget(target))
4351
+ );
4352
+ }
4353
+ async getArchive(target) {
4354
+ return await this.#runtime.run(
4355
+ async () => await (0, import_wiki_graph_core13.getWikiGraphLibraryArchive)(requireLibraryTarget(target))
4356
+ );
4357
+ }
4358
+ async moveArchive(options) {
4359
+ return await this.#runtime.run(
4360
+ async () => await (0, import_wiki_graph_core13.moveWikiGraphLibraryArchive)({
4361
+ target: requireLibraryTarget(options.target),
4362
+ to: options.to
4363
+ })
4364
+ );
4365
+ }
4366
+ async removeArchive(target) {
4367
+ return await this.#runtime.run(
4368
+ async () => await (0, import_wiki_graph_core13.removeWikiGraphLibraryArchive)({
4369
+ target: requireLibraryTarget(target)
4370
+ })
4371
+ );
4372
+ }
4373
+ async indexState(target) {
4374
+ return await this.#runtime.run(
4375
+ async () => await (0, import_wiki_graph_core13.readWikiGraphLibraryIndexState)(requireLibraryTarget(target))
4376
+ );
4377
+ }
4378
+ async rebuildIndex(target, onProgress) {
4379
+ return await this.#runtime.run(
4380
+ async () => await (0, import_wiki_graph_core13.rebuildWikiGraphLibraryIndex)(
4381
+ requireLibraryTarget(target),
4382
+ onProgress
4383
+ )
4384
+ );
4385
+ }
4386
+ async cleanIndex(target) {
4387
+ return await this.#runtime.run(
4388
+ async () => await (0, import_wiki_graph_core13.cleanWikiGraphLibraryIndex)(requireLibraryTarget(target))
4389
+ );
4390
+ }
4391
+ async getMetadata(target) {
4392
+ return await this.#runtime.run(
4393
+ async () => await (0, import_wiki_graph_core13.getWikiGraphLibraryMetadata)(requireLibraryTarget(target))
4394
+ );
4395
+ }
4396
+ async replaceMetadata(target, value) {
4397
+ return await this.#runtime.run(
4398
+ async () => await (0, import_wiki_graph_core13.replaceWikiGraphLibraryMetadata)(
4399
+ requireLibraryTarget(target),
4400
+ value
4401
+ )
4402
+ );
4403
+ }
4404
+ async putMetadata(target, key, value) {
4405
+ return await this.#runtime.run(
4406
+ async () => await (0, import_wiki_graph_core13.putWikiGraphLibraryMetadata)(
4407
+ requireLibraryTarget(target),
4408
+ key,
4409
+ value
4410
+ )
4411
+ );
4412
+ }
4413
+ async deleteMetadata(target, key) {
4414
+ return await this.#runtime.run(
4415
+ async () => await (0, import_wiki_graph_core13.deleteWikiGraphLibraryMetadataKey)(
4416
+ requireLibraryTarget(target),
4417
+ key
4418
+ )
4419
+ );
4420
+ }
4421
+ async clearMetadata(target) {
4422
+ return await this.#runtime.run(
4423
+ async () => await (0, import_wiki_graph_core13.clearWikiGraphLibraryMetadata)(requireLibraryTarget(target))
4424
+ );
4425
+ }
4426
+ async search(target, query, options = {}) {
4427
+ return await this.#runtime.run(async () => {
4428
+ const embeddingProvider = await createConfiguredEmbeddingProvider();
4429
+ return await (0, import_wiki_graph_core13.findWikiGraphLibraryObjects)(
4430
+ requireLibraryTarget(target),
4431
+ query,
4432
+ {
4433
+ ...options,
4434
+ ...embeddingProvider === void 0 ? {} : { embeddingProvider }
4435
+ }
4436
+ );
4437
+ });
4438
+ }
4439
+ async searchArchiveMembers(target, query, options = {}) {
4440
+ return await this.#runtime.run(async () => {
4441
+ const embeddingProvider = await createConfiguredEmbeddingProvider();
4442
+ return await (0, import_wiki_graph_core13.findWikiGraphLibraryArchiveMembers)(
4443
+ requireLibraryTarget(target),
4444
+ query,
4445
+ {
4446
+ ...options,
4447
+ ...embeddingProvider === void 0 ? {} : { embeddingProvider }
4448
+ }
4449
+ );
4450
+ });
4451
+ }
4452
+ async objects(target, options = {}) {
4453
+ return await this.#runtime.run(
4454
+ async () => await (0, import_wiki_graph_core13.listWikiGraphLibraryObjects)(
4455
+ requireLibraryTarget(target),
4456
+ options
4457
+ )
4458
+ );
4459
+ }
4460
+ async archiveMembers(target, options = {}) {
4461
+ return await this.#runtime.run(
4462
+ async () => await (0, import_wiki_graph_core13.listWikiGraphLibraryArchiveMembers)(
4463
+ requireLibraryTarget(target),
4464
+ options
4465
+ )
4466
+ );
4467
+ }
4468
+ async page(target, objectUri, options = {}) {
4469
+ return await this.#runtime.run(
4470
+ async () => await withConfiguredWikimediaResolver(
4471
+ objectUri,
4472
+ async (wikimediaOptions) => await (0, import_wiki_graph_core13.readWikiGraphLibraryPage)(
4473
+ requireLibraryTarget(target),
4474
+ objectUri,
4475
+ { ...options, ...wikimediaOptions }
4476
+ )
4477
+ )
4478
+ );
4479
+ }
4480
+ async related(target, objectUri, options = {}) {
4481
+ return await this.#runtime.run(
4482
+ async () => await (0, import_wiki_graph_core13.listRelatedWikiGraphLibraryObjects)(
4483
+ requireLibraryTarget(target),
4484
+ objectUri,
4485
+ options
4486
+ )
4487
+ );
4488
+ }
4489
+ async evidence(target, objectUri, options = {}) {
4490
+ return await this.#runtime.run(
4491
+ async () => await (0, import_wiki_graph_core13.listWikiGraphLibraryEvidence)(
4492
+ requireLibraryTarget(target),
4493
+ objectUri,
4494
+ options
4495
+ )
4496
+ );
4497
+ }
4498
+ async pack(target, objectUri, budget = 5e3) {
4499
+ return await this.#runtime.run(
4500
+ async () => await (0, import_wiki_graph_core13.packWikiGraphLibraryContext)(
4501
+ requireLibraryTarget(target),
4502
+ objectUri,
4503
+ budget
4504
+ )
4505
+ );
4506
+ }
4507
+ async list() {
4508
+ const records = await this.#runtime.run(
4509
+ async () => await (0, import_wiki_graph_core13.listWikiGraphLibraries)()
4510
+ );
4511
+ return records.map((record) => new WikiGraphLibrary(this.#runtime, record));
4512
+ }
4513
+ };
4514
+ var WikiGraphLibrary = class {
4515
+ #runtime;
4516
+ #record;
4517
+ constructor(runtime, record) {
4518
+ this.#runtime = runtime;
4519
+ this.#record = record;
4520
+ }
4521
+ get snapshot() {
4522
+ return this.#record;
4523
+ }
4524
+ get uri() {
4525
+ return (0, import_wiki_graph_core13.formatWikiGraphLibraryUri)(
4526
+ this.#record.isDefault ? void 0 : this.#record.publicId
4527
+ );
4528
+ }
4529
+ async archives() {
4530
+ const target = requireLibraryTarget(this.uri);
4531
+ return await this.#runtime.run(
4532
+ async () => await (0, import_wiki_graph_core13.listWikiGraphLibraryArchives)(target)
4533
+ );
4534
+ }
4535
+ async scan() {
4536
+ const target = requireLibraryTarget(this.uri);
4537
+ return await this.#runtime.run(
4538
+ async () => await (0, import_wiki_graph_core13.scanWikiGraphLibrary)(target)
4539
+ );
4540
+ }
4541
+ async remove() {
4542
+ const target = requireLibraryTarget(this.uri);
4543
+ return await this.#runtime.run(
4544
+ async () => await (0, import_wiki_graph_core13.removeWikiGraphLibrary)(target)
4545
+ );
4546
+ }
4547
+ };
4548
+ function requireLibraryTarget(value) {
4549
+ if (typeof value !== "string") return value;
4550
+ const target = (0, import_wiki_graph_core13.parseWikiGraphLibraryUri)(value);
4551
+ if (target === void 0) {
4552
+ throw new TypeError(`Invalid Wiki Graph library URI: ${value}`);
4553
+ }
4554
+ return target;
4555
+ }
4556
+
4557
+ // src/maintenance.ts
4558
+ var import_wiki_graph_core14 = require("wiki-graph-core");
4559
+ var import_path4 = require("path");
4560
+ var WikiGraphMaintenanceManager = class {
4561
+ #runtime;
4562
+ constructor(runtime) {
4563
+ this.#runtime = runtime;
4564
+ }
4565
+ async upgrade(target) {
4566
+ return await this.#runtime.run(
4567
+ async () => {
4568
+ if (target.kind === "home") {
4569
+ return await (0, import_wiki_graph_core14.upgradeWikiGraphMaintenanceTarget)({ kind: "home" });
4570
+ }
4571
+ if (target.kind === "library") {
4572
+ const parsed = (0, import_wiki_graph_core14.parseWikiGraphLibraryUri)(target.uri);
4573
+ if (parsed === void 0 || parsed.kind === "archive") {
4574
+ throw new Error(
4575
+ `Invalid Wiki Graph library upgrade target: ${target.uri}`
4576
+ );
4577
+ }
4578
+ return await (0, import_wiki_graph_core14.upgradeWikiGraphMaintenanceTarget)({
4579
+ kind: "library",
4580
+ target: parsed
4581
+ });
4582
+ }
4583
+ const path = resolveWikiGraphRuntimePath(target.path);
4584
+ return await (0, import_wiki_graph_core14.upgradeWikiGraphMaintenanceTarget)({
4585
+ file: new NodeFile(path),
4586
+ kind: "archive"
4587
+ });
4588
+ },
4589
+ void 0,
4590
+ { skipHomeBootstrap: target.kind === "home" }
4591
+ );
4592
+ }
4593
+ async migrateLegacy(inputPath, outputPath = defaultWikgOutputPath(inputPath)) {
4594
+ return await this.#runtime.run(async () => {
4595
+ const resolvedInputPath = resolveWikiGraphRuntimePath(inputPath);
4596
+ const resolvedOutputPath = resolveWikiGraphRuntimePath(outputPath);
4597
+ await (0, import_wiki_graph_core14.migrateLegacySdpubToWikg)(
4598
+ new NodeFile(resolvedInputPath),
4599
+ new NodeFile(resolvedOutputPath, (0, import_path4.basename)(resolvedOutputPath))
4600
+ );
4601
+ return {
4602
+ inputPath: resolvedInputPath,
4603
+ outputPath: resolvedOutputPath
4604
+ };
4605
+ });
4606
+ }
4607
+ };
4608
+ function defaultWikgOutputPath(inputPath) {
4609
+ return inputPath.toLowerCase().endsWith(".sdpub") ? `${inputPath.slice(0, -".sdpub".length)}.wikg` : `${inputPath}.wikg`;
4610
+ }
4611
+
4612
+ // src/sdk.ts
4613
+ var import_wiki_graph_core15 = require("wiki-graph-core");
4614
+ var WikiGraphSDK = class {
4615
+ #context;
4616
+ #homeReady;
4617
+ archives;
4618
+ conversions;
4619
+ config;
4620
+ continuations;
4621
+ jobs;
4622
+ libraries;
4623
+ maintenance;
4624
+ constructor(options = {}) {
4625
+ ensureNodeWikiGraphPlatform();
4626
+ this.#context = {
4627
+ cwd: options.cwd ?? process.cwd(),
4628
+ env: { ...process.env, ...options.env },
4629
+ envPolicy: options.envPolicy ?? "production",
4630
+ ...options.stateDir === void 0 ? {} : { stateDir: options.stateDir }
4631
+ };
4632
+ this.archives = new WikiGraphArchiveManager(this);
4633
+ this.conversions = new WikiGraphConversionManager(this);
4634
+ this.config = new WikiGraphConfigManager(this);
4635
+ this.continuations = new WikiGraphContinuationManager(this);
4636
+ this.jobs = new WikiGraphJobManager(this);
4637
+ this.libraries = new WikiGraphLibraryManager(this);
4638
+ this.maintenance = new WikiGraphMaintenanceManager(this);
4639
+ }
4640
+ async run(operation, signal, options = {}) {
4641
+ throwIfAborted(signal);
4642
+ const runWithContext = async () => await withWikiGraphSDKRuntimeContext(this.#context, async () => {
4643
+ if (options.skipHomeBootstrap !== true) {
4644
+ this.#homeReady ??= (0, import_wiki_graph_core15.ensureWikiGraphHomeSchemaCurrent)().catch(
4645
+ (error) => {
4646
+ this.#homeReady = void 0;
4647
+ throw error;
4648
+ }
4649
+ );
4650
+ await this.#homeReady;
4651
+ }
4652
+ return await operation();
4653
+ });
4654
+ const result = this.#context.stateDir === void 0 ? await runWithContext() : await withNodeWikiGraphStorage(
4655
+ this.#context.stateDir,
4656
+ runWithContext
4657
+ );
4658
+ throwIfAborted(signal);
4659
+ return result;
4660
+ }
4661
+ close() {
4662
+ this.jobs.close();
4663
+ }
4664
+ };
4665
+ function throwIfAborted(signal) {
4666
+ if (signal?.aborted !== true) return;
4667
+ throw signal.reason instanceof Error ? signal.reason : new Error("The Wiki Graph SDK operation was aborted.");
4668
+ }
4669
+ function createWikiGraphSDK(options = {}) {
4670
+ return new WikiGraphSDK(options);
4671
+ }
4672
+
4673
+ // src/index.ts
4674
+ var import_wiki_graph_core16 = require("wiki-graph-core");
4675
+ // Annotate the CommonJS export names for ESM import in node:
4676
+ 0 && (module.exports = {
4677
+ DANGEROUS_RUNTIME_ENV_NAMES,
4678
+ DEFAULT_EXTRACTION_PROMPT,
4679
+ DEFAULT_GENERATION_JOB_CONCURRENCY,
4680
+ DEFAULT_GENERATION_REQUEST_CONCURRENCY,
4681
+ DEFAULT_KNOWLEDGE_GRAPH_RECALL_PROMPT,
4682
+ LLMPaymentRequiredError,
4683
+ LOCAL_CONFIG_SECTIONS,
4684
+ NodeDirectory,
4685
+ NodeFile,
4686
+ WIKI_GRAPH_URI_PREFIX,
4687
+ WikiGraphArchiveExistsError,
4688
+ WikiGraphArchiveHandle,
4689
+ WikiGraphArchiveManager,
4690
+ WikiGraphConfigManager,
4691
+ WikiGraphContinuationManager,
4692
+ WikiGraphConversionManager,
4693
+ WikiGraphError,
4694
+ WikiGraphJob,
4695
+ WikiGraphJobManager,
4696
+ WikiGraphLibrary,
4697
+ WikiGraphLibraryManager,
4698
+ WikiGraphMaintenanceManager,
4699
+ WikiGraphSDK,
4700
+ buildSearchIndexEmbeddingProvider,
4701
+ buildWikiGraphLLMOptions,
4702
+ clearLocalConfigSection,
4703
+ createEntryRuntimeContext,
4704
+ createGenerationPerformanceHints,
4705
+ createNodeTemplateEnvironment,
4706
+ createNodeWikiGraphStorage,
4707
+ createStageLLM,
4708
+ createWikiGraphArchiveInspectReport,
4709
+ createWikiGraphSDK,
4710
+ deleteLocalConfigValue,
4711
+ embedQueryText,
4712
+ embedQueryTextWithLocalConfig,
4713
+ ensureNodeWikiGraphPlatform,
4714
+ formatError,
4715
+ formatGenerationPlanningDuration,
4716
+ formatGenerationPlanningModel,
4717
+ formatLocatedChapterResourceUri,
4718
+ formatLocatedChapterSourceCollectionUri,
4719
+ formatLocatedChapterUri,
4720
+ formatLocatedWikiGraphUri,
4721
+ formatWikiGraphCommandUri,
4722
+ formatWikiGraphLibraryUri,
4723
+ getNodeResourcePath,
4724
+ getWikiGraphSDKRuntimeContext,
4725
+ getWikiGraphSDKStateDir,
4726
+ installNodeWikiGraphPlatform,
4727
+ isSourceLocatorScopeUri,
4728
+ isWikiGraphLibraryUri,
4729
+ loadRequiredStageConfig,
4730
+ loadWikiGraphRuntimeConfig,
4731
+ maskLocalConfigSection,
4732
+ nodeWikiGraphPlatform,
4733
+ nodeWikispineCommandRunner,
4734
+ normalizeLocalConfigKey,
4735
+ parseChapterPath,
4736
+ parseChapterTreeInput,
4737
+ parseChapterUriPath,
4738
+ parseLocalConfigSection,
4739
+ parseLocatedWikiGraphUri,
4740
+ parseSourceTextJsonl,
4741
+ parseWikiGraphLibraryUri,
4742
+ parseWikiGraphUriSyntax,
4743
+ planGenerationTask,
4744
+ putLocalConfigValue,
4745
+ readLocalConfigSection,
4746
+ readWikiGraphEmbeddingConfig,
4747
+ replaceLocalConfigSection,
4748
+ requireKnowledgeGraphWikispineConfig,
4749
+ resolveExtractionPrompt,
4750
+ resolveHostedProviderEndpoint,
4751
+ resolveKnowledgeGraphRecallPrompt,
4752
+ resolveWikiGraphArchiveLocation,
4753
+ resolveWikispineConfig,
4754
+ runWikiGraphQueueWorker,
4755
+ validateLocalConfigSection,
4756
+ withNodeWikiGraphStorage,
4757
+ withWikiGraphSDKRuntimeContext,
4758
+ writeNodeFileHandle
4759
+ });
4760
+ //# sourceMappingURL=index.cjs.map