wiki-graph-sdk 0.8.1 → 0.8.2

package/dist/index.cjs

@@ -61,6 +61,7 @@ __export(src_exports, {
   createEntryRuntimeContext: () => createEntryRuntimeContext,
   createGenerationPerformanceHints: () => createGenerationPerformanceHints,
   createNodeTemplateEnvironment: () => createNodeTemplateEnvironment,
+  createNodeWikiGraphPlatform: () => createNodeWikiGraphPlatform,
   createNodeWikiGraphStorage: () => createNodeWikiGraphStorage,
   createStageLLM: () => createStageLLM,
   createWikiGraphSDK: () => createWikiGraphSDK,
@@ -75,6 +76,8 @@ __export(src_exports, {
   formatLocatedChapterSourceCollectionUri: () => import_wiki_graph_core19.formatLocatedChapterSourceCollectionUri,
   formatLocatedChapterUri: () => import_wiki_graph_core19.formatLocatedChapterUri,
   formatLocatedWikiGraphUri: () => import_wiki_graph_core19.formatLocatedWikiGraphUri,
+  formatSourceArtifactUri: () => import_wiki_graph_core19.formatSourceArtifactUri,
+  formatSourceLocatorFragment: () => import_wiki_graph_core19.formatSourceLocatorFragment,
   formatWikiGraphCommandUri: () => import_wiki_graph_core19.formatWikiGraphCommandUri,
   formatWikiGraphLibraryUri: () => import_wiki_graph_core19.formatWikiGraphLibraryUri,
   getNodeResourcePath: () => getNodeResourcePath,
@@ -84,7 +87,9 @@ __export(src_exports, {
   inspectWikiGraphArchive: () => inspectWikiGraphArchive,
   installNodeWikiGraphPlatform: () => installNodeWikiGraphPlatform,
   isSourceLocatorScopeUri: () => import_wiki_graph_core19.isSourceLocatorScopeUri,
+  isWikiGraphJobUri: () => import_wiki_graph_core19.isWikiGraphJobUri,
   isWikiGraphLibraryUri: () => import_wiki_graph_core19.isWikiGraphLibraryUri,
+  isWikiGraphUri: () => import_wiki_graph_core19.isWikiGraphUri,
   loadRequiredStageConfig: () => loadRequiredStageConfig,
   loadWikiGraphRuntimeConfig: () => loadWikiGraphRuntimeConfig,
   maskLocalConfigSection: () => maskLocalConfigSection,
@@ -96,6 +101,7 @@ __export(src_exports, {
   parseChapterUriPath: () => import_wiki_graph_core19.parseChapterUriPath,
   parseLocalConfigSection: () => parseLocalConfigSection,
   parseLocatedWikiGraphUri: () => import_wiki_graph_core19.parseLocatedWikiGraphUri,
+  parseSourceLocatorFragment: () => import_wiki_graph_core19.parseSourceLocatorFragment,
   parseSourceTextJsonl: () => import_wiki_graph_core19.parseSourceTextJsonl,
   parseWikiGraphLibraryUri: () => import_wiki_graph_core19.parseWikiGraphLibraryUri,
   parseWikiGraphUriSyntax: () => import_wiki_graph_core19.parseWikiGraphUriSyntax,
@@ -104,7 +110,10 @@ __export(src_exports, {
   readLocalConfigSection: () => readLocalConfigSection,
   readWikiGraphEmbeddingConfig: () => readWikiGraphEmbeddingConfig,
   replaceLocalConfigSection: () => replaceLocalConfigSection,
+  requireArchiveUri: () => import_wiki_graph_core19.requireArchiveUri,
   requireKnowledgeGraphWikispineConfig: () => requireKnowledgeGraphWikispineConfig,
+  requireLocatedObjectOrArchiveUri: () => import_wiki_graph_core19.requireLocatedObjectOrArchiveUri,
+  requireLocatedObjectUri: () => import_wiki_graph_core19.requireLocatedObjectUri,
   resolveExtractionPrompt: () => import_wiki_graph_core10.resolveExtractionPrompt,
   resolveHostedProviderEndpoint: () => resolveHostedProviderEndpoint,
   resolveKnowledgeGraphRecallPrompt: () => import_wiki_graph_core10.resolveKnowledgeGraphRecallPrompt,
@@ -836,6 +845,9 @@ var nodeWikiGraphPlatform = {
     write: writeNodeZip
   }
 };
+function createNodeWikiGraphPlatform(options = {}) {
+  return options.lifecycle === void 0 ? nodeWikiGraphPlatform : { ...nodeWikiGraphPlatform, lifecycle: options.lifecycle };
+}
 var NodeTemplateLoader = class extends import_nunjucks.Loader {
   #dataDirectory;
   constructor(dataDirectory) {
@@ -881,10 +893,15 @@ function resolveNodeDataDirectory() {
     directory = parent;
   }
 }
-function createNodeWikiGraphStorage(stateRoot = pathModule.join(os.homedir(), ".wikigraph")) {
+function createNodeWikiGraphStorage(stateRootOrOptions = pathModule.join(
+  os.homedir(),
+  ".wikigraph"
+)) {
+  const libraryRoot = typeof stateRootOrOptions === "string" ? stateRootOrOptions : stateRootOrOptions.libraryRoot;
+  const documentStoreRoot = typeof stateRootOrOptions === "string" ? pathModule.join(stateRootOrOptions, "documents") : stateRootOrOptions.documentStoreRoot;
   return {
-    library: new NodeDirectory(stateRoot),
-    documentStore: new NodeDirectory(pathModule.join(stateRoot, "documents"))
+    library: new NodeDirectory(libraryRoot),
+    documentStore: new NodeDirectory(documentStoreRoot)
   };
 }
 function installNodeWikiGraphPlatform(stateRoot) {
@@ -892,7 +909,11 @@ function installNodeWikiGraphPlatform(stateRoot) {
   (0, import_platform.installWikiGraphStorage)(createNodeWikiGraphStorage(stateRoot));
 }
 function ensureNodeWikiGraphPlatform() {
-  (0, import_platform.installWikiGraphPlatform)(nodeWikiGraphPlatform);
+  try {
+    (0, import_platform.getWikiGraphPlatform)();
+  } catch {
+    (0, import_platform.installWikiGraphPlatform)(nodeWikiGraphPlatform);
+  }
   try {
     (0, import_platform.getWikiGraphStorage)();
   } catch {
@@ -2364,6 +2385,13 @@ var WikiGraphArchiveHandle = class {
     );
     return status;
   }
+  /** Clear cached search sessions without changing the archive index. */
+  async clearSearchSessions(options = {}) {
+    await this.#runtime.run(
+      async () => await (0, import_wiki_graph_core7.deleteArchiveSearchSessions)(this.path),
+      options.signal
+    );
+  }
   async listChapters(options = {}) {
     return await this.#readDocument(
       async (document) => await (0, import_wiki_graph_core7.listChapters)(document),
@@ -3294,6 +3322,8 @@ async function createFindOptions(cursor, limit) {
     cursor: cursor.cursor,
     ...cursor.evidenceLimit === void 0 ? {} : { evidenceLimit: cursor.evidenceLimit },
     limit,
+    ...cursor.match === void 0 ? {} : { match: cursor.match },
+    ...cursor.order === void 0 ? {} : { order: cursor.order },
     ...cursor.queryMode === void 0 ? {} : { queryMode: cursor.queryMode },
     ...cursor.skipUnindexed === void 0 ? {} : { skipUnindexed: cursor.skipUnindexed },
     ...cursor.sourceContext === void 0 ? {} : { sourceContext: cursor.sourceContext },
@@ -3397,6 +3427,8 @@ function createContinuationPayload(context, cursor) {
         indexScope: context.indexScope,
         kind: "search",
         ...createLibraryQueryPayload(context),
+        ...context.match === void 0 ? {} : { match: context.match },
+        ...context.order === void 0 ? {} : { order: context.order },
         ...context.query === void 0 ? {} : { query: context.query },
         ...context.queryMode === void 0 ? {} : { queryMode: context.queryMode },
         ...context.skipUnindexed === void 0 ? {} : { skipUnindexed: context.skipUnindexed },
@@ -3599,12 +3631,41 @@ function getGenerationPlanningProfile(task) {
 // src/worker.ts
 var import_worker = require("wiki-graph-core/worker");
 var import_wiki_graph_core13 = require("wiki-graph-core");
+
+// src/host.ts
+var import_platform2 = require("wiki-graph-core/platform");
+function prepareWikiGraphSDKHost(host) {
+  if (host?.platform === void 0) {
+    ensureNodeWikiGraphPlatform();
+    return;
+  }
+  (0, import_platform2.installWikiGraphPlatform)(host.platform);
+}
+function assertWikiGraphSDKHostOptions(options) {
+  if (options.host?.storage !== void 0 && options.stateDir !== void 0) {
+    throw new TypeError(
+      "Wiki Graph SDK host.storage and stateDir cannot be used together."
+    );
+  }
+}
+async function withWikiGraphSDKHost(options, operation) {
+  assertWikiGraphSDKHostOptions(options);
+  prepareWikiGraphSDKHost(options.host);
+  if (options.host?.storage !== void 0) {
+    return await (0, import_platform2.withWikiGraphStorage)(options.host.storage, operation);
+  }
+  return await withNodeWikiGraphStorage(options.stateDir, operation);
+}
+
+// src/worker.ts
 var import_worker2 = require("wiki-graph-core/worker");
 async function runBuildJobWorker(options) {
-  installNodeWikiGraphPlatform();
-  const { stateDir, ...coreOptions } = options;
-  await withNodeWikiGraphStorage(
-    stateDir,
+  const { host, stateDir, ...coreOptions } = options;
+  await withWikiGraphSDKHost(
+    {
+      ...host === void 0 ? {} : { host },
+      ...stateDir === void 0 ? {} : { stateDir }
+    },
     async () => await (0, import_worker.runBuildJobWorker)(coreOptions)
   );
 }
@@ -5030,11 +5091,13 @@ var WikiGraphSDK = class {
   libraries;
   maintenance;
   constructor(options = {}) {
-    ensureNodeWikiGraphPlatform();
+    assertWikiGraphSDKHostOptions(options);
+    prepareWikiGraphSDKHost(options.host);
     this.#context = {
       cwd: options.cwd ?? process.cwd(),
       env: { ...process.env, ...options.env },
       envPolicy: options.envPolicy ?? "production",
+      ...options.host === void 0 ? {} : { host: options.host },
       providers: { ...options.providers },
       ...options.stateDir === void 0 ? {} : { stateDir: options.stateDir }
     };
@@ -5060,8 +5123,11 @@ var WikiGraphSDK = class {
       }
       return await operation();
     });
-    const result = this.#context.stateDir === void 0 ? await runWithContext() : await withNodeWikiGraphStorage(
-      this.#context.stateDir,
+    const result = await withWikiGraphSDKHost(
+      {
+        ...this.#context.host === void 0 ? {} : { host: this.#context.host },
+        ...this.#context.stateDir === void 0 ? {} : { stateDir: this.#context.stateDir }
+      },
       runWithContext
     );
     throwIfAborted2(signal);
@@ -5113,6 +5179,7 @@ var import_wiki_graph_core19 = require("wiki-graph-core");
   createEntryRuntimeContext,
   createGenerationPerformanceHints,
   createNodeTemplateEnvironment,
+  createNodeWikiGraphPlatform,
   createNodeWikiGraphStorage,
   createStageLLM,
   createWikiGraphSDK,
@@ -5127,6 +5194,8 @@ var import_wiki_graph_core19 = require("wiki-graph-core");
   formatLocatedChapterSourceCollectionUri,
   formatLocatedChapterUri,
   formatLocatedWikiGraphUri,
+  formatSourceArtifactUri,
+  formatSourceLocatorFragment,
   formatWikiGraphCommandUri,
   formatWikiGraphLibraryUri,
   getNodeResourcePath,
@@ -5136,7 +5205,9 @@ var import_wiki_graph_core19 = require("wiki-graph-core");
   inspectWikiGraphArchive,
   installNodeWikiGraphPlatform,
   isSourceLocatorScopeUri,
+  isWikiGraphJobUri,
   isWikiGraphLibraryUri,
+  isWikiGraphUri,
   loadRequiredStageConfig,
   loadWikiGraphRuntimeConfig,
   maskLocalConfigSection,
@@ -5148,6 +5219,7 @@ var import_wiki_graph_core19 = require("wiki-graph-core");
   parseChapterUriPath,
   parseLocalConfigSection,
   parseLocatedWikiGraphUri,
+  parseSourceLocatorFragment,
   parseSourceTextJsonl,
   parseWikiGraphLibraryUri,
   parseWikiGraphUriSyntax,
@@ -5156,7 +5228,10 @@ var import_wiki_graph_core19 = require("wiki-graph-core");
   readLocalConfigSection,
   readWikiGraphEmbeddingConfig,
   replaceLocalConfigSection,
+  requireArchiveUri,
   requireKnowledgeGraphWikispineConfig,
+  requireLocatedObjectOrArchiveUri,
+  requireLocatedObjectUri,
   resolveExtractionPrompt,
   resolveHostedProviderEndpoint,
   resolveKnowledgeGraphRecallPrompt,