memorix 1.2.2 → 1.2.3

package/src/memory/entity-extractor.ts

@@ -1,197 +1,197 @@
-/**
- * Entity Extractor
- *
- * Regex-based entity extraction from observation content.
- * Inspired by MemCP's RegexEntityExtractor (MAGMA paper).
- *
- * Extracts: file paths, module paths, URLs, @mentions, CamelCase identifiers.
- * Also detects causal patterns for automatic edge typing.
- */
-
-const ENTITY_PATTERNS: Array<{ kind: string; pattern: RegExp }> = [
-  // File paths (e.g. src/auth/jwt.ts, ./config.json)
-  { kind: 'file', pattern: /(?:^|[\s"'(])([.\w/-]+\.\w{1,10})(?:[\s"'),]|$)/g },
-  // Module/package paths (e.g. @orama/orama, memcp.core.graph)
-  { kind: 'module', pattern: /(?:^|[\s"'(,])(@[\w-]+\/[\w.-]+)|\b([a-zA-Z_]\w*(?:\.[a-zA-Z_]\w*){2,})\b/g },
-  // URLs
-  { kind: 'url', pattern: /https?:\/\/[^\s"'<>)]+/g },
-  // @mentions
-  { kind: 'mention', pattern: /@([a-zA-Z_]\w+)/g },
-  // CamelCase identifiers (likely class/type names)
-  { kind: 'identifier', pattern: /\b([A-Z][a-z]+(?:[A-Z][a-z]+)+)\b/g },
-  // Chinese identifiers: quoted Chinese terms or backtick-wrapped Chinese (e.g. 「认证模块」, `数据库连接`)
-  { kind: 'identifier', pattern: /[「『【]([一-鿿㐀-䶿]{2,}(?:[一-鿿㐀-䶿a-zA-Z0-9_-]*[一-鿿㐀-䶿])?)[」』】]/g },
-  { kind: 'identifier', pattern: /`([一-鿿㐀-䶿]{2,}[一-鿿㐀-䶿a-zA-Z0-9_-]*)`/g },
-];
-
-/**
- * Patterns that indicate causal relationships.
- * Used to auto-detect causal edge types in Knowledge Graph.
- */
-const CAUSAL_PATTERN = /\b(?:because|therefore|due to|caused by|as a result|decided to|chosen because|so that|in order to|leads to|results in|fixed by|resolved by)\b|(?:因为|所以|由于|导致|造成|结果|因此|为了|解决|修复|修正|决定|选择|采用)/i;
-
-export interface ExtractedEntities {
-  files: string[];
-  modules: string[];
-  urls: string[];
-  mentions: string[];
-  identifiers: string[];
-  hasCausalLanguage: boolean;
-}
-
-/**
- * Extract entities from text content.
- * Returns deduplicated lists of each entity type.
- */
-export function extractEntities(content: string): ExtractedEntities {
-  const result: ExtractedEntities = {
-    files: [],
-    modules: [],
-    urls: [],
-    mentions: [],
-    identifiers: [],
-    hasCausalLanguage: false,
-  };
-
-  const seen = new Set<string>();
-
-  for (const { kind, pattern } of ENTITY_PATTERNS) {
-    // Reset lastIndex for global regexes
-    pattern.lastIndex = 0;
-    let match: RegExpExecArray | null;
-    while ((match = pattern.exec(content)) !== null) {
-      const entity = (match[1] ?? match[2] ?? match[0]).trim().replace(/^["'(]+|["'),]+$/g, '');
-      if (entity.length < 3 || (kind === 'file' && entity.length < 5)) continue;
-
-      const key = `${kind}:${entity.toLowerCase()}`;
-      if (seen.has(key)) continue;
-      seen.add(key);
-
-      switch (kind) {
-        case 'file':
-          // Filter out property chains (e.g. config.llm.apiKey, process.env.FOO)
-          // Real files have path separators or known extensions
-          // Inline logic to prevent tree-shaking
-          const isLikelyFilePath = (() => {
-            // Has path separator → likely a file path
-            if (entity.includes('/') || entity.includes('\\')) return true;
-            // Starts with ./ or ../ → relative path
-            if (entity.startsWith('./') || entity.startsWith('../')) return true;
-            // Blocked property chains
-            if (/^(?:process\.env|config\.|options\.|params\.|props\.|this\.|self\.|module\.|exports\.|require\.|import\.)/i.test(entity)) return false;
-            // Multiple dots without path separator → likely property chain
-            const dotCount = (entity.match(/\./g) || []).length;
-            if (dotCount >= 2) return false;
-            // Single dot: check if extension is a known file extension
-            const ext = entity.split('.').pop()?.toLowerCase() ?? '';
-            const FILE_EXTENSIONS = new Set(['ts', 'js', 'tsx', 'jsx', 'mts', 'mjs', 'cjs', 'json', 'yaml', 'yml', 'toml', 'xml', 'csv', 'py', 'rb', 'go', 'rs', 'java', 'kt', 'swift', 'c', 'cpp', 'h', 'html', 'css', 'scss', 'less', 'vue', 'svelte', 'sh', 'bash', 'zsh', 'ps1', 'cmd', 'bat', 'md', 'txt', 'rst', 'log', 'sql', 'graphql', 'prisma', 'env', 'gitignore', 'dockerignore', 'dockerfile', 'lock', 'config']);
-            return FILE_EXTENSIONS.has(ext);
-          })();
-          if (isLikelyFilePath) {
-            result.files.push(entity);
-          }
-          break;
-        case 'module':
-          result.modules.push(entity);
-          break;
-        case 'url':
-          result.urls.push(entity);
-          break;
-        case 'mention':
-          result.mentions.push(entity);
-          break;
-        case 'identifier':
-          result.identifiers.push(entity);
-          break;
-      }
-    }
-  }
-
-  result.hasCausalLanguage = CAUSAL_PATTERN.test(content);
-
-  return result;
-}
-
-/** Known file extensions that indicate a real file path */
-const FILE_EXTENSIONS = new Set([
-  'ts', 'js', 'tsx', 'jsx', 'mts', 'mjs', 'cjs',
-  'json', 'yaml', 'yml', 'toml', 'xml', 'csv',
-  'py', 'rb', 'go', 'rs', 'java', 'kt', 'swift', 'c', 'cpp', 'h',
-  'html', 'css', 'scss', 'less', 'vue', 'svelte',
-  'sh', 'bash', 'zsh', 'ps1', 'cmd', 'bat',
-  'md', 'txt', 'rst', 'log',
-  'sql', 'graphql', 'prisma',
-  'env', 'gitignore', 'dockerignore', 'dockerfile',
-  'lock', 'config',
-]);
-
-/** Property-chain patterns that are NOT file paths */
-const PROPERTY_CHAIN_BLOCKLIST = /^(?:process\.env|config\.|options\.|params\.|props\.|this\.|self\.|module\.|exports\.|require\.|import\.)/i;
-
-/**
- * Determine if a matched string is likely a real file path vs a property chain.
- * 
- * Examples:
- *   "src/config.ts" → true (has path separator + known extension)
- *   "config.llm.apiKey" → false (property chain, no path separator)
- *   "./package.json" → true (relative path)
- *   "process.env.FOO" → false (property chain)
- */
-function isLikelyFilePath(candidate: string): boolean {
-  // Has path separator → likely a file path
-  if (candidate.includes('/') || candidate.includes('\\')) return true;
-
-  // Starts with ./ or ../ → relative path
-  if (candidate.startsWith('./') || candidate.startsWith('../')) return true;
-
-  // Blocked property chains
-  if (PROPERTY_CHAIN_BLOCKLIST.test(candidate)) return false;
-
-  // Multiple dots without path separator → likely property chain (e.g. config.llm.apiKey)
-  const dotCount = (candidate.match(/\./g) || []).length;
-  if (dotCount >= 2) return false;
-
-  // Single dot: check if extension is a known file extension
-  const ext = candidate.split('.').pop()?.toLowerCase() ?? '';
-  return FILE_EXTENSIONS.has(ext);
-}
-
-/**
- * Auto-generate concepts from extracted entities.
- * Merges with any user-provided concepts.
- */
-export function enrichConcepts(
-  userConcepts: string[],
-  extracted: ExtractedEntities,
-): string[] {
-  const all = new Set(userConcepts.map((c) => c.toLowerCase()));
-  const enriched = [...userConcepts];
-
-  // Add file names (without path) as concepts
-  for (const f of extracted.files) {
-    const name = f.split('/').pop()?.replace(/\.\w+$/, '') ?? '';
-    if (name.length >= 3 && !all.has(name.toLowerCase())) {
-      all.add(name.toLowerCase());
-      enriched.push(name);
-    }
-  }
-
-  // Add module names as concepts
-  for (const m of extracted.modules) {
-    const short = m.split(/[./]/).pop() ?? '';
-    if (short.length >= 3 && !all.has(short.toLowerCase())) {
-      all.add(short.toLowerCase());
-      enriched.push(short);
-    }
-  }
-
-  // Add CamelCase identifiers as concepts
-  for (const id of extracted.identifiers) {
-    if (!all.has(id.toLowerCase())) {
-      all.add(id.toLowerCase());
-      enriched.push(id);
-    }
-  }
-
-  return enriched;
-}
+/**
+ * Entity Extractor
+ *
+ * Regex-based entity extraction from observation content.
+ * Inspired by MemCP's RegexEntityExtractor (MAGMA paper).
+ *
+ * Extracts: file paths, module paths, URLs, @mentions, CamelCase identifiers.
+ * Also detects causal patterns for automatic edge typing.
+ */
+
+const ENTITY_PATTERNS: Array<{ kind: string; pattern: RegExp }> = [
+  // File paths (e.g. src/auth/jwt.ts, ./config.json)
+  { kind: 'file', pattern: /(?:^|[\s"'(])([.\w/-]+\.\w{1,10})(?:[\s"'),]|$)/g },
+  // Module/package paths (e.g. @orama/orama, memcp.core.graph)
+  { kind: 'module', pattern: /(?:^|[\s"'(,])(@[\w-]+\/[\w.-]+)|\b([a-zA-Z_]\w*(?:\.[a-zA-Z_]\w*){2,})\b/g },
+  // URLs
+  { kind: 'url', pattern: /https?:\/\/[^\s"'<>)]+/g },
+  // @mentions
+  { kind: 'mention', pattern: /@([a-zA-Z_]\w+)/g },
+  // CamelCase identifiers (likely class/type names)
+  { kind: 'identifier', pattern: /\b([A-Z][a-z]+(?:[A-Z][a-z]+)+)\b/g },
+  // Chinese identifiers: quoted Chinese terms or backtick-wrapped Chinese (e.g. 「认证模块」, `数据库连接`)
+  { kind: 'identifier', pattern: /[「『【]([一-鿿㐀-䶿]{2,}(?:[一-鿿㐀-䶿a-zA-Z0-9_-]*[一-鿿㐀-䶿])?)[」』】]/g },
+  { kind: 'identifier', pattern: /`([一-鿿㐀-䶿]{2,}[一-鿿㐀-䶿a-zA-Z0-9_-]*)`/g },
+];
+
+/**
+ * Patterns that indicate causal relationships.
+ * Used to auto-detect causal edge types in Knowledge Graph.
+ */
+const CAUSAL_PATTERN = /\b(?:because|therefore|due to|caused by|as a result|decided to|chosen because|so that|in order to|leads to|results in|fixed by|resolved by)\b|(?:因为|所以|由于|导致|造成|结果|因此|为了|解决|修复|修正|决定|选择|采用)/i;
+
+export interface ExtractedEntities {
+  files: string[];
+  modules: string[];
+  urls: string[];
+  mentions: string[];
+  identifiers: string[];
+  hasCausalLanguage: boolean;
+}
+
+/**
+ * Extract entities from text content.
+ * Returns deduplicated lists of each entity type.
+ */
+export function extractEntities(content: string): ExtractedEntities {
+  const result: ExtractedEntities = {
+    files: [],
+    modules: [],
+    urls: [],
+    mentions: [],
+    identifiers: [],
+    hasCausalLanguage: false,
+  };
+
+  const seen = new Set<string>();
+
+  for (const { kind, pattern } of ENTITY_PATTERNS) {
+    // Reset lastIndex for global regexes
+    pattern.lastIndex = 0;
+    let match: RegExpExecArray | null;
+    while ((match = pattern.exec(content)) !== null) {
+      const entity = (match[1] ?? match[2] ?? match[0]).trim().replace(/^["'(]+|["'),]+$/g, '');
+      if (entity.length < 3 || (kind === 'file' && entity.length < 5)) continue;
+
+      const key = `${kind}:${entity.toLowerCase()}`;
+      if (seen.has(key)) continue;
+      seen.add(key);
+
+      switch (kind) {
+        case 'file':
+          // Filter out property chains (e.g. config.llm.apiKey, process.env.FOO)
+          // Real files have path separators or known extensions
+          // Inline logic to prevent tree-shaking
+          const isLikelyFilePath = (() => {
+            // Has path separator → likely a file path
+            if (entity.includes('/') || entity.includes('\\')) return true;
+            // Starts with ./ or ../ → relative path
+            if (entity.startsWith('./') || entity.startsWith('../')) return true;
+            // Blocked property chains
+            if (/^(?:process\.env|config\.|options\.|params\.|props\.|this\.|self\.|module\.|exports\.|require\.|import\.)/i.test(entity)) return false;
+            // Multiple dots without path separator → likely property chain
+            const dotCount = (entity.match(/\./g) || []).length;
+            if (dotCount >= 2) return false;
+            // Single dot: check if extension is a known file extension
+            const ext = entity.split('.').pop()?.toLowerCase() ?? '';
+            const FILE_EXTENSIONS = new Set(['ts', 'js', 'tsx', 'jsx', 'mts', 'mjs', 'cjs', 'json', 'yaml', 'yml', 'toml', 'xml', 'csv', 'py', 'rb', 'go', 'rs', 'java', 'kt', 'swift', 'c', 'cpp', 'h', 'html', 'css', 'scss', 'less', 'vue', 'svelte', 'sh', 'bash', 'zsh', 'ps1', 'cmd', 'bat', 'md', 'txt', 'rst', 'log', 'sql', 'graphql', 'prisma', 'env', 'gitignore', 'dockerignore', 'dockerfile', 'lock', 'config']);
+            return FILE_EXTENSIONS.has(ext);
+          })();
+          if (isLikelyFilePath) {
+            result.files.push(entity);
+          }
+          break;
+        case 'module':
+          result.modules.push(entity);
+          break;
+        case 'url':
+          result.urls.push(entity);
+          break;
+        case 'mention':
+          result.mentions.push(entity);
+          break;
+        case 'identifier':
+          result.identifiers.push(entity);
+          break;
+      }
+    }
+  }
+
+  result.hasCausalLanguage = CAUSAL_PATTERN.test(content);
+
+  return result;
+}
+
+/** Known file extensions that indicate a real file path */
+const FILE_EXTENSIONS = new Set([
+  'ts', 'js', 'tsx', 'jsx', 'mts', 'mjs', 'cjs',
+  'json', 'yaml', 'yml', 'toml', 'xml', 'csv',
+  'py', 'rb', 'go', 'rs', 'java', 'kt', 'swift', 'c', 'cpp', 'h',
+  'html', 'css', 'scss', 'less', 'vue', 'svelte',
+  'sh', 'bash', 'zsh', 'ps1', 'cmd', 'bat',
+  'md', 'txt', 'rst', 'log',
+  'sql', 'graphql', 'prisma',
+  'env', 'gitignore', 'dockerignore', 'dockerfile',
+  'lock', 'config',
+]);
+
+/** Property-chain patterns that are NOT file paths */
+const PROPERTY_CHAIN_BLOCKLIST = /^(?:process\.env|config\.|options\.|params\.|props\.|this\.|self\.|module\.|exports\.|require\.|import\.)/i;
+
+/**
+ * Determine if a matched string is likely a real file path vs a property chain.
+ * 
+ * Examples:
+ *   "src/config.ts" → true (has path separator + known extension)
+ *   "config.llm.apiKey" → false (property chain, no path separator)
+ *   "./package.json" → true (relative path)
+ *   "process.env.FOO" → false (property chain)
+ */
+function isLikelyFilePath(candidate: string): boolean {
+  // Has path separator → likely a file path
+  if (candidate.includes('/') || candidate.includes('\\')) return true;
+
+  // Starts with ./ or ../ → relative path
+  if (candidate.startsWith('./') || candidate.startsWith('../')) return true;
+
+  // Blocked property chains
+  if (PROPERTY_CHAIN_BLOCKLIST.test(candidate)) return false;
+
+  // Multiple dots without path separator → likely property chain (e.g. config.llm.apiKey)
+  const dotCount = (candidate.match(/\./g) || []).length;
+  if (dotCount >= 2) return false;
+
+  // Single dot: check if extension is a known file extension
+  const ext = candidate.split('.').pop()?.toLowerCase() ?? '';
+  return FILE_EXTENSIONS.has(ext);
+}
+
+/**
+ * Auto-generate concepts from extracted entities.
+ * Merges with any user-provided concepts.
+ */
+export function enrichConcepts(
+  userConcepts: string[],
+  extracted: ExtractedEntities,
+): string[] {
+  const all = new Set(userConcepts.map((c) => c.toLowerCase()));
+  const enriched = [...userConcepts];
+
+  // Add file names (without path) as concepts
+  for (const f of extracted.files) {
+    const name = f.split('/').pop()?.replace(/\.\w+$/, '') ?? '';
+    if (name.length >= 3 && !all.has(name.toLowerCase())) {
+      all.add(name.toLowerCase());
+      enriched.push(name);
+    }
+  }
+
+  // Add module names as concepts
+  for (const m of extracted.modules) {
+    const short = m.split(/[./]/).pop() ?? '';
+    if (short.length >= 3 && !all.has(short.toLowerCase())) {
+      all.add(short.toLowerCase());
+      enriched.push(short);
+    }
+  }
+
+  // Add CamelCase identifiers as concepts
+  for (const id of extracted.identifiers) {
+    if (!all.has(id.toLowerCase())) {
+      all.add(id.toLowerCase());
+      enriched.push(id);
+    }
+  }
+
+  return enriched;
+}