@pcircle/memesh 2.9.2 → 2.9.4

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (142) hide show
  1. package/README.de.md +119 -78
  2. package/README.es.md +116 -75
  3. package/README.fr.md +116 -75
  4. package/README.id.md +115 -74
  5. package/README.ja.md +111 -70
  6. package/README.ko.md +116 -75
  7. package/README.md +113 -72
  8. package/README.th.md +118 -77
  9. package/README.vi.md +116 -75
  10. package/README.zh-CN.md +120 -79
  11. package/README.zh-TW.md +116 -75
  12. package/dist/core/GitCommandParser.d.ts +11 -0
  13. package/dist/core/GitCommandParser.d.ts.map +1 -0
  14. package/dist/core/GitCommandParser.js +43 -0
  15. package/dist/core/GitCommandParser.js.map +1 -0
  16. package/dist/core/HookIntegration.d.ts +0 -3
  17. package/dist/core/HookIntegration.d.ts.map +1 -1
  18. package/dist/core/HookIntegration.js +6 -30
  19. package/dist/core/HookIntegration.js.map +1 -1
  20. package/dist/db/ConnectionPool.d.ts.map +1 -1
  21. package/dist/db/ConnectionPool.js +17 -1
  22. package/dist/db/ConnectionPool.js.map +1 -1
  23. package/dist/db/adapters/BetterSqlite3Adapter.d.ts.map +1 -1
  24. package/dist/db/adapters/BetterSqlite3Adapter.js +22 -0
  25. package/dist/db/adapters/BetterSqlite3Adapter.js.map +1 -1
  26. package/dist/embeddings/EmbeddingService.d.ts +3 -0
  27. package/dist/embeddings/EmbeddingService.d.ts.map +1 -1
  28. package/dist/embeddings/EmbeddingService.js +28 -6
  29. package/dist/embeddings/EmbeddingService.js.map +1 -1
  30. package/dist/embeddings/InMemoryVectorAdapter.d.ts +15 -0
  31. package/dist/embeddings/InMemoryVectorAdapter.d.ts.map +1 -0
  32. package/dist/embeddings/InMemoryVectorAdapter.js +58 -0
  33. package/dist/embeddings/InMemoryVectorAdapter.js.map +1 -0
  34. package/dist/embeddings/SqliteVecAdapter.d.ts +15 -0
  35. package/dist/embeddings/SqliteVecAdapter.d.ts.map +1 -0
  36. package/dist/embeddings/SqliteVecAdapter.js +107 -0
  37. package/dist/embeddings/SqliteVecAdapter.js.map +1 -0
  38. package/dist/embeddings/VectorExtension.d.ts +2 -4
  39. package/dist/embeddings/VectorExtension.d.ts.map +1 -1
  40. package/dist/embeddings/VectorExtension.js.map +1 -1
  41. package/dist/embeddings/VectorSearchAdapter.d.ts +17 -0
  42. package/dist/embeddings/VectorSearchAdapter.d.ts.map +1 -0
  43. package/dist/embeddings/VectorSearchAdapter.js +2 -0
  44. package/dist/embeddings/VectorSearchAdapter.js.map +1 -0
  45. package/dist/embeddings/index.d.ts +6 -1
  46. package/dist/embeddings/index.d.ts.map +1 -1
  47. package/dist/embeddings/index.js +4 -1
  48. package/dist/embeddings/index.js.map +1 -1
  49. package/dist/knowledge-graph/ContentHasher.d.ts +4 -0
  50. package/dist/knowledge-graph/ContentHasher.d.ts.map +1 -0
  51. package/dist/knowledge-graph/ContentHasher.js +8 -0
  52. package/dist/knowledge-graph/ContentHasher.js.map +1 -0
  53. package/dist/knowledge-graph/KGSearchEngine.d.ts +37 -0
  54. package/dist/knowledge-graph/KGSearchEngine.d.ts.map +1 -0
  55. package/dist/knowledge-graph/KGSearchEngine.js +257 -0
  56. package/dist/knowledge-graph/KGSearchEngine.js.map +1 -0
  57. package/dist/knowledge-graph/index.d.ts +18 -7
  58. package/dist/knowledge-graph/index.d.ts.map +1 -1
  59. package/dist/knowledge-graph/index.js +147 -242
  60. package/dist/knowledge-graph/index.js.map +1 -1
  61. package/dist/mcp/ServerInitializer.d.ts.map +1 -1
  62. package/dist/mcp/ServerInitializer.js +1 -1
  63. package/dist/mcp/ServerInitializer.js.map +1 -1
  64. package/dist/mcp/StdinBufferManager.d.ts +11 -0
  65. package/dist/mcp/StdinBufferManager.d.ts.map +1 -0
  66. package/dist/mcp/StdinBufferManager.js +62 -0
  67. package/dist/mcp/StdinBufferManager.js.map +1 -0
  68. package/dist/mcp/ToolDefinitions.js +3 -3
  69. package/dist/mcp/ToolDefinitions.js.map +1 -1
  70. package/dist/mcp/daemon/StdioProxyClient.d.ts.map +1 -1
  71. package/dist/mcp/daemon/StdioProxyClient.js +6 -0
  72. package/dist/mcp/daemon/StdioProxyClient.js.map +1 -1
  73. package/dist/mcp/handlers/HookToolHandler.d.ts +10 -0
  74. package/dist/mcp/handlers/HookToolHandler.d.ts.map +1 -0
  75. package/dist/mcp/handlers/HookToolHandler.js +92 -0
  76. package/dist/mcp/handlers/HookToolHandler.js.map +1 -0
  77. package/dist/mcp/handlers/MemoryToolHandler.d.ts +21 -0
  78. package/dist/mcp/handlers/MemoryToolHandler.d.ts.map +1 -0
  79. package/dist/mcp/handlers/MemoryToolHandler.js +430 -0
  80. package/dist/mcp/handlers/MemoryToolHandler.js.map +1 -0
  81. package/dist/mcp/handlers/SystemToolHandler.d.ts +14 -0
  82. package/dist/mcp/handlers/SystemToolHandler.d.ts.map +1 -0
  83. package/dist/mcp/handlers/SystemToolHandler.js +224 -0
  84. package/dist/mcp/handlers/SystemToolHandler.js.map +1 -0
  85. package/dist/mcp/handlers/ToolHandlers.d.ts +4 -17
  86. package/dist/mcp/handlers/ToolHandlers.d.ts.map +1 -1
  87. package/dist/mcp/handlers/ToolHandlers.js +19 -689
  88. package/dist/mcp/handlers/ToolHandlers.js.map +1 -1
  89. package/dist/mcp/handlers/index.d.ts +3 -0
  90. package/dist/mcp/handlers/index.d.ts.map +1 -1
  91. package/dist/mcp/handlers/index.js +3 -0
  92. package/dist/mcp/handlers/index.js.map +1 -1
  93. package/dist/mcp/server-bootstrap.js +24 -59
  94. package/dist/mcp/server-bootstrap.js.map +1 -1
  95. package/dist/mcp/tools/buddy-do.d.ts.map +1 -1
  96. package/dist/mcp/tools/buddy-do.js +17 -1
  97. package/dist/mcp/tools/buddy-do.js.map +1 -1
  98. package/dist/mcp/tools/buddy-remember.d.ts +1 -1
  99. package/dist/mcp/tools/buddy-remember.d.ts.map +1 -1
  100. package/dist/mcp/tools/buddy-remember.js +58 -16
  101. package/dist/mcp/tools/buddy-remember.js.map +1 -1
  102. package/dist/mcp/tools/create-entities.d.ts.map +1 -1
  103. package/dist/mcp/tools/create-entities.js +18 -24
  104. package/dist/mcp/tools/create-entities.js.map +1 -1
  105. package/dist/memory/MemorySearchEngine.d.ts +17 -0
  106. package/dist/memory/MemorySearchEngine.d.ts.map +1 -0
  107. package/dist/memory/MemorySearchEngine.js +88 -0
  108. package/dist/memory/MemorySearchEngine.js.map +1 -0
  109. package/dist/memory/ProactiveRecaller.d.ts +26 -0
  110. package/dist/memory/ProactiveRecaller.d.ts.map +1 -0
  111. package/dist/memory/ProactiveRecaller.js +96 -0
  112. package/dist/memory/ProactiveRecaller.js.map +1 -0
  113. package/dist/memory/UnifiedMemoryStore.d.ts +1 -0
  114. package/dist/memory/UnifiedMemoryStore.d.ts.map +1 -1
  115. package/dist/memory/UnifiedMemoryStore.js +12 -64
  116. package/dist/memory/UnifiedMemoryStore.js.map +1 -1
  117. package/dist/memory/index.d.ts +3 -0
  118. package/dist/memory/index.d.ts.map +1 -1
  119. package/dist/memory/index.js +2 -0
  120. package/dist/memory/index.js.map +1 -1
  121. package/dist/ui/MetricsStore.d.ts.map +1 -1
  122. package/dist/ui/MetricsStore.js +21 -2
  123. package/dist/ui/MetricsStore.js.map +1 -1
  124. package/dist/utils/index.d.ts +0 -2
  125. package/dist/utils/index.d.ts.map +1 -1
  126. package/dist/utils/index.js +0 -2
  127. package/dist/utils/index.js.map +1 -1
  128. package/dist/utils/tracing/index.d.ts +0 -1
  129. package/dist/utils/tracing/index.d.ts.map +1 -1
  130. package/dist/utils/tracing/index.js +0 -1
  131. package/dist/utils/tracing/index.js.map +1 -1
  132. package/package.json +2 -12
  133. package/plugin.json +1 -1
  134. package/scripts/hooks/__tests__/post-tool-use-recall.test.js +192 -0
  135. package/scripts/hooks/__tests__/session-start-recall.test.js +86 -0
  136. package/scripts/hooks/post-tool-use-recall-utils.js +74 -0
  137. package/scripts/hooks/post-tool-use.js +83 -0
  138. package/scripts/hooks/pre-tool-use.js +30 -1
  139. package/scripts/hooks/session-start-recall-utils.js +40 -0
  140. package/scripts/hooks/session-start.js +66 -0
  141. package/scripts/hooks/templates/planning-template.md +46 -0
  142. package/scripts/postinstall-new.js +8 -1
