stratagate-dsh 0.3.4 → 0.3.5

package/dist/plugin.js

@@ -9047,25 +9047,56 @@ function lineText(value, maximum) {
 function renderMemoryDirectory(topics, events) {
   const { entries, categories: categories2 } = topicNavigation(topics, events);
   if (entries.length === 0) return "";
-  const header = "[StrataGate \u8BB0\u5FC6\u76EE\u5F55]\n\u4EC5\u4F9B\u5BFC\u822A\uFF0C\u6982\u89C8\u4E0D\u662F\u4E8B\u5B9E\u8BC1\u636E\u3002\u9700\u8981\u4E3B\u9898\u8109\u7EDC\u65F6\u7528 memory_expand_topic\uFF1B\u9700\u8981\u4E8B\u5B9E\u8BC1\u636E\u53EF\u76F4\u63A5 memory_search_events(topic_id)\uFF0C\u4E5F\u53EF\u76F4\u63A5\u641C\u7D22\u4E8B\u4EF6/\u56FE\u8C31\u3002";
-  const lines = entries.map((entry) => `- ${entry.id}\uFF1A${lineText(entry.title, 64)}${entry.description ? `\uFF1B${lineText(entry.description, 48)}` : ""}`);
-  const complete = `${header}
-${lines.join("\n")}`;
+  const header = "[StrataGate \u8BB0\u5FC6\u76EE\u5F55]\n\u4EC5\u4F9B\u5BFC\u822A\uFF0C\u4E0D\u662F\u4E8B\u5B9E\u8BC1\u636E\u3002\u9700\u8981\u4E3B\u9898\u8109\u7EDC\u65F6\u7528 memory_expand_topic\uFF1B\u9700\u8981\u4E8B\u5B9E\u8BC1\u636E\u53EF\u76F4\u63A5 memory_search_events(topic_id)\uFF0C\u4E5F\u53EF\u76F4\u63A5\u641C\u7D22\u4E8B\u4EF6/\u56FE\u8C31\u3002";
+  const byId = new Map(topics.map((topic) => [topic.id, topic]));
+  const chapters = entries.map((entry) => ({
+    ...entry,
+    line: `- ${entry.id}\uFF1A${lineText(entry.title, 64)}${entry.description ? `\uFF1B${lineText(entry.description, 48)}` : ""}`,
+    // Reuse the book's stable section order, including summary-free and legacy sections.
+    sections: memoryTopicSections(byId.get(entry.id)).map(({ title }) => `  - ${lineText(title, 64)}`)
+  }));
+  const render = (prefix, selected2) => `${prefix}
+${selected2.flatMap(({ line, sections }) => [line, ...sections]).join("\n")}`.trim();
+  const complete = render(header, chapters);
   if (estimateTokens(complete) <= MEMORY_DIRECTORY_TOKEN_BUDGET) return complete;
   const branches = categories2.map(({ id, label, count }) => `${id}\uFF08${label} ${count}\uFF09`).join("\uFF1B");
   const navigation = `${header}
-\u5171 ${entries.length} \u9879\uFF1B\u5206\u7C7B\uFF1A${branches}\u3002\u5B8C\u6574\u76EE\u5F55\u7528 memory_list_topics(category, offset) \u5206\u9875\uFF1Bquery \u53EF\u67E5\u4E3B\u9898\u3002\u4E0B\u5217\u4E3A\u90E8\u5206\u5165\u53E3\uFF1A`;
-  const selected = [];
-  for (const category of categories2) {
-    const entry = entries.find((item) => item.category === category.id);
-    const line = `- ${entry.id}\uFF1A${lineText(entry.title, 40)}`;
-    if (estimateTokens(`${navigation}
-${[...selected.map(({ line: line2 }) => line2), line].join("\n")}`) <= MEMORY_DIRECTORY_TOKEN_BUDGET) selected.push({ id: entry.id, line });
+\u5171 ${entries.length} \u9879\uFF1B\u5206\u7C7B\uFF1A${branches}\u3002\u76EE\u5F55\u5DF2\u88C1\u526A\uFF1B\u5B8C\u6574\u76EE\u5F55\u7528 memory_list_topics(category, offset) \u5206\u9875\uFF1Bquery \u53EF\u67E5\u4E3B\u9898\uFF0C\u5C0F\u8282\u7528 memory_expand_topic(id)\u3002`;
+  let selected = chapters.map((chapter) => ({ ...chapter, sections: [] }));
+  if (estimateTokens(render(navigation, selected)) > MEMORY_DIRECTORY_TOKEN_BUDGET) {
+    selected = [];
+    const admit = (chapter) => {
+      for (const line of [chapter.line, `- ${chapter.id}\uFF1A${lineText(chapter.title, 64)}`]) {
+        const candidate = { ...chapter, line, sections: [] };
+        if (estimateTokens(render(navigation, [...selected, candidate])) <= MEMORY_DIRECTORY_TOKEN_BUDGET) {
+          selected.push(candidate);
+          return;
+        }
+      }
+    };
+    for (const category of categories2) admit(chapters.find((chapter) => chapter.category === category.id));
+    for (const chapter of chapters) if (!selected.some(({ id }) => id === chapter.id)) admit(chapter);
   }
   const order = new Map(entries.map(({ id }, index) => [id, index]));
   selected.sort((a, b) => order.get(a.id) - order.get(b.id));
-  return `${navigation}
-${selected.map(({ line }) => line).join("\n")}`.trim();
+  const sectionLines = new Map(chapters.map(({ id, sections }) => [id, sections]));
+  const exhausted = /* @__PURE__ */ new Set();
+  for (let index = 0; exhausted.size < selected.length; index++) {
+    for (const chapter of selected) {
+      if (exhausted.has(chapter.id)) continue;
+      const line = sectionLines.get(chapter.id)[index];
+      if (line === void 0) {
+        exhausted.add(chapter.id);
+        continue;
+      }
+      chapter.sections.push(line);
+      if (estimateTokens(render(navigation, selected)) > MEMORY_DIRECTORY_TOKEN_BUDGET) {
+        chapter.sections.pop();
+        exhausted.add(chapter.id);
+      }
+    }
+  }
+  return render(navigation, selected);
 }
 function boundedTopic(topic, envelope = {}) {
   const { sections: _sections, ...summaryTopic } = topic;
@@ -9611,9 +9642,21 @@ var StrataGateRuntime = class {
   }
   async buildMemoryDirectory(session) {
     await this.flush();
-    const memory = await this.space(session);
-    await this.refreshExternalImportMemory(this.namespaceFor(session), memory);
-    return renderMemoryDirectory(this.visibleTopics(memory), memory.listAllEvents());
+    const namespace = this.namespaceFor(session);
+    const active = this.spaces.get(namespace);
+    if (active) {
+      const memory = await active;
+      await this.refreshExternalImportMemory(namespace, memory);
+      return renderMemoryDirectory(this.visibleTopics(memory), memory.listAllEvents());
+    }
+    if (this.config.database === ":memory:" || !existsSync(this.config.database)) return "";
+    const storage = new SqliteStorage({ filename: this.config.database, readonly: true });
+    try {
+      const memory = await StrataGate.openWithStorage({ storage, namespace });
+      return renderMemoryDirectory(this.visibleTopics(memory), memory.listAllEvents());
+    } finally {
+      await storage.close();
+    }
   }
   visibleTopics(memory) {
     if (this.agentMemoryRetrievalWeight > 0) return memory.listMemoryTopics();
@@ -9890,6 +9933,7 @@ var StrataGateRuntime = class {
   async buildAutoContext(session) {
     await this.flush();
     const memory = await this.space(session);
+    await this.refreshExternalImportMemory(this.namespaceFor(session), memory);
     const threadId = String(session.id);
     const blockContexts = memory.getBlockContext(threadId);
     if (this.syncDecayedBlockSurface(session, memory, blockContexts)) {
@@ -13863,20 +13907,22 @@ async function apply(ctx, config) {
     }
     const session = context.agent?.session;
     if (!session) return { ...assembled, contexts };
+    let autoMemory;
     try {
-      const directory = await runtime.buildMemoryDirectory(session);
-      if (directory) contexts.push({ name: "stratagate:memory-directory", text: directory });
+      const text3 = await runtime.buildAutoContext(session);
+      autoMemory = { name: "stratagate:auto-memory", text: text3 };
     } catch (error) {
-      ctx.logger.warn(`stratagate-memory directory failed: ${renderError(error)}`);
+      ctx.logger.warn(`stratagate-memory auto-context failed: ${renderError(error)}`);
       runtime.notePluginError(session, error);
     }
     try {
-      const text3 = await runtime.buildAutoContext(session);
-      contexts.push({ name: "stratagate:auto-memory", text: text3 });
+      const directory = await runtime.buildMemoryDirectory(session);
+      if (directory) contexts.push({ name: "stratagate:memory-directory", text: directory });
     } catch (error) {
-      ctx.logger.warn(`stratagate-memory auto-context failed: ${renderError(error)}`);
+      ctx.logger.warn(`stratagate-memory directory failed: ${renderError(error)}`);
       runtime.notePluginError(session, error);
     }
+    if (autoMemory) contexts.push(autoMemory);
     const feedbackSuggestion = runtime.takeFeedbackSuggestion(session);
     if (feedbackSuggestion) contexts.push({ name: "stratagate:feedback-suggestion", text: feedbackSuggestion });
     return { ...assembled, contexts };