@jcmrs/opencode-vmem 1.0.0

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 (140) hide show
  1. package/CHANGELOG.md +83 -0
  2. package/LICENSE +21 -0
  3. package/README.md +172 -0
  4. package/dist/capture/classification.d.ts +37 -0
  5. package/dist/capture/classification.d.ts.map +1 -0
  6. package/dist/capture/classification.js +181 -0
  7. package/dist/capture/classification.js.map +1 -0
  8. package/dist/capture/filtering.d.ts +63 -0
  9. package/dist/capture/filtering.d.ts.map +1 -0
  10. package/dist/capture/filtering.js +195 -0
  11. package/dist/capture/filtering.js.map +1 -0
  12. package/dist/capture/observation.d.ts +28 -0
  13. package/dist/capture/observation.d.ts.map +1 -0
  14. package/dist/capture/observation.js +188 -0
  15. package/dist/capture/observation.js.map +1 -0
  16. package/dist/commands/index.d.ts +38 -0
  17. package/dist/commands/index.d.ts.map +1 -0
  18. package/dist/commands/index.js +375 -0
  19. package/dist/commands/index.js.map +1 -0
  20. package/dist/config.d.ts +5 -0
  21. package/dist/config.d.ts.map +1 -0
  22. package/dist/config.js +41 -0
  23. package/dist/config.js.map +1 -0
  24. package/dist/context/builder.d.ts +19 -0
  25. package/dist/context/builder.d.ts.map +1 -0
  26. package/dist/context/builder.js +209 -0
  27. package/dist/context/builder.js.map +1 -0
  28. package/dist/context/format.d.ts +32 -0
  29. package/dist/context/format.d.ts.map +1 -0
  30. package/dist/context/format.js +211 -0
  31. package/dist/context/format.js.map +1 -0
  32. package/dist/embedding/index.d.ts +3 -0
  33. package/dist/embedding/index.d.ts.map +1 -0
  34. package/dist/embedding/index.js +3 -0
  35. package/dist/embedding/index.js.map +1 -0
  36. package/dist/embedding/provider.d.ts +15 -0
  37. package/dist/embedding/provider.d.ts.map +1 -0
  38. package/dist/embedding/provider.js +19 -0
  39. package/dist/embedding/provider.js.map +1 -0
  40. package/dist/embedding/transformers.d.ts +14 -0
  41. package/dist/embedding/transformers.d.ts.map +1 -0
  42. package/dist/embedding/transformers.js +55 -0
  43. package/dist/embedding/transformers.js.map +1 -0
  44. package/dist/extract/confidence.d.ts +48 -0
  45. package/dist/extract/confidence.d.ts.map +1 -0
  46. package/dist/extract/confidence.js +230 -0
  47. package/dist/extract/confidence.js.map +1 -0
  48. package/dist/extract/llm-extractor.d.ts +35 -0
  49. package/dist/extract/llm-extractor.d.ts.map +1 -0
  50. package/dist/extract/llm-extractor.js +204 -0
  51. package/dist/extract/llm-extractor.js.map +1 -0
  52. package/dist/extract/pipeline.d.ts +44 -0
  53. package/dist/extract/pipeline.d.ts.map +1 -0
  54. package/dist/extract/pipeline.js +171 -0
  55. package/dist/extract/pipeline.js.map +1 -0
  56. package/dist/extract/rules-engine.d.ts +14 -0
  57. package/dist/extract/rules-engine.d.ts.map +1 -0
  58. package/dist/extract/rules-engine.js +314 -0
  59. package/dist/extract/rules-engine.js.map +1 -0
  60. package/dist/extract/summarizer.d.ts +27 -0
  61. package/dist/extract/summarizer.d.ts.map +1 -0
  62. package/dist/extract/summarizer.js +148 -0
  63. package/dist/extract/summarizer.js.map +1 -0
  64. package/dist/hooks/session-idle.d.ts +38 -0
  65. package/dist/hooks/session-idle.d.ts.map +1 -0
  66. package/dist/hooks/session-idle.js +142 -0
  67. package/dist/hooks/session-idle.js.map +1 -0
  68. package/dist/hooks/system-transform.d.ts +41 -0
  69. package/dist/hooks/system-transform.d.ts.map +1 -0
  70. package/dist/hooks/system-transform.js +98 -0
  71. package/dist/hooks/system-transform.js.map +1 -0
  72. package/dist/hooks/tool-after.d.ts +66 -0
  73. package/dist/hooks/tool-after.d.ts.map +1 -0
  74. package/dist/hooks/tool-after.js +155 -0
  75. package/dist/hooks/tool-after.js.map +1 -0
  76. package/dist/index.d.ts +10 -0
  77. package/dist/index.d.ts.map +1 -0
  78. package/dist/index.js +10 -0
  79. package/dist/index.js.map +1 -0
  80. package/dist/mind.d.ts +69 -0
  81. package/dist/mind.d.ts.map +1 -0
  82. package/dist/mind.js +368 -0
  83. package/dist/mind.js.map +1 -0
  84. package/dist/procedural/index.d.ts +4 -0
  85. package/dist/procedural/index.d.ts.map +1 -0
  86. package/dist/procedural/index.js +4 -0
  87. package/dist/procedural/index.js.map +1 -0
  88. package/dist/procedural/induction.d.ts +23 -0
  89. package/dist/procedural/induction.d.ts.map +1 -0
  90. package/dist/procedural/induction.js +104 -0
  91. package/dist/procedural/induction.js.map +1 -0
  92. package/dist/procedural/miner.d.ts +45 -0
  93. package/dist/procedural/miner.d.ts.map +1 -0
  94. package/dist/procedural/miner.js +171 -0
  95. package/dist/procedural/miner.js.map +1 -0
  96. package/dist/procedural/suggester.d.ts +20 -0
  97. package/dist/procedural/suggester.d.ts.map +1 -0
  98. package/dist/procedural/suggester.js +69 -0
  99. package/dist/procedural/suggester.js.map +1 -0
  100. package/dist/store/archive.d.ts +31 -0
  101. package/dist/store/archive.d.ts.map +1 -0
  102. package/dist/store/archive.js +137 -0
  103. package/dist/store/archive.js.map +1 -0
  104. package/dist/store/backup.d.ts +36 -0
  105. package/dist/store/backup.d.ts.map +1 -0
  106. package/dist/store/backup.js +163 -0
  107. package/dist/store/backup.js.map +1 -0
  108. package/dist/store/memorycards.d.ts +31 -0
  109. package/dist/store/memorycards.d.ts.map +1 -0
  110. package/dist/store/memorycards.js +161 -0
  111. package/dist/store/memorycards.js.map +1 -0
  112. package/dist/store/mv2-adapter.d.ts +29 -0
  113. package/dist/store/mv2-adapter.d.ts.map +1 -0
  114. package/dist/store/mv2-adapter.js +359 -0
  115. package/dist/store/mv2-adapter.js.map +1 -0
  116. package/dist/store/procedures.d.ts +79 -0
  117. package/dist/store/procedures.d.ts.map +1 -0
  118. package/dist/store/procedures.js +183 -0
  119. package/dist/store/procedures.js.map +1 -0
  120. package/dist/store/tombstone.d.ts +27 -0
  121. package/dist/store/tombstone.d.ts.map +1 -0
  122. package/dist/store/tombstone.js +136 -0
  123. package/dist/store/tombstone.js.map +1 -0
  124. package/dist/tools/index.d.ts +36 -0
  125. package/dist/tools/index.d.ts.map +1 -0
  126. package/dist/tools/index.js +236 -0
  127. package/dist/tools/index.js.map +1 -0
  128. package/dist/types.d.ts +301 -0
  129. package/dist/types.d.ts.map +1 -0
  130. package/dist/types.js +69 -0
  131. package/dist/types.js.map +1 -0
  132. package/dist/utils/error.d.ts +57 -0
  133. package/dist/utils/error.d.ts.map +1 -0
  134. package/dist/utils/error.js +143 -0
  135. package/dist/utils/error.js.map +1 -0
  136. package/dist/utils/uuid.d.ts +2 -0
  137. package/dist/utils/uuid.d.ts.map +1 -0
  138. package/dist/utils/uuid.js +5 -0
  139. package/dist/utils/uuid.js.map +1 -0
  140. package/package.json +67 -0
