koishi-plugin-chatluna-long-memory 1.3.1 → 1.3.3

package/README.md

@@ -4,4 +4,4 @@
 
 > 提供长期记忆支持的插件
 
-[长期记忆文档](https://chatluna.chat/ecosystem/renderer/image.html)
+[长期记忆文档](https://chatluna.chat/ecosystem/plugin/long-term-memory.html)

package/lib/index.cjs

@@ -2001,6 +2001,153 @@ async function createVectorStoreRetriever(ctx, config, longMemoryId) {
 }
 __name(createVectorStoreRetriever, "createVectorStoreRetriever");
 
+// src/layers/emgas/extractor.ts
+var import_js_yaml2 = __toESM(require("js-yaml"), 1);
+var import_string3 = require("koishi-plugin-chatluna/utils/string");
+var GRAPH_ELEMENTS_EXTRACTION_PROMPT = /* @__PURE__ */ __name((text) => `
+You are an expert in knowledge extraction. Your task is to analyze the following text to identify key concepts and overarching topics.
+
+Text to analyze:
+"""
+${text}
+"""
+
+Respond with a single, valid YAML block with two keys: 'concepts' and 'topics'.
+
+Guidelines:
+1.  **concepts**: A list of the most important keywords, and named entities from the text. Use the base form or most common form.
+2.  **topics**: A list of 1-3 higher-level topics that categorize the concepts. These should be broader categories.
+3.  If no meaningful data can be extracted, return empty lists for both keys.
+
+Example:
+Text: "The user is asking about setting up a new React project with Vite. They are having trouble with the HMR (Hot Module Replacement) configuration."
+
+YAML Output:
+\`\`\`yaml
+concepts:
+  - React
+  - Vite
+  - HMR
+  - project setup
+topics:
+  - Frontend Development
+  - Build Tools
+\`\`\`
+
+YAML Output:
+`, "GRAPH_ELEMENTS_EXTRACTION_PROMPT");
+var STOP_WORDS = /* @__PURE__ */ new Set([
+  "a",
+  "an",
+  "and",
+  "are",
+  "as",
+  "at",
+  "be",
+  "by",
+  "for",
+  "from",
+  "how",
+  "in",
+  "is",
+  "it",
+  "of",
+  "on",
+  "or",
+  "that",
+  "the",
+  "their",
+  "they",
+  "this",
+  "to",
+  "with",
+  "用户",
+  "我们",
+  "你们",
+  "他们",
+  "以及",
+  "但是",
+  "如果",
+  "因为",
+  "所以"
+]);
+function format(items) {
+  return items.map((item) => String(item).trim().replace(/\s+/g, " ")).filter(Boolean).map(
+    (item) => /^[\x00-\x7F]+$/.test(item) ? item.toLowerCase() : item
+  );
+}
+__name(format, "format");
+function extractLocal(text) {
+  const counts = /* @__PURE__ */ new Map();
+  const groups = [];
+  for (const item of text.match(
+    /[A-Za-z][A-Za-z0-9+#./-]{1,}|[\u3400-\u9fff]{2,}/g
+  ) ?? []) {
+    if (/^[A-Za-z]/.test(item)) {
+      const word = item.toLowerCase();
+      if (STOP_WORDS.has(word)) {
+        continue;
+      }
+      counts.set(word, (counts.get(word) ?? 0) + 1);
+      if (word.length >= 4) {
+        groups.push(word);
+      }
+      continue;
+    }
+    if (item.length >= 2) {
+      groups.push(item.length > 12 ? item.slice(0, 12) : item);
+    }
+    if (item.length <= 8) {
+      counts.set(item, (counts.get(item) ?? 0) + 1);
+      continue;
+    }
+    for (let i = 0; i < item.length - 1; i++) {
+      const word = item.slice(i, i + 2);
+      counts.set(word, (counts.get(word) ?? 0) + 1);
+    }
+  }
+  const concepts = Array.from(counts.entries()).sort((a, b) => b[1] - a[1] || b[0].length - a[0].length).map(([item]) => item).slice(0, 12);
+  const topics = Array.from(new Set(format(groups))).filter((item) => !STOP_WORDS.has(item)).slice(0, 3);
+  return {
+    concepts,
+    topics: topics.length > 0 ? topics : concepts.slice(0, 3)
+  };
+}
+__name(extractLocal, "extractLocal");
+async function extractGraphElements(modelRef, text) {
+  const model = modelRef?.value;
+  if (model == null) {
+    return extractLocal(text);
+  }
+  try {
+    const prompt = GRAPH_ELEMENTS_EXTRACTION_PROMPT(text);
+    const res = await model.invoke(prompt);
+    const content = (0, import_string3.getMessageContent)(res.content).trim();
+    const yamlMatch = content.match(
+      /```(?:ya?ml)?\s*\r?\n([\s\S]*?)\r?\n```/i
+    );
+    const parsed = import_js_yaml2.default.load(yamlMatch?.[1] ?? content);
+    if (parsed && typeof parsed === "object") {
+      const concepts = format(
+        Array.isArray(parsed.concepts) ? parsed.concepts : []
+      );
+      const topics = format(
+        Array.isArray(parsed.topics) ? parsed.topics : []
+      );
+      if (concepts.length > 0 || topics.length > 0) {
+        return {
+          concepts,
+          topics: topics.length > 0 ? topics : concepts.slice(0, 3)
+        };
+      }
+    }
+    return extractLocal(text);
+  } catch {
+    return extractLocal(text);
+  }
+}
+__name(extractGraphElements, "extractGraphElements");
+
 // src/layers/emgas/graph.ts
 var MemoryGraph = class _MemoryGraph {
   static {
@@ -2209,12 +2356,16 @@ var MemoryGraph = class _MemoryGraph {
     return this.nodes.values();
   }
   toJSON() {
-    const nodes = Array.from(this.nodes.values()).map(
-      (node) => ({
-        ...node,
-        sourcePassageIds: node.sourcePassageIds ? Array.from(node.sourcePassageIds) : []
-      })
-    );
+    const nodes = Array.from(
+      this.nodes.values()
+    ).map((node) => ({
+      id: node.id,
+      type: node.type,
+      baseActivation: node.baseActivation,
+      createdAt: node.createdAt.toISOString(),
+      lastAccessed: node.lastAccessed.toISOString(),
+      sourcePassageIds: node.sourcePassageIds ? Array.from(node.sourcePassageIds) : []
+    }));
     const edges = [];
     const seenEdges = /* @__PURE__ */ new Set();
     for (const source of this.edges.values()) {
@@ -2239,9 +2390,11 @@ var MemoryGraph = class _MemoryGraph {
     if (!serializedGraph) return graph;
     for (const nodeData of serializedGraph.nodes || []) {
       graph.nodes.set(nodeData.id, {
-        ...nodeData,
         createdAt: new Date(nodeData.createdAt),
         lastAccessed: new Date(nodeData.lastAccessed),
+        id: nodeData.id,
+        type: nodeData.type,
+        baseActivation: nodeData.baseActivation,
         sourcePassageIds: new Set(nodeData.sourcePassageIds || [])
       });
     }
@@ -2260,87 +2413,7 @@ var import_crypto6 = require("crypto");
 var import_fs = require("fs");
 var path2 = __toESM(require("path"), 1);
 var import_vectorstores2 = require("koishi-plugin-chatluna/llm-core/vectorstores");
-
-// src/layers/emgas/extractor.ts
-var import_js_yaml2 = __toESM(require("js-yaml"), 1);
-var import_string3 = require("koishi-plugin-chatluna/utils/string");
-var import_error = require("koishi-plugin-chatluna/utils/error");
-var GRAPH_ELEMENTS_EXTRACTION_PROMPT = /* @__PURE__ */ __name((text) => `
-You are an expert in knowledge extraction. Your task is to analyze the following text to identify key concepts and overarching topics.
-
-Text to analyze:
-"""
-${text}
-"""
-
-Respond with a single, valid YAML block with two keys: 'concepts' and 'topics'.
-
-Guidelines:
-1.  **concepts**: A list of the most important keywords, and named entities from the text. Use the base form or most common form.
-2.  **topics**: A list of 1-3 higher-level topics that categorize the concepts. These should be broader categories.
-3.  If no meaningful data can be extracted, return empty lists for both keys.
-
-Example:
-Text: "The user is asking about setting up a new React project with Vite. They are having trouble with the HMR (Hot Module Replacement) configuration."
-
-YAML Output:
-\`\`\`yaml
-concepts:
-  - React
-  - Vite
-  - HMR
-  - project setup
-topics:
-  - Frontend Development
-  - Build Tools
-\`\`\`
-
-YAML Output:
-`, "GRAPH_ELEMENTS_EXTRACTION_PROMPT");
-async function extractGraphElements(modelRef, text) {
-  if (!modelRef.value) {
-    throw new import_error.ChatLunaError(
-      import_error.ChatLunaErrorCode.MODEL_NOT_FOUND,
-      new Error(
-        "LLM-based extractor is disabled. Cannot extract graph elements."
-      )
-    );
-  }
-  const model = modelRef.value;
-  try {
-    const prompt = GRAPH_ELEMENTS_EXTRACTION_PROMPT(text);
-    const res = await model.invoke(prompt);
-    const content = (0, import_string3.getMessageContent)(res.content);
-    const yamlMatch = content.match(
-      /```(?:ya?ml)?\s*\r?\n([\s\S]*?)\r?\n```/i
-    );
-    if (yamlMatch && yamlMatch[1]) {
-      const parsed = import_js_yaml2.default.load(yamlMatch[1]);
-      return {
-        concepts: [].concat(parsed.concepts || []).map((item) => String(item).trim()).filter(Boolean),
-        topics: [].concat(parsed.topics || []).map((item) => String(item).trim()).filter(Boolean)
-      };
-    }
-    throw new import_error.ChatLunaError(
-      import_error.ChatLunaErrorCode.MODEL_RESPONSE_IS_EMPTY,
-      new Error(
-        "LLM did not return a valid YAML for graph element extraction."
-      )
-    );
-  } catch (error) {
-    if (error instanceof import_error.ChatLunaError) {
-      throw error;
-    }
-    throw new import_error.ChatLunaError(
-      import_error.ChatLunaErrorCode.API_REQUEST_FAILED,
-      error
-    );
-  }
-}
-__name(extractGraphElements, "extractGraphElements");
-
-// src/layers/emgas/layer.ts
-function getGraphFilePath(baseDir, memoryId) {
+function getGraphFilePath(baseDir, memoryId, dir = "emgas") {
   if (!memoryId || typeof memoryId !== "string") {
     throw new Error("Invalid memoryId: must be a non-empty string.");
   }
@@ -2350,7 +2423,10 @@ function getGraphFilePath(baseDir, memoryId) {
   }
   return path2.join(
     baseDir,
-    "data/chatluna/long-memory/emgasa",
+    "data",
+    "chatluna",
+    "long-memory",
+    dir,
     `${sanitizedId}.json`
   );
 }
@@ -2374,28 +2450,48 @@ var EmgasMemoryLayer = class extends BaseMemoryRetrievalLayer {
       `Initializing EMGAS layer for memory ID: ${this.info.memoryId}`
     );
     const baseDir = this.ctx.baseDir || process.cwd();
-    const filePath = getGraphFilePath(baseDir, this.info.memoryId);
-    try {
-      const data = await import_fs.promises.readFile(filePath, "utf-8");
-      const serialized = JSON.parse(data);
-      this.memoryGraph = MemoryGraph.fromJSON(serialized);
-      logger2.debug(`EMGAS graph loaded from ${filePath}`);
-    } catch (error) {
-      const err = error;
-      if (err && err.code === "ENOENT") {
-        logger2.debug(
-          `No existing EMGAS graph found for ${this.info.memoryId}. A new one will be created.`
-        );
-      } else {
+    let loaded = false;
+    for (const filePath of [
+      getGraphFilePath(baseDir, this.info.memoryId),
+      getGraphFilePath(baseDir, this.info.memoryId, "emgasa")
+    ]) {
+      try {
+        const data = await import_fs.promises.readFile(filePath, "utf-8");
+        const serialized = JSON.parse(data);
+        this.memoryGraph = MemoryGraph.fromJSON(serialized);
+        logger2.debug(`EMGAS graph loaded from ${filePath}`);
+        loaded = true;
+        break;
+      } catch (error) {
+        const err = error;
+        if (err && err.code === "ENOENT") {
+          continue;
+        }
         logger2.error(
-          `Failed to load EMGAS graph from ${filePath}:`,
+          `Failed to load EMGAS graph for ${this.info.memoryId}:`,
           error
         );
+        break;
       }
     }
-    this.extractModel = await this.ctx.chatluna.createChatModel(
-      this.config.emgasExtractModel
-    );
+    if (!loaded) {
+      logger2.debug(
+        `No existing EMGAS graph found for ${this.info.memoryId}. A new one will be created.`
+      );
+    }
+    const modelName = this.config.emgasExtractModel !== "无" ? this.config.emgasExtractModel : this.config.longMemoryExtractModel !== "无" ? this.config.longMemoryExtractModel : this.config.hippoExtractModel !== "无" ? this.config.hippoExtractModel : void 0;
+    this.extractModel = void 0;
+    if (modelName?.includes("/")) {
+      this.extractModel = await this.ctx.chatluna.createChatModel(modelName);
+    } else if (modelName != null) {
+      logger2.warn(
+        `Invalid EMGAS extractor model: ${modelName}. Falling back to local concept extraction.`
+      );
+    } else {
+      logger2.debug(
+        "No EMGAS extractor model configured. Falling back to local concept extraction."
+      );
+    }
     this.docstore = new import_vectorstores2.DataBaseDocstore(
       this.ctx,
       resolveLongMemoryId(this.info)

package/lib/index.mjs

@@ -1964,6 +1964,153 @@ async function createVectorStoreRetriever(ctx, config, longMemoryId) {
 }
 __name(createVectorStoreRetriever, "createVectorStoreRetriever");
 
+// src/layers/emgas/extractor.ts
+import YAML2 from "js-yaml";
+import { getMessageContent as getMessageContent3 } from "koishi-plugin-chatluna/utils/string";
+var GRAPH_ELEMENTS_EXTRACTION_PROMPT = /* @__PURE__ */ __name((text) => `
+You are an expert in knowledge extraction. Your task is to analyze the following text to identify key concepts and overarching topics.
+
+Text to analyze:
+"""
+${text}
+"""
+
+Respond with a single, valid YAML block with two keys: 'concepts' and 'topics'.
+
+Guidelines:
+1.  **concepts**: A list of the most important keywords, and named entities from the text. Use the base form or most common form.
+2.  **topics**: A list of 1-3 higher-level topics that categorize the concepts. These should be broader categories.
+3.  If no meaningful data can be extracted, return empty lists for both keys.
+
+Example:
+Text: "The user is asking about setting up a new React project with Vite. They are having trouble with the HMR (Hot Module Replacement) configuration."
+
+YAML Output:
+\`\`\`yaml
+concepts:
+  - React
+  - Vite
+  - HMR
+  - project setup
+topics:
+  - Frontend Development
+  - Build Tools
+\`\`\`
+
+YAML Output:
+`, "GRAPH_ELEMENTS_EXTRACTION_PROMPT");
+var STOP_WORDS = /* @__PURE__ */ new Set([
+  "a",
+  "an",
+  "and",
+  "are",
+  "as",
+  "at",
+  "be",
+  "by",
+  "for",
+  "from",
+  "how",
+  "in",
+  "is",
+  "it",
+  "of",
+  "on",
+  "or",
+  "that",
+  "the",
+  "their",
+  "they",
+  "this",
+  "to",
+  "with",
+  "用户",
+  "我们",
+  "你们",
+  "他们",
+  "以及",
+  "但是",
+  "如果",
+  "因为",
+  "所以"
+]);
+function format(items) {
+  return items.map((item) => String(item).trim().replace(/\s+/g, " ")).filter(Boolean).map(
+    (item) => /^[\x00-\x7F]+$/.test(item) ? item.toLowerCase() : item
+  );
+}
+__name(format, "format");
+function extractLocal(text) {
+  const counts = /* @__PURE__ */ new Map();
+  const groups = [];
+  for (const item of text.match(
+    /[A-Za-z][A-Za-z0-9+#./-]{1,}|[\u3400-\u9fff]{2,}/g
+  ) ?? []) {
+    if (/^[A-Za-z]/.test(item)) {
+      const word = item.toLowerCase();
+      if (STOP_WORDS.has(word)) {
+        continue;
+      }
+      counts.set(word, (counts.get(word) ?? 0) + 1);
+      if (word.length >= 4) {
+        groups.push(word);
+      }
+      continue;
+    }
+    if (item.length >= 2) {
+      groups.push(item.length > 12 ? item.slice(0, 12) : item);
+    }
+    if (item.length <= 8) {
+      counts.set(item, (counts.get(item) ?? 0) + 1);
+      continue;
+    }
+    for (let i = 0; i < item.length - 1; i++) {
+      const word = item.slice(i, i + 2);
+      counts.set(word, (counts.get(word) ?? 0) + 1);
+    }
+  }
+  const concepts = Array.from(counts.entries()).sort((a, b) => b[1] - a[1] || b[0].length - a[0].length).map(([item]) => item).slice(0, 12);
+  const topics = Array.from(new Set(format(groups))).filter((item) => !STOP_WORDS.has(item)).slice(0, 3);
+  return {
+    concepts,
+    topics: topics.length > 0 ? topics : concepts.slice(0, 3)
+  };
+}
+__name(extractLocal, "extractLocal");
+async function extractGraphElements(modelRef, text) {
+  const model = modelRef?.value;
+  if (model == null) {
+    return extractLocal(text);
+  }
+  try {
+    const prompt = GRAPH_ELEMENTS_EXTRACTION_PROMPT(text);
+    const res = await model.invoke(prompt);
+    const content = getMessageContent3(res.content).trim();
+    const yamlMatch = content.match(
+      /```(?:ya?ml)?\s*\r?\n([\s\S]*?)\r?\n```/i
+    );
+    const parsed = YAML2.load(yamlMatch?.[1] ?? content);
+    if (parsed && typeof parsed === "object") {
+      const concepts = format(
+        Array.isArray(parsed.concepts) ? parsed.concepts : []
+      );
+      const topics = format(
+        Array.isArray(parsed.topics) ? parsed.topics : []
+      );
+      if (concepts.length > 0 || topics.length > 0) {
+        return {
+          concepts,
+          topics: topics.length > 0 ? topics : concepts.slice(0, 3)
+        };
+      }
+    }
+    return extractLocal(text);
+  } catch {
+    return extractLocal(text);
+  }
+}
+__name(extractGraphElements, "extractGraphElements");
+
 // src/layers/emgas/graph.ts
 var MemoryGraph = class _MemoryGraph {
   static {
@@ -2172,12 +2319,16 @@ var MemoryGraph = class _MemoryGraph {
     return this.nodes.values();
   }
   toJSON() {
-    const nodes = Array.from(this.nodes.values()).map(
-      (node) => ({
-        ...node,
-        sourcePassageIds: node.sourcePassageIds ? Array.from(node.sourcePassageIds) : []
-      })
-    );
+    const nodes = Array.from(
+      this.nodes.values()
+    ).map((node) => ({
+      id: node.id,
+      type: node.type,
+      baseActivation: node.baseActivation,
+      createdAt: node.createdAt.toISOString(),
+      lastAccessed: node.lastAccessed.toISOString(),
+      sourcePassageIds: node.sourcePassageIds ? Array.from(node.sourcePassageIds) : []
+    }));
     const edges = [];
     const seenEdges = /* @__PURE__ */ new Set();
     for (const source of this.edges.values()) {
@@ -2202,9 +2353,11 @@ var MemoryGraph = class _MemoryGraph {
     if (!serializedGraph) return graph;
     for (const nodeData of serializedGraph.nodes || []) {
       graph.nodes.set(nodeData.id, {
-        ...nodeData,
         createdAt: new Date(nodeData.createdAt),
         lastAccessed: new Date(nodeData.lastAccessed),
+        id: nodeData.id,
+        type: nodeData.type,
+        baseActivation: nodeData.baseActivation,
         sourcePassageIds: new Set(nodeData.sourcePassageIds || [])
       });
     }
@@ -2223,90 +2376,7 @@ import { randomUUID as randomUUID4 } from "crypto";
 import { promises as fs2 } from "fs";
 import * as path2 from "path";
 import { DataBaseDocstore } from "koishi-plugin-chatluna/llm-core/vectorstores";
-
-// src/layers/emgas/extractor.ts
-import YAML2 from "js-yaml";
-import { getMessageContent as getMessageContent3 } from "koishi-plugin-chatluna/utils/string";
-import {
-  ChatLunaError,
-  ChatLunaErrorCode
-} from "koishi-plugin-chatluna/utils/error";
-var GRAPH_ELEMENTS_EXTRACTION_PROMPT = /* @__PURE__ */ __name((text) => `
-You are an expert in knowledge extraction. Your task is to analyze the following text to identify key concepts and overarching topics.
-
-Text to analyze:
-"""
-${text}
-"""
-
-Respond with a single, valid YAML block with two keys: 'concepts' and 'topics'.
-
-Guidelines:
-1.  **concepts**: A list of the most important keywords, and named entities from the text. Use the base form or most common form.
-2.  **topics**: A list of 1-3 higher-level topics that categorize the concepts. These should be broader categories.
-3.  If no meaningful data can be extracted, return empty lists for both keys.
-
-Example:
-Text: "The user is asking about setting up a new React project with Vite. They are having trouble with the HMR (Hot Module Replacement) configuration."
-
-YAML Output:
-\`\`\`yaml
-concepts:
-  - React
-  - Vite
-  - HMR
-  - project setup
-topics:
-  - Frontend Development
-  - Build Tools
-\`\`\`
-
-YAML Output:
-`, "GRAPH_ELEMENTS_EXTRACTION_PROMPT");
-async function extractGraphElements(modelRef, text) {
-  if (!modelRef.value) {
-    throw new ChatLunaError(
-      ChatLunaErrorCode.MODEL_NOT_FOUND,
-      new Error(
-        "LLM-based extractor is disabled. Cannot extract graph elements."
-      )
-    );
-  }
-  const model = modelRef.value;
-  try {
-    const prompt = GRAPH_ELEMENTS_EXTRACTION_PROMPT(text);
-    const res = await model.invoke(prompt);
-    const content = getMessageContent3(res.content);
-    const yamlMatch = content.match(
-      /```(?:ya?ml)?\s*\r?\n([\s\S]*?)\r?\n```/i
-    );
-    if (yamlMatch && yamlMatch[1]) {
-      const parsed = YAML2.load(yamlMatch[1]);
-      return {
-        concepts: [].concat(parsed.concepts || []).map((item) => String(item).trim()).filter(Boolean),
-        topics: [].concat(parsed.topics || []).map((item) => String(item).trim()).filter(Boolean)
-      };
-    }
-    throw new ChatLunaError(
-      ChatLunaErrorCode.MODEL_RESPONSE_IS_EMPTY,
-      new Error(
-        "LLM did not return a valid YAML for graph element extraction."
-      )
-    );
-  } catch (error) {
-    if (error instanceof ChatLunaError) {
-      throw error;
-    }
-    throw new ChatLunaError(
-      ChatLunaErrorCode.API_REQUEST_FAILED,
-      error
-    );
-  }
-}
-__name(extractGraphElements, "extractGraphElements");
-
-// src/layers/emgas/layer.ts
-function getGraphFilePath(baseDir, memoryId) {
+function getGraphFilePath(baseDir, memoryId, dir = "emgas") {
   if (!memoryId || typeof memoryId !== "string") {
     throw new Error("Invalid memoryId: must be a non-empty string.");
   }
@@ -2316,7 +2386,10 @@ function getGraphFilePath(baseDir, memoryId) {
   }
   return path2.join(
     baseDir,
-    "data/chatluna/long-memory/emgasa",
+    "data",
+    "chatluna",
+    "long-memory",
+    dir,
     `${sanitizedId}.json`
   );
 }
@@ -2340,28 +2413,48 @@ var EmgasMemoryLayer = class extends BaseMemoryRetrievalLayer {
       `Initializing EMGAS layer for memory ID: ${this.info.memoryId}`
     );
     const baseDir = this.ctx.baseDir || process.cwd();
-    const filePath = getGraphFilePath(baseDir, this.info.memoryId);
-    try {
-      const data = await fs2.readFile(filePath, "utf-8");
-      const serialized = JSON.parse(data);
-      this.memoryGraph = MemoryGraph.fromJSON(serialized);
-      logger2.debug(`EMGAS graph loaded from ${filePath}`);
-    } catch (error) {
-      const err = error;
-      if (err && err.code === "ENOENT") {
-        logger2.debug(
-          `No existing EMGAS graph found for ${this.info.memoryId}. A new one will be created.`
-        );
-      } else {
+    let loaded = false;
+    for (const filePath of [
+      getGraphFilePath(baseDir, this.info.memoryId),
+      getGraphFilePath(baseDir, this.info.memoryId, "emgasa")
+    ]) {
+      try {
+        const data = await fs2.readFile(filePath, "utf-8");
+        const serialized = JSON.parse(data);
+        this.memoryGraph = MemoryGraph.fromJSON(serialized);
+        logger2.debug(`EMGAS graph loaded from ${filePath}`);
+        loaded = true;
+        break;
+      } catch (error) {
+        const err = error;
+        if (err && err.code === "ENOENT") {
+          continue;
+        }
         logger2.error(
-          `Failed to load EMGAS graph from ${filePath}:`,
+          `Failed to load EMGAS graph for ${this.info.memoryId}:`,
           error
         );
+        break;
       }
     }
-    this.extractModel = await this.ctx.chatluna.createChatModel(
-      this.config.emgasExtractModel
-    );
+    if (!loaded) {
+      logger2.debug(
+        `No existing EMGAS graph found for ${this.info.memoryId}. A new one will be created.`
+      );
+    }
+    const modelName = this.config.emgasExtractModel !== "无" ? this.config.emgasExtractModel : this.config.longMemoryExtractModel !== "无" ? this.config.longMemoryExtractModel : this.config.hippoExtractModel !== "无" ? this.config.hippoExtractModel : void 0;
+    this.extractModel = void 0;
+    if (modelName?.includes("/")) {
+      this.extractModel = await this.ctx.chatluna.createChatModel(modelName);
+    } else if (modelName != null) {
+      logger2.warn(
+        `Invalid EMGAS extractor model: ${modelName}. Falling back to local concept extraction.`
+      );
+    } else {
+      logger2.debug(
+        "No EMGAS extractor model configured. Falling back to local concept extraction."
+      );
+    }
     this.docstore = new DataBaseDocstore(
       this.ctx,
       resolveLongMemoryId(this.info)

package/lib/layers/emgas/extractor.d.ts

@@ -1,14 +1,11 @@
-import { ChatLunaChatModel } from 'koishi-plugin-chatluna/llm-core/platform/model';
 import { ComputedRef } from 'koishi-plugin-chatluna';
-export interface ExtractedGraphElements {
-    concepts: string[];
-    topics: string[];
-}
+import { ChatLunaChatModel } from 'koishi-plugin-chatluna/llm-core/platform/model';
+import { ExtractedGraphElements } from './types';
 /**
- * Extracts key concepts and topics from a text chunk using an LLM.
- * @param ctx The Koishi context.
- * @param config The plugin configuration.
+ * Extracts key concepts and topics from a text chunk.
+ * It prefers the configured LLM and falls back to local extraction.
+ * @param modelRef The configured extractor model.
  * @param text The text to analyze.
  * @returns A promise that resolves to an object containing concepts and topics.
  */
-export declare function extractGraphElements(modelRef: ComputedRef<ChatLunaChatModel>, text: string): Promise<ExtractedGraphElements>;
+export declare function extractGraphElements(modelRef: ComputedRef<ChatLunaChatModel | undefined> | undefined, text: string): Promise<ExtractedGraphElements>;

package/lib/layers/emgas/graph.d.ts

@@ -1,5 +1,4 @@
-import { ExtractedGraphElements } from './extractor';
-import { MemoryNode, SerializedMemoryGraph, SpreadingActivationOptions } from './types';
+import { ExtractedGraphElements, MemoryNode, SerializedMemoryGraph, SpreadingActivationOptions } from './types';
 /**
  * A class that implements the EMGAS (Episodic Memory Graph with Activation Spreading) framework.
  * It manages a dynamic graph of concepts, handles incremental updates, and performs retrieval

package/lib/layers/emgas/index.d.ts

@@ -1,3 +1,4 @@
 export * from './types';
+export { extractGraphElements } from './extractor';
 export { MemoryGraph } from './graph';
 export { EmgasMemoryLayer } from './layer';

package/lib/layers/emgas/types.d.ts

@@ -2,6 +2,10 @@
  * Core data structures for the EMGAS (Episodic Memory Graph with Activation Spreading) model.
  * Based on the TypeScript blueprint provided in the design document.
  */
+export interface ExtractedGraphElements {
+    concepts: string[];
+    topics: string[];
+}
 /**
  * Options for the Spreading Activation retrieval algorithm.
  */
@@ -62,6 +66,11 @@ export interface MemoryNode {
      */
     sourcePassageIds?: Set<string>;
 }
+export interface SerializedMemoryNode extends Omit<MemoryNode, 'createdAt' | 'lastAccessed' | 'sourcePassageIds'> {
+    createdAt: string;
+    lastAccessed: string;
+    sourcePassageIds: string[];
+}
 /**
  * Represents a weighted edge in the memory graph, connecting two nodes.
  */
@@ -90,7 +99,7 @@ export interface MemoryEdge {
  * Represents the entire memory graph, including nodes, edges, and persistence data.
  */
 export interface SerializedMemoryGraph {
-    nodes: MemoryNode[];
+    nodes: SerializedMemoryNode[];
     edges: MemoryEdge[];
     conceptCounts: [string, number][];
     pairCounts: [string, number][];

package/lib/plugins/tool.d.ts

@@ -35,27 +35,27 @@ export declare class MemoryAddTool extends StructuredTool {
             type: z.ZodNativeEnum<typeof MemoryType>;
             importance: z.ZodNumber;
         }, "strip", z.ZodTypeAny, {
-            type?: MemoryType;
             content?: string;
+            type?: MemoryType;
             importance?: number;
         }, {
-            type?: MemoryType;
             content?: string;
+            type?: MemoryType;
             importance?: number;
         }>, "many">;
         layer: z.ZodArray<z.ZodUnion<[z.ZodLiteral<"user">, z.ZodLiteral<"preset">, z.ZodLiteral<"guild">, z.ZodLiteral<"global">]>, "many">;
     }, "strip", z.ZodTypeAny, {
         layer?: ("global" | "preset" | "user" | "guild")[];
         memories?: {
-            type?: MemoryType;
             content?: string;
+            type?: MemoryType;
             importance?: number;
         }[];
     }, {
         layer?: ("global" | "preset" | "user" | "guild")[];
         memories?: {
-            type?: MemoryType;
             content?: string;
+            type?: MemoryType;
             importance?: number;
         }[];
     }>;
@@ -94,12 +94,12 @@ export declare class MemoryUpdateTool extends StructuredTool {
             type: z.ZodNativeEnum<typeof MemoryType>;
             importance: z.ZodNumber;
         }, "strip", z.ZodTypeAny, {
-            type?: MemoryType;
             content?: string;
+            type?: MemoryType;
             importance?: number;
         }, {
-            type?: MemoryType;
             content?: string;
+            type?: MemoryType;
             importance?: number;
         }>, "many">;
         layer: z.ZodArray<z.ZodUnion<[z.ZodLiteral<"user">, z.ZodLiteral<"preset">, z.ZodLiteral<"guild">, z.ZodLiteral<"global">]>, "many">;
@@ -107,16 +107,16 @@ export declare class MemoryUpdateTool extends StructuredTool {
         layer?: ("global" | "preset" | "user" | "guild")[];
         memoryIds?: string[];
         newMemories?: {
-            type?: MemoryType;
             content?: string;
+            type?: MemoryType;
             importance?: number;
         }[];
     }, {
         layer?: ("global" | "preset" | "user" | "guild")[];
         memoryIds?: string[];
         newMemories?: {
-            type?: MemoryType;
             content?: string;
+            type?: MemoryType;
             importance?: number;
         }[];
     }>;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-chatluna-long-memory",
   "description": "long memory for chatluna",
-  "version": "1.3.1",
+  "version": "1.3.3",
   "main": "lib/index.cjs",
   "module": "lib/index.mjs",
   "typings": "lib/index.d.ts",
@@ -62,7 +62,7 @@
   },
   "peerDependencies": {
     "koishi": "^4.18.9",
-    "koishi-plugin-chatluna": "^1.3.8"
+    "koishi-plugin-chatluna": "^1.3.33"
   },
   "resolutions": {
     "@langchain/core": "^0.3.80",