wiki-graph-sdk 0.7.0 → 0.7.1

package/dist/default-worker.cjs

@@ -35,7 +35,7 @@ __export(default_worker_exports, {
   runWikiGraphQueueWorker: () => runWikiGraphQueueWorker
 });
 module.exports = __toCommonJS(default_worker_exports);
-var import_wiki_graph_core5 = require("wiki-graph-core");
+var import_wiki_graph_core6 = require("wiki-graph-core");
 var import_worker3 = require("wiki-graph-core/worker");
 
 // src/embedding.ts
@@ -1444,6 +1444,7 @@ function abortError(signal) {
 
 // src/worker.ts
 var import_worker = require("wiki-graph-core/worker");
+var import_wiki_graph_core5 = require("wiki-graph-core");
 var import_worker2 = require("wiki-graph-core/worker");
 async function runBuildJobWorker(options) {
   installNodeWikiGraphPlatform();
@@ -1460,7 +1461,7 @@ async function runWikiGraphQueueWorker(options = {}) {
   await runBuildJobWorker({
     concurrency: config.concurrent?.job ?? DEFAULT_GENERATION_JOB_CONCURRENCY,
     executeJob: async (job, reporter, context) => {
-      await (0, import_wiki_graph_core5.withLoggingContext)(
+      await (0, import_wiki_graph_core6.withLoggingContext)(
         { logDirectory: job.log, operation: "build-job" },
         async () => await executeBuildJob(job, reporter, context)
       );
@@ -1483,7 +1484,7 @@ async function executeBuildJob(job, reporter, context) {
       let stage = await readChapterStage(job);
       if (stage === "sourced") {
         await executeStep(job, "reading-graph", execution, reporter, context, {
-          ...execution.extractionPrompt === void 0 ? {} : { extractionPrompt: execution.extractionPrompt }
+          ...execution.extractionPrompt === void 0 ? {} : { prompt: execution.extractionPrompt }
         });
         stage = await readChapterStage(job);
       }
@@ -1516,8 +1517,8 @@ async function executeBuildJob(job, reporter, context) {
       }
     }
     await executeStep(job, job.target, execution, reporter, context, {
-      ...job.target === "reading-graph" ? { extractionPrompt: execution.extractionPrompt } : {},
-      ...job.target === "knowledge-graph" ? { policyPrompt: execution.knowledgeGraphPrompt } : {}
+      ...job.target === "reading-graph" ? { prompt: execution.extractionPrompt } : {},
+      ...job.target === "knowledge-graph" ? { prompt: execution.knowledgeGraphPrompt } : {}
     });
   } finally {
     await execution.close();
@@ -1532,9 +1533,9 @@ async function executeStep(job, kind, execution, reporter, context, inputOptions
   const stepWorkspace = await resetStepWorkspace(job.workspace, kind);
   const inputFile = await stepWorkspace.createFile("input.jsonl");
   const outputWorkspace = await stepWorkspace.createDirectory("output");
-  const revision = await new import_wiki_graph_core5.WikiGraphArchiveFile(job.archive).readDocument(
+  const revision = await new import_wiki_graph_core6.WikiGraphArchiveFile(job.archive).readDocument(
     async (document) => {
-      await (0, import_wiki_graph_core5.getChapterDetails)(document, job.chapterId);
+      await (0, import_wiki_graph_core6.getChapterDetails)(document, job.chapterId);
       return await (0, import_worker3.writeChapterJobInputFile)(
         document,
         job.chapterId,
@@ -1544,7 +1545,7 @@ async function executeStep(job, kind, execution, reporter, context, inputOptions
       );
     }
   );
-  await (0, import_wiki_graph_core5.recordBuildJobInputRevision)({
+  await (0, import_wiki_graph_core6.recordBuildJobInputRevision)({
     currentRevision: revision,
     jobId: job.jobId,
     ownerId: requireRunningJobOwnerId(job)
@@ -1574,8 +1575,8 @@ async function executeStep(job, kind, execution, reporter, context, inputOptions
     total: 1,
     unit: "item"
   });
-  await new import_wiki_graph_core5.WikiGraphArchiveFile(job.archive).write(async (document) => {
-    assertJobStillRunning(await (0, import_wiki_graph_core5.getBuildJob)(job.jobId));
+  await new import_wiki_graph_core6.WikiGraphArchiveFile(job.archive).write(async (document) => {
+    assertJobStillRunning(await (0, import_wiki_graph_core6.getBuildJob)(job.jobId));
     await (0, import_worker3.applyChapterJobArtifactFile)(
       document,
       job.chapterId,
@@ -1591,7 +1592,7 @@ async function executeStep(job, kind, execution, reporter, context, inputOptions
     unit: "item"
   });
   await reporter.stepCompleted(kind);
-  assertJobStillRunning(await (0, import_wiki_graph_core5.getBuildJob)(job.jobId));
+  assertJobStillRunning(await (0, import_wiki_graph_core6.getBuildJob)(job.jobId));
 }
 async function openJobExecutor(job, config, signal, reporter) {
   const generation = !job.target.startsWith("index-");
@@ -1610,13 +1611,13 @@ async function openJobExecutor(job, config, signal, reporter) {
   const request = llm === void 0 ? void 0 : async (messages, index, maxRetries) => await llm.request(messages, {
     retryIndex: index,
     retryMax: maxRetries,
-    scope: import_wiki_graph_core5.WikiGraphScope.ReaderExtraction,
+    scope: import_wiki_graph_core6.WikiGraphScope.ReaderExtraction,
     signal
   });
   if (request !== void 0 && llm !== void 0) {
     request.lazy = async (operation) => await llm.request(async () => await operation(request));
   }
-  const wikimediaResolver = job.target !== "knowledge-graph" ? void 0 : await (0, import_wiki_graph_core5.openWikimediaResolver)(
+  const wikimediaResolver = job.target !== "knowledge-graph" ? void 0 : await (0, import_wiki_graph_core6.openWikimediaResolver)(
     config.wikimedia === void 0 ? {
       kind: "local",
       llmRequest: requireValue(
@@ -1659,12 +1660,12 @@ async function resetStepWorkspace(workspace, kind) {
   return await workspace.createDirectory(name);
 }
 async function readChapterStage(job) {
-  return await new import_wiki_graph_core5.WikiGraphArchiveFile(job.archive).readDocument(
-    async (document) => (await (0, import_wiki_graph_core5.getChapterDetails)(document, job.chapterId)).stage
+  return await new import_wiki_graph_core6.WikiGraphArchiveFile(job.archive).readDocument(
+    async (document) => (await (0, import_wiki_graph_core6.getChapterDetails)(document, job.chapterId)).stage
   );
 }
 async function assertCurrentFtsArtifact(job) {
-  await new import_wiki_graph_core5.WikiGraphArchiveFile(job.archive).readDocument(async (document) => {
+  await new import_wiki_graph_core6.WikiGraphArchiveFile(job.archive).readDocument(async (document) => {
     const revision = await document.serials.getRevision(job.chapterId);
     const artifact = await document.indexArtifacts.get(job.chapterId, "fts");
     if (artifact?.sourceRevision !== revision) {