@@ -0,0 +1,48 @@
1
+ import { MemoryCard, EnrichmentContext } from '../types.js';
2
+ export interface ConfidenceScore {
3
+ overall: number;
4
+ components: ConfidenceComponents;
5
+ factors: ConfidenceFactor[];
6
+ action: ConfidenceAction;
7
+ }
8
+ export interface ConfidenceComponents {
9
+ patternMatch: number;
10
+ textClarity: number;
11
+ extractionCertainty: number;
12
+ contextConsistency: number;
13
+ }
14
+ export interface ConfidenceFactor {
15
+ name: string;
16
+ impact: 'positive' | 'negative' | 'neutral';
17
+ weight: number;
18
+ description: string;
19
+ }
20
+ export type ConfidenceAction = 'accept' | 'flag' | 'tombstone' | 'reject';
21
+ export interface ConfidenceThresholds {
22
+ accept: number;
23
+ flag: number;
24
+ autoTombstone: number;
25
+ reject: number;
26
+ }
27
+ export declare class ConfidenceScorer {
28
+ private thresholds;
29
+ private existingCards;
30
+ constructor(thresholds?: Partial<ConfidenceThresholds>);
31
+ score(card: MemoryCard, ctx: EnrichmentContext, source: 'rules' | 'llm' | 'hybrid'): ConfidenceScore;
32
+ scoreBatch(cards: MemoryCard[], ctx: EnrichmentContext, source: 'rules' | 'llm' | 'hybrid'): Map<string, ConfidenceScore>;
33
+ private calculateComponents;
34
+ private scorePatternMatch;
35
+ private scoreTextClarity;
36
+ private scoreExtractionCertainty;
37
+ private scoreContextConsistency;
38
+ private calculateSimilarity;
39
+ private identifyFactors;
40
+ private calculateOverall;
41
+ determineAction(score: number): ConfidenceAction;
42
+ loadExistingCards(cards: MemoryCard[]): void;
43
+ shouldFlag(score: ConfidenceScore): boolean;
44
+ shouldTombstone(score: ConfidenceScore): boolean;
45
+ getThresholds(): ConfidenceThresholds;
46
+ setThresholds(thresholds: Partial<ConfidenceThresholds>): void;
47
+ }
48
+ //# sourceMappingURL=confidence.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"confidence.d.ts","sourceRoot":"","sources":["../../src/extract/confidence.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAE5D,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,oBAAoB,CAAC;IACjC,OAAO,EAAE,gBAAgB,EAAE,CAAC;IAC5B,MAAM,EAAE,gBAAgB,CAAC;CAC1B;AAED,MAAM,WAAW,oBAAoB;IACnC,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,kBAAkB,EAAE,MAAM,CAAC;CAC5B;AAED,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,UAAU,GAAG,UAAU,GAAG,SAAS,CAAC;IAC5C,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,MAAM,gBAAgB,GAAG,QAAQ,GAAG,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;AAE1E,MAAM,WAAW,oBAAoB;IACnC,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,MAAM,CAAC;CAChB;AASD,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,UAAU,CAAuB;IACzC,OAAO,CAAC,aAAa,CAA4B;gBAErC,UAAU,CAAC,EAAE,OAAO,CAAC,oBAAoB,CAAC;IAKtD,KAAK,CACH,IAAI,EAAE,UAAU,EAChB,GAAG,EAAE,iBAAiB,EACtB,MAAM,EAAE,OAAO,GAAG,KAAK,GAAG,QAAQ,GACjC,eAAe;IAclB,UAAU,CACR,KAAK,EAAE,UAAU,EAAE,EACnB,GAAG,EAAE,iBAAiB,EACtB,MAAM,EAAE,OAAO,GAAG,KAAK,GAAG,QAAQ,GACjC,GAAG,CAAC,MAAM,EAAE,eAAe,CAAC;IAU/B,OAAO,CAAC,mBAAmB;IAa3B,OAAO,CAAC,iBAAiB;IAWzB,OAAO,CAAC,gBAAgB;IA4BxB,OAAO,CAAC,wBAAwB;IA0BhC,OAAO,CAAC,uBAAuB;IAuB/B,OAAO,CAAC,mBAAmB;IAe3B,OAAO,CAAC,eAAe;IA+DvB,OAAO,CAAC,gBAAgB;IAqBxB,eAAe,CAAC,KAAK,EAAE,MAAM,GAAG,gBAAgB;IAYhD,iBAAiB,CAAC,KAAK,EAAE,UAAU,EAAE,GAAG,IAAI;IAY5C,UAAU,CAAC,KAAK,EAAE,eAAe,GAAG,OAAO;IAI3C,eAAe,CAAC,KAAK,EAAE,eAAe,GAAG,OAAO;IAIhD,aAAa,IAAI,oBAAoB;IAIrC,aAAa,CAAC,UAAU,EAAE,OAAO,CAAC,oBAAoB,CAAC,GAAG,IAAI;CAG/D"}
@@ -0,0 +1,230 @@
1
+ const DEFAULT_THRESHOLDS = {
2
+ accept: 0.8,
3
+ flag: 0.4,
4
+ autoTombstone: 0.4,
5
+ reject: 0.2
6
+ };
7
+ export class ConfidenceScorer {
8
+ thresholds;
9
+ existingCards;
10
+ constructor(thresholds) {
11
+ this.thresholds = { ...DEFAULT_THRESHOLDS, ...thresholds };
12
+ this.existingCards = new Map();
13
+ }
14
+ score(card, ctx, source) {
15
+ const components = this.calculateComponents(card, ctx, source);
16
+ const factors = this.identifyFactors(card, ctx, source, components);
17
+ const overall = this.calculateOverall(components, factors);
18
+ const action = this.determineAction(overall);
19
+ return {
20
+ overall,
21
+ components,
22
+ factors,
23
+ action
24
+ };
25
+ }
26
+ scoreBatch(cards, ctx, source) {
27
+ const results = new Map();
28
+ for (const card of cards) {
29
+ results.set(card.id, this.score(card, ctx, source));
30
+ }
31
+ return results;
32
+ }
33
+ calculateComponents(card, ctx, source) {
34
+ return {
35
+ patternMatch: this.scorePatternMatch(card, source),
36
+ textClarity: this.scoreTextClarity(ctx),
37
+ extractionCertainty: this.scoreExtractionCertainty(card),
38
+ contextConsistency: this.scoreContextConsistency(card)
39
+ };
40
+ }
41
+ scorePatternMatch(_card, source) {
42
+ switch (source) {
43
+ case 'rules':
44
+ return 0.9;
45
+ case 'hybrid':
46
+ return 0.85;
47
+ case 'llm':
48
+ return 0.7;
49
+ }
50
+ }
51
+ scoreTextClarity(ctx) {
52
+ const text = ctx.text;
53
+ if (!text || text.length < 10) {
54
+ return 0.3;
55
+ }
56
+ let score = 0.5;
57
+ if (text.length >= 50 && text.length <= 500) {
58
+ score += 0.2;
59
+ }
60
+ else if (text.length > 500) {
61
+ score -= 0.1;
62
+ }
63
+ const sentenceCount = (text.match(/[.!?]+/g) || []).length;
64
+ if (sentenceCount >= 1 && sentenceCount <= 5) {
65
+ score += 0.1;
66
+ }
67
+ const hasSpecialChars = /[<>{}[\]]/.test(text);
68
+ if (hasSpecialChars) {
69
+ score -= 0.1;
70
+ }
71
+ return Math.max(0, Math.min(1, score));
72
+ }
73
+ scoreExtractionCertainty(card) {
74
+ const baseConfidence = card.confidence ?? 0.5;
75
+ let score = baseConfidence;
76
+ if (card.entity && /^[a-z][a-z0-9._]*$/.test(card.entity)) {
77
+ score += 0.05;
78
+ }
79
+ else {
80
+ score -= 0.1;
81
+ }
82
+ if (card.slot && /^[a-z][a-z0-9_]*$/.test(card.slot)) {
83
+ score += 0.05;
84
+ }
85
+ else {
86
+ score -= 0.1;
87
+ }
88
+ if (card.value && card.value.length >= 2 && card.value.length <= 200) {
89
+ score += 0.05;
90
+ }
91
+ else if (!card.value || card.value.length < 2) {
92
+ score -= 0.15;
93
+ }
94
+ return Math.max(0, Math.min(1, score));
95
+ }
96
+ scoreContextConsistency(card) {
97
+ const key = `${card.entity}:${card.slot}`;
98
+ const existing = this.existingCards.get(key);
99
+ if (!existing || existing.length === 0) {
100
+ return 0.7;
101
+ }
102
+ const sameValue = existing.some(c => c.value === card.value);
103
+ if (sameValue) {
104
+ return 0.95;
105
+ }
106
+ const similarValue = existing.some(c => this.calculateSimilarity(c.value, card.value) > 0.8);
107
+ if (similarValue) {
108
+ return 0.85;
109
+ }
110
+ return 0.5;
111
+ }
112
+ calculateSimilarity(a, b) {
113
+ const aLower = a.toLowerCase();
114
+ const bLower = b.toLowerCase();
115
+ if (aLower === bLower)
116
+ return 1;
117
+ const aWords = new Set(aLower.split(/\s+/));
118
+ const bWords = new Set(bLower.split(/\s+/));
119
+ const intersection = new Set([...aWords].filter(x => bWords.has(x)));
120
+ const union = new Set([...aWords, ...bWords]);
121
+ return intersection.size / union.size;
122
+ }
123
+ identifyFactors(card, _ctx, source, components) {
124
+ const factors = [];
125
+ if (source === 'rules') {
126
+ factors.push({
127
+ name: 'rule_match',
128
+ impact: 'positive',
129
+ weight: 0.2,
130
+ description: 'Extracted via deterministic rule'
131
+ });
132
+ }
133
+ else if (source === 'llm') {
134
+ factors.push({
135
+ name: 'llm_extraction',
136
+ impact: 'neutral',
137
+ weight: 0,
138
+ description: 'Extracted via LLM (may have variance)'
139
+ });
140
+ }
141
+ if (card.confidence !== undefined) {
142
+ if (card.confidence >= 0.8) {
143
+ factors.push({
144
+ name: 'high_base_confidence',
145
+ impact: 'positive',
146
+ weight: 0.15,
147
+ description: `Base confidence: ${card.confidence.toFixed(2)}`
148
+ });
149
+ }
150
+ else if (card.confidence < 0.5) {
151
+ factors.push({
152
+ name: 'low_base_confidence',
153
+ impact: 'negative',
154
+ weight: -0.15,
155
+ description: `Low base confidence: ${card.confidence.toFixed(2)}`
156
+ });
157
+ }
158
+ }
159
+ if (components.textClarity < 0.5) {
160
+ factors.push({
161
+ name: 'unclear_text',
162
+ impact: 'negative',
163
+ weight: -0.1,
164
+ description: 'Source text is unclear or noisy'
165
+ });
166
+ }
167
+ if (card.value && card.value.length < 2) {
168
+ factors.push({
169
+ name: 'short_value',
170
+ impact: 'negative',
171
+ weight: -0.1,
172
+ description: 'Extracted value is very short'
173
+ });
174
+ }
175
+ return factors;
176
+ }
177
+ calculateOverall(components, factors) {
178
+ const weights = {
179
+ patternMatch: 0.25,
180
+ textClarity: 0.15,
181
+ extractionCertainty: 0.35,
182
+ contextConsistency: 0.25
183
+ };
184
+ let score = components.patternMatch * weights.patternMatch +
185
+ components.textClarity * weights.textClarity +
186
+ components.extractionCertainty * weights.extractionCertainty +
187
+ components.contextConsistency * weights.contextConsistency;
188
+ for (const factor of factors) {
189
+ score += factor.weight;
190
+ }
191
+ return Math.max(0, Math.min(1, score));
192
+ }
193
+ determineAction(score) {
194
+ if (score >= this.thresholds.accept) {
195
+ return 'accept';
196
+ }
197
+ else if (score >= this.thresholds.flag) {
198
+ return 'flag';
199
+ }
200
+ else if (score >= this.thresholds.reject) {
201
+ return 'tombstone';
202
+ }
203
+ else {
204
+ return 'reject';
205
+ }
206
+ }
207
+ loadExistingCards(cards) {
208
+ this.existingCards.clear();
209
+ for (const card of cards) {
210
+ const key = `${card.entity}:${card.slot}`;
211
+ if (!this.existingCards.has(key)) {
212
+ this.existingCards.set(key, []);
213
+ }
214
+ this.existingCards.get(key).push(card);
215
+ }
216
+ }
217
+ shouldFlag(score) {
218
+ return score.action === 'flag';
219
+ }
220
+ shouldTombstone(score) {
221
+ return score.action === 'tombstone' || score.action === 'reject';
222
+ }
223
+ getThresholds() {
224
+ return { ...this.thresholds };
225
+ }
226
+ setThresholds(thresholds) {
227
+ this.thresholds = { ...this.thresholds, ...thresholds };
228
+ }
229
+ }
230
+ //# sourceMappingURL=confidence.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"confidence.js","sourceRoot":"","sources":["../../src/extract/confidence.ts"],"names":[],"mappings":"AAgCA,MAAM,kBAAkB,GAAyB;IAC/C,MAAM,EAAE,GAAG;IACX,IAAI,EAAE,GAAG;IACT,aAAa,EAAE,GAAG;IAClB,MAAM,EAAE,GAAG;CACZ,CAAC;AAEF,MAAM,OAAO,gBAAgB;IACnB,UAAU,CAAuB;IACjC,aAAa,CAA4B;IAEjD,YAAY,UAA0C;QACpD,IAAI,CAAC,UAAU,GAAG,EAAE,GAAG,kBAAkB,EAAE,GAAG,UAAU,EAAE,CAAC;QAC3D,IAAI,CAAC,aAAa,GAAG,IAAI,GAAG,EAAE,CAAC;IACjC,CAAC;IAED,KAAK,CACH,IAAgB,EAChB,GAAsB,EACtB,MAAkC;QAElC,MAAM,UAAU,GAAG,IAAI,CAAC,mBAAmB,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QAC/D,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACpE,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAC3D,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QAE7C,OAAO;YACL,OAAO;YACP,UAAU;YACV,OAAO;YACP,MAAM;SACP,CAAC;IACJ,CAAC;IAED,UAAU,CACR,KAAmB,EACnB,GAAsB,EACtB,MAAkC;QAElC,MAAM,OAAO,GAAG,IAAI,GAAG,EAA2B,CAAC;QAEnD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC;QACtD,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,mBAAmB,CACzB,IAAgB,EAChB,GAAsB,EACtB,MAAkC;QAElC,OAAO;YACL,YAAY,EAAE,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,MAAM,CAAC;YAClD,WAAW,EAAE,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC;YACvC,mBAAmB,EAAE,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC;YACxD,kBAAkB,EAAE,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC;SACvD,CAAC;IACJ,CAAC;IAEO,iBAAiB,CAAC,KAAiB,EAAE,MAAkC;QAC7E,QAAQ,MAAM,EAAE,CAAC;YACf,KAAK,OAAO;gBACV,OAAO,GAAG,CAAC;YACb,KAAK,QAAQ;gBACX,OAAO,IAAI,CAAC;YACd,KAAK,KAAK;gBACR,OAAO,GAAG,CAAC;QACf,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,GAAsB;QAC7C,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC;QAEtB,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;YAC9B,OAAO,GAAG,CAAC;QACb,CAAC;QAED,IAAI,KAAK,GAAG,GAAG,CAAC;QAEhB,IAAI,IAAI,CAAC,MAAM,IAAI,EAAE,IAAI,IAAI,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;YAC5C,KAAK,IAAI,GAAG,CAAC;QACf,CAAC;aAAM,IAAI,IAAI,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC;YAC7B,KAAK,IAAI,GAAG,CAAC;QACf,CAAC;QAED,MAAM,aAAa,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;QAC3D,IAAI,aAAa,IAAI,CAAC,IAAI,aAAa,IAAI,CAAC,EAAE,CAAC;YAC7C,KAAK,IAAI,GAAG,CAAC;QACf,CAAC;QAED,MAAM,eAAe,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,IAAI,eAAe,EAAE,CAAC;YACpB,KAAK,IAAI,GAAG,CAAC;QACf,CAAC;QAED,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC;IACzC,CAAC;IAEO,wBAAwB,CAAC,IAAgB;QAC/C,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,IAAI,GAAG,CAAC;QAE9C,IAAI,KAAK,GAAG,cAAc,CAAC;QAE3B,IAAI,IAAI,CAAC,MAAM,IAAI,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1D,KAAK,IAAI,IAAI,CAAC;QAChB,CAAC;aAAM,CAAC;YACN,KAAK,IAAI,GAAG,CAAC;QACf,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,IAAI,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACrD,KAAK,IAAI,IAAI,CAAC;QAChB,CAAC;aAAM,CAAC;YACN,KAAK,IAAI,GAAG,CAAC;QACf,CAAC;QAED,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;YACrE,KAAK,IAAI,IAAI,CAAC;QAChB,CAAC;aAAM,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChD,KAAK,IAAI,IAAI,CAAC;QAChB,CAAC;QAED,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC;IACzC,CAAC;IAEO,uBAAuB,CAAC,IAAgB;QAC9C,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;QAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAE7C,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvC,OAAO,GAAG,CAAC;QACb,CAAC;QAED,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7D,IAAI,SAAS,EAAE,CAAC;YACd,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,YAAY,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CACrC,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,GAAG,CACpD,CAAC;QACF,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,GAAG,CAAC;IACb,CAAC;IAEO,mBAAmB,CAAC,CAAS,EAAE,CAAS;QAC9C,MAAM,MAAM,GAAG,CAAC,CAAC,WAAW,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,CAAC,CAAC,WAAW,EAAE,CAAC;QAE/B,IAAI,MAAM,KAAK,MAAM;YAAE,OAAO,CAAC,CAAC;QAEhC,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;QAC5C,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;QAE5C,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACrE,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC;QAE9C,OAAO,YAAY,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;IACxC,CAAC;IAEO,eAAe,CACrB,IAAgB,EAChB,IAAuB,EACvB,MAAkC,EAClC,UAAgC;QAEhC,MAAM,OAAO,GAAuB,EAAE,CAAC;QAEvC,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;YACvB,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,YAAY;gBAClB,MAAM,EAAE,UAAU;gBAClB,MAAM,EAAE,GAAG;gBACX,WAAW,EAAE,kCAAkC;aAChD,CAAC,CAAC;QACL,CAAC;aAAM,IAAI,MAAM,KAAK,KAAK,EAAE,CAAC;YAC5B,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,gBAAgB;gBACtB,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,CAAC;gBACT,WAAW,EAAE,uCAAuC;aACrD,CAAC,CAAC;QACL,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YAClC,IAAI,IAAI,CAAC,UAAU,IAAI,GAAG,EAAE,CAAC;gBAC3B,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,sBAAsB;oBAC5B,MAAM,EAAE,UAAU;oBAClB,MAAM,EAAE,IAAI;oBACZ,WAAW,EAAE,oBAAoB,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;iBAC9D,CAAC,CAAC;YACL,CAAC;iBAAM,IAAI,IAAI,CAAC,UAAU,GAAG,GAAG,EAAE,CAAC;gBACjC,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,qBAAqB;oBAC3B,MAAM,EAAE,UAAU;oBAClB,MAAM,EAAE,CAAC,IAAI;oBACb,WAAW,EAAE,wBAAwB,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;iBAClE,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,IAAI,UAAU,CAAC,WAAW,GAAG,GAAG,EAAE,CAAC;YACjC,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,cAAc;gBACpB,MAAM,EAAE,UAAU;gBAClB,MAAM,EAAE,CAAC,GAAG;gBACZ,WAAW,EAAE,iCAAiC;aAC/C,CAAC,CAAC;QACL,CAAC;QAED,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxC,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,aAAa;gBACnB,MAAM,EAAE,UAAU;gBAClB,MAAM,EAAE,CAAC,GAAG;gBACZ,WAAW,EAAE,+BAA+B;aAC7C,CAAC,CAAC;QACL,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,gBAAgB,CAAC,UAAgC,EAAE,OAA2B;QACpF,MAAM,OAAO,GAAG;YACd,YAAY,EAAE,IAAI;YAClB,WAAW,EAAE,IAAI;YACjB,mBAAmB,EAAE,IAAI;YACzB,kBAAkB,EAAE,IAAI;SACzB,CAAC;QAEF,IAAI,KAAK,GACP,UAAU,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY;YAC9C,UAAU,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW;YAC5C,UAAU,CAAC,mBAAmB,GAAG,OAAO,CAAC,mBAAmB;YAC5D,UAAU,CAAC,kBAAkB,GAAG,OAAO,CAAC,kBAAkB,CAAC;QAE7D,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC;QACzB,CAAC;QAED,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC;IACzC,CAAC;IAED,eAAe,CAAC,KAAa;QAC3B,IAAI,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YACpC,OAAO,QAAQ,CAAC;QAClB,CAAC;aAAM,IAAI,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;YACzC,OAAO,MAAM,CAAC;QAChB,CAAC;aAAM,IAAI,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YAC3C,OAAO,WAAW,CAAC;QACrB,CAAC;aAAM,CAAC;YACN,OAAO,QAAQ,CAAC;QAClB,CAAC;IACH,CAAC;IAED,iBAAiB,CAAC,KAAmB;QACnC,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;QAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAC1C,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBACjC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAClC,CAAC;YACD,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IAED,UAAU,CAAC,KAAsB;QAC/B,OAAO,KAAK,CAAC,MAAM,KAAK,MAAM,CAAC;IACjC,CAAC;IAED,eAAe,CAAC,KAAsB;QACpC,OAAO,KAAK,CAAC,MAAM,KAAK,WAAW,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,CAAC;IACnE,CAAC;IAED,aAAa;QACX,OAAO,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;IAChC,CAAC;IAED,aAAa,CAAC,UAAyC;QACrD,IAAI,CAAC,UAAU,GAAG,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,GAAG,UAAU,EAAE,CAAC;IAC1D,CAAC;CACF"}
@@ -0,0 +1,35 @@
1
+ import { EnrichmentContext, EnrichmentResult, MemoryCard, VmemConfig } from '../types.js';
2
+ export interface LLMExtractionConfig {
3
+ provider: 'zai' | 'openai';
4
+ model: string;
5
+ apiKey?: string;
6
+ maxTokens: number;
7
+ temperature: number;
8
+ timeout: number;
9
+ maxCardsPerObservation: number;
10
+ minConfidence: number;
11
+ }
12
+ export interface ExtractedMemory {
13
+ kind: MemoryCard['kind'];
14
+ entity: string;
15
+ slot: string;
16
+ value: string;
17
+ polarity?: 'positive' | 'negative';
18
+ confidence: number;
19
+ reasoning?: string;
20
+ }
21
+ export declare class LLMExtractor {
22
+ private config;
23
+ private globalConfig;
24
+ private apiKey;
25
+ constructor(globalConfig: VmemConfig, config?: Partial<LLMExtractionConfig>);
26
+ enrich(ctx: EnrichmentContext): Promise<EnrichmentResult>;
27
+ enrichBatch(ctxs: EnrichmentContext[]): Promise<EnrichmentResult[]>;
28
+ private callLLM;
29
+ private buildPromptText;
30
+ private buildCards;
31
+ private normalizeEntity;
32
+ private normalizeSlot;
33
+ private delay;
34
+ }
35
+ //# sourceMappingURL=llm-extractor.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"llm-extractor.d.ts","sourceRoot":"","sources":["../../src/extract/llm-extractor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAG1F,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,KAAK,GAAG,QAAQ,CAAC;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,sBAAsB,EAAE,MAAM,CAAC;IAC/B,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC;IACzB,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,UAAU,GAAG,UAAU,CAAC;IACnC,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAsCD,qBAAa,YAAY;IACvB,OAAO,CAAC,MAAM,CAAsB;IACpC,OAAO,CAAC,YAAY,CAAa;IACjC,OAAO,CAAC,MAAM,CAAgB;gBAElB,YAAY,EAAE,UAAU,EAAE,MAAM,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC;IAgBrE,MAAM,CAAC,GAAG,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAyCzD,WAAW,CAAC,IAAI,EAAE,iBAAiB,EAAE,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;YAa3D,OAAO;IAgFrB,OAAO,CAAC,eAAe;IAqBvB,OAAO,CAAC,UAAU;IAgBlB,OAAO,CAAC,eAAe;IAQvB,OAAO,CAAC,aAAa;IAQrB,OAAO,CAAC,KAAK;CAGd"}
@@ -0,0 +1,204 @@
1
+ import { uuid } from '../utils/uuid.js';
2
+ const DEFAULT_CONFIG = {
3
+ provider: 'zai',
4
+ model: 'glm-5',
5
+ maxTokens: 1000,
6
+ temperature: 0.3,
7
+ timeout: 30000,
8
+ maxCardsPerObservation: 5,
9
+ minConfidence: 0.5
10
+ };
11
+ const EXTRACTION_PROMPT = `You are a memory extraction system. Analyze the given text and extract structured memories.
12
+
13
+ For each memory, identify:
14
+ 1. kind: fact, preference, profile, goal, or other
15
+ 2. entity: who/what this is about (e.g., "user", "project")
16
+ 3. slot: the attribute name (e.g., "employer", "framework", "response_style")
17
+ 4. value: the extracted value
18
+ 5. confidence: how certain you are (0.0 to 1.0)
19
+
20
+ Rules:
21
+ - Only extract clear, factual information
22
+ - Do not hallucinate or infer beyond what is stated
23
+ - If uncertain, set confidence lower
24
+ - Maximum 5 memories per text
25
+
26
+ Respond with valid JSON only.`;
27
+ export class LLMExtractor {
28
+ config;
29
+ globalConfig;
30
+ apiKey;
31
+ constructor(globalConfig, config) {
32
+ this.globalConfig = globalConfig;
33
+ this.config = { ...DEFAULT_CONFIG, ...config };
34
+ // Get API key from: config > environment variable
35
+ this.apiKey = this.config.apiKey
36
+ || globalConfig.extraction.llmApiKey
37
+ || (config?.provider === 'openai'
38
+ ? process.env.OPENAI_API_KEY || null
39
+ : process.env.ZAI_API_KEY || process.env.OPENAI_API_KEY || null);
40
+ if (!this.apiKey && globalConfig.extraction.llmEnabled) {
41
+ console.warn('[vmem] LLM extraction enabled but no API key configured. Set ZAI_API_KEY or OPENAI_API_KEY environment variable.');
42
+ }
43
+ }
44
+ async enrich(ctx) {
45
+ const startTime = Date.now();
46
+ if (!this.globalConfig.extraction.llmEnabled) {
47
+ return {
48
+ cards: [],
49
+ matchedRules: [],
50
+ skippedRules: [],
51
+ errors: [{ rule: 'llm', error: 'LLM extraction disabled' }],
52
+ source: 'llm',
53
+ processingTimeMs: Date.now() - startTime
54
+ };
55
+ }
56
+ try {
57
+ const response = await this.callLLM(ctx);
58
+ const cards = this.buildCards(response.memories, ctx);
59
+ return {
60
+ cards,
61
+ matchedRules: ['llm'],
62
+ skippedRules: [],
63
+ errors: [],
64
+ source: 'llm',
65
+ processingTimeMs: response.latencyMs
66
+ };
67
+ }
68
+ catch (error) {
69
+ return {
70
+ cards: [],
71
+ matchedRules: [],
72
+ skippedRules: [],
73
+ errors: [{
74
+ rule: 'llm',
75
+ error: error instanceof Error ? error.message : String(error)
76
+ }],
77
+ source: 'llm',
78
+ processingTimeMs: Date.now() - startTime
79
+ };
80
+ }
81
+ }
82
+ async enrichBatch(ctxs) {
83
+ const results = [];
84
+ for (const ctx of ctxs) {
85
+ const result = await this.enrich(ctx);
86
+ results.push(result);
87
+ await this.delay(100);
88
+ }
89
+ return results;
90
+ }
91
+ async callLLM(ctx) {
92
+ const startTime = Date.now();
93
+ if (!this.apiKey) {
94
+ throw new Error('LLM API key not configured. Set ZAI_API_KEY or OPENAI_API_KEY environment variable.');
95
+ }
96
+ const text = this.buildPromptText(ctx);
97
+ const requestBody = {
98
+ model: this.config.model,
99
+ messages: [
100
+ { role: 'system', content: EXTRACTION_PROMPT },
101
+ { role: 'user', content: text }
102
+ ],
103
+ max_tokens: this.config.maxTokens,
104
+ temperature: this.config.temperature,
105
+ response_format: { type: 'json_object' }
106
+ };
107
+ const controller = new AbortController();
108
+ const timeoutId = setTimeout(() => controller.abort(), this.config.timeout);
109
+ try {
110
+ const baseUrl = this.config.provider === 'zai'
111
+ ? 'https://api.z.ai/v1'
112
+ : 'https://api.openai.com/v1';
113
+ const response = await fetch(`${baseUrl}/chat/completions`, {
114
+ method: 'POST',
115
+ headers: {
116
+ 'Content-Type': 'application/json',
117
+ 'Authorization': `Bearer ${this.apiKey}`
118
+ },
119
+ body: JSON.stringify(requestBody),
120
+ signal: controller.signal
121
+ });
122
+ clearTimeout(timeoutId);
123
+ if (!response.ok) {
124
+ throw new Error(`LLM API error: ${response.status} ${response.statusText}`);
125
+ }
126
+ const data = await response.json();
127
+ const content = data.choices?.[0]?.message?.content || '{}';
128
+ let memories;
129
+ try {
130
+ const parsed = JSON.parse(content);
131
+ memories = Array.isArray(parsed.memories) ? parsed.memories : [];
132
+ }
133
+ catch {
134
+ memories = [];
135
+ }
136
+ memories = memories
137
+ .filter(m => m && typeof m.confidence === 'number' && m.confidence >= this.config.minConfidence)
138
+ .slice(0, this.config.maxCardsPerObservation);
139
+ return {
140
+ memories,
141
+ usage: data.usage ? {
142
+ promptTokens: data.usage.prompt_tokens || 0,
143
+ completionTokens: data.usage.completion_tokens || 0
144
+ } : undefined,
145
+ latencyMs: Date.now() - startTime
146
+ };
147
+ }
148
+ catch (error) {
149
+ clearTimeout(timeoutId);
150
+ if (error instanceof Error && error.name === 'AbortError') {
151
+ throw new Error('LLM request timed out');
152
+ }
153
+ throw error;
154
+ }
155
+ }
156
+ buildPromptText(ctx) {
157
+ const parts = [];
158
+ if (ctx.title) {
159
+ parts.push(`Title: ${ctx.title}`);
160
+ }
161
+ parts.push(`Content: ${ctx.text}`);
162
+ if (ctx.output) {
163
+ parts.push(`Output: ${ctx.output.slice(0, 500)}`);
164
+ }
165
+ if (ctx.input) {
166
+ const inputStr = JSON.stringify(ctx.input).slice(0, 200);
167
+ parts.push(`Input: ${inputStr}`);
168
+ }
169
+ return parts.join('\n\n');
170
+ }
171
+ buildCards(memories, ctx) {
172
+ return memories.map(m => ({
173
+ id: uuid(),
174
+ frameId: ctx.frameId,
175
+ entity: this.normalizeEntity(m.entity),
176
+ slot: this.normalizeSlot(m.slot),
177
+ value: m.value.trim(),
178
+ kind: m.kind,
179
+ confidence: m.confidence,
180
+ sourceObservationId: ctx.frameId,
181
+ createdAt: Date.now(),
182
+ updatedAt: Date.now(),
183
+ metadata: m.reasoning ? { reasoning: m.reasoning } : undefined
184
+ }));
185
+ }
186
+ normalizeEntity(entity) {
187
+ return entity
188
+ .toLowerCase()
189
+ .replace(/[^a-z0-9._]/g, '_')
190
+ .replace(/_+/g, '_')
191
+ .replace(/^_|_$/g, '');
192
+ }
193
+ normalizeSlot(slot) {
194
+ return slot
195
+ .toLowerCase()
196
+ .replace(/[^a-z0-9_]/g, '_')
197
+ .replace(/_+/g, '_')
198
+ .replace(/^_|_$/g, '');
199
+ }
200
+ delay(ms) {
201
+ return new Promise(resolve => setTimeout(resolve, ms));
202
+ }
203
+ }
204
+ //# sourceMappingURL=llm-extractor.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"llm-extractor.js","sourceRoot":"","sources":["../../src/extract/llm-extractor.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAgCxC,MAAM,cAAc,GAAwB;IAC1C,QAAQ,EAAE,KAAK;IACf,KAAK,EAAE,OAAO;IACd,SAAS,EAAE,IAAI;IACf,WAAW,EAAE,GAAG;IAChB,OAAO,EAAE,KAAK;IACd,sBAAsB,EAAE,CAAC;IACzB,aAAa,EAAE,GAAG;CACnB,CAAC;AAEF,MAAM,iBAAiB,GAAG;;;;;;;;;;;;;;;8BAeI,CAAC;AAE/B,MAAM,OAAO,YAAY;IACf,MAAM,CAAsB;IAC5B,YAAY,CAAa;IACzB,MAAM,CAAgB;IAE9B,YAAY,YAAwB,EAAE,MAAqC;QACzE,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAC;QAE/C,kDAAkD;QAClD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM;eAC3B,YAAY,CAAC,UAAU,CAAC,SAAS;eACjC,CAAC,MAAM,EAAE,QAAQ,KAAK,QAAQ;gBAC/B,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,IAAI;gBACpC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,IAAI,CAAC,CAAC;QAErE,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,YAAY,CAAC,UAAU,CAAC,UAAU,EAAE,CAAC;YACvD,OAAO,CAAC,IAAI,CAAC,kHAAkH,CAAC,CAAC;QACnI,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,GAAsB;QACjC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,UAAU,EAAE,CAAC;YAC7C,OAAO;gBACL,KAAK,EAAE,EAAE;gBACT,YAAY,EAAE,EAAE;gBAChB,YAAY,EAAE,EAAE;gBAChB,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,yBAAyB,EAAE,CAAC;gBAC3D,MAAM,EAAE,KAAK;gBACb,gBAAgB,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aACzC,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACzC,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;YAEtD,OAAO;gBACL,KAAK;gBACL,YAAY,EAAE,CAAC,KAAK,CAAC;gBACrB,YAAY,EAAE,EAAE;gBAChB,MAAM,EAAE,EAAE;gBACV,MAAM,EAAE,KAAK;gBACb,gBAAgB,EAAE,QAAQ,CAAC,SAAS;aACrC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,KAAK,EAAE,EAAE;gBACT,YAAY,EAAE,EAAE;gBAChB,YAAY,EAAE,EAAE;gBAChB,MAAM,EAAE,CAAC;wBACP,IAAI,EAAE,KAAK;wBACX,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;qBAC9D,CAAC;gBACF,MAAM,EAAE,KAAK;gBACb,gBAAgB,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aACzC,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,IAAyB;QACzC,MAAM,OAAO,GAAuB,EAAE,CAAC;QAEvC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACtC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAErB,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACxB,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,KAAK,CAAC,OAAO,CAAC,GAAsB;QAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,qFAAqF,CAAC,CAAC;QACzG,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;QAEvC,MAAM,WAAW,GAAG;YAClB,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;YACxB,QAAQ,EAAE;gBACR,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,iBAAiB,EAAE;gBAC9C,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE;aAChC;YACD,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;YACjC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;YACpC,eAAe,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE;SACzC,CAAC;QAEF,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAE5E,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,KAAK,KAAK;gBAC5C,CAAC,CAAC,qBAAqB;gBACvB,CAAC,CAAC,2BAA2B,CAAC;YAEhC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,mBAAmB,EAAE;gBAC1D,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;iBACzC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;gBACjC,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;YAEH,YAAY,CAAC,SAAS,CAAC,CAAC;YAExB,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,kBAAkB,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;YAC9E,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAG/B,CAAC;YACF,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,IAAI,CAAC;YAE5D,IAAI,QAA2B,CAAC;YAChC,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAqC,CAAC;gBACvE,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;YACnE,CAAC;YAAC,MAAM,CAAC;gBACP,QAAQ,GAAG,EAAE,CAAC;YAChB,CAAC;YAED,QAAQ,GAAG,QAAQ;iBAChB,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,OAAO,CAAC,CAAC,UAAU,KAAK,QAAQ,IAAI,CAAC,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC;iBAC/F,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC;YAEhD,OAAO;gBACL,QAAQ;gBACR,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;oBAClB,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC;oBAC3C,gBAAgB,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB,IAAI,CAAC;iBACpD,CAAC,CAAC,CAAC,SAAS;gBACb,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aAClC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,YAAY,CAAC,SAAS,CAAC,CAAC;YAExB,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAC1D,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAC3C,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEO,eAAe,CAAC,GAAsB;QAC5C,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;YACd,KAAK,CAAC,IAAI,CAAC,UAAU,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC;QACpC,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,YAAY,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;QAEnC,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;YACf,KAAK,CAAC,IAAI,CAAC,WAAW,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;QACpD,CAAC;QAED,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;YACd,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YACzD,KAAK,CAAC,IAAI,CAAC,UAAU,QAAQ,EAAE,CAAC,CAAC;QACnC,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC5B,CAAC;IAEO,UAAU,CAAC,QAA2B,EAAE,GAAsB;QACpE,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACxB,EAAE,EAAE,IAAI,EAAE;YACV,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,MAAM,CAAC;YACtC,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC;YAChC,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,IAAI,EAAE;YACrB,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,UAAU,EAAE,CAAC,CAAC,UAAU;YACxB,mBAAmB,EAAE,GAAG,CAAC,OAAO;YAChC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,QAAQ,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,SAAS;SAC/D,CAAC,CAAC,CAAC;IACN,CAAC;IAEO,eAAe,CAAC,MAAc;QACpC,OAAO,MAAM;aACV,WAAW,EAAE;aACb,OAAO,CAAC,cAAc,EAAE,GAAG,CAAC;aAC5B,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC;aACnB,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;IAC3B,CAAC;IAEO,aAAa,CAAC,IAAY;QAChC,OAAO,IAAI;aACR,WAAW,EAAE;aACb,OAAO,CAAC,aAAa,EAAE,GAAG,CAAC;aAC3B,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC;aACnB,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;IAC3B,CAAC;IAEO,KAAK,CAAC,EAAU;QACtB,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;IACzD,CAAC;CACF"}
@@ -0,0 +1,44 @@
1
+ import { EnrichmentContext, MemoryCard, VmemConfig } from '../types.js';
2
+ import { RulesEngine } from './rules-engine.js';
3
+ import { ConfidenceScorer, ConfidenceScore } from './confidence.js';
4
+ export interface PipelineConfig {
5
+ rulesFirst: boolean;
6
+ llmFallback: boolean;
7
+ minRulesConfidence: number;
8
+ batchSize: number;
9
+ }
10
+ export interface PipelineResult {
11
+ cards: MemoryCard[];
12
+ scores: Map<string, ConfidenceScore>;
13
+ source: 'rules' | 'llm' | 'hybrid' | 'none';
14
+ stats: PipelineStats;
15
+ }
16
+ export interface PipelineStats {
17
+ totalObservations: number;
18
+ rulesMatches: number;
19
+ llmCalls: number;
20
+ cardsExtracted: number;
21
+ cardsAccepted: number;
22
+ cardsFlagged: number;
23
+ cardsTombstoned: number;
24
+ processingTimeMs: number;
25
+ }
26
+ export declare class ExtractionPipeline {
27
+ private rulesEngine;
28
+ private llmExtractor;
29
+ private confidenceScorer;
30
+ private config;
31
+ private globalConfig;
32
+ constructor(globalConfig: VmemConfig, config?: Partial<PipelineConfig>);
33
+ extract(ctx: EnrichmentContext): Promise<PipelineResult>;
34
+ extractBatch(ctxs: EnrichmentContext[]): Promise<PipelineResult[]>;
35
+ private scoreCards;
36
+ private applyActions;
37
+ private mergeResults;
38
+ private delay;
39
+ getRulesEngine(): RulesEngine;
40
+ getConfidenceScorer(): ConfidenceScorer;
41
+ loadExistingCards(cards: MemoryCard[]): void;
42
+ }
43
+ export declare function createPipeline(globalConfig: VmemConfig, config?: Partial<PipelineConfig>): ExtractionPipeline;
44
+ //# sourceMappingURL=pipeline.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"pipeline.d.ts","sourceRoot":"","sources":["../../src/extract/pipeline.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACxE,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAEhD,OAAO,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAEpE,MAAM,WAAW,cAAc;IAC7B,UAAU,EAAE,OAAO,CAAC;IACpB,WAAW,EAAE,OAAO,CAAC;IACrB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,UAAU,EAAE,CAAC;IACpB,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;IACrC,MAAM,EAAE,OAAO,GAAG,KAAK,GAAG,QAAQ,GAAG,MAAM,CAAC;IAC5C,KAAK,EAAE,aAAa,CAAC;CACtB;AAED,MAAM,WAAW,aAAa;IAC5B,iBAAiB,EAAE,MAAM,CAAC;IAC1B,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,cAAc,EAAE,MAAM,CAAC;IACvB,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,eAAe,EAAE,MAAM,CAAC;IACxB,gBAAgB,EAAE,MAAM,CAAC;CAC1B;AASD,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,gBAAgB,CAAmB;IAC3C,OAAO,CAAC,MAAM,CAAiB;IAC/B,OAAO,CAAC,YAAY,CAAa;gBAErB,YAAY,EAAE,UAAU,EAAE,MAAM,CAAC,EAAE,OAAO,CAAC,cAAc,CAAC;IAchE,OAAO,CAAC,GAAG,EAAE,iBAAiB,GAAG,OAAO,CAAC,cAAc,CAAC;IAgExD,YAAY,CAAC,IAAI,EAAE,iBAAiB,EAAE,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;YAoB1D,UAAU;IAiBxB,OAAO,CAAC,YAAY;IA+CpB,OAAO,CAAC,YAAY;IAsBpB,OAAO,CAAC,KAAK;IAIb,cAAc,IAAI,WAAW;IAI7B,mBAAmB,IAAI,gBAAgB;IAIvC,iBAAiB,CAAC,KAAK,EAAE,UAAU,EAAE,GAAG,IAAI;CAG7C;AAED,wBAAgB,cAAc,CAAC,YAAY,EAAE,UAAU,EAAE,MAAM,CAAC,EAAE,OAAO,CAAC,cAAc,CAAC,GAAG,kBAAkB,CAE7G"}