@chendpoc/pi-memory 0.1.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/README.md +180 -0
- package/dist/adapters/piComplete.d.ts +17 -0
- package/dist/adapters/piComplete.d.ts.map +1 -0
- package/dist/adapters/piComplete.js +169 -0
- package/dist/adapters/piComplete.js.map +1 -0
- package/dist/bundle/install.d.ts +34 -0
- package/dist/bundle/install.d.ts.map +1 -0
- package/dist/bundle/install.js +183 -0
- package/dist/bundle/install.js.map +1 -0
- package/dist/cli.d.ts +3 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +245 -0
- package/dist/cli.js.map +1 -0
- package/dist/config.d.ts +27 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +49 -0
- package/dist/config.js.map +1 -0
- package/dist/errclass.d.ts +7 -0
- package/dist/errclass.d.ts.map +1 -0
- package/dist/errclass.js +32 -0
- package/dist/errclass.js.map +1 -0
- package/dist/extension.d.ts +24 -0
- package/dist/extension.d.ts.map +1 -0
- package/dist/extension.js +7 -0
- package/dist/extension.js.map +1 -0
- package/dist/fallback/index.d.ts +11 -0
- package/dist/fallback/index.d.ts.map +1 -0
- package/dist/fallback/index.js +16 -0
- package/dist/fallback/index.js.map +1 -0
- package/dist/fallback/llmRerank.d.ts +19 -0
- package/dist/fallback/llmRerank.d.ts.map +1 -0
- package/dist/fallback/llmRerank.js +60 -0
- package/dist/fallback/llmRerank.js.map +1 -0
- package/dist/fallback/memoryMd.d.ts +6 -0
- package/dist/fallback/memoryMd.d.ts.map +1 -0
- package/dist/fallback/memoryMd.js +35 -0
- package/dist/fallback/memoryMd.js.map +1 -0
- package/dist/fallback/sessionIndex.d.ts +35 -0
- package/dist/fallback/sessionIndex.d.ts.map +1 -0
- package/dist/fallback/sessionIndex.js +222 -0
- package/dist/fallback/sessionIndex.js.map +1 -0
- package/dist/fallback/sessionSearch.d.ts +18 -0
- package/dist/fallback/sessionSearch.d.ts.map +1 -0
- package/dist/fallback/sessionSearch.js +161 -0
- package/dist/fallback/sessionSearch.js.map +1 -0
- package/dist/index.d.ts +25 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +24 -0
- package/dist/index.js.map +1 -0
- package/dist/paths.d.ts +7 -0
- package/dist/paths.d.ts.map +1 -0
- package/dist/paths.js +26 -0
- package/dist/paths.js.map +1 -0
- package/dist/pi-extension.d.ts +6 -0
- package/dist/pi-extension.d.ts.map +1 -0
- package/dist/pi-extension.js +224 -0
- package/dist/pi-extension.js.map +1 -0
- package/dist/preflight/detectIntents.d.ts +102 -0
- package/dist/preflight/detectIntents.d.ts.map +1 -0
- package/dist/preflight/detectIntents.js +624 -0
- package/dist/preflight/detectIntents.js.map +1 -0
- package/dist/preflight/hook.d.ts +58 -0
- package/dist/preflight/hook.d.ts.map +1 -0
- package/dist/preflight/hook.js +77 -0
- package/dist/preflight/hook.js.map +1 -0
- package/dist/preflight/render.d.ts +21 -0
- package/dist/preflight/render.d.ts.map +1 -0
- package/dist/preflight/render.js +132 -0
- package/dist/preflight/render.js.map +1 -0
- package/dist/preflight/strip.d.ts +11 -0
- package/dist/preflight/strip.d.ts.map +1 -0
- package/dist/preflight/strip.js +46 -0
- package/dist/preflight/strip.js.map +1 -0
- package/dist/service.d.ts +56 -0
- package/dist/service.d.ts.map +1 -0
- package/dist/service.js +158 -0
- package/dist/service.js.map +1 -0
- package/dist/sidecar/bundle.d.ts +19 -0
- package/dist/sidecar/bundle.d.ts.map +1 -0
- package/dist/sidecar/bundle.js +39 -0
- package/dist/sidecar/bundle.js.map +1 -0
- package/dist/sidecar/client.d.ts +17 -0
- package/dist/sidecar/client.d.ts.map +1 -0
- package/dist/sidecar/client.js +107 -0
- package/dist/sidecar/client.js.map +1 -0
- package/dist/sidecar/process.d.ts +14 -0
- package/dist/sidecar/process.d.ts.map +1 -0
- package/dist/sidecar/process.js +126 -0
- package/dist/sidecar/process.js.map +1 -0
- package/dist/tools/memoryAppend.d.ts +37 -0
- package/dist/tools/memoryAppend.d.ts.map +1 -0
- package/dist/tools/memoryAppend.js +99 -0
- package/dist/tools/memoryAppend.js.map +1 -0
- package/dist/tools/memoryRecall.d.ts +113 -0
- package/dist/tools/memoryRecall.d.ts.map +1 -0
- package/dist/tools/memoryRecall.js +325 -0
- package/dist/tools/memoryRecall.js.map +1 -0
- package/dist/trainer/bundleBuilder.d.ts +30 -0
- package/dist/trainer/bundleBuilder.d.ts.map +1 -0
- package/dist/trainer/bundleBuilder.js +106 -0
- package/dist/trainer/bundleBuilder.js.map +1 -0
- package/dist/trainer/bundleLoader.d.ts +12 -0
- package/dist/trainer/bundleLoader.d.ts.map +1 -0
- package/dist/trainer/bundleLoader.js +59 -0
- package/dist/trainer/bundleLoader.js.map +1 -0
- package/dist/trainer/deltaMerge.d.ts +38 -0
- package/dist/trainer/deltaMerge.d.ts.map +1 -0
- package/dist/trainer/deltaMerge.js +183 -0
- package/dist/trainer/deltaMerge.js.map +1 -0
- package/dist/trainer/entityResolver.d.ts +27 -0
- package/dist/trainer/entityResolver.d.ts.map +1 -0
- package/dist/trainer/entityResolver.js +92 -0
- package/dist/trainer/entityResolver.js.map +1 -0
- package/dist/trainer/extractFacts.d.ts +67 -0
- package/dist/trainer/extractFacts.d.ts.map +1 -0
- package/dist/trainer/extractFacts.js +213 -0
- package/dist/trainer/extractFacts.js.map +1 -0
- package/dist/trainer/index.d.ts +54 -0
- package/dist/trainer/index.d.ts.map +1 -0
- package/dist/trainer/index.js +82 -0
- package/dist/trainer/index.js.map +1 -0
- package/dist/trainer/llmExtractor.d.ts +16 -0
- package/dist/trainer/llmExtractor.d.ts.map +1 -0
- package/dist/trainer/llmExtractor.js +146 -0
- package/dist/trainer/llmExtractor.js.map +1 -0
- package/dist/trainer/marker.d.ts +10 -0
- package/dist/trainer/marker.d.ts.map +1 -0
- package/dist/trainer/marker.js +28 -0
- package/dist/trainer/marker.js.map +1 -0
- package/dist/trainer/scheduler.d.ts +31 -0
- package/dist/trainer/scheduler.d.ts.map +1 -0
- package/dist/trainer/scheduler.js +72 -0
- package/dist/trainer/scheduler.js.map +1 -0
- package/dist/trainer/sessionLoader.d.ts +23 -0
- package/dist/trainer/sessionLoader.d.ts.map +1 -0
- package/dist/trainer/sessionLoader.js +106 -0
- package/dist/trainer/sessionLoader.js.map +1 -0
- package/dist/types.d.ts +135 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +8 -0
- package/dist/types.js.map +1 -0
- package/package.json +78 -0
- package/src/adapters/piComplete.ts +233 -0
- package/src/bundle/install.ts +206 -0
- package/src/cli.ts +254 -0
- package/src/config.ts +92 -0
- package/src/errclass.ts +37 -0
- package/src/extension.ts +23 -0
- package/src/fallback/index.ts +24 -0
- package/src/fallback/llmRerank.ts +90 -0
- package/src/fallback/memoryMd.ts +36 -0
- package/src/fallback/sessionIndex.ts +289 -0
- package/src/fallback/sessionSearch.ts +181 -0
- package/src/index.ts +213 -0
- package/src/paths.ts +28 -0
- package/src/pi-extension.ts +276 -0
- package/src/preflight/detectIntents.ts +654 -0
- package/src/preflight/hook.ts +136 -0
- package/src/preflight/render.ts +185 -0
- package/src/preflight/strip.ts +50 -0
- package/src/service.ts +202 -0
- package/src/sidecar/bundle.ts +52 -0
- package/src/sidecar/client.ts +166 -0
- package/src/sidecar/process.ts +145 -0
- package/src/tools/memoryAppend.ts +113 -0
- package/src/tools/memoryRecall.ts +364 -0
- package/src/trainer/bundleBuilder.ts +192 -0
- package/src/trainer/bundleLoader.ts +105 -0
- package/src/trainer/deltaMerge.ts +221 -0
- package/src/trainer/entityResolver.ts +140 -0
- package/src/trainer/extractFacts.ts +312 -0
- package/src/trainer/index.ts +147 -0
- package/src/trainer/llmExtractor.ts +206 -0
- package/src/trainer/marker.ts +30 -0
- package/src/trainer/scheduler.ts +104 -0
- package/src/trainer/sessionLoader.ts +139 -0
- package/src/types.ts +168 -0
|
@@ -0,0 +1,183 @@
|
|
|
1
|
+
import { UPDATE_SIGNAL_RE } from "./extractFacts.js";
|
|
2
|
+
function edgeKey(headId, relation, tailId) {
|
|
3
|
+
return `${headId}\0${relation}\0${tailId}`;
|
|
4
|
+
}
|
|
5
|
+
function headRelKey(headId, relation) {
|
|
6
|
+
return `${headId}\0${relation}`;
|
|
7
|
+
}
|
|
8
|
+
/**
|
|
9
|
+
* Merge new extracted graph + events into an existing bundle graph,
|
|
10
|
+
* producing a combined result with delta operation log.
|
|
11
|
+
*
|
|
12
|
+
* Operations:
|
|
13
|
+
* - SKIP: entity+relation+tail already present with same value
|
|
14
|
+
* - ADD: genuinely new entity/edge/event
|
|
15
|
+
* - UPDATE: same head+relation but different tail, with update signal in evidence
|
|
16
|
+
* - DELETE: negated relation matches existing edge
|
|
17
|
+
*/
|
|
18
|
+
export function deltaMerge(existing, incoming) {
|
|
19
|
+
const delta = { entries: [], added: 0, updated: 0, deleted: 0, skipped: 0 };
|
|
20
|
+
function log(op, kind, detail) {
|
|
21
|
+
delta.entries.push({ op, kind, detail });
|
|
22
|
+
switch (op) {
|
|
23
|
+
case "add":
|
|
24
|
+
delta.added++;
|
|
25
|
+
break;
|
|
26
|
+
case "update":
|
|
27
|
+
delta.updated++;
|
|
28
|
+
break;
|
|
29
|
+
case "delete":
|
|
30
|
+
delta.deleted++;
|
|
31
|
+
break;
|
|
32
|
+
case "skip":
|
|
33
|
+
delta.skipped++;
|
|
34
|
+
break;
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
// ── Entities ──
|
|
38
|
+
const entityMap = new Map();
|
|
39
|
+
for (const ent of existing.graph.entities) {
|
|
40
|
+
entityMap.set(ent.id, { ...ent, mentions: [...ent.mentions] });
|
|
41
|
+
}
|
|
42
|
+
for (const ent of incoming.graph.entities) {
|
|
43
|
+
const prev = entityMap.get(ent.id);
|
|
44
|
+
if (prev) {
|
|
45
|
+
const newMentions = ent.mentions.filter((m) => !prev.mentions.some((pm) => pm.sessionId === m.sessionId && pm.turnIndex === m.turnIndex));
|
|
46
|
+
if (newMentions.length > 0) {
|
|
47
|
+
prev.mentions.push(...newMentions);
|
|
48
|
+
log("update", "entity", `${ent.canonicalName}: +${newMentions.length} mentions`);
|
|
49
|
+
}
|
|
50
|
+
else {
|
|
51
|
+
log("skip", "entity", ent.canonicalName);
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
else {
|
|
55
|
+
entityMap.set(ent.id, { ...ent, mentions: [...ent.mentions] });
|
|
56
|
+
log("add", "entity", ent.canonicalName);
|
|
57
|
+
}
|
|
58
|
+
}
|
|
59
|
+
// ── Edges: collect negations and normal edges ──
|
|
60
|
+
const existingEdgeMap = new Map();
|
|
61
|
+
const existingByHeadRel = new Map();
|
|
62
|
+
for (const edge of existing.graph.relations) {
|
|
63
|
+
const key = edgeKey(edge.headEntityId, edge.relation, edge.tailEntityId);
|
|
64
|
+
existingEdgeMap.set(key, edge);
|
|
65
|
+
const hrKey = headRelKey(edge.headEntityId, edge.relation);
|
|
66
|
+
let arr = existingByHeadRel.get(hrKey);
|
|
67
|
+
if (!arr) {
|
|
68
|
+
arr = [];
|
|
69
|
+
existingByHeadRel.set(hrKey, arr);
|
|
70
|
+
}
|
|
71
|
+
arr.push(edge);
|
|
72
|
+
}
|
|
73
|
+
const negatedRelations = [];
|
|
74
|
+
const normalRelations = [];
|
|
75
|
+
for (const rel of incoming.graph.relations) {
|
|
76
|
+
if (rel.negated) {
|
|
77
|
+
negatedRelations.push(rel);
|
|
78
|
+
}
|
|
79
|
+
else {
|
|
80
|
+
normalRelations.push(rel);
|
|
81
|
+
}
|
|
82
|
+
}
|
|
83
|
+
// Process negations → DELETE from existing
|
|
84
|
+
const deletedKeys = new Set();
|
|
85
|
+
for (const neg of negatedRelations) {
|
|
86
|
+
if (neg.headEntityId) {
|
|
87
|
+
const key = edgeKey(neg.headEntityId, neg.relation, neg.tailEntityId);
|
|
88
|
+
if (existingEdgeMap.has(key)) {
|
|
89
|
+
deletedKeys.add(key);
|
|
90
|
+
const ent = entityMap.get(neg.tailEntityId);
|
|
91
|
+
log("delete", "edge", `${neg.headEntityId}:${neg.relation}:${ent?.canonicalName ?? neg.tailEntityId}`);
|
|
92
|
+
}
|
|
93
|
+
}
|
|
94
|
+
else {
|
|
95
|
+
for (const [key, edge] of existingEdgeMap) {
|
|
96
|
+
if (edge.relation === neg.relation && edge.tailEntityId === neg.tailEntityId) {
|
|
97
|
+
deletedKeys.add(key);
|
|
98
|
+
const headEnt = entityMap.get(edge.headEntityId);
|
|
99
|
+
const tailEnt = entityMap.get(edge.tailEntityId);
|
|
100
|
+
log("delete", "edge", `${headEnt?.canonicalName ?? edge.headEntityId}:${neg.relation}:${tailEnt?.canonicalName ?? edge.tailEntityId}`);
|
|
101
|
+
}
|
|
102
|
+
}
|
|
103
|
+
}
|
|
104
|
+
}
|
|
105
|
+
// Process normal edges → ADD / SKIP / UPDATE
|
|
106
|
+
const mergedEdges = [];
|
|
107
|
+
for (const [key, edge] of existingEdgeMap) {
|
|
108
|
+
if (!deletedKeys.has(key)) {
|
|
109
|
+
mergedEdges.push(edge);
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
for (const rel of normalRelations) {
|
|
113
|
+
const key = edgeKey(rel.headEntityId, rel.relation, rel.tailEntityId);
|
|
114
|
+
if (existingEdgeMap.has(key) && !deletedKeys.has(key)) {
|
|
115
|
+
const tailEnt = entityMap.get(rel.tailEntityId);
|
|
116
|
+
log("skip", "edge", `${rel.headEntityId}:${rel.relation}:${tailEnt?.canonicalName ?? rel.tailEntityId}`);
|
|
117
|
+
continue;
|
|
118
|
+
}
|
|
119
|
+
if (deletedKeys.has(key)) {
|
|
120
|
+
// Re-adding a previously-deleted edge is still an add
|
|
121
|
+
}
|
|
122
|
+
const hasUpdateSignal = UPDATE_SIGNAL_RE.test(rel.evidence);
|
|
123
|
+
if (hasUpdateSignal) {
|
|
124
|
+
const hrKey = headRelKey(rel.headEntityId, rel.relation);
|
|
125
|
+
const existingForHR = existingByHeadRel.get(hrKey);
|
|
126
|
+
if (existingForHR) {
|
|
127
|
+
for (const oldEdge of existingForHR) {
|
|
128
|
+
if (oldEdge.tailEntityId !== rel.tailEntityId) {
|
|
129
|
+
const oldKey = edgeKey(oldEdge.headEntityId, oldEdge.relation, oldEdge.tailEntityId);
|
|
130
|
+
if (!deletedKeys.has(oldKey)) {
|
|
131
|
+
deletedKeys.add(oldKey);
|
|
132
|
+
const idx = mergedEdges.findIndex((e) => e.headEntityId === oldEdge.headEntityId &&
|
|
133
|
+
e.relation === oldEdge.relation &&
|
|
134
|
+
e.tailEntityId === oldEdge.tailEntityId);
|
|
135
|
+
if (idx !== -1)
|
|
136
|
+
mergedEdges.splice(idx, 1);
|
|
137
|
+
const oldTailEnt = entityMap.get(oldEdge.tailEntityId);
|
|
138
|
+
const newTailEnt = entityMap.get(rel.tailEntityId);
|
|
139
|
+
log("update", "edge", `${rel.relation}: ${oldTailEnt?.canonicalName ?? oldEdge.tailEntityId} → ${newTailEnt?.canonicalName ?? rel.tailEntityId}`);
|
|
140
|
+
}
|
|
141
|
+
}
|
|
142
|
+
}
|
|
143
|
+
}
|
|
144
|
+
}
|
|
145
|
+
if (!mergedEdges.some((e) => e.headEntityId === rel.headEntityId &&
|
|
146
|
+
e.relation === rel.relation &&
|
|
147
|
+
e.tailEntityId === rel.tailEntityId)) {
|
|
148
|
+
mergedEdges.push(rel);
|
|
149
|
+
if (!hasUpdateSignal || !existingByHeadRel.has(headRelKey(rel.headEntityId, rel.relation))) {
|
|
150
|
+
const tailEnt = entityMap.get(rel.tailEntityId);
|
|
151
|
+
log("add", "edge", `${rel.headEntityId}:${rel.relation}:${tailEnt?.canonicalName ?? rel.tailEntityId}`);
|
|
152
|
+
}
|
|
153
|
+
}
|
|
154
|
+
}
|
|
155
|
+
// ── Events ──
|
|
156
|
+
const eventSigSet = new Set();
|
|
157
|
+
const mergedEvents = [];
|
|
158
|
+
for (const ev of existing.events) {
|
|
159
|
+
const sig = `${ev.sessionId}\0${ev.turnIndex}\0${ev.description.slice(0, 80)}`;
|
|
160
|
+
eventSigSet.add(sig);
|
|
161
|
+
mergedEvents.push(ev);
|
|
162
|
+
}
|
|
163
|
+
for (const ev of incoming.events) {
|
|
164
|
+
const sig = `${ev.sessionId}\0${ev.turnIndex}\0${ev.description.slice(0, 80)}`;
|
|
165
|
+
if (eventSigSet.has(sig)) {
|
|
166
|
+
log("skip", "event", ev.description.slice(0, 60));
|
|
167
|
+
}
|
|
168
|
+
else {
|
|
169
|
+
eventSigSet.add(sig);
|
|
170
|
+
mergedEvents.push(ev);
|
|
171
|
+
log("add", "event", ev.description.slice(0, 60));
|
|
172
|
+
}
|
|
173
|
+
}
|
|
174
|
+
return {
|
|
175
|
+
graph: {
|
|
176
|
+
entities: [...entityMap.values()],
|
|
177
|
+
relations: mergedEdges,
|
|
178
|
+
},
|
|
179
|
+
events: mergedEvents,
|
|
180
|
+
delta,
|
|
181
|
+
};
|
|
182
|
+
}
|
|
183
|
+
//# sourceMappingURL=deltaMerge.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"deltaMerge.js","sourceRoot":"","sources":["../../src/trainer/deltaMerge.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAwBrD,SAAS,OAAO,CAAC,MAAc,EAAE,QAAgB,EAAE,MAAc;IAC/D,OAAO,GAAG,MAAM,KAAK,QAAQ,KAAK,MAAM,EAAE,CAAC;AAC7C,CAAC;AAED,SAAS,UAAU,CAAC,MAAc,EAAE,QAAgB;IAClD,OAAO,GAAG,MAAM,KAAK,QAAQ,EAAE,CAAC;AAClC,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,UAAU,UAAU,CACxB,QAA4D,EAC5D,QAA4D;IAE5D,MAAM,KAAK,GAAa,EAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;IAEtF,SAAS,GAAG,CAAC,EAAW,EAAE,IAA2B,EAAE,MAAc;QACnE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QACzC,QAAQ,EAAE,EAAE,CAAC;YACX,KAAK,KAAK;gBAAE,KAAK,CAAC,KAAK,EAAE,CAAC;gBAAC,MAAM;YACjC,KAAK,QAAQ;gBAAE,KAAK,CAAC,OAAO,EAAE,CAAC;gBAAC,MAAM;YACtC,KAAK,QAAQ;gBAAE,KAAK,CAAC,OAAO,EAAE,CAAC;gBAAC,MAAM;YACtC,KAAK,MAAM;gBAAE,KAAK,CAAC,OAAO,EAAE,CAAC;gBAAC,MAAM;QACtC,CAAC;IACH,CAAC;IAED,iBAAiB;IACjB,MAAM,SAAS,GAAG,IAAI,GAAG,EAA0B,CAAC;IACpD,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;QAC1C,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,GAAG,EAAE,QAAQ,EAAE,CAAC,GAAG,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACjE,CAAC;IAED,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACnC,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,WAAW,GAAG,GAAG,CAAC,QAAQ,CAAC,MAAM,CACrC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CACxB,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,SAAS,KAAK,CAAC,CAAC,SAAS,IAAI,EAAE,CAAC,SAAS,KAAK,CAAC,CAAC,SAAS,CACrE,CACF,CAAC;YACF,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,CAAC;gBACnC,GAAG,CAAC,QAAQ,EAAE,QAAQ,EAAE,GAAG,GAAG,CAAC,aAAa,MAAM,WAAW,CAAC,MAAM,WAAW,CAAC,CAAC;YACnF,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,MAAM,EAAE,QAAQ,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC;aAAM,CAAC;YACN,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,GAAG,EAAE,QAAQ,EAAE,CAAC,GAAG,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC/D,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IAED,kDAAkD;IAClD,MAAM,eAAe,GAAG,IAAI,GAAG,EAA4B,CAAC;IAC5D,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAA8B,CAAC;IAChE,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;QAC5C,MAAM,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QACzE,eAAe,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAC/B,MAAM,KAAK,GAAG,UAAU,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC3D,IAAI,GAAG,GAAG,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,CAAC,GAAG,EAAE,CAAC;YAAC,GAAG,GAAG,EAAE,CAAC;YAAC,iBAAiB,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAAC,CAAC;QAC1D,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACjB,CAAC;IAED,MAAM,gBAAgB,GAAuB,EAAE,CAAC;IAChD,MAAM,eAAe,GAAuB,EAAE,CAAC;IAC/C,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;QAC3C,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YAChB,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,CAAC;aAAM,CAAC;YACN,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,2CAA2C;IAC3C,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IACtC,KAAK,MAAM,GAAG,IAAI,gBAAgB,EAAE,CAAC;QACnC,IAAI,GAAG,CAAC,YAAY,EAAE,CAAC;YACrB,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,YAAY,CAAC,CAAC;YACtE,IAAI,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC7B,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBACrB,MAAM,GAAG,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;gBAC5C,GAAG,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,GAAG,CAAC,YAAY,IAAI,GAAG,CAAC,QAAQ,IAAI,GAAG,EAAE,aAAa,IAAI,GAAG,CAAC,YAAY,EAAE,CAAC,CAAC;YACzG,CAAC;QACH,CAAC;aAAM,CAAC;YACN,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;gBAC1C,IAAI,IAAI,CAAC,QAAQ,KAAK,GAAG,CAAC,QAAQ,IAAI,IAAI,CAAC,YAAY,KAAK,GAAG,CAAC,YAAY,EAAE,CAAC;oBAC7E,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;oBACrB,MAAM,OAAO,GAAG,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;oBACjD,MAAM,OAAO,GAAG,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;oBACjD,GAAG,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,OAAO,EAAE,aAAa,IAAI,IAAI,CAAC,YAAY,IAAI,GAAG,CAAC,QAAQ,IAAI,OAAO,EAAE,aAAa,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;gBACzI,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,6CAA6C;IAC7C,MAAM,WAAW,GAAuB,EAAE,CAAC;IAE3C,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;QAC1C,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAC1B,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED,KAAK,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;QAClC,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,YAAY,CAAC,CAAC;QAEtE,IAAI,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YACtD,MAAM,OAAO,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;YAChD,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,GAAG,CAAC,YAAY,IAAI,GAAG,CAAC,QAAQ,IAAI,OAAO,EAAE,aAAa,IAAI,GAAG,CAAC,YAAY,EAAE,CAAC,CAAC;YACzG,SAAS;QACX,CAAC;QAED,IAAI,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YACzB,sDAAsD;QACxD,CAAC;QAED,MAAM,eAAe,GAAG,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC5D,IAAI,eAAe,EAAE,CAAC;YACpB,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,YAAY,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC;YACzD,MAAM,aAAa,GAAG,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACnD,IAAI,aAAa,EAAE,CAAC;gBAClB,KAAK,MAAM,OAAO,IAAI,aAAa,EAAE,CAAC;oBACpC,IAAI,OAAO,CAAC,YAAY,KAAK,GAAG,CAAC,YAAY,EAAE,CAAC;wBAC9C,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;wBACrF,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;4BAC7B,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;4BACxB,MAAM,GAAG,GAAG,WAAW,CAAC,SAAS,CAC/B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,KAAK,OAAO,CAAC,YAAY;gCACvC,CAAC,CAAC,QAAQ,KAAK,OAAO,CAAC,QAAQ;gCAC/B,CAAC,CAAC,YAAY,KAAK,OAAO,CAAC,YAAY,CAC/C,CAAC;4BACF,IAAI,GAAG,KAAK,CAAC,CAAC;gCAAE,WAAW,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;4BAC3C,MAAM,UAAU,GAAG,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;4BACvD,MAAM,UAAU,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;4BACnD,GAAG,CAAC,QAAQ,EAAE,MAAM,EAClB,GAAG,GAAG,CAAC,QAAQ,KAAK,UAAU,EAAE,aAAa,IAAI,OAAO,CAAC,YAAY,MAAM,UAAU,EAAE,aAAa,IAAI,GAAG,CAAC,YAAY,EAAE,CAAC,CAAC;wBAChI,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,IAAI,CACnB,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,KAAK,GAAG,CAAC,YAAY;YACnC,CAAC,CAAC,QAAQ,KAAK,GAAG,CAAC,QAAQ;YAC3B,CAAC,CAAC,YAAY,KAAK,GAAG,CAAC,YAAY,CAC3C,EAAE,CAAC;YACF,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACtB,IAAI,CAAC,eAAe,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,YAAY,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC;gBAC3F,MAAM,OAAO,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;gBAChD,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,GAAG,CAAC,YAAY,IAAI,GAAG,CAAC,QAAQ,IAAI,OAAO,EAAE,aAAa,IAAI,GAAG,CAAC,YAAY,EAAE,CAAC,CAAC;YAC1G,CAAC;QACH,CAAC;IACH,CAAC;IAED,eAAe;IACf,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IACtC,MAAM,YAAY,GAAqB,EAAE,CAAC;IAE1C,KAAK,MAAM,EAAE,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;QACjC,MAAM,GAAG,GAAG,GAAG,EAAE,CAAC,SAAS,KAAK,EAAE,CAAC,SAAS,KAAK,EAAE,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;QAC/E,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACrB,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACxB,CAAC;IAED,KAAK,MAAM,EAAE,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;QACjC,MAAM,GAAG,GAAG,GAAG,EAAE,CAAC,SAAS,KAAK,EAAE,CAAC,SAAS,KAAK,EAAE,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;QAC/E,IAAI,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YACzB,GAAG,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QACpD,CAAC;aAAM,CAAC;YACN,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACrB,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACtB,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAED,OAAO;QACL,KAAK,EAAE;YACL,QAAQ,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC;YACjC,SAAS,EAAE,WAAW;SACvB;QACD,MAAM,EAAE,YAAY;QACpB,KAAK;KACN,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
import type { ExtractedEntity, ExtractedRelation, EntityMention, EntityType } from "./extractFacts.js";
|
|
2
|
+
export interface ResolvedEntity {
|
|
3
|
+
id: string;
|
|
4
|
+
canonicalName: string;
|
|
5
|
+
type: EntityType;
|
|
6
|
+
aliases: string[];
|
|
7
|
+
mentions: EntityMention[];
|
|
8
|
+
}
|
|
9
|
+
export interface ResolvedRelation {
|
|
10
|
+
headEntityId: string;
|
|
11
|
+
relation: string;
|
|
12
|
+
tailEntityId: string;
|
|
13
|
+
sessionId: string;
|
|
14
|
+
turnIndex: number;
|
|
15
|
+
evidence: string;
|
|
16
|
+
/** True when the session explicitly negates this relation. */
|
|
17
|
+
negated?: boolean;
|
|
18
|
+
}
|
|
19
|
+
export interface ResolvedGraph {
|
|
20
|
+
entities: ResolvedEntity[];
|
|
21
|
+
relations: ResolvedRelation[];
|
|
22
|
+
}
|
|
23
|
+
/**
|
|
24
|
+
* Cross-session entity dedup: normalize names, merge by key, assign stable IDs.
|
|
25
|
+
*/
|
|
26
|
+
export declare function resolveEntities(entities: ExtractedEntity[], relations: ExtractedRelation[]): ResolvedGraph;
|
|
27
|
+
//# sourceMappingURL=entityResolver.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"entityResolver.d.ts","sourceRoot":"","sources":["../../src/trainer/entityResolver.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,eAAe,EACf,iBAAiB,EACjB,aAAa,EACb,UAAU,EACX,MAAM,mBAAmB,CAAC;AAE3B,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,aAAa,EAAE,MAAM,CAAC;IACtB,IAAI,EAAE,UAAU,CAAC;IACjB,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,QAAQ,EAAE,aAAa,EAAE,CAAC;CAC3B;AAED,MAAM,WAAW,gBAAgB;IAC/B,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACjB,8DAA8D;IAC9D,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,aAAa;IAC5B,QAAQ,EAAE,cAAc,EAAE,CAAC;IAC3B,SAAS,EAAE,gBAAgB,EAAE,CAAC;CAC/B;AA4BD;;GAEG;AACH,wBAAgB,eAAe,CAC7B,QAAQ,EAAE,eAAe,EAAE,EAC3B,SAAS,EAAE,iBAAiB,EAAE,GAC7B,aAAa,CA0Ef"}
|
|
@@ -0,0 +1,92 @@
|
|
|
1
|
+
import { createHash } from "node:crypto";
|
|
2
|
+
function normalizeKey(name) {
|
|
3
|
+
return name
|
|
4
|
+
.toLowerCase()
|
|
5
|
+
.replace(/[''`]/g, "'")
|
|
6
|
+
.replace(/[""]/g, '"')
|
|
7
|
+
.replace(/\s+/g, " ")
|
|
8
|
+
.replace(/[.\-_]/g, "")
|
|
9
|
+
.trim();
|
|
10
|
+
}
|
|
11
|
+
function entityHash(normalizedKey) {
|
|
12
|
+
return "ent_" + createHash("sha256").update(normalizedKey).digest("hex").slice(0, 12);
|
|
13
|
+
}
|
|
14
|
+
function pickBestType(types) {
|
|
15
|
+
const priority = [
|
|
16
|
+
"Person", "Company", "Organization", "Project", "Tool",
|
|
17
|
+
"Language", "Product", "Platform", "Website", "Location",
|
|
18
|
+
"Document", "File", "Concept", "Unknown",
|
|
19
|
+
];
|
|
20
|
+
for (const t of priority) {
|
|
21
|
+
if (types.includes(t))
|
|
22
|
+
return t;
|
|
23
|
+
}
|
|
24
|
+
return "Unknown";
|
|
25
|
+
}
|
|
26
|
+
/**
|
|
27
|
+
* Cross-session entity dedup: normalize names, merge by key, assign stable IDs.
|
|
28
|
+
*/
|
|
29
|
+
export function resolveEntities(entities, relations) {
|
|
30
|
+
const keyMap = new Map();
|
|
31
|
+
for (const ent of entities) {
|
|
32
|
+
const key = normalizeKey(ent.name);
|
|
33
|
+
if (!key)
|
|
34
|
+
continue;
|
|
35
|
+
let bucket = keyMap.get(key);
|
|
36
|
+
if (!bucket) {
|
|
37
|
+
bucket = { names: new Set(), types: [], mentions: [] };
|
|
38
|
+
keyMap.set(key, bucket);
|
|
39
|
+
}
|
|
40
|
+
bucket.names.add(ent.name);
|
|
41
|
+
bucket.types.push(ent.type);
|
|
42
|
+
bucket.mentions.push(...ent.mentions);
|
|
43
|
+
}
|
|
44
|
+
const nameToId = new Map();
|
|
45
|
+
const resolved = [];
|
|
46
|
+
for (const [key, bucket] of keyMap) {
|
|
47
|
+
const id = entityHash(key);
|
|
48
|
+
const namesArr = [...bucket.names];
|
|
49
|
+
const canonical = namesArr.sort((a, b) => b.length - a.length)[0] ?? key;
|
|
50
|
+
for (const name of namesArr) {
|
|
51
|
+
nameToId.set(normalizeKey(name), id);
|
|
52
|
+
}
|
|
53
|
+
resolved.push({
|
|
54
|
+
id,
|
|
55
|
+
canonicalName: canonical,
|
|
56
|
+
type: pickBestType(bucket.types),
|
|
57
|
+
aliases: namesArr.filter((n) => n !== canonical),
|
|
58
|
+
mentions: bucket.mentions,
|
|
59
|
+
});
|
|
60
|
+
}
|
|
61
|
+
const resolvedRelations = [];
|
|
62
|
+
for (const rel of relations) {
|
|
63
|
+
const tailId = nameToId.get(normalizeKey(rel.tailName));
|
|
64
|
+
if (!tailId)
|
|
65
|
+
continue;
|
|
66
|
+
if (rel.negated) {
|
|
67
|
+
resolvedRelations.push({
|
|
68
|
+
headEntityId: rel.headName ? (nameToId.get(normalizeKey(rel.headName)) ?? "") : "",
|
|
69
|
+
relation: rel.relation,
|
|
70
|
+
tailEntityId: tailId,
|
|
71
|
+
sessionId: rel.sessionId,
|
|
72
|
+
turnIndex: rel.turnIndex,
|
|
73
|
+
evidence: rel.evidence,
|
|
74
|
+
negated: true,
|
|
75
|
+
});
|
|
76
|
+
continue;
|
|
77
|
+
}
|
|
78
|
+
const headId = nameToId.get(normalizeKey(rel.headName));
|
|
79
|
+
if (!headId)
|
|
80
|
+
continue;
|
|
81
|
+
resolvedRelations.push({
|
|
82
|
+
headEntityId: headId,
|
|
83
|
+
relation: rel.relation,
|
|
84
|
+
tailEntityId: tailId,
|
|
85
|
+
sessionId: rel.sessionId,
|
|
86
|
+
turnIndex: rel.turnIndex,
|
|
87
|
+
evidence: rel.evidence,
|
|
88
|
+
});
|
|
89
|
+
}
|
|
90
|
+
return { entities: resolved, relations: resolvedRelations };
|
|
91
|
+
}
|
|
92
|
+
//# sourceMappingURL=entityResolver.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"entityResolver.js","sourceRoot":"","sources":["../../src/trainer/entityResolver.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAiCzC,SAAS,YAAY,CAAC,IAAY;IAChC,OAAO,IAAI;SACR,WAAW,EAAE;SACb,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC;SACtB,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC;SACrB,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC;SACpB,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;SACtB,IAAI,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,UAAU,CAAC,aAAqB;IACvC,OAAO,MAAM,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;AACxF,CAAC;AAED,SAAS,YAAY,CAAC,KAAmB;IACvC,MAAM,QAAQ,GAAiB;QAC7B,QAAQ,EAAE,SAAS,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM;QACtD,UAAU,EAAE,SAAS,EAAE,UAAU,EAAE,SAAS,EAAE,UAAU;QACxD,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS;KACzC,CAAC;IACF,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;QACzB,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;YAAE,OAAO,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,eAAe,CAC7B,QAA2B,EAC3B,SAA8B;IAE9B,MAAM,MAAM,GAAG,IAAI,GAAG,EAIlB,CAAC;IAEL,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,MAAM,GAAG,GAAG,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACnC,IAAI,CAAC,GAAG;YAAE,SAAS;QAEnB,IAAI,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC7B,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,GAAG,EAAE,KAAK,EAAE,IAAI,GAAG,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;YACvD,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAC1B,CAAC;QACD,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC3B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC5B,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC3C,MAAM,QAAQ,GAAqB,EAAE,CAAC;IAEtC,KAAK,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;QACnC,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;QAC3B,MAAM,QAAQ,GAAG,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC;QAEzE,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;YAC5B,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;QACvC,CAAC;QAED,QAAQ,CAAC,IAAI,CAAC;YACZ,EAAE;YACF,aAAa,EAAE,SAAS;YACxB,IAAI,EAAE,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC;YAChC,OAAO,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC;YAChD,QAAQ,EAAE,MAAM,CAAC,QAAQ;SAC1B,CAAC,CAAC;IACL,CAAC;IAED,MAAM,iBAAiB,GAAuB,EAAE,CAAC;IACjD,KAAK,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;QAC5B,MAAM,MAAM,GAAG,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QACxD,IAAI,CAAC,MAAM;YAAE,SAAS;QAEtB,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YAChB,iBAAiB,CAAC,IAAI,CAAC;gBACrB,YAAY,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE;gBAClF,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,YAAY,EAAE,MAAM;gBACpB,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,OAAO,EAAE,IAAI;aACd,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QAED,MAAM,MAAM,GAAG,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QACxD,IAAI,CAAC,MAAM;YAAE,SAAS;QAEtB,iBAAiB,CAAC,IAAI,CAAC;YACrB,YAAY,EAAE,MAAM;YACpB,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,YAAY,EAAE,MAAM;YACpB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,QAAQ,EAAE,GAAG,CAAC,QAAQ;SACvB,CAAC,CAAC;IACL,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,iBAAiB,EAAE,CAAC;AAC9D,CAAC"}
|
|
@@ -0,0 +1,67 @@
|
|
|
1
|
+
import type { SessionTurn, LoadedSession } from "./sessionLoader.js";
|
|
2
|
+
/** Relation catalog mirroring Kocoro compactMemoryRelationCatalog. */
|
|
3
|
+
export declare const RELATION_CATALOG: {
|
|
4
|
+
readonly people_and_social: readonly ["employed_at", "previously_employed_at", "works_on", "affiliated_with", "studied_under", "studied_at", "collaborates_with", "follows_person", "followed_by_person", "commented_on", "knows_about", "has_handle_on", "has_email"];
|
|
5
|
+
readonly ownership_and_company: readonly ["created", "created_by", "maintained_by", "develops", "developed_by_org", "owns", "owned_by", "acquired", "acquired_by", "subsidiary_of", "parent_of", "founded", "founded_by", "invested_in", "received_investment_from", "customer_of", "has_customer", "competes_with", "banking_relationship"];
|
|
6
|
+
readonly technical_and_project: readonly ["uses", "used_by", "depends_on", "implemented_in", "runs_on", "integrates_with", "supports", "powered_by", "loaded_via", "has_component", "part_of", "has_property", "has_path", "stored_at", "monitors", "targets", "enables", "enabled_by", "generates", "generated_from", "implements", "implemented_by", "excludes", "deleted_from"];
|
|
7
|
+
readonly content_and_metadata: readonly ["published_on", "released", "latest_release_tag", "forked_from", "inspired_by", "succeeds", "preceded_by", "describes", "described_in", "category", "has_alias", "has_url", "located_in", "scheduled_for", "ranked_on", "listed_on", "features_project"];
|
|
8
|
+
readonly generic_fallback: readonly ["related_to", "other"];
|
|
9
|
+
};
|
|
10
|
+
export declare const ALL_RELATIONS: ("created" | "founded" | "owns" | "uses" | "supports" | "released" | "monitors" | "employed_at" | "previously_employed_at" | "works_on" | "affiliated_with" | "studied_under" | "studied_at" | "collaborates_with" | "follows_person" | "followed_by_person" | "commented_on" | "knows_about" | "has_handle_on" | "has_email" | "created_by" | "maintained_by" | "develops" | "developed_by_org" | "owned_by" | "acquired" | "acquired_by" | "subsidiary_of" | "parent_of" | "founded_by" | "invested_in" | "received_investment_from" | "customer_of" | "has_customer" | "competes_with" | "banking_relationship" | "used_by" | "depends_on" | "implemented_in" | "runs_on" | "integrates_with" | "powered_by" | "loaded_via" | "has_component" | "part_of" | "has_property" | "has_path" | "stored_at" | "targets" | "enables" | "enabled_by" | "generates" | "generated_from" | "implements" | "implemented_by" | "excludes" | "deleted_from" | "published_on" | "latest_release_tag" | "forked_from" | "inspired_by" | "succeeds" | "preceded_by" | "describes" | "described_in" | "category" | "has_alias" | "has_url" | "located_in" | "scheduled_for" | "ranked_on" | "listed_on" | "features_project" | "related_to" | "other")[];
|
|
11
|
+
export type EntityType = "Person" | "Company" | "Organization" | "Project" | "Tool" | "Language" | "Concept" | "Document" | "File" | "Location" | "Product" | "Website" | "Platform" | "Unknown";
|
|
12
|
+
export interface ExtractedEntity {
|
|
13
|
+
name: string;
|
|
14
|
+
type: EntityType;
|
|
15
|
+
mentions: EntityMention[];
|
|
16
|
+
}
|
|
17
|
+
export interface EntityMention {
|
|
18
|
+
sessionId: string;
|
|
19
|
+
turnIndex: number;
|
|
20
|
+
snippet: string;
|
|
21
|
+
}
|
|
22
|
+
export interface ExtractedRelation {
|
|
23
|
+
headName: string;
|
|
24
|
+
relation: string;
|
|
25
|
+
tailName: string;
|
|
26
|
+
sessionId: string;
|
|
27
|
+
turnIndex: number;
|
|
28
|
+
evidence: string;
|
|
29
|
+
/** True when the session explicitly negates this relation ("no longer uses X"). */
|
|
30
|
+
negated?: boolean;
|
|
31
|
+
}
|
|
32
|
+
export interface ExtractedEvent {
|
|
33
|
+
description: string;
|
|
34
|
+
sessionId: string;
|
|
35
|
+
timestamp: string;
|
|
36
|
+
turnIndex: number;
|
|
37
|
+
}
|
|
38
|
+
export interface ExtractionResult {
|
|
39
|
+
entities: ExtractedEntity[];
|
|
40
|
+
relations: ExtractedRelation[];
|
|
41
|
+
events: ExtractedEvent[];
|
|
42
|
+
}
|
|
43
|
+
/**
|
|
44
|
+
* Optional LLM-backed extractor interface. The default implementation
|
|
45
|
+
* uses regex/heuristic patterns only and does not require an LLM.
|
|
46
|
+
*/
|
|
47
|
+
export interface LLMFactExtractor {
|
|
48
|
+
extractFacts(turns: SessionTurn[], sessionId: string): Promise<{
|
|
49
|
+
entities: ExtractedEntity[];
|
|
50
|
+
relations: ExtractedRelation[];
|
|
51
|
+
events: ExtractedEvent[];
|
|
52
|
+
}>;
|
|
53
|
+
}
|
|
54
|
+
export interface ExtractFactsOptions {
|
|
55
|
+
llmExtractor?: LLMFactExtractor | null;
|
|
56
|
+
}
|
|
57
|
+
export declare const UPDATE_SIGNAL_RE: RegExp;
|
|
58
|
+
/**
|
|
59
|
+
* Extract structured facts from a single session's turns using
|
|
60
|
+
* regex/heuristic patterns. For deeper extraction, pass an LLMFactExtractor.
|
|
61
|
+
*/
|
|
62
|
+
export declare function extractFacts(session: LoadedSession, opts?: ExtractFactsOptions): Promise<ExtractionResult>;
|
|
63
|
+
/**
|
|
64
|
+
* Extract facts from multiple sessions and merge results.
|
|
65
|
+
*/
|
|
66
|
+
export declare function extractFactsFromSessions(sessions: LoadedSession[], opts?: ExtractFactsOptions): Promise<ExtractionResult>;
|
|
67
|
+
//# sourceMappingURL=extractFacts.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"extractFacts.d.ts","sourceRoot":"","sources":["../../src/trainer/extractFacts.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAErE,sEAAsE;AACtE,eAAO,MAAM,gBAAgB;;;;;;CA6BnB,CAAC;AAEX,eAAO,MAAM,aAAa,2qCAAyC,CAAC;AAEpE,MAAM,MAAM,UAAU,GAClB,QAAQ,GAAG,SAAS,GAAG,cAAc,GAAG,SAAS,GAAG,MAAM,GAC1D,UAAU,GAAG,SAAS,GAAG,UAAU,GAAG,MAAM,GAAG,UAAU,GACzD,SAAS,GAAG,SAAS,GAAG,UAAU,GAAG,SAAS,CAAC;AAEnD,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,UAAU,CAAC;IACjB,QAAQ,EAAE,aAAa,EAAE,CAAC;CAC3B;AAED,MAAM,WAAW,aAAa;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACjB,mFAAmF;IACnF,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,cAAc;IAC7B,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,EAAE,eAAe,EAAE,CAAC;IAC5B,SAAS,EAAE,iBAAiB,EAAE,CAAC;IAC/B,MAAM,EAAE,cAAc,EAAE,CAAC;CAC1B;AAED;;;GAGG;AACH,MAAM,WAAW,gBAAgB;IAC/B,YAAY,CACV,KAAK,EAAE,WAAW,EAAE,EACpB,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC;QACT,QAAQ,EAAE,eAAe,EAAE,CAAC;QAC5B,SAAS,EAAE,iBAAiB,EAAE,CAAC;QAC/B,MAAM,EAAE,cAAc,EAAE,CAAC;KAC1B,CAAC,CAAC;CACJ;AAED,MAAM,WAAW,mBAAmB;IAClC,YAAY,CAAC,EAAE,gBAAgB,GAAG,IAAI,CAAC;CACxC;AA2CD,eAAO,MAAM,gBAAgB,QAAyG,CAAC;AAyCvI;;;GAGG;AACH,wBAAsB,YAAY,CAChC,OAAO,EAAE,aAAa,EACtB,IAAI,CAAC,EAAE,mBAAmB,GACzB,OAAO,CAAC,gBAAgB,CAAC,CAqG3B;AAED;;GAEG;AACH,wBAAsB,wBAAwB,CAC5C,QAAQ,EAAE,aAAa,EAAE,EACzB,IAAI,CAAC,EAAE,mBAAmB,GACzB,OAAO,CAAC,gBAAgB,CAAC,CAiB3B"}
|
|
@@ -0,0 +1,213 @@
|
|
|
1
|
+
/** Relation catalog mirroring Kocoro compactMemoryRelationCatalog. */
|
|
2
|
+
export const RELATION_CATALOG = {
|
|
3
|
+
people_and_social: [
|
|
4
|
+
"employed_at", "previously_employed_at", "works_on", "affiliated_with",
|
|
5
|
+
"studied_under", "studied_at", "collaborates_with", "follows_person",
|
|
6
|
+
"followed_by_person", "commented_on", "knows_about", "has_handle_on",
|
|
7
|
+
"has_email",
|
|
8
|
+
],
|
|
9
|
+
ownership_and_company: [
|
|
10
|
+
"created", "created_by", "maintained_by", "develops",
|
|
11
|
+
"developed_by_org", "owns", "owned_by", "acquired", "acquired_by",
|
|
12
|
+
"subsidiary_of", "parent_of", "founded", "founded_by", "invested_in",
|
|
13
|
+
"received_investment_from", "customer_of", "has_customer",
|
|
14
|
+
"competes_with", "banking_relationship",
|
|
15
|
+
],
|
|
16
|
+
technical_and_project: [
|
|
17
|
+
"uses", "used_by", "depends_on", "implemented_in", "runs_on",
|
|
18
|
+
"integrates_with", "supports", "powered_by", "loaded_via",
|
|
19
|
+
"has_component", "part_of", "has_property", "has_path", "stored_at",
|
|
20
|
+
"monitors", "targets", "enables", "enabled_by", "generates",
|
|
21
|
+
"generated_from", "implements", "implemented_by", "excludes",
|
|
22
|
+
"deleted_from",
|
|
23
|
+
],
|
|
24
|
+
content_and_metadata: [
|
|
25
|
+
"published_on", "released", "latest_release_tag", "forked_from",
|
|
26
|
+
"inspired_by", "succeeds", "preceded_by", "describes", "described_in",
|
|
27
|
+
"category", "has_alias", "has_url", "located_in", "scheduled_for",
|
|
28
|
+
"ranked_on", "listed_on", "features_project",
|
|
29
|
+
],
|
|
30
|
+
generic_fallback: ["related_to", "other"],
|
|
31
|
+
};
|
|
32
|
+
export const ALL_RELATIONS = Object.values(RELATION_CATALOG).flat();
|
|
33
|
+
// ── Heuristic patterns ──
|
|
34
|
+
const CAPITALIZED_NAME_RE = /\b([A-Z][a-z]+(?:\s+[A-Z][a-z]+)+)\b/g;
|
|
35
|
+
const TOOL_PROJECT_RE = /\b(React|Vue|Angular|Svelte|Next\.js|Nuxt|Django|Flask|Rails|Spring|Express|FastAPI|Rust|Go|Python|TypeScript|JavaScript|Node\.js|Docker|Kubernetes|Redis|PostgreSQL|MySQL|MongoDB|SQLite|Terraform|AWS|GCP|Azure|GitHub|GitLab|Slack|Figma|Notion|Linear|Jira|Vercel|Netlify|Supabase|Firebase|Anthropic|OpenAI|Claude|GPT|Gemini|Ollama|Tailwind|GSAP|Prisma|Drizzle|Vite|Webpack|ESLint|Prettier|Vitest|Jest|Playwright|Cypress)\b/gi;
|
|
36
|
+
const RELATION_PATTERNS = [
|
|
37
|
+
{ re: /\b(\w[\w\s]*?)\s+(?:uses?|is using|使用)\s+(\w[\w\s.]*?\b)/gi, relation: "uses", headGroup: 1, tailGroup: 2 },
|
|
38
|
+
{ re: /\b(\w[\w\s]*?)\s+(?:created|built|made|开发了|创建了)\s+(\w[\w\s.]*?\b)/gi, relation: "created", headGroup: 1, tailGroup: 2 },
|
|
39
|
+
{ re: /\b(\w[\w\s]*?)\s+(?:works?\s+(?:at|for)|在.+工作)\s+(\w[\w\s.]*?\b)/gi, relation: "employed_at", headGroup: 1, tailGroup: 2 },
|
|
40
|
+
{ re: /\b(\w[\w\s]*?)\s+(?:works?\s+on|负责)\s+(\w[\w\s.]*?\b)/gi, relation: "works_on", headGroup: 1, tailGroup: 2 },
|
|
41
|
+
{ re: /\b(\w[\w\s]*?)\s+(?:collaborates?\s+with|合作)\s+(\w[\w\s.]*?\b)/gi, relation: "collaborates_with", headGroup: 1, tailGroup: 2 },
|
|
42
|
+
{ re: /\b(\w[\w\s]*?)\s+(?:depends?\s+on|依赖)\s+(\w[\w\s.]*?\b)/gi, relation: "depends_on", headGroup: 1, tailGroup: 2 },
|
|
43
|
+
{ re: /\b(\w[\w\s]*?)\s+(?:runs?\s+on|运行在)\s+(\w[\w\s.]*?\b)/gi, relation: "runs_on", headGroup: 1, tailGroup: 2 },
|
|
44
|
+
{ re: /\b(\w[\w\s]*?)\s+(?:integrates?\s+with|集成)\s+(\w[\w\s.]*?\b)/gi, relation: "integrates_with", headGroup: 1, tailGroup: 2 },
|
|
45
|
+
{ re: /\b(\w[\w\s]*?)\s+(?:founded|创立了)\s+(\w[\w\s.]*?\b)/gi, relation: "founded", headGroup: 1, tailGroup: 2 },
|
|
46
|
+
{ re: /\b(\w[\w\s]*?)\s+(?:implements?|实现了)\s+(\w[\w\s.]*?\b)/gi, relation: "implements", headGroup: 1, tailGroup: 2 },
|
|
47
|
+
];
|
|
48
|
+
const NEGATION_PATTERNS = [
|
|
49
|
+
{ re: /\b(?:no longer|stopped|quit|dropped|gave up|abandoned)\s+(?:using|uses?)\s+(\w[\w\s.]*?\b)/gi, relation: "uses", entityGroup: 1 },
|
|
50
|
+
{ re: /(?:不再|停止|放弃)(?:使用|用)\s*([A-Za-z][\w\s.]*\w)/gi, relation: "uses", entityGroup: 1 },
|
|
51
|
+
{ re: /\b(?:left|quit|resigned from|no longer (?:works?|employed) (?:at|for))\s+(\w[\w\s.]*?\b)/gi, relation: "employed_at", entityGroup: 1 },
|
|
52
|
+
{ re: /(?:不再|离开了|辞去了?)(?:在)?\s*([A-Za-z][\w\s.]*\w)/gi, relation: "employed_at", entityGroup: 1 },
|
|
53
|
+
{ re: /\b(?:stopped|quit|no longer)\s+(?:working on|developing|maintaining)\s+(\w[\w\s.]*?\b)/gi, relation: "works_on", entityGroup: 1 },
|
|
54
|
+
{ re: /(?:不再|停止)\s*(?:负责|开发|维护)\s*([A-Za-z][\w\s.]*\w)/gi, relation: "works_on", entityGroup: 1 },
|
|
55
|
+
{ re: /\b(?:removed|dropped|deleted|uninstalled)\s+(\w[\w\s.]*?\b)/gi, relation: "uses", entityGroup: 1 },
|
|
56
|
+
{ re: /\b(?:switched|migrated|moved)\s+(?:from)\s+(\w[\w\s.]*?)\s+(?:to)\s+\w/gi, relation: "uses", entityGroup: 1 },
|
|
57
|
+
{ re: /(?:从)\s*([A-Za-z][\w\s.]*\w)\s*(?:迁移|切换|换)(?:到|成)/gi, relation: "uses", entityGroup: 1 },
|
|
58
|
+
];
|
|
59
|
+
export const UPDATE_SIGNAL_RE = /\b(?:now|switched to|migrated to|moved to|changed to|upgraded to|replaced with|改用|换成|切换到|迁移到|升级到)\b/i;
|
|
60
|
+
const EVENT_PATTERNS = [
|
|
61
|
+
/\b(?:decided|agreed|shipped|released|launched|deployed|migrated|completed|finished|merged|approved|resolved)\b/i,
|
|
62
|
+
/\b(?:决定|完成|上线|发布|部署|迁移|合并|通过)\b/,
|
|
63
|
+
];
|
|
64
|
+
const NOISE_WORDS = new Set([
|
|
65
|
+
"the", "a", "an", "this", "that", "it", "i", "we", "they", "he", "she",
|
|
66
|
+
"my", "our", "their", "you", "your", "its",
|
|
67
|
+
"is", "are", "was", "were", "be", "been", "being",
|
|
68
|
+
"has", "have", "had", "do", "does", "did",
|
|
69
|
+
"will", "would", "could", "should", "can", "may", "might",
|
|
70
|
+
"not", "no", "yes", "ok", "sure", "thanks", "thank",
|
|
71
|
+
"if", "but", "and", "or", "so", "then", "also",
|
|
72
|
+
]);
|
|
73
|
+
function isNoiseName(name) {
|
|
74
|
+
const lower = name.toLowerCase().trim();
|
|
75
|
+
if (lower.length < 2 || lower.length > 60)
|
|
76
|
+
return true;
|
|
77
|
+
const words = lower.split(/\s+/);
|
|
78
|
+
return words.every((w) => NOISE_WORDS.has(w));
|
|
79
|
+
}
|
|
80
|
+
function inferEntityType(name) {
|
|
81
|
+
const lower = name.toLowerCase();
|
|
82
|
+
const toolMatch = name.match(TOOL_PROJECT_RE);
|
|
83
|
+
if (toolMatch && toolMatch[0].toLowerCase() === lower)
|
|
84
|
+
return "Tool";
|
|
85
|
+
if (/\b(?:inc|corp|ltd|llc|gmbh|co|company|group)\b/i.test(name))
|
|
86
|
+
return "Company";
|
|
87
|
+
if (/\b(?:university|institute|foundation|org)\b/i.test(name))
|
|
88
|
+
return "Organization";
|
|
89
|
+
if (/^[A-Z][a-z]+(?:\s+[A-Z][a-z]+)+$/.test(name))
|
|
90
|
+
return "Person";
|
|
91
|
+
return "Unknown";
|
|
92
|
+
}
|
|
93
|
+
function snippet(text, maxLen = 120) {
|
|
94
|
+
if (text.length <= maxLen)
|
|
95
|
+
return text;
|
|
96
|
+
return text.slice(0, maxLen) + "...";
|
|
97
|
+
}
|
|
98
|
+
/**
|
|
99
|
+
* Extract structured facts from a single session's turns using
|
|
100
|
+
* regex/heuristic patterns. For deeper extraction, pass an LLMFactExtractor.
|
|
101
|
+
*/
|
|
102
|
+
export async function extractFacts(session, opts) {
|
|
103
|
+
if (opts?.llmExtractor) {
|
|
104
|
+
return opts.llmExtractor.extractFacts(session.turns, session.id);
|
|
105
|
+
}
|
|
106
|
+
const entMap = new Map();
|
|
107
|
+
const relations = [];
|
|
108
|
+
const events = [];
|
|
109
|
+
function addEntity(name, type, turn) {
|
|
110
|
+
const trimmed = name.trim();
|
|
111
|
+
if (isNoiseName(trimmed))
|
|
112
|
+
return;
|
|
113
|
+
const key = trimmed.toLowerCase();
|
|
114
|
+
let ent = entMap.get(key);
|
|
115
|
+
if (!ent) {
|
|
116
|
+
ent = { name: trimmed, type, mentions: [] };
|
|
117
|
+
entMap.set(key, ent);
|
|
118
|
+
}
|
|
119
|
+
if (type !== "Unknown" && ent.type === "Unknown") {
|
|
120
|
+
ent.type = type;
|
|
121
|
+
}
|
|
122
|
+
ent.mentions.push({
|
|
123
|
+
sessionId: session.id,
|
|
124
|
+
turnIndex: turn.turnIndex,
|
|
125
|
+
snippet: snippet(turn.content),
|
|
126
|
+
});
|
|
127
|
+
}
|
|
128
|
+
for (const turn of session.turns) {
|
|
129
|
+
const text = turn.content;
|
|
130
|
+
// Extract capitalized multi-word names (likely Person names)
|
|
131
|
+
for (const m of text.matchAll(CAPITALIZED_NAME_RE)) {
|
|
132
|
+
addEntity(m[1], inferEntityType(m[1]), turn);
|
|
133
|
+
}
|
|
134
|
+
// Extract known tools/projects
|
|
135
|
+
for (const m of text.matchAll(TOOL_PROJECT_RE)) {
|
|
136
|
+
addEntity(m[0], "Tool", turn);
|
|
137
|
+
}
|
|
138
|
+
// Extract relations
|
|
139
|
+
for (const pattern of RELATION_PATTERNS) {
|
|
140
|
+
for (const m of text.matchAll(pattern.re)) {
|
|
141
|
+
const head = m[pattern.headGroup]?.trim() ?? "";
|
|
142
|
+
const tail = m[pattern.tailGroup]?.trim() ?? "";
|
|
143
|
+
if (isNoiseName(head) || isNoiseName(tail))
|
|
144
|
+
continue;
|
|
145
|
+
addEntity(head, inferEntityType(head), turn);
|
|
146
|
+
addEntity(tail, inferEntityType(tail), turn);
|
|
147
|
+
relations.push({
|
|
148
|
+
headName: head,
|
|
149
|
+
relation: pattern.relation,
|
|
150
|
+
tailName: tail,
|
|
151
|
+
sessionId: session.id,
|
|
152
|
+
turnIndex: turn.turnIndex,
|
|
153
|
+
evidence: snippet(text),
|
|
154
|
+
});
|
|
155
|
+
}
|
|
156
|
+
}
|
|
157
|
+
// Extract negated relations ("no longer uses X", "stopped using X")
|
|
158
|
+
for (const pattern of NEGATION_PATTERNS) {
|
|
159
|
+
for (const m of text.matchAll(pattern.re)) {
|
|
160
|
+
const entityName = m[pattern.entityGroup]?.trim() ?? "";
|
|
161
|
+
if (isNoiseName(entityName))
|
|
162
|
+
continue;
|
|
163
|
+
addEntity(entityName, inferEntityType(entityName), turn);
|
|
164
|
+
relations.push({
|
|
165
|
+
headName: "",
|
|
166
|
+
relation: pattern.relation,
|
|
167
|
+
tailName: entityName,
|
|
168
|
+
sessionId: session.id,
|
|
169
|
+
turnIndex: turn.turnIndex,
|
|
170
|
+
evidence: snippet(text),
|
|
171
|
+
negated: true,
|
|
172
|
+
});
|
|
173
|
+
}
|
|
174
|
+
}
|
|
175
|
+
// Detect events (decisions, milestones)
|
|
176
|
+
for (const re of EVENT_PATTERNS) {
|
|
177
|
+
if (re.test(text)) {
|
|
178
|
+
events.push({
|
|
179
|
+
description: snippet(text, 200),
|
|
180
|
+
sessionId: session.id,
|
|
181
|
+
timestamp: session.createdAt || new Date().toISOString(),
|
|
182
|
+
turnIndex: turn.turnIndex,
|
|
183
|
+
});
|
|
184
|
+
break;
|
|
185
|
+
}
|
|
186
|
+
}
|
|
187
|
+
}
|
|
188
|
+
return {
|
|
189
|
+
entities: [...entMap.values()],
|
|
190
|
+
relations,
|
|
191
|
+
events,
|
|
192
|
+
};
|
|
193
|
+
}
|
|
194
|
+
/**
|
|
195
|
+
* Extract facts from multiple sessions and merge results.
|
|
196
|
+
*/
|
|
197
|
+
export async function extractFactsFromSessions(sessions, opts) {
|
|
198
|
+
const allEntities = [];
|
|
199
|
+
const allRelations = [];
|
|
200
|
+
const allEvents = [];
|
|
201
|
+
for (const session of sessions) {
|
|
202
|
+
const result = await extractFacts(session, opts);
|
|
203
|
+
allEntities.push(...result.entities);
|
|
204
|
+
allRelations.push(...result.relations);
|
|
205
|
+
allEvents.push(...result.events);
|
|
206
|
+
}
|
|
207
|
+
return {
|
|
208
|
+
entities: allEntities,
|
|
209
|
+
relations: allRelations,
|
|
210
|
+
events: allEvents,
|
|
211
|
+
};
|
|
212
|
+
}
|
|
213
|
+
//# sourceMappingURL=extractFacts.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"extractFacts.js","sourceRoot":"","sources":["../../src/trainer/extractFacts.ts"],"names":[],"mappings":"AAEA,sEAAsE;AACtE,MAAM,CAAC,MAAM,gBAAgB,GAAG;IAC9B,iBAAiB,EAAE;QACjB,aAAa,EAAE,wBAAwB,EAAE,UAAU,EAAE,iBAAiB;QACtE,eAAe,EAAE,YAAY,EAAE,mBAAmB,EAAE,gBAAgB;QACpE,oBAAoB,EAAE,cAAc,EAAE,aAAa,EAAE,eAAe;QACpE,WAAW;KACZ;IACD,qBAAqB,EAAE;QACrB,SAAS,EAAE,YAAY,EAAE,eAAe,EAAE,UAAU;QACpD,kBAAkB,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,aAAa;QACjE,eAAe,EAAE,WAAW,EAAE,SAAS,EAAE,YAAY,EAAE,aAAa;QACpE,0BAA0B,EAAE,aAAa,EAAE,cAAc;QACzD,eAAe,EAAE,sBAAsB;KACxC;IACD,qBAAqB,EAAE;QACrB,MAAM,EAAE,SAAS,EAAE,YAAY,EAAE,gBAAgB,EAAE,SAAS;QAC5D,iBAAiB,EAAE,UAAU,EAAE,YAAY,EAAE,YAAY;QACzD,eAAe,EAAE,SAAS,EAAE,cAAc,EAAE,UAAU,EAAE,WAAW;QACnE,UAAU,EAAE,SAAS,EAAE,SAAS,EAAE,YAAY,EAAE,WAAW;QAC3D,gBAAgB,EAAE,YAAY,EAAE,gBAAgB,EAAE,UAAU;QAC5D,cAAc;KACf;IACD,oBAAoB,EAAE;QACpB,cAAc,EAAE,UAAU,EAAE,oBAAoB,EAAE,aAAa;QAC/D,aAAa,EAAE,UAAU,EAAE,aAAa,EAAE,WAAW,EAAE,cAAc;QACrE,UAAU,EAAE,WAAW,EAAE,SAAS,EAAE,YAAY,EAAE,eAAe;QACjE,WAAW,EAAE,WAAW,EAAE,kBAAkB;KAC7C;IACD,gBAAgB,EAAE,CAAC,YAAY,EAAE,OAAO,CAAC;CACjC,CAAC;AAEX,MAAM,CAAC,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,EAAE,CAAC;AA8DpE,2BAA2B;AAE3B,MAAM,mBAAmB,GAAG,uCAAuC,CAAC;AAEpE,MAAM,eAAe,GACnB,yaAAya,CAAC;AAE5a,MAAM,iBAAiB,GAKlB;IACH,EAAE,EAAE,EAAE,4DAA4D,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;IAClH,EAAE,EAAE,EAAE,qEAAqE,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;IAC9H,EAAE,EAAE,EAAE,oEAAoE,EAAE,QAAQ,EAAE,aAAa,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;IACjI,EAAE,EAAE,EAAE,yDAAyD,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;IACnH,EAAE,EAAE,EAAE,kEAAkE,EAAE,QAAQ,EAAE,mBAAmB,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;IACrI,EAAE,EAAE,EAAE,2DAA2D,EAAE,QAAQ,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;IACvH,EAAE,EAAE,EAAE,yDAAyD,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;IAClH,EAAE,EAAE,EAAE,gEAAgE,EAAE,QAAQ,EAAE,iBAAiB,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;IACjI,EAAE,EAAE,EAAE,sDAAsD,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;IAC/G,EAAE,EAAE,EAAE,0DAA0D,EAAE,QAAQ,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;CACvH,CAAC;AAEF,MAAM,iBAAiB,GAIlB;IACH,EAAE,EAAE,EAAE,8FAA8F,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,EAAE;IACxI,EAAE,EAAE,EAAE,+CAA+C,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,EAAE;IACzF,EAAE,EAAE,EAAE,4FAA4F,EAAE,QAAQ,EAAE,aAAa,EAAE,WAAW,EAAE,CAAC,EAAE;IAC7I,EAAE,EAAE,EAAE,gDAAgD,EAAE,QAAQ,EAAE,aAAa,EAAE,WAAW,EAAE,CAAC,EAAE;IACjG,EAAE,EAAE,EAAE,0FAA0F,EAAE,QAAQ,EAAE,UAAU,EAAE,WAAW,EAAE,CAAC,EAAE;IACxI,EAAE,EAAE,EAAE,mDAAmD,EAAE,QAAQ,EAAE,UAAU,EAAE,WAAW,EAAE,CAAC,EAAE;IACjG,EAAE,EAAE,EAAE,+DAA+D,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,EAAE;IACzG,EAAE,EAAE,EAAE,0EAA0E,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,EAAE;IACpH,EAAE,EAAE,EAAE,qDAAqD,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,EAAE;CAChG,CAAC;AAEF,MAAM,CAAC,MAAM,gBAAgB,GAAG,sGAAsG,CAAC;AAEvI,MAAM,cAAc,GAAa;IAC/B,iHAAiH;IACjH,iCAAiC;CAClC,CAAC;AAEF,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC;IAC1B,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK;IACtE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK;IAC1C,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO;IACjD,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK;IACzC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO;IACzD,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO;IACnD,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM;CAC/C,CAAC,CAAC;AAEH,SAAS,WAAW,CAAC,IAAY;IAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;IACxC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,KAAK,CAAC,MAAM,GAAG,EAAE;QAAE,OAAO,IAAI,CAAC;IACvD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACjC,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;AAChD,CAAC;AAED,SAAS,eAAe,CAAC,IAAY;IACnC,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IACjC,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;IAC9C,IAAI,SAAS,IAAI,SAAS,CAAC,CAAC,CAAE,CAAC,WAAW,EAAE,KAAK,KAAK;QAAE,OAAO,MAAM,CAAC;IAEtE,IAAI,iDAAiD,CAAC,IAAI,CAAC,IAAI,CAAC;QAAE,OAAO,SAAS,CAAC;IACnF,IAAI,8CAA8C,CAAC,IAAI,CAAC,IAAI,CAAC;QAAE,OAAO,cAAc,CAAC;IACrF,IAAI,kCAAkC,CAAC,IAAI,CAAC,IAAI,CAAC;QAAE,OAAO,QAAQ,CAAC;IAEnE,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,OAAO,CAAC,IAAY,EAAE,MAAM,GAAG,GAAG;IACzC,IAAI,IAAI,CAAC,MAAM,IAAI,MAAM;QAAE,OAAO,IAAI,CAAC;IACvC,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,GAAG,KAAK,CAAC;AACvC,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,OAAsB,EACtB,IAA0B;IAE1B,IAAI,IAAI,EAAE,YAAY,EAAE,CAAC;QACvB,OAAO,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,GAAG,EAA2B,CAAC;IAClD,MAAM,SAAS,GAAwB,EAAE,CAAC;IAC1C,MAAM,MAAM,GAAqB,EAAE,CAAC;IAEpC,SAAS,SAAS,CAChB,IAAY,EACZ,IAAgB,EAChB,IAAiB;QAEjB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAC5B,IAAI,WAAW,CAAC,OAAO,CAAC;YAAE,OAAO;QACjC,MAAM,GAAG,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QAClC,IAAI,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC1B,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,GAAG,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;YAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACvB,CAAC;QACD,IAAI,IAAI,KAAK,SAAS,IAAI,GAAG,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YACjD,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC;QAClB,CAAC;QACD,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC;YAChB,SAAS,EAAE,OAAO,CAAC,EAAE;YACrB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC;SAC/B,CAAC,CAAC;IACL,CAAC;IAED,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QACjC,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC;QAE1B,6DAA6D;QAC7D,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;YACnD,SAAS,CAAC,CAAC,CAAC,CAAC,CAAE,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC,CAAE,CAAC,EAAE,IAAI,CAAC,CAAC;QACjD,CAAC;QAED,+BAA+B;QAC/B,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YAC/C,SAAS,CAAC,CAAC,CAAC,CAAC,CAAE,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;QACjC,CAAC;QAED,oBAAoB;QACpB,KAAK,MAAM,OAAO,IAAI,iBAAiB,EAAE,CAAC;YACxC,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;gBAC1C,MAAM,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;gBAChD,MAAM,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;gBAChD,IAAI,WAAW,CAAC,IAAI,CAAC,IAAI,WAAW,CAAC,IAAI,CAAC;oBAAE,SAAS;gBACrD,SAAS,CAAC,IAAI,EAAE,eAAe,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC;gBAC7C,SAAS,CAAC,IAAI,EAAE,eAAe,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC;gBAC7C,SAAS,CAAC,IAAI,CAAC;oBACb,QAAQ,EAAE,IAAI;oBACd,QAAQ,EAAE,OAAO,CAAC,QAAQ;oBAC1B,QAAQ,EAAE,IAAI;oBACd,SAAS,EAAE,OAAO,CAAC,EAAE;oBACrB,SAAS,EAAE,IAAI,CAAC,SAAS;oBACzB,QAAQ,EAAE,OAAO,CAAC,IAAI,CAAC;iBACxB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,oEAAoE;QACpE,KAAK,MAAM,OAAO,IAAI,iBAAiB,EAAE,CAAC;YACxC,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;gBAC1C,MAAM,UAAU,GAAG,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;gBACxD,IAAI,WAAW,CAAC,UAAU,CAAC;oBAAE,SAAS;gBACtC,SAAS,CAAC,UAAU,EAAE,eAAe,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,CAAC;gBACzD,SAAS,CAAC,IAAI,CAAC;oBACb,QAAQ,EAAE,EAAE;oBACZ,QAAQ,EAAE,OAAO,CAAC,QAAQ;oBAC1B,QAAQ,EAAE,UAAU;oBACpB,SAAS,EAAE,OAAO,CAAC,EAAE;oBACrB,SAAS,EAAE,IAAI,CAAC,SAAS;oBACzB,QAAQ,EAAE,OAAO,CAAC,IAAI,CAAC;oBACvB,OAAO,EAAE,IAAI;iBACd,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,wCAAwC;QACxC,KAAK,MAAM,EAAE,IAAI,cAAc,EAAE,CAAC;YAChC,IAAI,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAClB,MAAM,CAAC,IAAI,CAAC;oBACV,WAAW,EAAE,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC;oBAC/B,SAAS,EAAE,OAAO,CAAC,EAAE;oBACrB,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;oBACxD,SAAS,EAAE,IAAI,CAAC,SAAS;iBAC1B,CAAC,CAAC;gBACH,MAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO;QACL,QAAQ,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;QAC9B,SAAS;QACT,MAAM;KACP,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAC5C,QAAyB,EACzB,IAA0B;IAE1B,MAAM,WAAW,GAAsB,EAAE,CAAC;IAC1C,MAAM,YAAY,GAAwB,EAAE,CAAC;IAC7C,MAAM,SAAS,GAAqB,EAAE,CAAC;IAEvC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACjD,WAAW,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;QACrC,YAAY,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC;QACvC,SAAS,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAED,OAAO;QACL,QAAQ,EAAE,WAAW;QACrB,SAAS,EAAE,YAAY;QACvB,MAAM,EAAE,SAAS;KAClB,CAAC;AACJ,CAAC"}
|