@@ -46,6 +46,28 @@ export class BetterSqlite3Adapter {
46
46
  try {
47
47
  const Database = (await import('better-sqlite3')).default;
48
48
  const db = new Database(dbPath, options);
49
+ if (dbPath !== ':memory:') {
50
+ try {
51
+ const result = db.pragma('quick_check', { simple: true });
52
+ if (result !== 'ok') {
53
+ logger.warn('[BetterSqlite3Adapter] Database quick_check failed', {
54
+ dbPath,
55
+ result,
56
+ });
57
+ db.close();
58
+ throw new Error(`Database integrity check failed: ${result}`);
59
+ }
60
+ }
61
+ catch (error) {
62
+ if (error instanceof Error && error.message.startsWith('Database integrity check failed')) {
63
+ throw error;
64
+ }
65
+ logger.warn('[BetterSqlite3Adapter] Could not run quick_check pragma', {
66
+ dbPath,
67
+ error: error instanceof Error ? error.message : String(error),
68
+ });
69
+ }
70
+ }
49
71
  logger.debug('[BetterSqlite3Adapter] Successfully loaded native module', {
50
72
  dbPath,
51
73
  inMemory: dbPath === ':memory:',
@@ -1 +1 @@
1
- {"version":3,"file":"BetterSqlite3Adapter.js","sourceRoot":"","sources":["../../../src/db/adapters/BetterSqlite3Adapter.ts"],"names":[],"mappings":"AAyBA,OAAO,EAAE,MAAM,EAAE,MAAM,uBAAuB,CAAC;AAgB/C,MAAM,CAAC,KAAK,UAAU,8BAA8B;IAClD,IAAI,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAG9C,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,EAAE,CAAC;QAEf,MAAM,CAAC,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACjE,OAAO;YACL,SAAS,EAAE,IAAI;YACf,IAAI,EAAE,gBAAgB;SACvB,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAG5E,IAAI,kBAAkB,GAAG,EAAE,CAAC;QAE5B,IAAI,YAAY,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;YAChD,kBAAkB,GAAG,iCAAiC,CAAC;QACzD,CAAC;aAAM,IAAI,YAAY,CAAC,QAAQ,CAAC,sBAAsB,CAAC,EAAE,CAAC;YACzD,kBAAkB,GAAG,iCAAiC,CAAC;QACzD,CAAC;aAAM,IAAI,YAAY,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;YACrF,kBAAkB,GAAG,gFAAgF,CAAC;QACxG,CAAC;aAAM,CAAC;YACN,kBAAkB,GAAG,qEAAqE,CAAC;QAC7F,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,kDAAkD,EAAE;YAC9D,KAAK,EAAE,YAAY;YACnB,UAAU,EAAE,kBAAkB;SAC/B,CAAC,CAAC;QAEH,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,IAAI,EAAE,gBAAgB;YACtB,KAAK,EAAE,YAAY;YACnB,kBAAkB;SACnB,CAAC;IACJ,CAAC;AACH,CAAC;AAOD,MAAM,OAAO,oBAAoB;IACvB,EAAE,CAAoB;IAK9B,YAAoB,EAAqB;QACvC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;IACf,CAAC;IAaD,MAAM,CAAC,KAAK,CAAC,MAAM,CACjB,MAAc,EACd,OAA0B;QAE1B,IAAI,CAAC;YAEH,MAAM,QAAQ,GAAG,CAAC,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC,CAAC,OAAO,CAAC;YAE1D,MAAM,EAAE,GAAG,IAAI,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAEzC,MAAM,CAAC,KAAK,CAAC,0DAA0D,EAAE;gBACvE,MAAM;gBACN,QAAQ,EAAE,MAAM,KAAK,UAAU;aAChC,CAAC,CAAC;YAEH,OAAO,IAAI,oBAAoB,CAAC,EAAE,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAE5E,MAAM,CAAC,KAAK,CAAC,sDAAsD,EAAE;gBACnE,KAAK,EAAE,YAAY;gBACnB,MAAM;aACP,CAAC,CAAC;YAEH,MAAM,IAAI,KAAK,CACb,kCAAkC,YAAY,IAAI;gBAClD,gFAAgF,CACjF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO,CAAC,GAAW;QACjB,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAGlC,OAAO;YACL,GAAG,EAAE,CAAU,GAAG,MAAa,EAAO,EAAE;gBACtC,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAQ,CAAC;YACpC,CAAC;YAED,GAAG,EAAE,CAAU,GAAG,MAAa,EAAiB,EAAE;gBAChD,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAkB,CAAC;YAC9C,CAAC;YAED,GAAG,EAAE,CAAC,GAAG,MAAa,EAAc,EAAE;gBACpC,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC;gBACnC,OAAO;oBACL,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,eAAe,EAAE,MAAM,CAAC,eAAe;iBACxC,CAAC;YACJ,CAAC;YAED,OAAO,EAAE,CAAU,GAAG,MAAa,EAAuB,EAAE;gBAC1D,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,MAAM,CAAwB,CAAC;YACxD,CAAC;SACF,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,GAAW;QACd,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACpB,CAAC;IAED,KAAK;QACH,IAAI,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YACjB,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,KAAK,CAAC,mDAAmD,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC;IACxB,CAAC;IAED,IAAI,IAAI;QACN,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;IACtB,CAAC;CACF"}
1
+ {"version":3,"file":"BetterSqlite3Adapter.js","sourceRoot":"","sources":["../../../src/db/adapters/BetterSqlite3Adapter.ts"],"names":[],"mappings":"AAyBA,OAAO,EAAE,MAAM,EAAE,MAAM,uBAAuB,CAAC;AAgB/C,MAAM,CAAC,KAAK,UAAU,8BAA8B;IAClD,IAAI,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAG9C,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,EAAE,CAAC;QAEf,MAAM,CAAC,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACjE,OAAO;YACL,SAAS,EAAE,IAAI;YACf,IAAI,EAAE,gBAAgB;SACvB,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAG5E,IAAI,kBAAkB,GAAG,EAAE,CAAC;QAE5B,IAAI,YAAY,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;YAChD,kBAAkB,GAAG,iCAAiC,CAAC;QACzD,CAAC;aAAM,IAAI,YAAY,CAAC,QAAQ,CAAC,sBAAsB,CAAC,EAAE,CAAC;YACzD,kBAAkB,GAAG,iCAAiC,CAAC;QACzD,CAAC;aAAM,IAAI,YAAY,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;YACrF,kBAAkB,GAAG,gFAAgF,CAAC;QACxG,CAAC;aAAM,CAAC;YACN,kBAAkB,GAAG,qEAAqE,CAAC;QAC7F,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,kDAAkD,EAAE;YAC9D,KAAK,EAAE,YAAY;YACnB,UAAU,EAAE,kBAAkB;SAC/B,CAAC,CAAC;QAEH,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,IAAI,EAAE,gBAAgB;YACtB,KAAK,EAAE,YAAY;YACnB,kBAAkB;SACnB,CAAC;IACJ,CAAC;AACH,CAAC;AAOD,MAAM,OAAO,oBAAoB;IACvB,EAAE,CAAoB;IAK9B,YAAoB,EAAqB;QACvC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;IACf,CAAC;IAaD,MAAM,CAAC,KAAK,CAAC,MAAM,CACjB,MAAc,EACd,OAA0B;QAE1B,IAAI,CAAC;YAEH,MAAM,QAAQ,GAAG,CAAC,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC,CAAC,OAAO,CAAC;YAE1D,MAAM,EAAE,GAAG,IAAI,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAGzC,IAAI,MAAM,KAAK,UAAU,EAAE,CAAC;gBAC1B,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,EAAE,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAW,CAAC;oBACpE,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;wBACpB,MAAM,CAAC,IAAI,CAAC,oDAAoD,EAAE;4BAChE,MAAM;4BACN,MAAM;yBACP,CAAC,CAAC;wBACH,EAAE,CAAC,KAAK,EAAE,CAAC;wBACX,MAAM,IAAI,KAAK,CAAC,oCAAoC,MAAM,EAAE,CAAC,CAAC;oBAChE,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,iCAAiC,CAAC,EAAE,CAAC;wBAC1F,MAAM,KAAK,CAAC;oBACd,CAAC;oBACD,MAAM,CAAC,IAAI,CAAC,yDAAyD,EAAE;wBACrE,MAAM;wBACN,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;qBAC9D,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAED,MAAM,CAAC,KAAK,CAAC,0DAA0D,EAAE;gBACvE,MAAM;gBACN,QAAQ,EAAE,MAAM,KAAK,UAAU;aAChC,CAAC,CAAC;YAEH,OAAO,IAAI,oBAAoB,CAAC,EAAE,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAE5E,MAAM,CAAC,KAAK,CAAC,sDAAsD,EAAE;gBACnE,KAAK,EAAE,YAAY;gBACnB,MAAM;aACP,CAAC,CAAC;YAEH,MAAM,IAAI,KAAK,CACb,kCAAkC,YAAY,IAAI;gBAClD,gFAAgF,CACjF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO,CAAC,GAAW;QACjB,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAGlC,OAAO;YACL,GAAG,EAAE,CAAU,GAAG,MAAa,EAAO,EAAE;gBACtC,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAQ,CAAC;YACpC,CAAC;YAED,GAAG,EAAE,CAAU,GAAG,MAAa,EAAiB,EAAE;gBAChD,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAkB,CAAC;YAC9C,CAAC;YAED,GAAG,EAAE,CAAC,GAAG,MAAa,EAAc,EAAE;gBACpC,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC;gBACnC,OAAO;oBACL,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,eAAe,EAAE,MAAM,CAAC,eAAe;iBACxC,CAAC;YACJ,CAAC;YAED,OAAO,EAAE,CAAU,GAAG,MAAa,EAAuB,EAAE;gBAC1D,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,MAAM,CAAwB,CAAC;YACxD,CAAC;SACF,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,GAAW;QACd,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACpB,CAAC;IAED,KAAK;QACH,IAAI,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YACjB,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,KAAK,CAAC,mDAAmD,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC;IACxB,CAAC;IAED,IAAI,IAAI;QACN,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;IACtB,CAAC;CACF"}
@@ -1,8 +1,10 @@
1
1
  export declare class EmbeddingService {
2
2
  private static readonly DIMENSIONS;
3
3
  private static readonly MODEL_NAME;
4
+ private static readonly MAX_CACHE_SIZE;
4
5
  private extractor;
5
6
  private initialized;
7
+ private cache;
6
8
  initialize(): Promise<void>;
7
9
  isInitialized(): boolean;
8
10
  encode(text: string): Promise<Float32Array>;
@@ -15,6 +17,7 @@ export declare class LazyEmbeddingService {
15
17
  private static instance;
16
18
  private static initPromise;
17
19
  static get(): Promise<EmbeddingService>;
20
+ static preload(): Promise<void>;
18
21
  static dispose(): Promise<void>;
19
22
  }
20
23
  //# sourceMappingURL=EmbeddingService.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"EmbeddingService.d.ts","sourceRoot":"","sources":["../../src/embeddings/EmbeddingService.ts"],"names":[],"mappings":"AA+CA,qBAAa,gBAAgB;IAE3B,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAO;IAGzC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAA6B;IAG/D,OAAO,CAAC,SAAS,CAA0C;IAG3D,OAAO,CAAC,WAAW,CAAS;IAWtB,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IA4BjC,aAAa,IAAI,OAAO;IAoBlB,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IA+B3C,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IA+B3D,gBAAgB,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,YAAY,GAAG,MAAM;IA4B1D,aAAa,IAAI,MAAM;IAUjB,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CAK/B;AAmBD,qBAAa,oBAAoB;IAE/B,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAiC;IAGxD,OAAO,CAAC,MAAM,CAAC,WAAW,CAA0C;WAYvD,GAAG,IAAI,OAAO,CAAC,gBAAgB,CAAC;WA+BhC,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CAOtC"}
1
+ {"version":3,"file":"EmbeddingService.d.ts","sourceRoot":"","sources":["../../src/embeddings/EmbeddingService.ts"],"names":[],"mappings":"AAgDA,qBAAa,gBAAgB;IAE3B,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAO;IAGzC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAA6B;IAG/D,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAO;IAG7C,OAAO,CAAC,SAAS,CAA0C;IAG3D,OAAO,CAAC,WAAW,CAAS;IAG5B,OAAO,CAAC,KAAK,CAA4E;IAWnF,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IA4BjC,aAAa,IAAI,OAAO;IAoBlB,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IA0C3C,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAiC3D,gBAAgB,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,YAAY,GAAG,MAAM;IA4B1D,aAAa,IAAI,MAAM;IAUjB,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CAM/B;AAmBD,qBAAa,oBAAoB;IAE/B,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAiC;IAGxD,OAAO,CAAC,MAAM,CAAC,WAAW,CAA0C;WAYvD,GAAG,IAAI,OAAO,CAAC,gBAAgB,CAAC;WAgChC,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;WAiBxB,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CAOtC"}
@@ -2,14 +2,17 @@ import path from 'path';
2
2
  import { pipeline, env } from '@xenova/transformers';
3
3
  import { ModelManager } from './ModelManager.js';
4
4
  import { logger } from '../utils/logger.js';
5
+ import { LRUCache } from '../utils/lru-cache.js';
5
6
  const modelDir = new ModelManager().getModelDir();
6
7
  env.cacheDir = path.dirname(modelDir);
7
8
  env.allowRemoteModels = true;
8
9
  export class EmbeddingService {
9
10
  static DIMENSIONS = 384;
10
11
  static MODEL_NAME = 'Xenova/all-MiniLM-L6-v2';
12
+ static MAX_CACHE_SIZE = 500;
11
13
  extractor = null;
12
14
  initialized = false;
15
+ cache = new LRUCache({ maxSize: EmbeddingService.MAX_CACHE_SIZE });
13
16
  async initialize() {
14
17
  if (this.initialized) {
15
18
  return;
@@ -34,19 +37,27 @@ export class EmbeddingService {
34
37
  if (!this.extractor) {
35
38
  throw new Error('EmbeddingService not initialized. Call initialize() first.');
36
39
  }
40
+ const cached = this.cache.get(text);
41
+ if (cached) {
42
+ return new Float32Array(cached);
43
+ }
37
44
  const result = await this.extractor(text, {
38
45
  pooling: 'mean',
39
46
  normalize: true,
40
47
  });
41
- return new Float32Array(result.data);
48
+ const embedding = new Float32Array(result.data);
49
+ this.cache.set(text, new Float32Array(embedding));
50
+ return embedding;
42
51
  }
43
52
  async encodeBatch(texts) {
44
- const embeddings = [];
45
- for (const text of texts) {
46
- const embedding = await this.encode(text);
47
- embeddings.push(embedding);
53
+ const chunkSize = 10;
54
+ const results = [];
55
+ for (let i = 0; i < texts.length; i += chunkSize) {
56
+ const chunk = texts.slice(i, i + chunkSize);
57
+ const chunkResults = await Promise.all(chunk.map(text => this.encode(text)));
58
+ results.push(...chunkResults);
48
59
  }
49
- return embeddings;
60
+ return results;
50
61
  }
51
62
  cosineSimilarity(a, b) {
52
63
  if (a.length !== b.length) {
@@ -72,6 +83,7 @@ export class EmbeddingService {
72
83
  async dispose() {
73
84
  this.extractor = null;
74
85
  this.initialized = false;
86
+ this.cache.clear();
75
87
  logger.debug('EmbeddingService disposed');
76
88
  }
77
89
  }
@@ -98,6 +110,16 @@ export class LazyEmbeddingService {
98
110
  }
99
111
  return this.initPromise;
100
112
  }
113
+ static async preload() {
114
+ try {
115
+ await LazyEmbeddingService.get();
116
+ }
117
+ catch (err) {
118
+ logger.debug('ONNX model preload failed (will load on first use)', {
119
+ error: err instanceof Error ? err.message : String(err),
120
+ });
121
+ }
122
+ }
101
123
  static async dispose() {
102
124
  if (this.instance) {
103
125
  await this.instance.dispose();
@@ -1 +1 @@
1
- {"version":3,"file":"EmbeddingService.js","sourceRoot":"","sources":["../../src/embeddings/EmbeddingService.ts"],"names":[],"mappings":"AA6BA,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,QAAQ,EAAE,GAAG,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAG5C,MAAM,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC,WAAW,EAAE,CAAC;AAClD,GAAG,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;AACtC,GAAG,CAAC,iBAAiB,GAAG,IAAI,CAAC;AAU7B,MAAM,OAAO,gBAAgB;IAEnB,MAAM,CAAU,UAAU,GAAG,GAAG,CAAC;IAGjC,MAAM,CAAU,UAAU,GAAG,yBAAyB,CAAC;IAGvD,SAAS,GAAqC,IAAI,CAAC;IAGnD,WAAW,GAAG,KAAK,CAAC;IAW5B,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAO;QACT,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,kCAAkC,EAAE,EAAE,KAAK,EAAE,gBAAgB,CAAC,UAAU,EAAE,CAAC,CAAC;QAExF,IAAI,CAAC;YAEH,IAAI,CAAC,SAAS,GAAG,MAAM,QAAQ,CAAC,oBAAoB,EAAE,gBAAgB,CAAC,UAAU,EAAE;gBACjF,SAAS,EAAE,KAAK;aACjB,CAAC,CAAC;YAEH,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YACxB,MAAM,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAC;QAC3D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,uCAAuC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YACjE,MAAM,IAAI,KAAK,CACb,2CAA2C,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,EAAE,CAC5F,CAAC;QACJ,CAAC;IACH,CAAC;IAOD,aAAa;QACX,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAkBD,KAAK,CAAC,MAAM,CAAC,IAAY;QACvB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;QAChF,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE;YACxC,OAAO,EAAE,MAAM;YACf,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;QAGH,OAAO,IAAI,YAAY,CAAC,MAAM,CAAC,IAAyB,CAAC,CAAC;IAC5D,CAAC;IAmBD,KAAK,CAAC,WAAW,CAAC,KAAe;QAC/B,MAAM,UAAU,GAAmB,EAAE,CAAC;QAEtC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC1C,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC7B,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAsBD,gBAAgB,CAAC,CAAe,EAAE,CAAe;QAC/C,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC,MAAM,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/E,CAAC;QAED,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAClC,UAAU,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAC1B,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACrB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACvB,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtD,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;YACpB,OAAO,CAAC,CAAC;QACX,CAAC;QAED,OAAO,UAAU,GAAG,SAAS,CAAC;IAChC,CAAC;IAOD,aAAa;QACX,OAAO,gBAAgB,CAAC,UAAU,CAAC;IACrC,CAAC;IAQD,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QACzB,MAAM,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;IAC5C,CAAC;;AAoBH,MAAM,OAAO,oBAAoB;IAEvB,MAAM,CAAC,QAAQ,GAA4B,IAAI,CAAC;IAGhD,MAAM,CAAC,WAAW,GAAqC,IAAI,CAAC;IAYpE,MAAM,CAAC,KAAK,CAAC,GAAG;QAEd,IAAI,IAAI,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE,CAAC;YACnC,OAAO,IAAI,CAAC,QAAQ,CAAC;QACvB,CAAC;QAGD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,IAAI,CAAC,WAAW,GAAG,CAAC,KAAK,IAAI,EAAE;gBAC7B,IAAI,CAAC;oBACH,IAAI,CAAC,QAAQ,GAAG,IAAI,gBAAgB,EAAE,CAAC;oBACvC,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC;oBACjC,OAAO,IAAI,CAAC,QAAQ,CAAC;gBACvB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBAEf,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;oBACxB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;oBACrB,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC,CAAC,EAAE,CAAC;QACP,CAAC;QAED,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAQD,MAAM,CAAC,KAAK,CAAC,OAAO;QAClB,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YAC9B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;YACrB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAC1B,CAAC;IACH,CAAC"}
1
+ {"version":3,"file":"EmbeddingService.js","sourceRoot":"","sources":["../../src/embeddings/EmbeddingService.ts"],"names":[],"mappings":"AA6BA,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,QAAQ,EAAE,GAAG,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AAGjD,MAAM,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC,WAAW,EAAE,CAAC;AAClD,GAAG,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;AACtC,GAAG,CAAC,iBAAiB,GAAG,IAAI,CAAC;AAU7B,MAAM,OAAO,gBAAgB;IAEnB,MAAM,CAAU,UAAU,GAAG,GAAG,CAAC;IAGjC,MAAM,CAAU,UAAU,GAAG,yBAAyB,CAAC;IAGvD,MAAM,CAAU,cAAc,GAAG,GAAG,CAAC;IAGrC,SAAS,GAAqC,IAAI,CAAC;IAGnD,WAAW,GAAG,KAAK,CAAC;IAGpB,KAAK,GAAG,IAAI,QAAQ,CAAe,EAAE,OAAO,EAAE,gBAAgB,CAAC,cAAc,EAAE,CAAC,CAAC;IAWzF,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAO;QACT,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,kCAAkC,EAAE,EAAE,KAAK,EAAE,gBAAgB,CAAC,UAAU,EAAE,CAAC,CAAC;QAExF,IAAI,CAAC;YAEH,IAAI,CAAC,SAAS,GAAG,MAAM,QAAQ,CAAC,oBAAoB,EAAE,gBAAgB,CAAC,UAAU,EAAE;gBACjF,SAAS,EAAE,KAAK;aACjB,CAAC,CAAC;YAEH,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YACxB,MAAM,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAC;QAC3D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,uCAAuC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YACjE,MAAM,IAAI,KAAK,CACb,2CAA2C,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,EAAE,CAC5F,CAAC;QACJ,CAAC;IACH,CAAC;IAOD,aAAa;QACX,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAkBD,KAAK,CAAC,MAAM,CAAC,IAAY;QACvB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;QAChF,CAAC;QAGD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACpC,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE;YACxC,OAAO,EAAE,MAAM;YACf,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;QAGH,MAAM,SAAS,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,IAAyB,CAAC,CAAC;QAGrE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC;QAElD,OAAO,SAAS,CAAC;IACnB,CAAC;IAmBD,KAAK,CAAC,WAAW,CAAC,KAAe;QAC/B,MAAM,SAAS,GAAG,EAAE,CAAC;QACrB,MAAM,OAAO,GAAmB,EAAE,CAAC;QAEnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;YACjD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC;YAC5C,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC7E,OAAO,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAsBD,gBAAgB,CAAC,CAAe,EAAE,CAAe;QAC/C,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC,MAAM,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/E,CAAC;QAED,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAClC,UAAU,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAC1B,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACrB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACvB,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtD,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;YACpB,OAAO,CAAC,CAAC;QACX,CAAC;QAED,OAAO,UAAU,GAAG,SAAS,CAAC;IAChC,CAAC;IAOD,aAAa;QACX,OAAO,gBAAgB,CAAC,UAAU,CAAC;IACrC,CAAC;IAQD,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QACzB,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACnB,MAAM,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;IAC5C,CAAC;;AAoBH,MAAM,OAAO,oBAAoB;IAEvB,MAAM,CAAC,QAAQ,GAA4B,IAAI,CAAC;IAGhD,MAAM,CAAC,WAAW,GAAqC,IAAI,CAAC;IAYpE,MAAM,CAAC,KAAK,CAAC,GAAG;QAEd,IAAI,IAAI,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE,CAAC;YACnC,OAAO,IAAI,CAAC,QAAQ,CAAC;QACvB,CAAC;QAGD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,IAAI,CAAC,WAAW,GAAG,CAAC,KAAK,IAAI,EAAE;gBAC7B,IAAI,CAAC;oBACH,IAAI,CAAC,QAAQ,GAAG,IAAI,gBAAgB,EAAE,CAAC;oBACvC,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC;oBACjC,OAAO,IAAI,CAAC,QAAQ,CAAC;gBACvB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBAEf,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;oBACxB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;oBACrB,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC,CAAC,EAAE,CAAC;QACP,CAAC;QAED,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IASD,MAAM,CAAC,KAAK,CAAC,OAAO;QAClB,IAAI,CAAC;YACH,MAAM,oBAAoB,CAAC,GAAG,EAAE,CAAC;QACnC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YAEb,MAAM,CAAC,KAAK,CAAC,oDAAoD,EAAE;gBACjE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;aACxD,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAQD,MAAM,CAAC,KAAK,CAAC,OAAO;QAClB,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YAC9B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;YACrB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAC1B,CAAC;IACH,CAAC"}
@@ -0,0 +1,15 @@
1
+ import type Database from 'better-sqlite3';
2
+ import type { VectorSearchAdapter, KnnSearchResult } from './VectorSearchAdapter.js';
3
+ export declare class InMemoryVectorAdapter implements VectorSearchAdapter {
4
+ private vectors;
5
+ loadExtension(_db: Database.Database): void;
6
+ createVectorTable(_db: Database.Database, _dimensions?: number): void;
7
+ insertEmbedding(_db: Database.Database, entityName: string, embedding: Float32Array, dimensions?: number): void;
8
+ deleteEmbedding(_db: Database.Database, entityName: string): void;
9
+ knnSearch(_db: Database.Database, queryEmbedding: Float32Array, k: number, dimensions?: number): KnnSearchResult[];
10
+ getEmbedding(_db: Database.Database, entityName: string): Float32Array | null;
11
+ hasEmbedding(_db: Database.Database, entityName: string): boolean;
12
+ getEmbeddingCount(_db: Database.Database): number;
13
+ private cosineDistance;
14
+ }
15
+ //# sourceMappingURL=InMemoryVectorAdapter.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"InMemoryVectorAdapter.d.ts","sourceRoot":"","sources":["../../src/embeddings/InMemoryVectorAdapter.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,QAAQ,MAAM,gBAAgB,CAAC;AAC3C,OAAO,KAAK,EAAE,mBAAmB,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAGrF,qBAAa,qBAAsB,YAAW,mBAAmB;IAC/D,OAAO,CAAC,OAAO,CAAmC;IAElD,aAAa,CAAC,GAAG,EAAE,QAAQ,CAAC,QAAQ,GAAG,IAAI;IAI3C,iBAAiB,CAAC,GAAG,EAAE,QAAQ,CAAC,QAAQ,EAAE,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI;IAIrE,eAAe,CACb,GAAG,EAAE,QAAQ,CAAC,QAAQ,EACtB,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,YAAY,EACvB,UAAU,GAAE,MAAqC,GAChD,IAAI;IASP,eAAe,CAAC,GAAG,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,GAAG,IAAI;IAIjE,SAAS,CACP,GAAG,EAAE,QAAQ,CAAC,QAAQ,EACtB,cAAc,EAAE,YAAY,EAC5B,CAAC,EAAE,MAAM,EACT,UAAU,GAAE,MAAqC,GAChD,eAAe,EAAE;IAkBpB,YAAY,CAAC,GAAG,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,GAAG,YAAY,GAAG,IAAI;IAM7E,YAAY,CAAC,GAAG,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO;IAIjE,iBAAiB,CAAC,GAAG,EAAE,QAAQ,CAAC,QAAQ,GAAG,MAAM;IAIjD,OAAO,CAAC,cAAc;CAavB"}
@@ -0,0 +1,58 @@
1
+ import { DEFAULT_EMBEDDING_DIMENSIONS } from './VectorSearchAdapter.js';
2
+ export class InMemoryVectorAdapter {
3
+ vectors = new Map();
4
+ loadExtension(_db) {
5
+ }
6
+ createVectorTable(_db, _dimensions) {
7
+ }
8
+ insertEmbedding(_db, entityName, embedding, dimensions = DEFAULT_EMBEDDING_DIMENSIONS) {
9
+ if (embedding.length !== dimensions) {
10
+ throw new Error(`Invalid embedding dimensions: expected ${dimensions}, got ${embedding.length}. Entity: ${entityName}`);
11
+ }
12
+ this.vectors.set(entityName, new Float32Array(embedding));
13
+ }
14
+ deleteEmbedding(_db, entityName) {
15
+ this.vectors.delete(entityName);
16
+ }
17
+ knnSearch(_db, queryEmbedding, k, dimensions = DEFAULT_EMBEDDING_DIMENSIONS) {
18
+ if (queryEmbedding.length !== dimensions) {
19
+ throw new Error(`Invalid query embedding dimensions: expected ${dimensions}, got ${queryEmbedding.length}`);
20
+ }
21
+ const results = [];
22
+ for (const [name, vec] of this.vectors) {
23
+ results.push({
24
+ entityName: name,
25
+ distance: this.cosineDistance(queryEmbedding, vec),
26
+ });
27
+ }
28
+ results.sort((a, b) => a.distance - b.distance);
29
+ return results.slice(0, k);
30
+ }
31
+ getEmbedding(_db, entityName) {
32
+ const vec = this.vectors.get(entityName);
33
+ if (!vec)
34
+ return null;
35
+ return new Float32Array(vec);
36
+ }
37
+ hasEmbedding(_db, entityName) {
38
+ return this.vectors.has(entityName);
39
+ }
40
+ getEmbeddingCount(_db) {
41
+ return this.vectors.size;
42
+ }
43
+ cosineDistance(a, b) {
44
+ let dot = 0;
45
+ let normA = 0;
46
+ let normB = 0;
47
+ for (let i = 0; i < a.length; i++) {
48
+ dot += a[i] * b[i];
49
+ normA += a[i] * a[i];
50
+ normB += b[i] * b[i];
51
+ }
52
+ const denom = Math.sqrt(normA) * Math.sqrt(normB);
53
+ if (denom === 0)
54
+ return 1;
55
+ return 1 - dot / denom;
56
+ }
57
+ }
58
+ //# sourceMappingURL=InMemoryVectorAdapter.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"InMemoryVectorAdapter.js","sourceRoot":"","sources":["../../src/embeddings/InMemoryVectorAdapter.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,4BAA4B,EAAE,MAAM,0BAA0B,CAAC;AAExE,MAAM,OAAO,qBAAqB;IACxB,OAAO,GAAG,IAAI,GAAG,EAAwB,CAAC;IAElD,aAAa,CAAC,GAAsB;IAEpC,CAAC;IAED,iBAAiB,CAAC,GAAsB,EAAE,WAAoB;IAE9D,CAAC;IAED,eAAe,CACb,GAAsB,EACtB,UAAkB,EAClB,SAAuB,EACvB,aAAqB,4BAA4B;QAEjD,IAAI,SAAS,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;YACpC,MAAM,IAAI,KAAK,CACb,0CAA0C,UAAU,SAAS,SAAS,CAAC,MAAM,aAAa,UAAU,EAAE,CACvG,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC;IAC5D,CAAC;IAED,eAAe,CAAC,GAAsB,EAAE,UAAkB;QACxD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,SAAS,CACP,GAAsB,EACtB,cAA4B,EAC5B,CAAS,EACT,aAAqB,4BAA4B;QAEjD,IAAI,cAAc,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CACb,gDAAgD,UAAU,SAAS,cAAc,CAAC,MAAM,EAAE,CAC3F,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAsB,EAAE,CAAC;QACtC,KAAK,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACvC,OAAO,CAAC,IAAI,CAAC;gBACX,UAAU,EAAE,IAAI;gBAChB,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,cAAc,EAAE,GAAG,CAAC;aACnD,CAAC,CAAC;QACL,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;QAChD,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC7B,CAAC;IAED,YAAY,CAAC,GAAsB,EAAE,UAAkB;QACrD,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACzC,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAC;QACtB,OAAO,IAAI,YAAY,CAAC,GAAG,CAAC,CAAC;IAC/B,CAAC;IAED,YAAY,CAAC,GAAsB,EAAE,UAAkB;QACrD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACtC,CAAC;IAED,iBAAiB,CAAC,GAAsB;QACtC,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IAC3B,CAAC;IAEO,cAAc,CAAC,CAAe,EAAE,CAAe;QACrD,IAAI,GAAG,GAAG,CAAC,CAAC;QACZ,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAClC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACnB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACrB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACvB,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAClD,IAAI,KAAK,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QAC1B,OAAO,CAAC,GAAG,GAAG,GAAG,KAAK,CAAC;IACzB,CAAC;CACF"}
@@ -0,0 +1,15 @@
1
+ import Database from 'better-sqlite3';
2
+ import type { VectorSearchAdapter, KnnSearchResult } from './VectorSearchAdapter.js';
3
+ export declare class SqliteVecAdapter implements VectorSearchAdapter {
4
+ private extensionLoaded;
5
+ loadExtension(db: Database.Database): void;
6
+ createVectorTable(db: Database.Database, dimensions?: number): void;
7
+ insertEmbedding(db: Database.Database, entityName: string, embedding: Float32Array, dimensions?: number): void;
8
+ deleteEmbedding(db: Database.Database, entityName: string): void;
9
+ knnSearch(db: Database.Database, queryEmbedding: Float32Array, k: number, dimensions?: number): KnnSearchResult[];
10
+ getEmbedding(db: Database.Database, entityName: string): Float32Array | null;
11
+ hasEmbedding(db: Database.Database, entityName: string): boolean;
12
+ getEmbeddingCount(db: Database.Database): number;
13
+ private ensureExtensionLoaded;
14
+ }
15
+ //# sourceMappingURL=SqliteVecAdapter.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"SqliteVecAdapter.d.ts","sourceRoot":"","sources":["../../src/embeddings/SqliteVecAdapter.ts"],"names":[],"mappings":"AAQA,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AAGtC,OAAO,KAAK,EAAE,mBAAmB,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAqBrF,qBAAa,gBAAiB,YAAW,mBAAmB;IAE1D,OAAO,CAAC,eAAe,CAAoC;IAE3D,aAAa,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,GAAG,IAAI;IAiB1C,iBAAiB,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,GAAE,MAAqC,GAAG,IAAI;IAajG,eAAe,CACb,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,YAAY,EACvB,UAAU,GAAE,MAAqC,GAChD,IAAI;IAwBP,eAAe,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,GAAG,IAAI;IAMhE,SAAS,CACP,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,cAAc,EAAE,YAAY,EAC5B,CAAC,EAAE,MAAM,EACT,UAAU,GAAE,MAAqC,GAChD,eAAe,EAAE;IA4BpB,YAAY,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,GAAG,YAAY,GAAG,IAAI;IAsB5E,YAAY,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO;IAOhE,iBAAiB,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,GAAG,MAAM;IAShD,OAAO,CAAC,qBAAqB;CAK9B"}
@@ -0,0 +1,107 @@
1
+ import * as sqliteVec from 'sqlite-vec';
2
+ import { logger } from '../utils/logger.js';
3
+ import { DEFAULT_EMBEDDING_DIMENSIONS } from './VectorSearchAdapter.js';
4
+ function bufferToFloat32Array(buffer) {
5
+ const arrayBuffer = buffer.buffer.slice(buffer.byteOffset, buffer.byteOffset + buffer.byteLength);
6
+ return new Float32Array(arrayBuffer);
7
+ }
8
+ export class SqliteVecAdapter {
9
+ extensionLoaded = new WeakSet();
10
+ loadExtension(db) {
11
+ if (this.extensionLoaded.has(db)) {
12
+ return;
13
+ }
14
+ try {
15
+ sqliteVec.load(db);
16
+ this.extensionLoaded.add(db);
17
+ logger.debug('sqlite-vec extension loaded successfully');
18
+ }
19
+ catch (error) {
20
+ logger.error('Failed to load sqlite-vec extension', { error });
21
+ throw new Error(`Failed to load sqlite-vec: ${error instanceof Error ? error.message : error}`);
22
+ }
23
+ }
24
+ createVectorTable(db, dimensions = DEFAULT_EMBEDDING_DIMENSIONS) {
25
+ this.ensureExtensionLoaded(db);
26
+ db.exec(`
27
+ CREATE VIRTUAL TABLE IF NOT EXISTS entity_embeddings USING vec0(
28
+ entity_name TEXT PRIMARY KEY,
29
+ embedding float[${dimensions}] distance_metric=cosine
30
+ )
31
+ `);
32
+ logger.debug('Vector table created', { dimensions });
33
+ }
34
+ insertEmbedding(db, entityName, embedding, dimensions = DEFAULT_EMBEDDING_DIMENSIONS) {
35
+ if (embedding.length !== dimensions) {
36
+ throw new Error(`Invalid embedding dimensions: expected ${dimensions}, got ${embedding.length}. ` +
37
+ `Entity: ${entityName}`);
38
+ }
39
+ this.ensureExtensionLoaded(db);
40
+ const deleteStmt = db.prepare('DELETE FROM entity_embeddings WHERE entity_name = ?');
41
+ deleteStmt.run(entityName);
42
+ const insertStmt = db.prepare(`
43
+ INSERT INTO entity_embeddings (entity_name, embedding)
44
+ VALUES (?, ?)
45
+ `);
46
+ const vectorJson = JSON.stringify(Array.from(embedding));
47
+ insertStmt.run(entityName, vectorJson);
48
+ }
49
+ deleteEmbedding(db, entityName) {
50
+ this.ensureExtensionLoaded(db);
51
+ const stmt = db.prepare('DELETE FROM entity_embeddings WHERE entity_name = ?');
52
+ stmt.run(entityName);
53
+ }
54
+ knnSearch(db, queryEmbedding, k, dimensions = DEFAULT_EMBEDDING_DIMENSIONS) {
55
+ if (queryEmbedding.length !== dimensions) {
56
+ throw new Error(`Invalid query embedding dimensions: expected ${dimensions}, got ${queryEmbedding.length}`);
57
+ }
58
+ this.ensureExtensionLoaded(db);
59
+ const queryJson = JSON.stringify(Array.from(queryEmbedding));
60
+ const stmt = db.prepare(`
61
+ SELECT
62
+ entity_name,
63
+ distance
64
+ FROM entity_embeddings
65
+ WHERE embedding MATCH ?
66
+ AND k = ?
67
+ `);
68
+ const rows = stmt.all(queryJson, k);
69
+ return rows.map(row => ({
70
+ entityName: row.entity_name,
71
+ distance: row.distance
72
+ }));
73
+ }
74
+ getEmbedding(db, entityName) {
75
+ this.ensureExtensionLoaded(db);
76
+ const stmt = db.prepare('SELECT embedding FROM entity_embeddings WHERE entity_name = ?');
77
+ const row = stmt.get(entityName);
78
+ if (!row) {
79
+ return null;
80
+ }
81
+ if (Buffer.isBuffer(row.embedding)) {
82
+ return bufferToFloat32Array(row.embedding);
83
+ }
84
+ if (typeof row.embedding === 'string') {
85
+ return new Float32Array(JSON.parse(row.embedding));
86
+ }
87
+ logger.warn('Unexpected embedding format for entity', { entityName });
88
+ return null;
89
+ }
90
+ hasEmbedding(db, entityName) {
91
+ this.ensureExtensionLoaded(db);
92
+ const stmt = db.prepare('SELECT 1 FROM entity_embeddings WHERE entity_name = ? LIMIT 1');
93
+ const row = stmt.get(entityName);
94
+ return row !== undefined;
95
+ }
96
+ getEmbeddingCount(db) {
97
+ const stmt = db.prepare('SELECT COUNT(*) as cnt FROM entity_embeddings');
98
+ const row = stmt.get();
99
+ return row.cnt;
100
+ }
101
+ ensureExtensionLoaded(db) {
102
+ if (!this.extensionLoaded.has(db)) {
103
+ this.loadExtension(db);
104
+ }
105
+ }
106
+ }
107
+ //# sourceMappingURL=SqliteVecAdapter.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"SqliteVecAdapter.js","sourceRoot":"","sources":["../../src/embeddings/SqliteVecAdapter.ts"],"names":[],"mappings":"AASA,OAAO,KAAK,SAAS,MAAM,YAAY,CAAC;AACxC,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAE5C,OAAO,EAAE,4BAA4B,EAAE,MAAM,0BAA0B,CAAC;AAMxE,SAAS,oBAAoB,CAAC,MAAc;IAC1C,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,CACrC,MAAM,CAAC,UAAU,EACjB,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CACtC,CAAC;IACF,OAAO,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;AACvC,CAAC;AAQD,MAAM,OAAO,gBAAgB;IAEnB,eAAe,GAAG,IAAI,OAAO,EAAqB,CAAC;IAE3D,aAAa,CAAC,EAAqB;QACjC,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;YACjC,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACnB,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC7B,MAAM,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC3D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,qCAAqC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAC/D,MAAM,IAAI,KAAK,CACb,8BAA8B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,EAAE,CAC/E,CAAC;QACJ,CAAC;IACH,CAAC;IAED,iBAAiB,CAAC,EAAqB,EAAE,aAAqB,4BAA4B;QACxF,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAE/B,EAAE,CAAC,IAAI,CAAC;;;0BAGc,UAAU;;KAE/B,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,sBAAsB,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;IACvD,CAAC;IAED,eAAe,CACb,EAAqB,EACrB,UAAkB,EAClB,SAAuB,EACvB,aAAqB,4BAA4B;QAEjD,IAAI,SAAS,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;YACpC,MAAM,IAAI,KAAK,CACb,0CAA0C,UAAU,SAAS,SAAS,CAAC,MAAM,IAAI;gBACjF,WAAW,UAAU,EAAE,CACxB,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAG/B,MAAM,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC,qDAAqD,CAAC,CAAC;QACrF,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAE3B,MAAM,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC;;;KAG7B,CAAC,CAAC;QAGH,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;QACzD,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;IACzC,CAAC;IAED,eAAe,CAAC,EAAqB,EAAE,UAAkB;QACvD,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAC/B,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,qDAAqD,CAAC,CAAC;QAC/E,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACvB,CAAC;IAED,SAAS,CACP,EAAqB,EACrB,cAA4B,EAC5B,CAAS,EACT,aAAqB,4BAA4B;QAEjD,IAAI,cAAc,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CACb,gDAAgD,UAAU,SAAS,cAAc,CAAC,MAAM,EAAE,CAC3F,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAE/B,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC;QAE7D,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC;;;;;;;KAOvB,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,CAAqD,CAAC;QAExF,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACtB,UAAU,EAAE,GAAG,CAAC,WAAW;YAC3B,QAAQ,EAAE,GAAG,CAAC,QAAQ;SACvB,CAAC,CAAC,CAAC;IACN,CAAC;IAED,YAAY,CAAC,EAAqB,EAAE,UAAkB;QACpD,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAE/B,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,+DAA+D,CAAC,CAAC;QACzF,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAA+C,CAAC;QAE/E,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YACnC,OAAO,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,CAAC;QAED,IAAI,OAAO,GAAG,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;YACtC,OAAO,IAAI,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,wCAAwC,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;QACtE,OAAO,IAAI,CAAC;IACd,CAAC;IAED,YAAY,CAAC,EAAqB,EAAE,UAAkB;QACpD,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAC/B,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,+DAA+D,CAAC,CAAC;QACzF,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACjC,OAAO,GAAG,KAAK,SAAS,CAAC;IAC3B,CAAC;IAED,iBAAiB,CAAC,EAAqB;QACrC,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,+CAA+C,CAAC,CAAC;QACzE,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAqB,CAAC;QAC1C,OAAO,GAAG,CAAC,GAAG,CAAC;IACjB,CAAC;IAKO,qBAAqB,CAAC,EAAqB;QACjD,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;YAClC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;CACF"}
@@ -1,8 +1,6 @@
1
1
  import Database from 'better-sqlite3';
2
- export interface KnnSearchResult {
3
- entityName: string;
4
- distance: number;
5
- }
2
+ import type { KnnSearchResult } from './VectorSearchAdapter.js';
3
+ export type { KnnSearchResult } from './VectorSearchAdapter.js';
6
4
  export declare const DEFAULT_EMBEDDING_DIMENSIONS = 384;
7
5
  export declare class VectorExtension {
8
6
  private static extensionLoaded;
@@ -1 +1 @@
1
- {"version":3,"file":"VectorExtension.d.ts","sourceRoot":"","sources":["../../src/embeddings/VectorExtension.ts"],"names":[],"mappings":"AA6BA,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AAOtC,MAAM,WAAW,eAAe;IAE9B,UAAU,EAAE,MAAM,CAAC;IAEnB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAKD,eAAO,MAAM,4BAA4B,MAAM,CAAC;AAoBhD,qBAAa,eAAe;IAE1B,OAAO,CAAC,MAAM,CAAC,eAAe,CAAoC;IAWlE,MAAM,CAAC,aAAa,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,GAAG,IAAI;IAwBjD,MAAM,CAAC,iBAAiB,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,GAAE,MAAY,GAAG,IAAI;IA2B/E,MAAM,CAAC,eAAe,CACpB,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,YAAY,EACvB,kBAAkB,GAAE,MAAqC,GACxD,IAAI;IAiCP,MAAM,CAAC,eAAe,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,GAAG,IAAI;IAmBvE,MAAM,CAAC,SAAS,CACd,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,cAAc,EAAE,YAAY,EAC5B,CAAC,EAAE,MAAM,EACT,kBAAkB,GAAE,MAAqC,GACxD,eAAe,EAAE;IAuCpB,MAAM,CAAC,YAAY,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,GAAG,YAAY,GAAG,IAAI;IA8BnF,MAAM,CAAC,YAAY,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO;IAavE,MAAM,CAAC,iBAAiB,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,GAAG,MAAM;IAYvD,OAAO,CAAC,MAAM,CAAC,qBAAqB;CAKrC"}
1
+ {"version":3,"file":"VectorExtension.d.ts","sourceRoot":"","sources":["../../src/embeddings/VectorExtension.ts"],"names":[],"mappings":"AA6BA,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AAKtC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAChE,YAAY,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAKhE,eAAO,MAAM,4BAA4B,MAAM,CAAC;AAoBhD,qBAAa,eAAe;IAE1B,OAAO,CAAC,MAAM,CAAC,eAAe,CAAoC;IAWlE,MAAM,CAAC,aAAa,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,GAAG,IAAI;IAwBjD,MAAM,CAAC,iBAAiB,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,GAAE,MAAY,GAAG,IAAI;IA2B/E,MAAM,CAAC,eAAe,CACpB,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,YAAY,EACvB,kBAAkB,GAAE,MAAqC,GACxD,IAAI;IAiCP,MAAM,CAAC,eAAe,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,GAAG,IAAI;IAmBvE,MAAM,CAAC,SAAS,CACd,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,cAAc,EAAE,YAAY,EAC5B,CAAC,EAAE,MAAM,EACT,kBAAkB,GAAE,MAAqC,GACxD,eAAe,EAAE;IAuCpB,MAAM,CAAC,YAAY,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,GAAG,YAAY,GAAG,IAAI;IA8BnF,MAAM,CAAC,YAAY,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO;IAavE,MAAM,CAAC,iBAAiB,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,GAAG,MAAM;IAYvD,OAAO,CAAC,MAAM,CAAC,qBAAqB;CAKrC"}
@@ -1 +1 @@
1
- {"version":3,"file":"VectorExtension.js","sourceRoot":"","sources":["../../src/embeddings/VectorExtension.ts"],"names":[],"mappings":"AA8BA,OAAO,KAAK,SAAS,MAAM,YAAY,CAAC;AACxC,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAe5C,MAAM,CAAC,MAAM,4BAA4B,GAAG,GAAG,CAAC;AAMhD,SAAS,oBAAoB,CAAC,MAAc;IAC1C,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,CACrC,MAAM,CAAC,UAAU,EACjB,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CACtC,CAAC;IACF,OAAO,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;AACvC,CAAC;AAQD,MAAM,OAAO,eAAe;IAElB,MAAM,CAAC,eAAe,GAAG,IAAI,OAAO,EAAqB,CAAC;IAWlE,MAAM,CAAC,aAAa,CAAC,EAAqB;QACxC,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;YACjC,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACnB,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC7B,MAAM,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC3D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,qCAAqC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAC/D,MAAM,IAAI,KAAK,CAAC,8BAA8B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;QAClG,CAAC;IACH,CAAC;IAWD,MAAM,CAAC,iBAAiB,CAAC,EAAqB,EAAE,aAAqB,GAAG;QACtE,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAI/B,EAAE,CAAC,IAAI,CAAC;;;0BAGc,UAAU;;KAE/B,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,sBAAsB,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;IACvD,CAAC;IAcD,MAAM,CAAC,eAAe,CACpB,EAAqB,EACrB,UAAkB,EAClB,SAAuB,EACvB,qBAA6B,4BAA4B;QAGzD,IAAI,SAAS,CAAC,MAAM,KAAK,kBAAkB,EAAE,CAAC;YAC5C,MAAM,IAAI,KAAK,CACb,0CAA0C,kBAAkB,SAAS,SAAS,CAAC,MAAM,IAAI;gBACzF,WAAW,UAAU,EAAE,CACxB,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAI/B,MAAM,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC,qDAAqD,CAAC,CAAC;QACrF,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAE3B,MAAM,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC;;;KAG7B,CAAC,CAAC;QAIH,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;QACzD,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;IACzC,CAAC;IAQD,MAAM,CAAC,eAAe,CAAC,EAAqB,EAAE,UAAkB;QAC9D,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAC/B,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,qDAAqD,CAAC,CAAC;QAC/E,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACvB,CAAC;IAeD,MAAM,CAAC,SAAS,CACd,EAAqB,EACrB,cAA4B,EAC5B,CAAS,EACT,qBAA6B,4BAA4B;QAGzD,IAAI,cAAc,CAAC,MAAM,KAAK,kBAAkB,EAAE,CAAC;YACjD,MAAM,IAAI,KAAK,CACb,gDAAgD,kBAAkB,SAAS,cAAc,CAAC,MAAM,EAAE,CACnG,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAG/B,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC;QAI7D,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC;;;;;;;KAOvB,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,CAAqD,CAAC;QAExF,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACtB,UAAU,EAAE,GAAG,CAAC,WAAW;YAC3B,QAAQ,EAAE,GAAG,CAAC,QAAQ;SACvB,CAAC,CAAC,CAAC;IACN,CAAC;IASD,MAAM,CAAC,YAAY,CAAC,EAAqB,EAAE,UAAkB;QAC3D,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAE/B,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,+DAA+D,CAAC,CAAC;QACzF,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAA+C,CAAC;QAE/E,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,OAAO,IAAI,CAAC;QACd,CAAC;QAGD,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YACnC,OAAO,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,CAAC;QAED,IAAI,OAAO,GAAG,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;YACtC,OAAO,IAAI,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,wCAAwC,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;QACtE,OAAO,IAAI,CAAC;IACd,CAAC;IASD,MAAM,CAAC,YAAY,CAAC,EAAqB,EAAE,UAAkB;QAC3D,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAC/B,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,+DAA+D,CAAC,CAAC;QACzF,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACjC,OAAO,GAAG,KAAK,SAAS,CAAC;IAC3B,CAAC;IAQD,MAAM,CAAC,iBAAiB,CAAC,EAAqB;QAC5C,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,+CAA+C,CAAC,CAAC;QACzE,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAqB,CAAC;QAC1C,OAAO,GAAG,CAAC,GAAG,CAAC;IACjB,CAAC;IAQO,MAAM,CAAC,qBAAqB,CAAC,EAAqB;QACxD,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;YAClC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;QACzB,CAAC;IACH,CAAC"}
1
+ {"version":3,"file":"VectorExtension.js","sourceRoot":"","sources":["../../src/embeddings/VectorExtension.ts"],"names":[],"mappings":"AA8BA,OAAO,KAAK,SAAS,MAAM,YAAY,CAAC;AACxC,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAS5C,MAAM,CAAC,MAAM,4BAA4B,GAAG,GAAG,CAAC;AAMhD,SAAS,oBAAoB,CAAC,MAAc;IAC1C,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,CACrC,MAAM,CAAC,UAAU,EACjB,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CACtC,CAAC;IACF,OAAO,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC;AACvC,CAAC;AAQD,MAAM,OAAO,eAAe;IAElB,MAAM,CAAC,eAAe,GAAG,IAAI,OAAO,EAAqB,CAAC;IAWlE,MAAM,CAAC,aAAa,CAAC,EAAqB;QACxC,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;YACjC,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACnB,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC7B,MAAM,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC3D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,qCAAqC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAC/D,MAAM,IAAI,KAAK,CAAC,8BAA8B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;QAClG,CAAC;IACH,CAAC;IAWD,MAAM,CAAC,iBAAiB,CAAC,EAAqB,EAAE,aAAqB,GAAG;QACtE,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAI/B,EAAE,CAAC,IAAI,CAAC;;;0BAGc,UAAU;;KAE/B,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,sBAAsB,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;IACvD,CAAC;IAcD,MAAM,CAAC,eAAe,CACpB,EAAqB,EACrB,UAAkB,EAClB,SAAuB,EACvB,qBAA6B,4BAA4B;QAGzD,IAAI,SAAS,CAAC,MAAM,KAAK,kBAAkB,EAAE,CAAC;YAC5C,MAAM,IAAI,KAAK,CACb,0CAA0C,kBAAkB,SAAS,SAAS,CAAC,MAAM,IAAI;gBACzF,WAAW,UAAU,EAAE,CACxB,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAI/B,MAAM,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC,qDAAqD,CAAC,CAAC;QACrF,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAE3B,MAAM,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC;;;KAG7B,CAAC,CAAC;QAIH,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;QACzD,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;IACzC,CAAC;IAQD,MAAM,CAAC,eAAe,CAAC,EAAqB,EAAE,UAAkB;QAC9D,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAC/B,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,qDAAqD,CAAC,CAAC;QAC/E,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACvB,CAAC;IAeD,MAAM,CAAC,SAAS,CACd,EAAqB,EACrB,cAA4B,EAC5B,CAAS,EACT,qBAA6B,4BAA4B;QAGzD,IAAI,cAAc,CAAC,MAAM,KAAK,kBAAkB,EAAE,CAAC;YACjD,MAAM,IAAI,KAAK,CACb,gDAAgD,kBAAkB,SAAS,cAAc,CAAC,MAAM,EAAE,CACnG,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAG/B,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC;QAI7D,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC;;;;;;;KAOvB,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,CAAqD,CAAC;QAExF,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACtB,UAAU,EAAE,GAAG,CAAC,WAAW;YAC3B,QAAQ,EAAE,GAAG,CAAC,QAAQ;SACvB,CAAC,CAAC,CAAC;IACN,CAAC;IASD,MAAM,CAAC,YAAY,CAAC,EAAqB,EAAE,UAAkB;QAC3D,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAE/B,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,+DAA+D,CAAC,CAAC;QACzF,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAA+C,CAAC;QAE/E,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,OAAO,IAAI,CAAC;QACd,CAAC;QAGD,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YACnC,OAAO,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,CAAC;QAED,IAAI,OAAO,GAAG,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;YACtC,OAAO,IAAI,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,wCAAwC,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;QACtE,OAAO,IAAI,CAAC;IACd,CAAC;IASD,MAAM,CAAC,YAAY,CAAC,EAAqB,EAAE,UAAkB;QAC3D,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAC/B,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,+DAA+D,CAAC,CAAC;QACzF,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACjC,OAAO,GAAG,KAAK,SAAS,CAAC;IAC3B,CAAC;IAQD,MAAM,CAAC,iBAAiB,CAAC,EAAqB;QAC5C,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,+CAA+C,CAAC,CAAC;QACzE,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAqB,CAAC;QAC1C,OAAO,GAAG,CAAC,GAAG,CAAC;IACjB,CAAC;IAQO,MAAM,CAAC,qBAAqB,CAAC,EAAqB;QACxD,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;YAClC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;QACzB,CAAC;IACH,CAAC"}
@@ -0,0 +1,17 @@
1
+ import type Database from 'better-sqlite3';
2
+ export interface KnnSearchResult {
3
+ entityName: string;
4
+ distance: number;
5
+ }
6
+ export declare const DEFAULT_EMBEDDING_DIMENSIONS = 384;
7
+ export interface VectorSearchAdapter {
8
+ loadExtension(db: Database.Database): void;
9
+ createVectorTable(db: Database.Database, dimensions?: number): void;
10
+ insertEmbedding(db: Database.Database, entityName: string, embedding: Float32Array, dimensions?: number): void;
11
+ deleteEmbedding(db: Database.Database, entityName: string): void;
12
+ knnSearch(db: Database.Database, queryEmbedding: Float32Array, k: number, dimensions?: number): KnnSearchResult[];
13
+ getEmbedding(db: Database.Database, entityName: string): Float32Array | null;
14
+ hasEmbedding(db: Database.Database, entityName: string): boolean;
15
+ getEmbeddingCount(db: Database.Database): number;
16
+ }
17
+ //# sourceMappingURL=VectorSearchAdapter.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"VectorSearchAdapter.d.ts","sourceRoot":"","sources":["../../src/embeddings/VectorSearchAdapter.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,QAAQ,MAAM,gBAAgB,CAAC;AAK3C,MAAM,WAAW,eAAe;IAE9B,UAAU,EAAE,MAAM,CAAC;IAEnB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAKD,eAAO,MAAM,4BAA4B,MAAM,CAAC;AAWhD,MAAM,WAAW,mBAAmB;IAElC,aAAa,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,GAAG,IAAI,CAAC;IAG3C,iBAAiB,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAGpE,eAAe,CACb,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,YAAY,EACvB,UAAU,CAAC,EAAE,MAAM,GAClB,IAAI,CAAC;IAGR,eAAe,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAGjE,SAAS,CACP,EAAE,EAAE,QAAQ,CAAC,QAAQ,EACrB,cAAc,EAAE,YAAY,EAC5B,CAAC,EAAE,MAAM,EACT,UAAU,CAAC,EAAE,MAAM,GAClB,eAAe,EAAE,CAAC;IAGrB,YAAY,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,GAAG,YAAY,GAAG,IAAI,CAAC;IAG7E,YAAY,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC;IAGjE,iBAAiB,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,GAAG,MAAM,CAAC;CAClD"}
@@ -0,0 +1,2 @@
1
+ export const DEFAULT_EMBEDDING_DIMENSIONS = 384;
2
+ //# sourceMappingURL=VectorSearchAdapter.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"VectorSearchAdapter.js","sourceRoot":"","sources":["../../src/embeddings/VectorSearchAdapter.ts"],"names":[],"mappings":"AAsBA,MAAM,CAAC,MAAM,4BAA4B,GAAG,GAAG,CAAC"}
@@ -1,4 +1,9 @@
1
1
  export { ModelManager } from './ModelManager.js';
2
2
  export { EmbeddingService, LazyEmbeddingService } from './EmbeddingService.js';
3
- export { VectorExtension, type KnnSearchResult, DEFAULT_EMBEDDING_DIMENSIONS, } from './VectorExtension.js';
3
+ export type { VectorSearchAdapter } from './VectorSearchAdapter.js';
4
+ export { DEFAULT_EMBEDDING_DIMENSIONS } from './VectorSearchAdapter.js';
5
+ export type { KnnSearchResult } from './VectorSearchAdapter.js';
6
+ export { SqliteVecAdapter } from './SqliteVecAdapter.js';
7
+ export { InMemoryVectorAdapter } from './InMemoryVectorAdapter.js';
8
+ export { VectorExtension } from './VectorExtension.js';
4
9
  //# sourceMappingURL=index.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/embeddings/index.ts"],"names":[],"mappings":"AAqCA,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC/E,OAAO,EACL,eAAe,EACf,KAAK,eAAe,EACpB,4BAA4B,GAC7B,MAAM,sBAAsB,CAAC"}
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/embeddings/index.ts"],"names":[],"mappings":"AAeA,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAG/E,YAAY,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AACpE,OAAO,EAAE,4BAA4B,EAAE,MAAM,0BAA0B,CAAC;AACxE,YAAY,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAChE,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAGnE,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC"}
@@ -1,4 +1,7 @@
1
1
  export { ModelManager } from './ModelManager.js';
2
2
  export { EmbeddingService, LazyEmbeddingService } from './EmbeddingService.js';
3
- export { VectorExtension, DEFAULT_EMBEDDING_DIMENSIONS, } from './VectorExtension.js';
3
+ export { DEFAULT_EMBEDDING_DIMENSIONS } from './VectorSearchAdapter.js';
4
+ export { SqliteVecAdapter } from './SqliteVecAdapter.js';
5
+ export { InMemoryVectorAdapter } from './InMemoryVectorAdapter.js';
6
+ export { VectorExtension } from './VectorExtension.js';
4
7
  //# sourceMappingURL=index.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/embeddings/index.ts"],"names":[],"mappings":"AAqCA,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC/E,OAAO,EACL,eAAe,EAEf,4BAA4B,GAC7B,MAAM,sBAAsB,CAAC"}
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/embeddings/index.ts"],"names":[],"mappings":"AAeA,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAI/E,OAAO,EAAE,4BAA4B,EAAE,MAAM,0BAA0B,CAAC;AAExE,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAGnE,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC"}
@@ -0,0 +1,4 @@
1
+ export declare class ContentHasher {
2
+ static hashEmbeddingSource(entityName: string, observations?: string[]): string;
3
+ }
4
+ //# sourceMappingURL=ContentHasher.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ContentHasher.d.ts","sourceRoot":"","sources":["../../src/knowledge-graph/ContentHasher.ts"],"names":[],"mappings":"AAOA,qBAAa,aAAa;IASxB,MAAM,CAAC,mBAAmB,CAAC,UAAU,EAAE,MAAM,EAAE,YAAY,GAAE,MAAM,EAAO,GAAG,MAAM;CAIpF"}
@@ -0,0 +1,8 @@
1
+ import { createHash } from 'crypto';
2
+ export class ContentHasher {
3
+ static hashEmbeddingSource(entityName, observations = []) {
4
+ const text = [entityName, ...observations].join(' ');
5
+ return createHash('sha256').update(text).digest('hex').substring(0, 16);
6
+ }
7
+ }
8
+ //# sourceMappingURL=ContentHasher.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ContentHasher.js","sourceRoot":"","sources":["../../src/knowledge-graph/ContentHasher.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAOpC,MAAM,OAAO,aAAa;IASxB,MAAM,CAAC,mBAAmB,CAAC,UAAkB,EAAE,eAAyB,EAAE;QACxE,MAAM,IAAI,GAAG,CAAC,UAAU,EAAE,GAAG,YAAY,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrD,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAC1E,CAAC;CACF"}
@@ -0,0 +1,37 @@
1
+ import type Database from 'better-sqlite3';
2
+ import type { Entity, SearchQuery } from './types.js';
3
+ import type { VectorSearchAdapter } from '../embeddings/VectorSearchAdapter.js';
4
+ import { QueryCache } from '../db/QueryCache.js';
5
+ export interface SemanticSearchResult {
6
+ entity: Entity;
7
+ similarity: number;
8
+ }
9
+ export declare const CONTROL_CHAR_PATTERN: RegExp;
10
+ export interface KGSearchEngineContext {
11
+ db: Database.Database;
12
+ getVectorAdapter: () => VectorSearchAdapter | null;
13
+ isVectorEnabled: () => boolean;
14
+ getVectorInitPromise: () => Promise<void> | null;
15
+ getEntity: (name: string) => Entity | null;
16
+ queryCache: QueryCache<string, any>;
17
+ }
18
+ export declare const MAX_FTS5_QUERY_LENGTH = 500;
19
+ export declare class KGSearchEngine {
20
+ private readonly ctx;
21
+ constructor(context: KGSearchEngineContext);
22
+ escapeLikePattern(pattern: string): string;
23
+ searchFTS5(query: string, limit: number): number[];
24
+ prepareFTS5Query(query: string): string;
25
+ searchEntities(query: SearchQuery): Entity[];
26
+ semanticSearch(query: string, options?: {
27
+ limit?: number;
28
+ minSimilarity?: number;
29
+ entityTypes?: string[];
30
+ }): Promise<SemanticSearchResult[]>;
31
+ hybridSearch(query: string, options?: {
32
+ limit?: number;
33
+ minSimilarity?: number;
34
+ }): Promise<SemanticSearchResult[]>;
35
+ keywordSearchAsSemanticResults(query: string, limit: number): SemanticSearchResult[];
36
+ }
37
+ //# sourceMappingURL=KGSearchEngine.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"KGSearchEngine.d.ts","sourceRoot":"","sources":["../../src/knowledge-graph/KGSearchEngine.ts"],"names":[],"mappings":"AAUA,OAAO,KAAK,QAAQ,MAAM,gBAAgB,CAAC;AAC3C,OAAO,KAAK,EAAE,MAAM,EAAE,WAAW,EAAc,MAAM,YAAY,CAAC;AAClE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAGhF,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAWjD,MAAM,WAAW,oBAAoB;IACnC,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;CACpB;AAMD,eAAO,MAAM,oBAAoB,QAA8C,CAAC;AAMhF,MAAM,WAAW,qBAAqB;IAEpC,EAAE,EAAE,QAAQ,CAAC,QAAQ,CAAC;IAEtB,gBAAgB,EAAE,MAAM,mBAAmB,GAAG,IAAI,CAAC;IAEnD,eAAe,EAAE,MAAM,OAAO,CAAC;IAE/B,oBAAoB,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAEjD,SAAS,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,MAAM,GAAG,IAAI,CAAC;IAE3C,UAAU,EAAE,UAAU,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CACrC;AAGD,eAAO,MAAM,qBAAqB,MAAM,CAAC;AAEzC,qBAAa,cAAc;IACzB,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAwB;gBAEhC,OAAO,EAAE,qBAAqB;IAY1C,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IAoB1C,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE;IAqClD,gBAAgB,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM;IA2DvC,cAAc,CAAC,KAAK,EAAE,WAAW,GAAG,MAAM,EAAE;IA+ItC,cAAc,CAClB,KAAK,EAAE,MAAM,EACb,OAAO,GAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,aAAa,CAAC,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,EAAE,CAAA;KAAO,GAC/E,OAAO,CAAC,oBAAoB,EAAE,CAAC;IAgD5B,YAAY,CAChB,KAAK,EAAE,MAAM,EACb,OAAO,GAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,aAAa,CAAC,EAAE,MAAM,CAAA;KAAO,GACvD,OAAO,CAAC,oBAAoB,EAAE,CAAC;IAwClC,8BAA8B,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,oBAAoB,EAAE;CAIrF"}