aimemory-core 1.0.1

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 (86) hide show
  1. package/.eslintrc.json +22 -0
  2. package/.github/workflows/ci.yml +57 -0
  3. package/.prettierrc +8 -0
  4. package/README.md +197 -0
  5. package/dist/config.d.ts +20 -0
  6. package/dist/config.d.ts.map +1 -0
  7. package/dist/config.js +41 -0
  8. package/dist/config.js.map +1 -0
  9. package/dist/contextBuilder.d.ts +16 -0
  10. package/dist/contextBuilder.d.ts.map +1 -0
  11. package/dist/contextBuilder.js +139 -0
  12. package/dist/contextBuilder.js.map +1 -0
  13. package/dist/index.d.ts +45 -0
  14. package/dist/index.d.ts.map +1 -0
  15. package/dist/index.js +88 -0
  16. package/dist/index.js.map +1 -0
  17. package/dist/licensing.d.ts +45 -0
  18. package/dist/licensing.d.ts.map +1 -0
  19. package/dist/licensing.js +99 -0
  20. package/dist/licensing.js.map +1 -0
  21. package/dist/memoryManager.d.ts +35 -0
  22. package/dist/memoryManager.d.ts.map +1 -0
  23. package/dist/memoryManager.js +265 -0
  24. package/dist/memoryManager.js.map +1 -0
  25. package/dist/metadataStore.d.ts +24 -0
  26. package/dist/metadataStore.d.ts.map +1 -0
  27. package/dist/metadataStore.js +247 -0
  28. package/dist/metadataStore.js.map +1 -0
  29. package/dist/planManager.d.ts +80 -0
  30. package/dist/planManager.d.ts.map +1 -0
  31. package/dist/planManager.js +327 -0
  32. package/dist/planManager.js.map +1 -0
  33. package/dist/rateLimiter.d.ts +49 -0
  34. package/dist/rateLimiter.d.ts.map +1 -0
  35. package/dist/rateLimiter.js +142 -0
  36. package/dist/rateLimiter.js.map +1 -0
  37. package/dist/storage/index.d.ts +3 -0
  38. package/dist/storage/index.d.ts.map +1 -0
  39. package/dist/storage/index.js +3 -0
  40. package/dist/storage/index.js.map +1 -0
  41. package/dist/storage/postgres.d.ts +31 -0
  42. package/dist/storage/postgres.d.ts.map +1 -0
  43. package/dist/storage/postgres.js +171 -0
  44. package/dist/storage/postgres.js.map +1 -0
  45. package/dist/storage/redis.d.ts +34 -0
  46. package/dist/storage/redis.d.ts.map +1 -0
  47. package/dist/storage/redis.js +101 -0
  48. package/dist/storage/redis.js.map +1 -0
  49. package/dist/types.d.ts +95 -0
  50. package/dist/types.d.ts.map +1 -0
  51. package/dist/types.js +2 -0
  52. package/dist/types.js.map +1 -0
  53. package/dist/usageTracker.d.ts +63 -0
  54. package/dist/usageTracker.d.ts.map +1 -0
  55. package/dist/usageTracker.js +238 -0
  56. package/dist/usageTracker.js.map +1 -0
  57. package/dist/vectorStore.d.ts +18 -0
  58. package/dist/vectorStore.d.ts.map +1 -0
  59. package/dist/vectorStore.js +97 -0
  60. package/dist/vectorStore.js.map +1 -0
  61. package/examples/advanced.ts +164 -0
  62. package/examples/basic.ts +87 -0
  63. package/package.json +60 -0
  64. package/src/config.ts +65 -0
  65. package/src/contextBuilder.ts +184 -0
  66. package/src/index.ts +209 -0
  67. package/src/licensing.ts +138 -0
  68. package/src/memoryManager.ts +340 -0
  69. package/src/metadataStore.ts +298 -0
  70. package/src/planManager.ts +417 -0
  71. package/src/rateLimiter.ts +186 -0
  72. package/src/storage/index.ts +2 -0
  73. package/src/storage/postgres.ts +209 -0
  74. package/src/storage/redis.ts +117 -0
  75. package/src/types.ts +114 -0
  76. package/src/usageTracker.ts +325 -0
  77. package/src/vectorStore.ts +116 -0
  78. package/tests/aibrain.test.ts +171 -0
  79. package/tests/contextBuilder.test.ts +138 -0
  80. package/tests/memoryManager.test.ts +205 -0
  81. package/tests/metadataStore.test.ts +131 -0
  82. package/tests/rateLimiter.test.ts +57 -0
  83. package/tests/usageTracker.test.ts +62 -0
  84. package/tests/vectorStore.test.ts +106 -0
  85. package/tsconfig.json +25 -0
  86. package/vitest.config.ts +12 -0
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAkB,cAAc,EAAE,MAAM,UAAU,CAAC;AAC1D,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAoBlD,OAAO,EAEL,cAAc,EACd,cAAc,GACf,MAAM,UAAU,CAAC;AAElB,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,eAAe,CAAC;AAEvB,OAAO,EACL,aAAa,GACd,MAAM,iBAAiB,CAAC;AAEzB,OAAO,EACL,aAAa,EACb,mBAAmB,GACpB,MAAM,iBAAiB,CAAC;AAEzB,OAAO,EACL,cAAc,EACd,oBAAoB,GACrB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EACL,WAAW,EAIX,sBAAsB,EACtB,iBAAiB,EACjB,mBAAmB,GACpB,MAAM,eAAe,CAAC;AAEvB,OAAO,EACL,YAAY,EAMZ,kBAAkB,EAClB,kBAAkB,EAClB,oBAAoB,GACrB,MAAM,gBAAgB,CAAC;AAExB,OAAO,EACL,WAAW,EACX,WAAW,IAAI,WAAW,EAS1B,KAAK,EACL,OAAO,EACP,aAAa,EACb,iBAAiB,EACjB,iBAAiB,GAClB,MAAM,eAAe,CAAC;AAEvB,OAAO,EACL,mBAAmB,EACnB,kBAAkB,EAGlB,sBAAsB,EACtB,wBAAwB,GACzB,MAAM,WAAW,CAAC;AAEnB,OAAO,EACL,cAAc,EAId,oBAAoB,EACpB,iBAAiB,GAClB,MAAM,aAAa,CAAC;AAErB,MAAM,OAAO,QAAQ;IACX,aAAa,CAAgB;IAC7B,cAAc,CAAiB;IAEvC,YAAY,MAAgC;QAC1C,MAAM,eAAe,GAAG,cAAc,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QACrD,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,CAAC,eAAe,CAAC,CAAC;QACxD,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,CACtC,IAAI,CAAC,aAAa,EAClB,eAAe,CAAC,OAAO,CACxB,CAAC;IACJ,CAAC;IAED,oBAAoB,CAAC,EAAqB;QACxC,IAAI,CAAC,aAAa,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAe,EAAE,QAAkC;QAChE,OAAO,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IACzD,CAAC;IAED,KAAK,CAAC,oBAAoB,CACxB,OAAe,EACf,IAA0B,EAC1B,QAAkC;QAElC,OAAO,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,OAAO,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;IACrE,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,OAAe,EAAE,QAAkC;QACpE,OAAO,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,OAAe,EAAE,QAAkC;QAC1E,OAAO,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IAC7D,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,OAAe,EAAE,QAAkC;QAC3E,OAAO,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,OAAgD;QAC1E,OAAO,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC;YAC/B,IAAI,EAAE,KAAK;YACX,KAAK,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE;YAC3B,SAAS,EAAE,OAAO,EAAE,SAAS;SAC9B,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAa;QAC5B,OAAO,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,EAAU;QACxB,OAAO,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,EAAU,EAAE,OAAwB;QACrD,OAAO,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,OAAO,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,KAAc;QACpC,OAAO,IAAI,CAAC,aAAa,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,OAAO,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,KAAK;QACT,OAAO,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,OAAO;QACX,OAAO,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,UAAU;QACd,OAAO,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,QAAkB;QACjC,OAAO,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;IACjD,CAAC;IAED,gBAAgB;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,iBAAiB;QACf,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC;CACF;AAED,MAAM,UAAU,cAAc,CAAC,MAAgC;IAC7D,OAAO,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;AAC9B,CAAC;AAED,OAAO,EAAE,cAAc,IAAI,aAAa,EAAE,CAAC"}
@@ -0,0 +1,45 @@
1
+ import { ProductPlan, PlanConfig } from './planManager.js';
2
+ export interface LicenseConfig {
3
+ apiKey?: string;
4
+ userId?: string;
5
+ environment: 'development' | 'production';
6
+ }
7
+ export interface LicenseValidation {
8
+ valid: boolean;
9
+ plan: ProductPlan;
10
+ features: PlanConfig['features'];
11
+ error?: string;
12
+ }
13
+ export interface AIMemoryLicense {
14
+ validate(config: LicenseConfig): Promise<LicenseValidation>;
15
+ getPlan(userId: string): ProductPlan;
16
+ setPlan(userId: string, plan: ProductPlan): void;
17
+ requiresApiKey(): boolean;
18
+ isDevelopmentMode(): boolean;
19
+ }
20
+ export declare class LicenseManager implements AIMemoryLicense {
21
+ private userPlans;
22
+ private developmentMode;
23
+ private apiKeys;
24
+ constructor(options?: {
25
+ developmentMode?: boolean;
26
+ apiKeys?: Map<string, {
27
+ userId: string;
28
+ plan: ProductPlan;
29
+ }>;
30
+ });
31
+ validate(config: LicenseConfig): Promise<LicenseValidation>;
32
+ getPlan(userId: string): ProductPlan;
33
+ setPlan(userId: string, plan: ProductPlan): void;
34
+ registerApiKey(apiKey: string, userId: string, plan: ProductPlan): void;
35
+ revokeApiKey(apiKey: string): boolean;
36
+ requiresApiKey(): boolean;
37
+ isDevelopmentMode(): boolean;
38
+ enableProductionMode(): void;
39
+ disableProductionMode(): void;
40
+ }
41
+ export declare function createLicenseManager(options?: {
42
+ developmentMode?: boolean;
43
+ }): LicenseManager;
44
+ export declare function getLicenseManager(): LicenseManager | null;
45
+ //# sourceMappingURL=licensing.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"licensing.d.ts","sourceRoot":"","sources":["../src/licensing.ts"],"names":[],"mappings":"AAAA,OAAO,EAAS,WAAW,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAElE,MAAM,WAAW,aAAa;IAC5B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,aAAa,GAAG,YAAY,CAAC;CAC3C;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE,OAAO,CAAC;IACf,IAAI,EAAE,WAAW,CAAC;IAClB,QAAQ,EAAE,UAAU,CAAC,UAAU,CAAC,CAAC;IACjC,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,MAAM,EAAE,aAAa,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAC5D,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,WAAW,CAAC;IACrC,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,GAAG,IAAI,CAAC;IACjD,cAAc,IAAI,OAAO,CAAC;IAC1B,iBAAiB,IAAI,OAAO,CAAC;CAC9B;AAED,qBAAa,cAAe,YAAW,eAAe;IACpD,OAAO,CAAC,SAAS,CAAuC;IACxD,OAAO,CAAC,eAAe,CAAU;IACjC,OAAO,CAAC,OAAO,CAAiE;gBAEpE,OAAO,CAAC,EAAE;QAAE,eAAe,CAAC,EAAE,OAAO,CAAC;QAAC,OAAO,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE;YAAE,MAAM,EAAE,MAAM,CAAC;YAAC,IAAI,EAAE,WAAW,CAAA;SAAE,CAAC,CAAA;KAAE;IAO3G,QAAQ,CAAC,MAAM,EAAE,aAAa,GAAG,OAAO,CAAC,iBAAiB,CAAC;IA4DjE,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,WAAW;IAIpC,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,GAAG,IAAI;IAIhD,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,GAAG,IAAI;IAIvE,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO;IAIrC,cAAc,IAAI,OAAO;IAIzB,iBAAiB,IAAI,OAAO;IAI5B,oBAAoB,IAAI,IAAI;IAI5B,qBAAqB,IAAI,IAAI;CAG9B;AAID,wBAAgB,oBAAoB,CAAC,OAAO,CAAC,EAAE;IAAE,eAAe,CAAC,EAAE,OAAO,CAAA;CAAE,GAAG,cAAc,CAG5F;AAED,wBAAgB,iBAAiB,IAAI,cAAc,GAAG,IAAI,CAEzD"}
@@ -0,0 +1,99 @@
1
+ import { PLANS } from './planManager.js';
2
+ export class LicenseManager {
3
+ userPlans = new Map();
4
+ developmentMode;
5
+ apiKeys = new Map();
6
+ constructor(options) {
7
+ this.developmentMode = options?.developmentMode ?? true;
8
+ if (options?.apiKeys) {
9
+ this.apiKeys = options.apiKeys;
10
+ }
11
+ }
12
+ async validate(config) {
13
+ if (config.environment === 'development' && this.developmentMode) {
14
+ return {
15
+ valid: true,
16
+ plan: 'free',
17
+ features: PLANS.free.features,
18
+ };
19
+ }
20
+ if (config.environment === 'production') {
21
+ if (!config.apiKey && !this.developmentMode) {
22
+ return {
23
+ valid: false,
24
+ plan: 'free',
25
+ features: PLANS.free.features,
26
+ error: 'API key required for production use. Get one at https://aimemory.dev/dashboard',
27
+ };
28
+ }
29
+ if (config.apiKey) {
30
+ const keyData = this.apiKeys.get(config.apiKey);
31
+ if (!keyData) {
32
+ return {
33
+ valid: false,
34
+ plan: 'free',
35
+ features: PLANS.free.features,
36
+ error: 'Invalid API key',
37
+ };
38
+ }
39
+ return {
40
+ valid: true,
41
+ plan: keyData.plan,
42
+ features: PLANS[keyData.plan].features,
43
+ };
44
+ }
45
+ if (!config.userId) {
46
+ return {
47
+ valid: false,
48
+ plan: 'free',
49
+ features: PLANS.free.features,
50
+ error: 'User ID required',
51
+ };
52
+ }
53
+ const userPlan = this.getPlan(config.userId);
54
+ return {
55
+ valid: true,
56
+ plan: userPlan,
57
+ features: PLANS[userPlan].features,
58
+ };
59
+ }
60
+ return {
61
+ valid: true,
62
+ plan: 'free',
63
+ features: PLANS.free.features,
64
+ };
65
+ }
66
+ getPlan(userId) {
67
+ return this.userPlans.get(userId) || 'free';
68
+ }
69
+ setPlan(userId, plan) {
70
+ this.userPlans.set(userId, plan);
71
+ }
72
+ registerApiKey(apiKey, userId, plan) {
73
+ this.apiKeys.set(apiKey, { userId, plan });
74
+ }
75
+ revokeApiKey(apiKey) {
76
+ return this.apiKeys.delete(apiKey);
77
+ }
78
+ requiresApiKey() {
79
+ return !this.developmentMode;
80
+ }
81
+ isDevelopmentMode() {
82
+ return this.developmentMode;
83
+ }
84
+ enableProductionMode() {
85
+ this.developmentMode = false;
86
+ }
87
+ disableProductionMode() {
88
+ this.developmentMode = true;
89
+ }
90
+ }
91
+ let globalLicenseManager = null;
92
+ export function createLicenseManager(options) {
93
+ globalLicenseManager = new LicenseManager(options);
94
+ return globalLicenseManager;
95
+ }
96
+ export function getLicenseManager() {
97
+ return globalLicenseManager;
98
+ }
99
+ //# sourceMappingURL=licensing.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"licensing.js","sourceRoot":"","sources":["../src/licensing.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAA2B,MAAM,kBAAkB,CAAC;AAuBlE,MAAM,OAAO,cAAc;IACjB,SAAS,GAA6B,IAAI,GAAG,EAAE,CAAC;IAChD,eAAe,CAAU;IACzB,OAAO,GAAuD,IAAI,GAAG,EAAE,CAAC;IAEhF,YAAY,OAAqG;QAC/G,IAAI,CAAC,eAAe,GAAG,OAAO,EAAE,eAAe,IAAI,IAAI,CAAC;QACxD,IAAI,OAAO,EAAE,OAAO,EAAE,CAAC;YACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QACjC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,MAAqB;QAClC,IAAI,MAAM,CAAC,WAAW,KAAK,aAAa,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACjE,OAAO;gBACL,KAAK,EAAE,IAAI;gBACX,IAAI,EAAE,MAAM;gBACZ,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ;aAC9B,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,CAAC,WAAW,KAAK,YAAY,EAAE,CAAC;YACxC,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;gBAC5C,OAAO;oBACL,KAAK,EAAE,KAAK;oBACZ,IAAI,EAAE,MAAM;oBACZ,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ;oBAC7B,KAAK,EAAE,gFAAgF;iBACxF,CAAC;YACJ,CAAC;YAED,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBAClB,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBAChD,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,OAAO;wBACL,KAAK,EAAE,KAAK;wBACZ,IAAI,EAAE,MAAM;wBACZ,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ;wBAC7B,KAAK,EAAE,iBAAiB;qBACzB,CAAC;gBACJ,CAAC;gBACD,OAAO;oBACL,KAAK,EAAE,IAAI;oBACX,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ;iBACvC,CAAC;YACJ,CAAC;YAED,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBACnB,OAAO;oBACL,KAAK,EAAE,KAAK;oBACZ,IAAI,EAAE,MAAM;oBACZ,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ;oBAC7B,KAAK,EAAE,kBAAkB;iBAC1B,CAAC;YACJ,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC7C,OAAO;gBACL,KAAK,EAAE,IAAI;gBACX,IAAI,EAAE,QAAQ;gBACd,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,QAAQ;aACnC,CAAC;QACJ,CAAC;QAED,OAAO;YACL,KAAK,EAAE,IAAI;YACX,IAAI,EAAE,MAAM;YACZ,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ;SAC9B,CAAC;IACJ,CAAC;IAED,OAAO,CAAC,MAAc;QACpB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC;IAC9C,CAAC;IAED,OAAO,CAAC,MAAc,EAAE,IAAiB;QACvC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,cAAc,CAAC,MAAc,EAAE,MAAc,EAAE,IAAiB;QAC9D,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IAC7C,CAAC;IAED,YAAY,CAAC,MAAc;QACzB,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC;IAED,cAAc;QACZ,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC;IAC/B,CAAC;IAED,iBAAiB;QACf,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,oBAAoB;QAClB,IAAI,CAAC,eAAe,GAAG,KAAK,CAAC;IAC/B,CAAC;IAED,qBAAqB;QACnB,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;IAC9B,CAAC;CACF;AAED,IAAI,oBAAoB,GAA0B,IAAI,CAAC;AAEvD,MAAM,UAAU,oBAAoB,CAAC,OAAuC;IAC1E,oBAAoB,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,CAAC;IACnD,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,iBAAiB;IAC/B,OAAO,oBAAoB,CAAC;AAC9B,CAAC"}
@@ -0,0 +1,35 @@
1
+ import { Memory, MemoryMetadata, MemoryType, MemoryQuery, SearchResult, MemoryStats, EmbeddingFunction } from './types.js';
2
+ import { AIMemoryConfig } from './config.js';
3
+ export declare class MemoryManager {
4
+ private vectorStore;
5
+ private metadataStore;
6
+ private config;
7
+ private embeddingFunction;
8
+ constructor(config?: Partial<AIMemoryConfig>);
9
+ setEmbeddingFunction(fn: EmbeddingFunction): void;
10
+ addMemory(content: string, metadata?: Partial<MemoryMetadata>): Promise<Memory>;
11
+ addConversation(content: string, role: 'user' | 'assistant', metadata?: Partial<MemoryMetadata>): Promise<Memory>;
12
+ addFact(content: string, metadata?: Partial<MemoryMetadata>): Promise<Memory>;
13
+ addPreference(content: string, metadata?: Partial<MemoryMetadata>): Promise<Memory>;
14
+ addInstruction(content: string, metadata?: Partial<MemoryMetadata>): Promise<Memory>;
15
+ search(query: MemoryQuery): Promise<SearchResult[]>;
16
+ private buildVectorFilter;
17
+ private matchesFilter;
18
+ getMemory(id: string): Promise<Memory | null>;
19
+ updateMemory(id: string, updates: Partial<Memory>): Promise<Memory | null>;
20
+ deleteMemory(id: string): Promise<void>;
21
+ getRecentMemories(limit?: number): Promise<Memory[]>;
22
+ getMemoriesByType(type: MemoryType, limit?: number): Promise<Memory[]>;
23
+ getMemoriesByTag(tag: string, limit?: number): Promise<Memory[]>;
24
+ getMemoriesByUser(userId: string, limit?: number): Promise<Memory[]>;
25
+ getMemoriesBySession(sessionId: string, limit?: number): Promise<Memory[]>;
26
+ getStats(): Promise<MemoryStats>;
27
+ clear(): Promise<void>;
28
+ private enforceMaxMemories;
29
+ cleanup(): Promise<number>;
30
+ getConfig(): Readonly<AIMemoryConfig>;
31
+ exportData(): Promise<Memory[]>;
32
+ importData(memories: Memory[]): Promise<void>;
33
+ }
34
+ export declare function createMemoryManager(config?: Partial<AIMemoryConfig>): MemoryManager;
35
+ //# sourceMappingURL=memoryManager.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"memoryManager.d.ts","sourceRoot":"","sources":["../src/memoryManager.ts"],"names":[],"mappings":"AACA,OAAO,EACL,MAAM,EACN,cAAc,EACd,UAAU,EACV,WAAW,EACX,YAAY,EACZ,WAAW,EACX,iBAAiB,EAClB,MAAM,YAAY,CAAC;AAGpB,OAAO,EAAE,cAAc,EAAkB,MAAM,aAAa,CAAC;AAE7D,qBAAa,aAAa;IACxB,OAAO,CAAC,WAAW,CAAsB;IACzC,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,MAAM,CAAiB;IAC/B,OAAO,CAAC,iBAAiB,CAAkC;gBAE/C,MAAM,GAAE,OAAO,CAAC,cAAc,CAAM;IAMhD,oBAAoB,CAAC,EAAE,EAAE,iBAAiB,GAAG,IAAI;IAI3C,SAAS,CACb,OAAO,EAAE,MAAM,EACf,QAAQ,GAAE,OAAO,CAAC,cAAc,CAAM,GACrC,OAAO,CAAC,MAAM,CAAC;IA8CZ,eAAe,CACnB,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,GAAG,WAAW,EAC1B,QAAQ,GAAE,OAAO,CAAC,cAAc,CAAM,GACrC,OAAO,CAAC,MAAM,CAAC;IAQZ,OAAO,CACX,OAAO,EAAE,MAAM,EACf,QAAQ,GAAE,OAAO,CAAC,cAAc,CAAM,GACrC,OAAO,CAAC,MAAM,CAAC;IAQZ,aAAa,CACjB,OAAO,EAAE,MAAM,EACf,QAAQ,GAAE,OAAO,CAAC,cAAc,CAAM,GACrC,OAAO,CAAC,MAAM,CAAC;IAQZ,cAAc,CAClB,OAAO,EAAE,MAAM,EACf,QAAQ,GAAE,OAAO,CAAC,cAAc,CAAM,GACrC,OAAO,CAAC,MAAM,CAAC;IAQZ,MAAM,CAAC,KAAK,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAuCzD,OAAO,CAAC,iBAAiB;IAgBzB,OAAO,CAAC,aAAa;IAiCf,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAI7C,YAAY,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IA+B1E,YAAY,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAKvC,iBAAiB,CAAC,KAAK,GAAE,MAAW,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAIxD,iBAAiB,CAAC,IAAI,EAAE,UAAU,EAAE,KAAK,GAAE,MAAW,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAI1E,gBAAgB,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,GAAE,MAAW,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAIpE,iBAAiB,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,GAAE,MAAW,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAIxE,oBAAoB,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,GAAE,MAAW,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAI9E,QAAQ,IAAI,OAAO,CAAC,WAAW,CAAC;IAIhC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;YAKd,kBAAkB;IAe1B,OAAO,IAAI,OAAO,CAAC,MAAM,CAAC;IAkBhC,SAAS,IAAI,QAAQ,CAAC,cAAc,CAAC;IAI/B,UAAU,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAI/B,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CAapD;AAED,wBAAgB,mBAAmB,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,cAAc,CAAC,GAAG,aAAa,CAEnF"}
@@ -0,0 +1,265 @@
1
+ import { v4 as uuidv4 } from 'uuid';
2
+ import { InMemoryVectorStore } from './vectorStore.js';
3
+ import { MetadataStore } from './metadataStore.js';
4
+ import { validateConfig } from './config.js';
5
+ export class MemoryManager {
6
+ vectorStore;
7
+ metadataStore;
8
+ config;
9
+ embeddingFunction = null;
10
+ constructor(config = {}) {
11
+ this.config = validateConfig(config);
12
+ this.vectorStore = new InMemoryVectorStore(this.config.embeddingDimension);
13
+ this.metadataStore = new MetadataStore();
14
+ }
15
+ setEmbeddingFunction(fn) {
16
+ this.embeddingFunction = fn;
17
+ }
18
+ async addMemory(content, metadata = {}) {
19
+ const id = uuidv4();
20
+ const timestamp = Date.now();
21
+ const fullMetadata = {
22
+ type: metadata.type || 'context',
23
+ importance: metadata.importance ?? this.config.defaultImportance,
24
+ tags: metadata.tags || [],
25
+ userId: metadata.userId,
26
+ sessionId: metadata.sessionId,
27
+ source: metadata.source,
28
+ ...metadata,
29
+ };
30
+ let embedding;
31
+ if (this.embeddingFunction) {
32
+ try {
33
+ embedding = await this.embeddingFunction(content);
34
+ }
35
+ catch (error) {
36
+ console.warn('Failed to generate embedding:', error);
37
+ }
38
+ }
39
+ const memory = {
40
+ id,
41
+ content,
42
+ embedding,
43
+ timestamp,
44
+ metadata: fullMetadata,
45
+ };
46
+ await this.metadataStore.add(memory);
47
+ if (embedding) {
48
+ await this.vectorStore.add([{
49
+ id,
50
+ vector: embedding,
51
+ metadata: { ...memory },
52
+ }]);
53
+ }
54
+ await this.enforceMaxMemories();
55
+ return memory;
56
+ }
57
+ async addConversation(content, role, metadata = {}) {
58
+ const prefix = role === 'user' ? 'User: ' : 'Assistant: ';
59
+ return this.addMemory(prefix + content, {
60
+ type: 'conversation',
61
+ ...metadata,
62
+ });
63
+ }
64
+ async addFact(content, metadata = {}) {
65
+ return this.addMemory(content, {
66
+ type: 'fact',
67
+ importance: Math.max(metadata.importance ?? 0.7, 0.7),
68
+ ...metadata,
69
+ });
70
+ }
71
+ async addPreference(content, metadata = {}) {
72
+ return this.addMemory(content, {
73
+ type: 'preference',
74
+ importance: Math.max(metadata.importance ?? 0.6, 0.6),
75
+ ...metadata,
76
+ });
77
+ }
78
+ async addInstruction(content, metadata = {}) {
79
+ return this.addMemory(content, {
80
+ type: 'instruction',
81
+ importance: Math.max(metadata.importance ?? 0.9, 0.9),
82
+ ...metadata,
83
+ });
84
+ }
85
+ async search(query) {
86
+ const { text, embedding, filter, limit = 10, threshold } = query;
87
+ if (!embedding && !text) {
88
+ throw new Error('Either text or embedding must be provided');
89
+ }
90
+ let searchEmbedding = embedding;
91
+ if (text && !searchEmbedding && this.embeddingFunction) {
92
+ try {
93
+ searchEmbedding = await this.embeddingFunction(text);
94
+ }
95
+ catch (error) {
96
+ console.warn('Failed to generate embedding for search:', error);
97
+ const memories = await this.metadataStore.find(filter || {}, limit);
98
+ return memories.map(m => ({ memory: m, score: 1 }));
99
+ }
100
+ }
101
+ if (searchEmbedding) {
102
+ const vectorResults = await this.vectorStore.search(searchEmbedding, {
103
+ limit: limit * 2,
104
+ threshold,
105
+ filter: filter ? this.buildVectorFilter(filter) : undefined,
106
+ });
107
+ let results = vectorResults;
108
+ if (filter) {
109
+ results = vectorResults.filter(r => this.matchesFilter(r.memory, filter));
110
+ }
111
+ return results.slice(0, limit);
112
+ }
113
+ const memories = await this.metadataStore.find(filter || {}, limit);
114
+ return memories.map(m => ({ memory: m, score: 1 }));
115
+ }
116
+ buildVectorFilter(filter) {
117
+ const vectorFilter = {};
118
+ if (filter.types) {
119
+ vectorFilter['metadata.type'] = { $in: filter.types };
120
+ }
121
+ if (filter.userId) {
122
+ vectorFilter['metadata.userId'] = filter.userId;
123
+ }
124
+ if (filter.sessionId) {
125
+ vectorFilter['metadata.sessionId'] = filter.sessionId;
126
+ }
127
+ return vectorFilter;
128
+ }
129
+ matchesFilter(memory, filter) {
130
+ if (filter.types && !filter.types.includes(memory.metadata.type)) {
131
+ return false;
132
+ }
133
+ if (filter.tags) {
134
+ const hasTag = filter.tags.some((tag) => memory.metadata.tags.includes(tag));
135
+ if (!hasTag)
136
+ return false;
137
+ }
138
+ if (filter.userId && memory.metadata.userId !== filter.userId) {
139
+ return false;
140
+ }
141
+ if (filter.sessionId && memory.metadata.sessionId !== filter.sessionId) {
142
+ return false;
143
+ }
144
+ if (filter.minImportance !== undefined &&
145
+ memory.metadata.importance < filter.minImportance) {
146
+ return false;
147
+ }
148
+ if (filter.maxImportance !== undefined &&
149
+ memory.metadata.importance > filter.maxImportance) {
150
+ return false;
151
+ }
152
+ if (filter.startDate !== undefined && memory.timestamp < filter.startDate) {
153
+ return false;
154
+ }
155
+ if (filter.endDate !== undefined && memory.timestamp > filter.endDate) {
156
+ return false;
157
+ }
158
+ return true;
159
+ }
160
+ async getMemory(id) {
161
+ return this.metadataStore.getById(id);
162
+ }
163
+ async updateMemory(id, updates) {
164
+ const existing = await this.metadataStore.getById(id);
165
+ if (!existing)
166
+ return null;
167
+ const updated = {
168
+ ...existing,
169
+ ...updates,
170
+ id,
171
+ metadata: {
172
+ ...existing.metadata,
173
+ ...updates.metadata,
174
+ },
175
+ };
176
+ await this.metadataStore.update(updated);
177
+ if (updates.embedding || updates.content) {
178
+ const embedding = updates.embedding || existing.embedding;
179
+ if (embedding) {
180
+ await this.vectorStore.delete([id]);
181
+ await this.vectorStore.add([{
182
+ id,
183
+ vector: embedding,
184
+ metadata: { ...updated },
185
+ }]);
186
+ }
187
+ }
188
+ return updated;
189
+ }
190
+ async deleteMemory(id) {
191
+ await this.metadataStore.delete(id);
192
+ await this.vectorStore.delete([id]);
193
+ }
194
+ async getRecentMemories(limit = 10) {
195
+ return this.metadataStore.getRecent(limit);
196
+ }
197
+ async getMemoriesByType(type, limit = 50) {
198
+ return this.metadataStore.find({ types: [type] }, limit);
199
+ }
200
+ async getMemoriesByTag(tag, limit = 50) {
201
+ return this.metadataStore.find({ tags: [tag] }, limit);
202
+ }
203
+ async getMemoriesByUser(userId, limit = 50) {
204
+ return this.metadataStore.find({ userId }, limit);
205
+ }
206
+ async getMemoriesBySession(sessionId, limit = 50) {
207
+ return this.metadataStore.find({ sessionId }, limit);
208
+ }
209
+ async getStats() {
210
+ return this.metadataStore.getStats();
211
+ }
212
+ async clear() {
213
+ await this.metadataStore.clear();
214
+ await this.vectorStore.clear();
215
+ }
216
+ async enforceMaxMemories() {
217
+ const size = await this.metadataStore.size();
218
+ if (size > this.config.maxMemories) {
219
+ const memories = await this.metadataStore.getRecent(size);
220
+ const toDelete = memories
221
+ .filter(m => m.metadata.importance < this.config.autoCleanup.minImportance)
222
+ .slice(this.config.maxMemories - size);
223
+ for (const memory of toDelete) {
224
+ await this.deleteMemory(memory.id);
225
+ }
226
+ }
227
+ }
228
+ async cleanup() {
229
+ if (!this.config.autoCleanup.enabled)
230
+ return 0;
231
+ const cutoff = Date.now() - this.config.autoCleanup.maxAge;
232
+ const allMemories = await this.metadataStore.getAll();
233
+ let deletedCount = 0;
234
+ for (const memory of allMemories) {
235
+ if (memory.timestamp < cutoff &&
236
+ memory.metadata.importance < this.config.autoCleanup.minImportance) {
237
+ await this.deleteMemory(memory.id);
238
+ deletedCount++;
239
+ }
240
+ }
241
+ return deletedCount;
242
+ }
243
+ getConfig() {
244
+ return { ...this.config };
245
+ }
246
+ async exportData() {
247
+ return this.metadataStore.getAll();
248
+ }
249
+ async importData(memories) {
250
+ for (const memory of memories) {
251
+ await this.metadataStore.add(memory);
252
+ if (memory.embedding) {
253
+ await this.vectorStore.add([{
254
+ id: memory.id,
255
+ vector: memory.embedding,
256
+ metadata: { ...memory },
257
+ }]);
258
+ }
259
+ }
260
+ }
261
+ }
262
+ export function createMemoryManager(config) {
263
+ return new MemoryManager(config);
264
+ }
265
+ //# sourceMappingURL=memoryManager.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"memoryManager.js","sourceRoot":"","sources":["../src/memoryManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,IAAI,MAAM,EAAE,MAAM,MAAM,CAAC;AAUpC,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAkB,cAAc,EAAE,MAAM,aAAa,CAAC;AAE7D,MAAM,OAAO,aAAa;IAChB,WAAW,CAAsB;IACjC,aAAa,CAAgB;IAC7B,MAAM,CAAiB;IACvB,iBAAiB,GAA6B,IAAI,CAAC;IAE3D,YAAY,SAAkC,EAAE;QAC9C,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;QACrC,IAAI,CAAC,WAAW,GAAG,IAAI,mBAAmB,CAAC,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAC3E,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;IAC3C,CAAC;IAED,oBAAoB,CAAC,EAAqB;QACxC,IAAI,CAAC,iBAAiB,GAAG,EAAE,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,SAAS,CACb,OAAe,EACf,WAAoC,EAAE;QAEtC,MAAM,EAAE,GAAG,MAAM,EAAE,CAAC;QACpB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,MAAM,YAAY,GAAmB;YACnC,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,SAAS;YAChC,UAAU,EAAE,QAAQ,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB;YAChE,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,EAAE;YACzB,MAAM,EAAE,QAAQ,CAAC,MAAM;YACvB,SAAS,EAAE,QAAQ,CAAC,SAAS;YAC7B,MAAM,EAAE,QAAQ,CAAC,MAAM;YACvB,GAAG,QAAQ;SACZ,CAAC;QAEF,IAAI,SAA+B,CAAC;QACpC,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,IAAI,CAAC;gBACH,SAAS,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YACpD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,IAAI,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;QAED,MAAM,MAAM,GAAW;YACrB,EAAE;YACF,OAAO;YACP,SAAS;YACT,SAAS;YACT,QAAQ,EAAE,YAAY;SACvB,CAAC;QAEF,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAErC,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;oBAC1B,EAAE;oBACF,MAAM,EAAE,SAAS;oBACjB,QAAQ,EAAE,EAAE,GAAG,MAAM,EAAE;iBACxB,CAAC,CAAC,CAAC;QACN,CAAC;QAED,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAEhC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,OAAe,EACf,IAA0B,EAC1B,WAAoC,EAAE;QAEtC,MAAM,MAAM,GAAG,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,aAAa,CAAC;QAC1D,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,OAAO,EAAE;YACtC,IAAI,EAAE,cAAc;YACpB,GAAG,QAAQ;SACZ,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,OAAO,CACX,OAAe,EACf,WAAoC,EAAE;QAEtC,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE;YAC7B,IAAI,EAAE,MAAM;YACZ,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,UAAU,IAAI,GAAG,EAAE,GAAG,CAAC;YACrD,GAAG,QAAQ;SACZ,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,aAAa,CACjB,OAAe,EACf,WAAoC,EAAE;QAEtC,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE;YAC7B,IAAI,EAAE,YAAY;YAClB,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,UAAU,IAAI,GAAG,EAAE,GAAG,CAAC;YACrD,GAAG,QAAQ;SACZ,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,OAAe,EACf,WAAoC,EAAE;QAEtC,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE;YAC7B,IAAI,EAAE,aAAa;YACnB,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,UAAU,IAAI,GAAG,EAAE,GAAG,CAAC;YACrD,GAAG,QAAQ;SACZ,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAkB;QAC7B,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,GAAG,EAAE,EAAE,SAAS,EAAE,GAAG,KAAK,CAAC;QAEjE,IAAI,CAAC,SAAS,IAAI,CAAC,IAAI,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;QAC/D,CAAC;QAED,IAAI,eAAe,GAAG,SAAS,CAAC;QAEhC,IAAI,IAAI,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACvD,IAAI,CAAC;gBACH,eAAe,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YACvD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,IAAI,CAAC,0CAA0C,EAAE,KAAK,CAAC,CAAC;gBAChE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,EAAE,KAAK,CAAC,CAAC;gBACpE,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;YACtD,CAAC;QACH,CAAC;QAED,IAAI,eAAe,EAAE,CAAC;YACpB,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,eAAe,EAAE;gBACnE,KAAK,EAAE,KAAK,GAAG,CAAC;gBAChB,SAAS;gBACT,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS;aAC5D,CAAC,CAAC;YAEH,IAAI,OAAO,GAAG,aAAa,CAAC;YAE5B,IAAI,MAAM,EAAE,CAAC;gBACX,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;YAC5E,CAAC;YAED,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QACjC,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,EAAE,KAAK,CAAC,CAAC;QACpE,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IACtD,CAAC;IAEO,iBAAiB,CAAC,MAAW;QACnC,MAAM,YAAY,GAA4B,EAAE,CAAC;QAEjD,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,YAAY,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;QACxD,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,YAAY,CAAC,iBAAiB,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC;QAClD,CAAC;QACD,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;YACrB,YAAY,CAAC,oBAAoB,CAAC,GAAG,MAAM,CAAC,SAAS,CAAC;QACxD,CAAC;QAED,OAAO,YAAY,CAAC;IACtB,CAAC;IAEO,aAAa,CAAC,MAAc,EAAE,MAAW;QAC/C,IAAI,MAAM,CAAC,KAAK,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YACjE,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YAChB,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAW,EAAE,EAAE,CAC9C,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CACnC,CAAC;YACF,IAAI,CAAC,MAAM;gBAAE,OAAO,KAAK,CAAC;QAC5B,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC;YAC9D,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,QAAQ,CAAC,SAAS,KAAK,MAAM,CAAC,SAAS,EAAE,CAAC;YACvE,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,MAAM,CAAC,aAAa,KAAK,SAAS;YAClC,MAAM,CAAC,QAAQ,CAAC,UAAU,GAAG,MAAM,CAAC,aAAa,EAAE,CAAC;YACtD,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,MAAM,CAAC,aAAa,KAAK,SAAS;YAClC,MAAM,CAAC,QAAQ,CAAC,UAAU,GAAG,MAAM,CAAC,aAAa,EAAE,CAAC;YACtD,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,MAAM,CAAC,SAAS,KAAK,SAAS,IAAI,MAAM,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;YAC1E,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,MAAM,CAAC,OAAO,KAAK,SAAS,IAAI,MAAM,CAAC,SAAS,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;YACtE,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,EAAU;QACxB,OAAO,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,EAAU,EAAE,OAAwB;QACrD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACtD,IAAI,CAAC,QAAQ;YAAE,OAAO,IAAI,CAAC;QAE3B,MAAM,OAAO,GAAW;YACtB,GAAG,QAAQ;YACX,GAAG,OAAO;YACV,EAAE;YACF,QAAQ,EAAE;gBACR,GAAG,QAAQ,CAAC,QAAQ;gBACpB,GAAG,OAAO,CAAC,QAAQ;aACpB;SACF,CAAC;QAEF,MAAM,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAEzC,IAAI,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACzC,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,QAAQ,CAAC,SAAS,CAAC;YAC1D,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gBACpC,MAAM,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;wBAC1B,EAAE;wBACF,MAAM,EAAE,SAAS;wBACjB,QAAQ,EAAE,EAAE,GAAG,OAAO,EAAE;qBACzB,CAAC,CAAC,CAAC;YACN,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,EAAU;QAC3B,MAAM,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACpC,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,QAAgB,EAAE;QACxC,OAAO,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,IAAgB,EAAE,QAAgB,EAAE;QAC1D,OAAO,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,GAAW,EAAE,QAAgB,EAAE;QACpD,OAAO,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;IACzD,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,MAAc,EAAE,QAAgB,EAAE;QACxD,OAAO,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,KAAK,CAAC,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,SAAiB,EAAE,QAAgB,EAAE;QAC9D,OAAO,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,EAAE,KAAK,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,OAAO,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;QACjC,MAAM,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;IACjC,CAAC;IAEO,KAAK,CAAC,kBAAkB;QAC9B,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;QAE7C,IAAI,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;YACnC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAC1D,MAAM,QAAQ,GAAG,QAAQ;iBACtB,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC;iBAC1E,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,IAAI,CAAC,CAAC;YAEzC,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;gBAC9B,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO;YAAE,OAAO,CAAC,CAAC;QAE/C,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC;QAC3D,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;QAEtD,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,KAAK,MAAM,MAAM,IAAI,WAAW,EAAE,CAAC;YACjC,IAAI,MAAM,CAAC,SAAS,GAAG,MAAM;gBACzB,MAAM,CAAC,QAAQ,CAAC,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,aAAa,EAAE,CAAC;gBACvE,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACnC,YAAY,EAAE,CAAC;YACjB,CAAC;QACH,CAAC;QAED,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,SAAS;QACP,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,UAAU;QACd,OAAO,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,QAAkB;QACjC,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;YAC9B,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAErC,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;gBACrB,MAAM,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;wBAC1B,EAAE,EAAE,MAAM,CAAC,EAAE;wBACb,MAAM,EAAE,MAAM,CAAC,SAAS;wBACxB,QAAQ,EAAE,EAAE,GAAG,MAAM,EAAE;qBACxB,CAAC,CAAC,CAAC;YACN,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAED,MAAM,UAAU,mBAAmB,CAAC,MAAgC;IAClE,OAAO,IAAI,aAAa,CAAC,MAAM,CAAC,CAAC;AACnC,CAAC"}
@@ -0,0 +1,24 @@
1
+ import { Memory, MemoryFilter, MemoryStats } from './types.js';
2
+ export declare class MetadataStore {
3
+ private memories;
4
+ private indexByType;
5
+ private indexByTag;
6
+ private indexByUser;
7
+ private indexBySession;
8
+ private indexByTime;
9
+ constructor();
10
+ private initializeIndexes;
11
+ add(memory: Memory): Promise<void>;
12
+ private insertByTime;
13
+ update(memory: Memory): Promise<void>;
14
+ delete(id: string): Promise<void>;
15
+ getById(id: string): Promise<Memory | null>;
16
+ getAll(): Promise<Memory[]>;
17
+ find(filter: MemoryFilter, limit?: number): Promise<Memory[]>;
18
+ getByTimeRange(startDate: number, endDate: number): Promise<Memory[]>;
19
+ getRecent(limit?: number): Promise<Memory[]>;
20
+ getStats(): Promise<MemoryStats>;
21
+ clear(): Promise<void>;
22
+ size(): Promise<number>;
23
+ }
24
+ //# sourceMappingURL=metadataStore.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"metadataStore.d.ts","sourceRoot":"","sources":["../src/metadataStore.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,YAAY,EAAc,WAAW,EAAE,MAAM,YAAY,CAAC;AAE3E,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAkC;IAClD,OAAO,CAAC,WAAW,CAA2C;IAC9D,OAAO,CAAC,UAAU,CAAuC;IACzD,OAAO,CAAC,WAAW,CAAuC;IAC1D,OAAO,CAAC,cAAc,CAAuC;IAC7D,OAAO,CAAC,WAAW,CAAgB;;IAMnC,OAAO,CAAC,iBAAiB;IAOnB,GAAG,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAmCxC,OAAO,CAAC,YAAY;IAkBd,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA+BrC,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA0BjC,OAAO,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAM3C,MAAM,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAI3B,IAAI,CAAC,MAAM,EAAE,YAAY,EAAE,KAAK,GAAE,MAAY,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAsFlE,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAerE,SAAS,CAAC,KAAK,GAAE,MAAW,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAKhD,QAAQ,IAAI,OAAO,CAAC,WAAW,CAAC;IAsChC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAStB,IAAI,IAAI,OAAO,CAAC,MAAM,CAAC;CAG9B"}