@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.
- package/CHANGELOG.md +83 -0
- package/LICENSE +21 -0
- package/README.md +172 -0
- package/dist/capture/classification.d.ts +37 -0
- package/dist/capture/classification.d.ts.map +1 -0
- package/dist/capture/classification.js +181 -0
- package/dist/capture/classification.js.map +1 -0
- package/dist/capture/filtering.d.ts +63 -0
- package/dist/capture/filtering.d.ts.map +1 -0
- package/dist/capture/filtering.js +195 -0
- package/dist/capture/filtering.js.map +1 -0
- package/dist/capture/observation.d.ts +28 -0
- package/dist/capture/observation.d.ts.map +1 -0
- package/dist/capture/observation.js +188 -0
- package/dist/capture/observation.js.map +1 -0
- package/dist/commands/index.d.ts +38 -0
- package/dist/commands/index.d.ts.map +1 -0
- package/dist/commands/index.js +375 -0
- package/dist/commands/index.js.map +1 -0
- package/dist/config.d.ts +5 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +41 -0
- package/dist/config.js.map +1 -0
- package/dist/context/builder.d.ts +19 -0
- package/dist/context/builder.d.ts.map +1 -0
- package/dist/context/builder.js +209 -0
- package/dist/context/builder.js.map +1 -0
- package/dist/context/format.d.ts +32 -0
- package/dist/context/format.d.ts.map +1 -0
- package/dist/context/format.js +211 -0
- package/dist/context/format.js.map +1 -0
- package/dist/embedding/index.d.ts +3 -0
- package/dist/embedding/index.d.ts.map +1 -0
- package/dist/embedding/index.js +3 -0
- package/dist/embedding/index.js.map +1 -0
- package/dist/embedding/provider.d.ts +15 -0
- package/dist/embedding/provider.d.ts.map +1 -0
- package/dist/embedding/provider.js +19 -0
- package/dist/embedding/provider.js.map +1 -0
- package/dist/embedding/transformers.d.ts +14 -0
- package/dist/embedding/transformers.d.ts.map +1 -0
- package/dist/embedding/transformers.js +55 -0
- package/dist/embedding/transformers.js.map +1 -0
- package/dist/extract/confidence.d.ts +48 -0
- package/dist/extract/confidence.d.ts.map +1 -0
- package/dist/extract/confidence.js +230 -0
- package/dist/extract/confidence.js.map +1 -0
- package/dist/extract/llm-extractor.d.ts +35 -0
- package/dist/extract/llm-extractor.d.ts.map +1 -0
- package/dist/extract/llm-extractor.js +204 -0
- package/dist/extract/llm-extractor.js.map +1 -0
- package/dist/extract/pipeline.d.ts +44 -0
- package/dist/extract/pipeline.d.ts.map +1 -0
- package/dist/extract/pipeline.js +171 -0
- package/dist/extract/pipeline.js.map +1 -0
- package/dist/extract/rules-engine.d.ts +14 -0
- package/dist/extract/rules-engine.d.ts.map +1 -0
- package/dist/extract/rules-engine.js +314 -0
- package/dist/extract/rules-engine.js.map +1 -0
- package/dist/extract/summarizer.d.ts +27 -0
- package/dist/extract/summarizer.d.ts.map +1 -0
- package/dist/extract/summarizer.js +148 -0
- package/dist/extract/summarizer.js.map +1 -0
- package/dist/hooks/session-idle.d.ts +38 -0
- package/dist/hooks/session-idle.d.ts.map +1 -0
- package/dist/hooks/session-idle.js +142 -0
- package/dist/hooks/session-idle.js.map +1 -0
- package/dist/hooks/system-transform.d.ts +41 -0
- package/dist/hooks/system-transform.d.ts.map +1 -0
- package/dist/hooks/system-transform.js +98 -0
- package/dist/hooks/system-transform.js.map +1 -0
- package/dist/hooks/tool-after.d.ts +66 -0
- package/dist/hooks/tool-after.d.ts.map +1 -0
- package/dist/hooks/tool-after.js +155 -0
- package/dist/hooks/tool-after.js.map +1 -0
- package/dist/index.d.ts +10 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +10 -0
- package/dist/index.js.map +1 -0
- package/dist/mind.d.ts +69 -0
- package/dist/mind.d.ts.map +1 -0
- package/dist/mind.js +368 -0
- package/dist/mind.js.map +1 -0
- package/dist/procedural/index.d.ts +4 -0
- package/dist/procedural/index.d.ts.map +1 -0
- package/dist/procedural/index.js +4 -0
- package/dist/procedural/index.js.map +1 -0
- package/dist/procedural/induction.d.ts +23 -0
- package/dist/procedural/induction.d.ts.map +1 -0
- package/dist/procedural/induction.js +104 -0
- package/dist/procedural/induction.js.map +1 -0
- package/dist/procedural/miner.d.ts +45 -0
- package/dist/procedural/miner.d.ts.map +1 -0
- package/dist/procedural/miner.js +171 -0
- package/dist/procedural/miner.js.map +1 -0
- package/dist/procedural/suggester.d.ts +20 -0
- package/dist/procedural/suggester.d.ts.map +1 -0
- package/dist/procedural/suggester.js +69 -0
- package/dist/procedural/suggester.js.map +1 -0
- package/dist/store/archive.d.ts +31 -0
- package/dist/store/archive.d.ts.map +1 -0
- package/dist/store/archive.js +137 -0
- package/dist/store/archive.js.map +1 -0
- package/dist/store/backup.d.ts +36 -0
- package/dist/store/backup.d.ts.map +1 -0
- package/dist/store/backup.js +163 -0
- package/dist/store/backup.js.map +1 -0
- package/dist/store/memorycards.d.ts +31 -0
- package/dist/store/memorycards.d.ts.map +1 -0
- package/dist/store/memorycards.js +161 -0
- package/dist/store/memorycards.js.map +1 -0
- package/dist/store/mv2-adapter.d.ts +29 -0
- package/dist/store/mv2-adapter.d.ts.map +1 -0
- package/dist/store/mv2-adapter.js +359 -0
- package/dist/store/mv2-adapter.js.map +1 -0
- package/dist/store/procedures.d.ts +79 -0
- package/dist/store/procedures.d.ts.map +1 -0
- package/dist/store/procedures.js +183 -0
- package/dist/store/procedures.js.map +1 -0
- package/dist/store/tombstone.d.ts +27 -0
- package/dist/store/tombstone.d.ts.map +1 -0
- package/dist/store/tombstone.js +136 -0
- package/dist/store/tombstone.js.map +1 -0
- package/dist/tools/index.d.ts +36 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +236 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/types.d.ts +301 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +69 -0
- package/dist/types.js.map +1 -0
- package/dist/utils/error.d.ts +57 -0
- package/dist/utils/error.d.ts.map +1 -0
- package/dist/utils/error.js +143 -0
- package/dist/utils/error.js.map +1 -0
- package/dist/utils/uuid.d.ts +2 -0
- package/dist/utils/uuid.d.ts.map +1 -0
- package/dist/utils/uuid.js +5 -0
- package/dist/utils/uuid.js.map +1 -0
- 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"}
|