memo-grafter 0.1.0 → 0.2.0

package/dist/pipeline/RetrieverPipeline.d.ts

@@ -0,0 +1,11 @@
+import type { GraphStore } from "../store/index.js";
+import type { EmbedAdapter, RetrievalResult, RetrieverConfig } from "../types.js";
+export declare class RetrieverPipeline {
+    private store;
+    private embedder;
+    private config;
+    constructor(store: GraphStore, embedder: EmbedAdapter, config: RetrieverConfig);
+    run(query: string, sessionId: string): Promise<RetrievalResult>;
+    private buildBlocks;
+}
+//# sourceMappingURL=RetrieverPipeline.d.ts.map

package/dist/pipeline/RetrieverPipeline.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"RetrieverPipeline.d.ts","sourceRoot":"","sources":["../../src/pipeline/RetrieverPipeline.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,KAAK,EACV,YAAY,EAEZ,eAAe,EACf,eAAe,EAEhB,MAAM,aAAa,CAAC;AAWrB,qBAAa,iBAAiB;IAE1B,OAAO,CAAC,KAAK;IACb,OAAO,CAAC,QAAQ;IAChB,OAAO,CAAC,MAAM;gBAFN,KAAK,EAAE,UAAU,EACjB,QAAQ,EAAE,YAAY,EACtB,MAAM,EAAE,eAAe;IAG3B,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC;YAsDvD,WAAW;CA8B1B"}

package/dist/pipeline/RetrieverPipeline.js

@@ -0,0 +1,73 @@
+import { buildFactRetrievalPrompt, formatFactBlock, } from "../prompts/factRetrievalPrompt.js";
+import { countApproxTokens } from "../utils/text/tokenCount.js";
+export class RetrieverPipeline {
+    store;
+    embedder;
+    config;
+    constructor(store, embedder, config) {
+        this.store = store;
+        this.embedder = embedder;
+        this.config = config;
+    }
+    async run(query, sessionId) {
+        const limit = this.config.limit ?? 10;
+        const minSimilarity = this.config.minSimilarity ?? 0.6;
+        const tokenBudget = this.config.tokenBudget ?? 1200;
+        const embedding = await this.embedder.embed(query);
+        const searchedFacts = await this.store.searchMemories(embedding, sessionId, limit, minSimilarity);
+        const activeFacts = searchedFacts.filter((fact) => fact.decayed === false && fact.supersededBy == null);
+        if (activeFacts.length === 0) {
+            return {
+                facts: [],
+                nodes: [],
+                systemPrompt: buildFactRetrievalPrompt([]),
+                tokenCount: 0,
+            };
+        }
+        const rankedBlocks = (await this.buildBlocks(activeFacts, sessionId))
+            .sort((a, b) => b.score - a.score);
+        const includedBlocks = [];
+        const facts = [];
+        const nodes = [];
+        let tokenCount = 0;
+        for (const block of rankedBlocks) {
+            const formattedBlock = formatFactBlock(block.facts, block.parentNode);
+            const blockTokenCount = countApproxTokens(formattedBlock);
+            if (tokenCount + blockTokenCount > tokenBudget) {
+                break;
+            }
+            includedBlocks.push(formattedBlock);
+            facts.push(...block.facts);
+            nodes.push(block.parentNode);
+            tokenCount += blockTokenCount;
+        }
+        return {
+            facts,
+            nodes,
+            systemPrompt: buildFactRetrievalPrompt(includedBlocks),
+            tokenCount,
+        };
+    }
+    async buildBlocks(facts, sessionId) {
+        const factsByTopic = new Map();
+        for (const fact of facts) {
+            const topicFacts = factsByTopic.get(fact.topicNodeId) ?? [];
+            topicFacts.push(fact);
+            factsByTopic.set(fact.topicNodeId, topicFacts);
+        }
+        const blocks = [];
+        for (const [topicNodeId, topicFacts] of factsByTopic) {
+            const parentNode = await this.store.getTopicNode(topicNodeId, sessionId);
+            if (!parentNode) {
+                continue;
+            }
+            blocks.push({
+                facts: topicFacts,
+                parentNode,
+                score: Math.max(...topicFacts.map((fact) => fact.similarity)),
+            });
+        }
+        return blocks;
+    }
+}
+//# sourceMappingURL=RetrieverPipeline.js.map

package/dist/pipeline/RetrieverPipeline.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"RetrieverPipeline.js","sourceRoot":"","sources":["../../src/pipeline/RetrieverPipeline.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,wBAAwB,EACxB,eAAe,GAChB,MAAM,mCAAmC,CAAC;AAS3C,OAAO,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AAUhE,MAAM,OAAO,iBAAiB;IAElB;IACA;IACA;IAHV,YACU,KAAiB,EACjB,QAAsB,EACtB,MAAuB;QAFvB,UAAK,GAAL,KAAK,CAAY;QACjB,aAAQ,GAAR,QAAQ,CAAc;QACtB,WAAM,GAAN,MAAM,CAAiB;IAC9B,CAAC;IAEJ,KAAK,CAAC,GAAG,CAAC,KAAa,EAAE,SAAiB;QACxC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC;QACtC,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,GAAG,CAAC;QACvD,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,IAAI,CAAC;QAEpD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACnD,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,cAAc,CACnD,SAAS,EACT,SAAS,EACT,KAAK,EACL,aAAa,CACd,CAAC;QACF,MAAM,WAAW,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAChD,IAAI,CAAC,OAAO,KAAK,KAAK,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CACpD,CAAC;QAEF,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,OAAO;gBACL,KAAK,EAAE,EAAE;gBACT,KAAK,EAAE,EAAE;gBACT,YAAY,EAAE,wBAAwB,CAAC,EAAE,CAAC;gBAC1C,UAAU,EAAE,CAAC;aACd,CAAC;QACJ,CAAC;QAED,MAAM,YAAY,GAAG,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;aAClE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QACrC,MAAM,cAAc,GAAa,EAAE,CAAC;QACpC,MAAM,KAAK,GAAuB,EAAE,CAAC;QACrC,MAAM,KAAK,GAAgB,EAAE,CAAC;QAC9B,IAAI,UAAU,GAAG,CAAC,CAAC;QAEnB,KAAK,MAAM,KAAK,IAAI,YAAY,EAAE,CAAC;YACjC,MAAM,cAAc,GAAG,eAAe,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;YACtE,MAAM,eAAe,GAAG,iBAAiB,CAAC,cAAc,CAAC,CAAC;YAE1D,IAAI,UAAU,GAAG,eAAe,GAAG,WAAW,EAAE,CAAC;gBAC/C,MAAM;YACR,CAAC;YAED,cAAc,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACpC,KAAK,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;YAC3B,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC7B,UAAU,IAAI,eAAe,CAAC;QAChC,CAAC;QAED,OAAO;YACL,KAAK;YACL,KAAK;YACL,YAAY,EAAE,wBAAwB,CAAC,cAAc,CAAC;YACtD,UAAU;SACX,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,WAAW,CACvB,KAAyB,EACzB,SAAiB;QAEjB,MAAM,YAAY,GAAG,IAAI,GAAG,EAA8B,CAAC;QAE3D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC;YAC5D,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtB,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;QACjD,CAAC;QAED,MAAM,MAAM,GAAqB,EAAE,CAAC;QAEpC,KAAK,MAAM,CAAC,WAAW,EAAE,UAAU,CAAC,IAAI,YAAY,EAAE,CAAC;YACrD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;YAEzE,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,SAAS;YACX,CAAC;YAED,MAAM,CAAC,IAAI,CAAC;gBACV,KAAK,EAAE,UAAU;gBACjB,UAAU;gBACV,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;aAC9D,CAAC,CAAC;QACL,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;CACF"}

package/dist/pipeline/SegmentProcessor.d.ts

@@ -1,4 +1,4 @@
-import type { GraphStore } from "../store/GraphStore.js";
+import type { GraphStore } from "../store/index.js";
 import type { EmbedAdapter, LLMAdapter, Message, TopicNode } from "../types.js";
 import type { DriftSegment } from "./TopicDriftDetector.js";
 export declare class SegmentProcessor {
@@ -6,6 +6,7 @@ export declare class SegmentProcessor {
     private readonly llm;
     private readonly embedder;
     private readonly config;
+    private lastExtraction;
     constructor(store: GraphStore, llm: LLMAdapter, embedder: EmbedAdapter, config: {
         topK: number;
         semanticThreshold: number;
@@ -14,6 +15,6 @@ export declare class SegmentProcessor {
     private createSegment;
     private nodeRunner;
     private nodeProcessor;
-    private parseExtraction;
+    private processMemories;
 }
 //# sourceMappingURL=SegmentProcessor.d.ts.map

package/dist/pipeline/SegmentProcessor.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"SegmentProcessor.d.ts","sourceRoot":"","sources":["../../src/pipeline/SegmentProcessor.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,KAAK,EAAE,YAAY,EAAE,UAAU,EAAE,OAAO,EAAE,SAAS,EAAgB,MAAM,aAAa,CAAC;AAE9F,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAE5D,qBAAa,gBAAgB;IAEzB,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,GAAG;IACpB,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,MAAM;gBAHN,KAAK,EAAE,UAAU,EACjB,GAAG,EAAE,UAAU,EACf,QAAQ,EAAE,YAAY,EACtB,MAAM,EAAE;QACvB,IAAI,EAAE,MAAM,CAAC;QACb,iBAAiB,EAAE,MAAM,CAAC;KAC3B;IAGG,OAAO,CAAC,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC;YAKlF,aAAa;YAYb,UAAU;YAMV,aAAa;IA8C3B,OAAO,CAAC,eAAe;CAiBxB"}
+{"version":3,"file":"SegmentProcessor.d.ts","sourceRoot":"","sources":["../../src/pipeline/SegmentProcessor.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,KAAK,EACV,YAAY,EAEZ,UAAU,EAEV,OAAO,EAEP,SAAS,EAEV,MAAM,aAAa,CAAC;AAMrB,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAE5D,qBAAa,gBAAgB;IAUzB,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,GAAG;IACpB,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,MAAM;IAZzB,OAAO,CAAC,cAAc,CAMpB;gBAGiB,KAAK,EAAE,UAAU,EACjB,GAAG,EAAE,UAAU,EACf,QAAQ,EAAE,YAAY,EACtB,MAAM,EAAE;QACvB,IAAI,EAAE,MAAM,CAAC;QACb,iBAAiB,EAAE,MAAM,CAAC;KAC3B;IAGG,OAAO,CAAC,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC;YAOlF,aAAa;YAYb,UAAU;YAMV,aAAa;YA0Bb,eAAe;CAwC9B"}

package/dist/pipeline/SegmentProcessor.js

@@ -1,10 +1,18 @@
 import { randomUUID } from "node:crypto";
-import { normalizeText } from "../utils/normalizeText.js";
+import { buildSegmentExtractionPrompt } from "../prompts/segmentExtractionPrompt.js";
+import { buildSegmentSummary, formatMemoryEmbeddingText, parseSegmentExtraction, } from "../utils/extraction/segmentExtraction.js";
 export class SegmentProcessor {
     store;
     llm;
     embedder;
     config;
+    lastExtraction = {
+        label: "Unknown",
+        userIntent: "",
+        outcome: "",
+        open: null,
+        memories: [],
+    };
     constructor(store, llm, embedder, config) {
         this.store = store;
         this.llm = llm;
@@ -13,7 +21,9 @@ export class SegmentProcessor {
     }
     async process(segment, messages, sessionId) {
         const savedSegment = await this.createSegment(segment, sessionId);
-        return this.nodeRunner(savedSegment, messages);
+        const topicNode = await this.nodeRunner(savedSegment, messages);
+        await this.processMemories(this.lastExtraction.memories, savedSegment, topicNode);
+        return topicNode;
     }
     async createSegment(segment, sessionId) {
         return this.store.saveSegment({
@@ -33,36 +43,18 @@ export class SegmentProcessor {
     }
     async nodeProcessor(messages, segment) {
         const segmentMessages = messages.slice(segment.startIndex, segment.endIndex + 1);
-        const messageContent = segmentMessages
-            .map((message, index) => `Message ${index + 1}:\n[${message.role}] ${normalizeText(message.content) ?? message.content}`)
-            .join("\n\n");
-        const extractionPrompt = [
-            "Analyze this conversation segment and extract structured memory for a future chatbot.",
-            "",
-            "Identify:",
-            "- The core topic being discussed",
-            "- What the user wanted to know, do, or accomplish",
-            "- What the assistant provided, decided, or concluded",
-            "- Any unresolved questions or open threads",
-            "",
-            "Respond only in this exact format with no extra text:",
-            "LABEL: <3-6 word topic label>",
-            "USER_INTENT: <two-three sentences describing what the user was trying to achieve or understand>",
-            "OUTCOME: <one-two sentences describing what was concluded, decided, or provided>",
-            "OPEN: <one sentence describing any unresolved question or follow-up, or None if fully resolved>",
-            "",
-            "Conversation segment:",
-            messageContent,
-        ].join("\n");
+        const extractionPrompt = buildSegmentExtractionPrompt(segmentMessages);
         const raw = await this.llm.complete([{ role: "user", content: extractionPrompt }]);
-        const extracted = this.parseExtraction(raw);
-        const embedding = await this.embedder.embed(extracted.summary);
+        const extracted = parseSegmentExtraction(raw);
+        this.lastExtraction = extracted;
+        const summary = buildSegmentSummary(extracted);
+        const embedding = await this.embedder.embed(summary);
         return {
             id: randomUUID(),
             sessionId: segment.sessionId,
             segmentId: segment.id,
             label: extracted.label,
-            summary: extracted.summary,
+            summary,
             embedding,
             messageRange: [segment.startIndex, segment.endIndex],
             topicOrder: segment.topicOrder,
@@ -73,20 +65,40 @@ export class SegmentProcessor {
             createdAt: new Date(),
         };
     }
-    parseExtraction(raw) {
-        const label = raw.match(/^LABEL:\s*(.+)$/im)?.[1]?.trim() ?? "Unknown";
-        const userIntent = raw.match(/^USER_INTENT:\s*(.+)$/im)?.[1]?.trim() ?? "";
-        const outcome = raw.match(/^OUTCOME:\s*(.+)$/im)?.[1]?.trim() ?? "";
-        const open = raw.match(/^OPEN:\s*(.+)$/im)?.[1]?.trim() ?? "";
-        const parts = [
-            userIntent && `User wanted: ${userIntent}`,
-            outcome && `Outcome: ${outcome}`,
-            open && open.toLowerCase() !== "none" && `Still open: ${open}`,
-        ].filter(Boolean);
-        return {
-            label,
-            summary: parts.join(" "),
-        };
+    async processMemories(memories, segment, topicNode) {
+        if (memories.length === 0)
+            return;
+        try {
+            const nodes = [];
+            for (const memory of memories) {
+                const embedding = await this.embedder.embed(formatMemoryEmbeddingText(memory));
+                nodes.push({
+                    id: randomUUID(),
+                    segmentId: segment.id,
+                    topicNodeId: topicNode.id,
+                    sessionId: segment.sessionId,
+                    agentId: topicNode.agentId,
+                    agentColor: topicNode.agentColor,
+                    fleetId: topicNode.fleetId,
+                    memoryType: memory.memoryType,
+                    sourceType: "conversation",
+                    subject: memory.subject,
+                    predicate: memory.predicate,
+                    value: memory.value,
+                    confidence: memory.confidence,
+                    embedding,
+                    sourceUrl: null,
+                    sourceTitle: null,
+                    supersededBy: null,
+                    decayed: false,
+                });
+            }
+            await this.store.insertMemories(nodes);
+            await this.store.buildMemoryEdges(topicNode.id, segment.sessionId, this.config.semanticThreshold);
+        }
+        catch (error) {
+            console.warn("SegmentProcessor memory processing warning:", error);
+        }
     }
 }
 //# sourceMappingURL=SegmentProcessor.js.map

package/dist/pipeline/SegmentProcessor.js.map

@@ -1 +1 @@
-{"version":3,"file":"SegmentProcessor.js","sourceRoot":"","sources":["../../src/pipeline/SegmentProcessor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGzC,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAG1D,MAAM,OAAO,gBAAgB;IAER;IACA;IACA;IACA;IAJnB,YACmB,KAAiB,EACjB,GAAe,EACf,QAAsB,EACtB,MAGhB;QANgB,UAAK,GAAL,KAAK,CAAY;QACjB,QAAG,GAAH,GAAG,CAAY;QACf,aAAQ,GAAR,QAAQ,CAAc;QACtB,WAAM,GAAN,MAAM,CAGtB;IACA,CAAC;IAEJ,KAAK,CAAC,OAAO,CAAC,OAAqB,EAAE,QAAmB,EAAE,SAAiB;QACzE,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAClE,OAAO,IAAI,CAAC,UAAU,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;IACjD,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,OAAqB,EAAE,SAAiB;QAClE,OAAO,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC;YAC5B,EAAE,EAAE,UAAU,EAAE;YAChB,SAAS;YACT,UAAU,EAAE,OAAO,CAAC,KAAK;YACzB,QAAQ,EAAE,OAAO,CAAC,GAAG;YACrB,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,OAAqB,EAAE,QAAmB;QACjE,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACzD,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAChC,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,QAAmB,EAAE,OAAqB;QACpE,MAAM,eAAe,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC;QACjF,MAAM,cAAc,GAAG,eAAe;aACnC,GAAG,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC,WAAW,KAAK,GAAG,CAAC,OAAO,OAAO,CAAC,IAAI,KAAK,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;aACxH,IAAI,CAAC,MAAM,CAAC,CAAC;QAEhB,MAAM,gBAAgB,GAAG;YACvB,uFAAuF;YACvF,EAAE;YACF,WAAW;YACX,kCAAkC;YAClC,mDAAmD;YACnD,sDAAsD;YACtD,4CAA4C;YAC5C,EAAE;YACF,uDAAuD;YACvD,+BAA+B;YAC/B,iGAAiG;YACjG,kFAAkF;YAClF,iGAAiG;YACjG,EAAE;YACF,uBAAuB;YACvB,cAAc;SACf,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC;QACnF,MAAM,SAAS,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;QAC5C,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAE/D,OAAO;YACL,EAAE,EAAE,UAAU,EAAE;YAChB,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,SAAS,EAAE,OAAO,CAAC,EAAE;YACrB,KAAK,EAAE,SAAS,CAAC,KAAK;YACtB,OAAO,EAAE,SAAS,CAAC,OAAO;YAC1B,SAAS;YACT,YAAY,EAAE,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,QAAQ,CAAC;YACpD,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,UAAU,EAAE,IAAI;YAChB,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,IAAI;YACb,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;IACJ,CAAC;IAEO,eAAe,CAAC,GAAW;QACjC,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,mBAAmB,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,SAAS,CAAC;QACvE,MAAM,UAAU,GAAG,GAAG,CAAC,KAAK,CAAC,yBAAyB,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAC3E,MAAM,OAAO,GAAG,GAAG,CAAC,KAAK,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QACpE,MAAM,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAE9D,MAAM,KAAK,GAAG;YACZ,UAAU,IAAI,gBAAgB,UAAU,EAAE;YAC1C,OAAO,IAAI,YAAY,OAAO,EAAE;YAChC,IAAI,IAAI,IAAI,CAAC,WAAW,EAAE,KAAK,MAAM,IAAI,eAAe,IAAI,EAAE;SAC/D,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAElB,OAAO;YACL,KAAK;YACL,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC;SACzB,CAAC;IACJ,CAAC;CACF"}
+{"version":3,"file":"SegmentProcessor.js","sourceRoot":"","sources":["../../src/pipeline/SegmentProcessor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,4BAA4B,EAAE,MAAM,uCAAuC,CAAC;AAYrF,OAAO,EACL,mBAAmB,EACnB,yBAAyB,EACzB,sBAAsB,GACvB,MAAM,0CAA0C,CAAC;AAGlD,MAAM,OAAO,gBAAgB;IAUR;IACA;IACA;IACA;IAZX,cAAc,GAA4B;QAChD,KAAK,EAAE,SAAS;QAChB,UAAU,EAAE,EAAE;QACd,OAAO,EAAE,EAAE;QACX,IAAI,EAAE,IAAI;QACV,QAAQ,EAAE,EAAE;KACb,CAAC;IAEF,YACmB,KAAiB,EACjB,GAAe,EACf,QAAsB,EACtB,MAGhB;QANgB,UAAK,GAAL,KAAK,CAAY;QACjB,QAAG,GAAH,GAAG,CAAY;QACf,aAAQ,GAAR,QAAQ,CAAc;QACtB,WAAM,GAAN,MAAM,CAGtB;IACA,CAAC;IAEJ,KAAK,CAAC,OAAO,CAAC,OAAqB,EAAE,QAAmB,EAAE,SAAiB;QACzE,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAClE,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAChE,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;QAClF,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,OAAqB,EAAE,SAAiB;QAClE,OAAO,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC;YAC5B,EAAE,EAAE,UAAU,EAAE;YAChB,SAAS;YACT,UAAU,EAAE,OAAO,CAAC,KAAK;YACzB,QAAQ,EAAE,OAAO,CAAC,GAAG;YACrB,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,OAAqB,EAAE,QAAmB;QACjE,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACzD,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAChC,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,QAAmB,EAAE,OAAqB;QACpE,MAAM,eAAe,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC;QACjF,MAAM,gBAAgB,GAAG,4BAA4B,CAAC,eAAe,CAAC,CAAC;QACvE,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC;QACnF,MAAM,SAAS,GAAG,sBAAsB,CAAC,GAAG,CAAC,CAAC;QAC9C,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;QAChC,MAAM,OAAO,GAAG,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAC/C,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAErD,OAAO;YACL,EAAE,EAAE,UAAU,EAAE;YAChB,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,SAAS,EAAE,OAAO,CAAC,EAAE;YACrB,KAAK,EAAE,SAAS,CAAC,KAAK;YACtB,OAAO;YACP,SAAS;YACT,YAAY,EAAE,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,QAAQ,CAAC;YACpD,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,UAAU,EAAE,IAAI;YAChB,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,IAAI;YACb,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,eAAe,CAC3B,QAA2B,EAC3B,OAAqB,EACrB,SAAoB;QAEpB,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAElC,IAAI,CAAC;YACH,MAAM,KAAK,GAAuB,EAAE,CAAC;YAErC,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;gBAC9B,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,yBAAyB,CAAC,MAAM,CAAC,CAAC,CAAC;gBAC/E,KAAK,CAAC,IAAI,CAAC;oBACT,EAAE,EAAE,UAAU,EAAE;oBAChB,SAAS,EAAE,OAAO,CAAC,EAAE;oBACrB,WAAW,EAAE,SAAS,CAAC,EAAE;oBACzB,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,OAAO,EAAE,SAAS,CAAC,OAAO;oBAC1B,UAAU,EAAE,SAAS,CAAC,UAAU;oBAChC,OAAO,EAAE,SAAS,CAAC,OAAO;oBAC1B,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,UAAU,EAAE,cAAc;oBAC1B,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,SAAS,EAAE,MAAM,CAAC,SAAS;oBAC3B,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,SAAS;oBACT,SAAS,EAAE,IAAI;oBACf,WAAW,EAAE,IAAI;oBACjB,YAAY,EAAE,IAAI;oBAClB,OAAO,EAAE,KAAK;iBACf,CAAC,CAAC;YACL,CAAC;YAED,MAAM,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YACvC,MAAM,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,SAAS,CAAC,EAAE,EAAE,OAAO,CAAC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;QACpG,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,IAAI,CAAC,6CAA6C,EAAE,KAAK,CAAC,CAAC;QACrE,CAAC;IACH,CAAC;CACF"}

package/dist/pipeline/TopicDriftDetector.d.ts

@@ -1,22 +1,40 @@
-import type { DriftMode, Message } from "../types.js";
+import type { DriftMode, LLMAdapter, Message, TopicNode } from "../types.js";
 export interface DriftSegment {
     start: number;
     end: number;
     topicOrder: number;
     driftScore: number;
 }
+export interface DriftDetectionResult {
+    segments: DriftSegment[];
+    reentryMap: Map<number, string>;
+}
 export declare class TopicDriftDetector {
     private readonly config;
+    private readonly llm?;
     constructor(config: {
         windowSize: number;
         threshold: number;
         mode: DriftMode;
         minSegmentMessages: number;
-    });
+        llmAmbiguityDetection: boolean;
+        reentryDetection: boolean;
+        reentryThreshold: number;
+    }, llm?: LLMAdapter | undefined);
+    /**
+     * @deprecated Use detectSegments() instead. This sync wrapper does not run LLM
+     * ambiguity detection or reentry detection.
+     */
     detect(messages: Message[], embeddings: number[][]): number[];
-    detectSegments(messages: Message[], embeddings: number[][]): DriftSegment[];
+    detectSegments(messages: Message[], embeddings: number[][], existingNodes?: TopicNode[]): Promise<DriftDetectionResult>;
     private detectBoundaries;
+    private detectBoundariesSync;
     private detectIntentBoundaries;
     private detectWindowBoundaries;
+    private detectIntentBoundariesSync;
+    private detectWindowBoundariesSync;
+    private resolveAmbiguousScore;
+    private classifyIntentShift;
+    private maybeCheckReentry;
 }
 //# sourceMappingURL=TopicDriftDetector.d.ts.map

package/dist/pipeline/TopicDriftDetector.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"TopicDriftDetector.d.ts","sourceRoot":"","sources":["../../src/pipeline/TopicDriftDetector.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAItD,MAAM,WAAW,YAAY;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACpB;AAOD,qBAAa,kBAAkB;IAE3B,OAAO,CAAC,QAAQ,CAAC,MAAM;gBAAN,MAAM,EAAE;QACvB,UAAU,EAAE,MAAM,CAAC;QACnB,SAAS,EAAE,MAAM,CAAC;QAClB,IAAI,EAAE,SAAS,CAAC;QAChB,kBAAkB,EAAE,MAAM,CAAC;KAC5B;IAGH,MAAM,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,EAAE,GAAG,MAAM,EAAE;IAI7D,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,EAAE,GAAG,YAAY,EAAE;IA8B3E,OAAO,CAAC,gBAAgB;IAQxB,OAAO,CAAC,sBAAsB;IA+B9B,OAAO,CAAC,sBAAsB;CA8B/B"}
+{"version":3,"file":"TopicDriftDetector.d.ts","sourceRoot":"","sources":["../../src/pipeline/TopicDriftDetector.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAM7E,MAAM,WAAW,YAAY;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,oBAAoB;IACnC,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,UAAU,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACjC;AAQD,qBAAa,kBAAkB;IAE3B,OAAO,CAAC,QAAQ,CAAC,MAAM;IASvB,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC;gBATJ,MAAM,EAAE;QACvB,UAAU,EAAE,MAAM,CAAC;QACnB,SAAS,EAAE,MAAM,CAAC;QAClB,IAAI,EAAE,SAAS,CAAC;QAChB,kBAAkB,EAAE,MAAM,CAAC;QAC3B,qBAAqB,EAAE,OAAO,CAAC;QAC/B,gBAAgB,EAAE,OAAO,CAAC;QAC1B,gBAAgB,EAAE,MAAM,CAAC;KAC1B,EACgB,GAAG,CAAC,EAAE,UAAU,YAAA;IAGnC;;;OAGG;IACH,MAAM,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,EAAE,GAAG,MAAM,EAAE;IAIvD,cAAc,CAClB,QAAQ,EAAE,OAAO,EAAE,EACnB,UAAU,EAAE,MAAM,EAAE,EAAE,EACtB,aAAa,GAAE,SAAS,EAAO,GAC9B,OAAO,CAAC,oBAAoB,CAAC;YAuClB,gBAAgB;IAY9B,OAAO,CAAC,oBAAoB;YAQd,sBAAsB;YAmDtB,sBAAsB;IAyDpC,OAAO,CAAC,0BAA0B;IAyClC,OAAO,CAAC,0BAA0B;YA2CpB,qBAAqB;YAerB,mBAAmB;IAWjC,OAAO,CAAC,iBAAiB;CAI1B"}

package/dist/pipeline/TopicDriftDetector.js

@@ -1,18 +1,28 @@
-import { cosineSimilarity } from "../utils/drift/cosineSimilarity.js";
+import { buildIntentShiftPrompt } from "../prompts/intentShiftPrompt.js";
+import { computeDriftScore } from "../utils/drift/driftScore.js";
+import { findReentryNodeId } from "../utils/drift/reentryMatch.js";
 import { avg } from "../utils/drift/vectorAvg.js";
 export class TopicDriftDetector {
     config;
-    constructor(config) {
+    llm;
+    constructor(config, llm) {
         this.config = config;
+        this.llm = llm;
     }
+    /**
+     * @deprecated Use detectSegments() instead. This sync wrapper does not run LLM
+     * ambiguity detection or reentry detection.
+     */
     detect(messages, embeddings) {
-        return this.detectBoundaries(messages, embeddings).map((boundary) => boundary.index);
+        return this.detectBoundariesSync(messages, embeddings).map((boundary) => boundary.index);
     }
-    detectSegments(messages, embeddings) {
-        if (messages.length === 0)
-            return [];
-        const boundaries = this.detectBoundaries(messages, embeddings);
+    async detectSegments(messages, embeddings, existingNodes = []) {
+        if (messages.length === 0) {
+            return { segments: [], reentryMap: new Map() };
+        }
+        const boundaries = await this.detectBoundaries(messages, embeddings, existingNodes);
         const segments = [];
+        const reentryMap = new Map();
         let start = 0;
         for (const [index, boundary] of boundaries.entries()) {
             const end = boundary.index - 1;
@@ -24,6 +34,10 @@ export class TopicDriftDetector {
                     driftScore: boundary.score,
                 });
             }
+            const nextTopicOrder = index + 2;
+            if (boundary.reentryNodeId) {
+                reentryMap.set(nextTopicOrder, boundary.reentryNodeId);
+            }
             start = boundary.index;
         }
         segments.push({
@@ -32,18 +46,98 @@ export class TopicDriftDetector {
             topicOrder: segments.length + 1,
             driftScore: 0,
         });
-        return segments;
+        return { segments, reentryMap };
+    }
+    async detectBoundaries(messages, embeddings, existingNodes) {
+        if (this.config.mode === "intent") {
+            return this.detectIntentBoundaries(messages, embeddings, existingNodes);
+        }
+        return this.detectWindowBoundaries(messages, embeddings, existingNodes);
     }
-    detectBoundaries(messages, embeddings) {
+    detectBoundariesSync(messages, embeddings) {
         if (this.config.mode === "intent") {
-            return this.detectIntentBoundaries(messages, embeddings);
+            return this.detectIntentBoundariesSync(messages, embeddings);
+        }
+        return this.detectWindowBoundariesSync(messages, embeddings);
+    }
+    async detectIntentBoundaries(messages, embeddings, existingNodes) {
+        const boundaries = [];
+        let segmentStart = 0;
+        let topicEmbeddings = [];
+        let previousUserEmbedding = null;
+        for (let index = 0; index < messages.length; index += 1) {
+            const message = messages[index];
+            const embedding = embeddings[index];
+            if (!message || !embedding || message.role !== "user")
+                continue;
+            if (topicEmbeddings.length === 0) {
+                topicEmbeddings = [embedding];
+                previousUserEmbedding = embedding;
+                continue;
+            }
+            let driftScore = computeDriftScore(embedding, message, avg(topicEmbeddings), previousUserEmbedding, this.config.threshold);
+            driftScore = await this.resolveAmbiguousScore(driftScore, messages.slice(Math.max(0, index - 3), index), message);
+            const segmentLength = index - segmentStart;
+            if (driftScore > this.config.threshold && segmentLength >= this.config.minSegmentMessages) {
+                const reentryNodeId = this.maybeCheckReentry(embedding, existingNodes);
+                boundaries.push({
+                    index,
+                    score: driftScore,
+                    ...(reentryNodeId ? { reentryNodeId } : {}),
+                });
+                segmentStart = index;
+                topicEmbeddings = [embedding];
+                previousUserEmbedding = embedding;
+                continue;
+            }
+            topicEmbeddings.push(embedding);
+            previousUserEmbedding = embedding;
+        }
+        return boundaries;
+    }
+    async detectWindowBoundaries(messages, embeddings, existingNodes) {
+        const boundaries = [];
+        const halfWindowSize = Math.floor(this.config.windowSize / 2);
+        if (halfWindowSize < 1)
+            return boundaries;
+        let segmentStart = 0;
+        let window = [];
+        for (let index = 0; index < messages.length; index += 1) {
+            const embedding = embeddings[index];
+            if (!embedding)
+                continue;
+            window = [...window, { index, embedding }].slice(-this.config.windowSize);
+            if (window.length < this.config.windowSize)
+                continue;
+            const previous = window.slice(0, halfWindowSize);
+            const current = window.slice(-halfWindowSize);
+            const boundaryIndex = window.at(-halfWindowSize)?.index ?? index;
+            const currentMessage = messages[boundaryIndex];
+            const currentEmbedding = embeddings[boundaryIndex] ?? avg(current.map((item) => item.embedding));
+            const previousEmbedding = previous.at(-1)?.embedding ?? null;
+            if (!currentMessage)
+                continue;
+            let driftScore = computeDriftScore(currentEmbedding, currentMessage, avg(previous.map((item) => item.embedding)), previousEmbedding, this.config.threshold);
+            driftScore = await this.resolveAmbiguousScore(driftScore, messages.slice(Math.max(0, boundaryIndex - 3), boundaryIndex), currentMessage);
+            const segmentLength = boundaryIndex - segmentStart;
+            if (driftScore > this.config.threshold && segmentLength >= this.config.minSegmentMessages) {
+                const reentryNodeId = this.maybeCheckReentry(currentEmbedding, existingNodes);
+                boundaries.push({
+                    index: boundaryIndex,
+                    score: driftScore,
+                    ...(reentryNodeId ? { reentryNodeId } : {}),
+                });
+                segmentStart = boundaryIndex;
+                window = window.filter((item) => item.index >= boundaryIndex);
+            }
         }
-        return this.detectWindowBoundaries(messages, embeddings);
+        return boundaries;
     }
-    detectIntentBoundaries(messages, embeddings) {
+    detectIntentBoundariesSync(messages, embeddings) {
         const boundaries = [];
         let segmentStart = 0;
         let topicEmbeddings = [];
+        let previousUserEmbedding = null;
         for (let index = 0; index < messages.length; index += 1) {
             const message = messages[index];
             const embedding = embeddings[index];
@@ -51,21 +145,24 @@ export class TopicDriftDetector {
                 continue;
             if (topicEmbeddings.length === 0) {
                 topicEmbeddings = [embedding];
+                previousUserEmbedding = embedding;
                 continue;
             }
-            const driftScore = 1 - cosineSimilarity(avg(topicEmbeddings), embedding);
+            const driftScore = computeDriftScore(embedding, message, avg(topicEmbeddings), previousUserEmbedding, this.config.threshold);
             const segmentLength = index - segmentStart;
             if (driftScore > this.config.threshold && segmentLength >= this.config.minSegmentMessages) {
                 boundaries.push({ index, score: driftScore });
                 segmentStart = index;
                 topicEmbeddings = [embedding];
+                previousUserEmbedding = embedding;
                 continue;
             }
             topicEmbeddings.push(embedding);
+            previousUserEmbedding = embedding;
         }
         return boundaries;
     }
-    detectWindowBoundaries(messages, embeddings) {
+    detectWindowBoundariesSync(messages, embeddings) {
         const boundaries = [];
         const halfWindowSize = Math.floor(this.config.windowSize / 2);
         if (halfWindowSize < 1)
@@ -79,10 +176,15 @@ export class TopicDriftDetector {
             window = [...window, { index, embedding }].slice(-this.config.windowSize);
             if (window.length < this.config.windowSize)
                 continue;
-            const previous = window.slice(0, halfWindowSize).map((item) => item.embedding);
-            const current = window.slice(-halfWindowSize).map((item) => item.embedding);
-            const driftScore = 1 - cosineSimilarity(avg(previous), avg(current));
+            const previous = window.slice(0, halfWindowSize);
+            const current = window.slice(-halfWindowSize);
             const boundaryIndex = window.at(-halfWindowSize)?.index ?? index;
+            const currentMessage = messages[boundaryIndex];
+            const currentEmbedding = embeddings[boundaryIndex] ?? avg(current.map((item) => item.embedding));
+            const previousEmbedding = previous.at(-1)?.embedding ?? null;
+            if (!currentMessage)
+                continue;
+            const driftScore = computeDriftScore(currentEmbedding, currentMessage, avg(previous.map((item) => item.embedding)), previousEmbedding, this.config.threshold);
             const segmentLength = boundaryIndex - segmentStart;
             if (driftScore > this.config.threshold && segmentLength >= this.config.minSegmentMessages) {
                 boundaries.push({ index: boundaryIndex, score: driftScore });
@@ -92,5 +194,29 @@ export class TopicDriftDetector {
         }
         return boundaries;
     }
+    async resolveAmbiguousScore(driftScore, recentMessages, currentMessage) {
+        const ambiguousLow = this.config.threshold * 0.6;
+        const ambiguousHigh = this.config.threshold * 1.2;
+        const isAmbiguous = driftScore >= ambiguousLow && driftScore <= ambiguousHigh;
+        if (!isAmbiguous || !this.config.llmAmbiguityDetection || !this.llm)
+            return driftScore;
+        const isShift = await this.classifyIntentShift(recentMessages, currentMessage);
+        return isShift ? this.config.threshold * 1.5 : this.config.threshold * 0.4;
+    }
+    async classifyIntentShift(recentMessages, currentMessage) {
+        const prompt = buildIntentShiftPrompt(recentMessages, currentMessage);
+        try {
+            const response = await this.llm?.complete([{ role: "user", content: prompt }]);
+            return response?.includes("NEW_TOPIC") ?? false;
+        }
+        catch {
+            return false;
+        }
+    }
+    maybeCheckReentry(embedding, existingNodes) {
+        if (!this.config.reentryDetection || existingNodes.length === 0)
+            return undefined;
+        return findReentryNodeId(embedding, existingNodes, this.config.reentryThreshold) ?? undefined;
+    }
 }
 //# sourceMappingURL=TopicDriftDetector.js.map

package/dist/pipeline/TopicDriftDetector.js.map

@@ -1 +1 @@
-{"version":3,"file":"TopicDriftDetector.js","sourceRoot":"","sources":["../../src/pipeline/TopicDriftDetector.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,MAAM,oCAAoC,CAAC;AACtE,OAAO,EAAE,GAAG,EAAE,MAAM,6BAA6B,CAAC;AAclD,MAAM,OAAO,kBAAkB;IAEV;IADnB,YACmB,MAKhB;QALgB,WAAM,GAAN,MAAM,CAKtB;IACA,CAAC;IAEJ,MAAM,CAAC,QAAmB,EAAE,UAAsB;QAChD,OAAO,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACvF,CAAC;IAED,cAAc,CAAC,QAAmB,EAAE,UAAsB;QACxD,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAErC,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAC/D,MAAM,QAAQ,GAAmB,EAAE,CAAC;QACpC,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC;YACrD,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,GAAG,CAAC,CAAC;YAC/B,IAAI,GAAG,IAAI,KAAK,EAAE,CAAC;gBACjB,QAAQ,CAAC,IAAI,CAAC;oBACZ,KAAK;oBACL,GAAG;oBACH,UAAU,EAAE,KAAK,GAAG,CAAC;oBACrB,UAAU,EAAE,QAAQ,CAAC,KAAK;iBAC3B,CAAC,CAAC;YACL,CAAC;YACD,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;QACzB,CAAC;QAED,QAAQ,CAAC,IAAI,CAAC;YACZ,KAAK;YACL,GAAG,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC;YACxB,UAAU,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC;YAC/B,UAAU,EAAE,CAAC;SACd,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,gBAAgB,CAAC,QAAmB,EAAE,UAAsB;QAClE,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAClC,OAAO,IAAI,CAAC,sBAAsB,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAC3D,CAAC;QAED,OAAO,IAAI,CAAC,sBAAsB,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;IAC3D,CAAC;IAEO,sBAAsB,CAAC,QAAmB,EAAE,UAAsB;QACxE,MAAM,UAAU,GAAoB,EAAE,CAAC;QACvC,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,eAAe,GAAe,EAAE,CAAC;QAErC,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;YACxD,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;YAChC,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;YACpC,IAAI,CAAC,OAAO,IAAI,CAAC,SAAS,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM;gBAAE,SAAS;YAEhE,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACjC,eAAe,GAAG,CAAC,SAAS,CAAC,CAAC;gBAC9B,SAAS;YACX,CAAC;YAED,MAAM,UAAU,GAAG,CAAC,GAAG,gBAAgB,CAAC,GAAG,CAAC,eAAe,CAAC,EAAE,SAAS,CAAC,CAAC;YACzE,MAAM,aAAa,GAAG,KAAK,GAAG,YAAY,CAAC;YAE3C,IAAI,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;gBAC1F,UAAU,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;gBAC9C,YAAY,GAAG,KAAK,CAAC;gBACrB,eAAe,GAAG,CAAC,SAAS,CAAC,CAAC;gBAC9B,SAAS;YACX,CAAC;YAED,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAClC,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,sBAAsB,CAAC,QAAmB,EAAE,UAAsB;QACxE,MAAM,UAAU,GAAoB,EAAE,CAAC;QACvC,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;QAC9D,IAAI,cAAc,GAAG,CAAC;YAAE,OAAO,UAAU,CAAC;QAE1C,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,MAAM,GAAkD,EAAE,CAAC;QAE/D,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;YACxD,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;YACpC,IAAI,CAAC,SAAS;gBAAE,SAAS;YAEzB,MAAM,GAAG,CAAC,GAAG,MAAM,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAC1E,IAAI,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU;gBAAE,SAAS;YAErD,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC/E,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC5E,MAAM,UAAU,GAAG,CAAC,GAAG,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;YACrE,MAAM,aAAa,GAAG,MAAM,CAAC,EAAE,CAAC,CAAC,cAAc,CAAC,EAAE,KAAK,IAAI,KAAK,CAAC;YACjE,MAAM,aAAa,GAAG,aAAa,GAAG,YAAY,CAAC;YAEnD,IAAI,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;gBAC1F,UAAU,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;gBAC7D,YAAY,GAAG,aAAa,CAAC;gBAC7B,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,IAAI,aAAa,CAAC,CAAC;YAChE,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;CACF"}
+{"version":3,"file":"TopicDriftDetector.js","sourceRoot":"","sources":["../../src/pipeline/TopicDriftDetector.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,sBAAsB,EAAE,MAAM,iCAAiC,CAAC;AACzE,OAAO,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AACjE,OAAO,EAAE,iBAAiB,EAAE,MAAM,gCAAgC,CAAC;AACnE,OAAO,EAAE,GAAG,EAAE,MAAM,6BAA6B,CAAC;AAoBlD,MAAM,OAAO,kBAAkB;IAEV;IASA;IAVnB,YACmB,MAQhB,EACgB,GAAgB;QAThB,WAAM,GAAN,MAAM,CAQtB;QACgB,QAAG,GAAH,GAAG,CAAa;IAChC,CAAC;IAEJ;;;OAGG;IACH,MAAM,CAAC,QAAmB,EAAE,UAAsB;QAChD,OAAO,IAAI,CAAC,oBAAoB,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAC3F,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,QAAmB,EACnB,UAAsB,EACtB,gBAA6B,EAAE;QAE/B,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,GAAG,EAAE,EAAE,CAAC;QACjD,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,UAAU,EAAE,aAAa,CAAC,CAAC;QACpF,MAAM,QAAQ,GAAmB,EAAE,CAAC;QACpC,MAAM,UAAU,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC7C,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC;YACrD,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,GAAG,CAAC,CAAC;YAC/B,IAAI,GAAG,IAAI,KAAK,EAAE,CAAC;gBACjB,QAAQ,CAAC,IAAI,CAAC;oBACZ,KAAK;oBACL,GAAG;oBACH,UAAU,EAAE,KAAK,GAAG,CAAC;oBACrB,UAAU,EAAE,QAAQ,CAAC,KAAK;iBAC3B,CAAC,CAAC;YACL,CAAC;YAED,MAAM,cAAc,GAAG,KAAK,GAAG,CAAC,CAAC;YACjC,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;gBAC3B,UAAU,CAAC,GAAG,CAAC,cAAc,EAAE,QAAQ,CAAC,aAAa,CAAC,CAAC;YACzD,CAAC;YAED,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;QACzB,CAAC;QAED,QAAQ,CAAC,IAAI,CAAC;YACZ,KAAK;YACL,GAAG,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC;YACxB,UAAU,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC;YAC/B,UAAU,EAAE,CAAC;SACd,CAAC,CAAC;QAEH,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC;IAClC,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC5B,QAAmB,EACnB,UAAsB,EACtB,aAA0B;QAE1B,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAClC,OAAO,IAAI,CAAC,sBAAsB,CAAC,QAAQ,EAAE,UAAU,EAAE,aAAa,CAAC,CAAC;QAC1E,CAAC;QAED,OAAO,IAAI,CAAC,sBAAsB,CAAC,QAAQ,EAAE,UAAU,EAAE,aAAa,CAAC,CAAC;IAC1E,CAAC;IAEO,oBAAoB,CAAC,QAAmB,EAAE,UAAsB;QACtE,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAClC,OAAO,IAAI,CAAC,0BAA0B,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAC/D,CAAC;QAED,OAAO,IAAI,CAAC,0BAA0B,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;IAC/D,CAAC;IAEO,KAAK,CAAC,sBAAsB,CAClC,QAAmB,EACnB,UAAsB,EACtB,aAA0B;QAE1B,MAAM,UAAU,GAAoB,EAAE,CAAC;QACvC,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,eAAe,GAAe,EAAE,CAAC;QACrC,IAAI,qBAAqB,GAAoB,IAAI,CAAC;QAElD,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;YACxD,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;YAChC,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;YACpC,IAAI,CAAC,OAAO,IAAI,CAAC,SAAS,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM;gBAAE,SAAS;YAEhE,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACjC,eAAe,GAAG,CAAC,SAAS,CAAC,CAAC;gBAC9B,qBAAqB,GAAG,SAAS,CAAC;gBAClC,SAAS;YACX,CAAC;YAED,IAAI,UAAU,GAAG,iBAAiB,CAChC,SAAS,EACT,OAAO,EACP,GAAG,CAAC,eAAe,CAAC,EACpB,qBAAqB,EACrB,IAAI,CAAC,MAAM,CAAC,SAAS,CACtB,CAAC;YACF,UAAU,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,UAAU,EAAE,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,OAAO,CAAC,CAAC;YAElH,MAAM,aAAa,GAAG,KAAK,GAAG,YAAY,CAAC;YAC3C,IAAI,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;gBAC1F,MAAM,aAAa,GAAG,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;gBACvE,UAAU,CAAC,IAAI,CAAC;oBACd,KAAK;oBACL,KAAK,EAAE,UAAU;oBACjB,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC5C,CAAC,CAAC;gBACH,YAAY,GAAG,KAAK,CAAC;gBACrB,eAAe,GAAG,CAAC,SAAS,CAAC,CAAC;gBAC9B,qBAAqB,GAAG,SAAS,CAAC;gBAClC,SAAS;YACX,CAAC;YAED,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAChC,qBAAqB,GAAG,SAAS,CAAC;QACpC,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,KAAK,CAAC,sBAAsB,CAClC,QAAmB,EACnB,UAAsB,EACtB,aAA0B;QAE1B,MAAM,UAAU,GAAoB,EAAE,CAAC;QACvC,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;QAC9D,IAAI,cAAc,GAAG,CAAC;YAAE,OAAO,UAAU,CAAC;QAE1C,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,MAAM,GAAkD,EAAE,CAAC;QAE/D,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;YACxD,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;YACpC,IAAI,CAAC,SAAS;gBAAE,SAAS;YAEzB,MAAM,GAAG,CAAC,GAAG,MAAM,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAC1E,IAAI,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU;gBAAE,SAAS;YAErD,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC;YACjD,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,CAAC;YAC9C,MAAM,aAAa,GAAG,MAAM,CAAC,EAAE,CAAC,CAAC,cAAc,CAAC,EAAE,KAAK,IAAI,KAAK,CAAC;YACjE,MAAM,cAAc,GAAG,QAAQ,CAAC,aAAa,CAAC,CAAC;YAC/C,MAAM,gBAAgB,GAAG,UAAU,CAAC,aAAa,CAAC,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;YACjG,MAAM,iBAAiB,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,SAAS,IAAI,IAAI,CAAC;YAE7D,IAAI,CAAC,cAAc;gBAAE,SAAS;YAE9B,IAAI,UAAU,GAAG,iBAAiB,CAChC,gBAAgB,EAChB,cAAc,EACd,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAC3C,iBAAiB,EACjB,IAAI,CAAC,MAAM,CAAC,SAAS,CACtB,CAAC;YACF,UAAU,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAC3C,UAAU,EACV,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,aAAa,GAAG,CAAC,CAAC,EAAE,aAAa,CAAC,EAC7D,cAAc,CACf,CAAC;YAEF,MAAM,aAAa,GAAG,aAAa,GAAG,YAAY,CAAC;YACnD,IAAI,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;gBAC1F,MAAM,aAAa,GAAG,IAAI,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,aAAa,CAAC,CAAC;gBAC9E,UAAU,CAAC,IAAI,CAAC;oBACd,KAAK,EAAE,aAAa;oBACpB,KAAK,EAAE,UAAU;oBACjB,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC5C,CAAC,CAAC;gBACH,YAAY,GAAG,aAAa,CAAC;gBAC7B,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,IAAI,aAAa,CAAC,CAAC;YAChE,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,0BAA0B,CAAC,QAAmB,EAAE,UAAsB;QAC5E,MAAM,UAAU,GAAoB,EAAE,CAAC;QACvC,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,eAAe,GAAe,EAAE,CAAC;QACrC,IAAI,qBAAqB,GAAoB,IAAI,CAAC;QAElD,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;YACxD,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;YAChC,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;YACpC,IAAI,CAAC,OAAO,IAAI,CAAC,SAAS,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM;gBAAE,SAAS;YAEhE,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACjC,eAAe,GAAG,CAAC,SAAS,CAAC,CAAC;gBAC9B,qBAAqB,GAAG,SAAS,CAAC;gBAClC,SAAS;YACX,CAAC;YAED,MAAM,UAAU,GAAG,iBAAiB,CAClC,SAAS,EACT,OAAO,EACP,GAAG,CAAC,eAAe,CAAC,EACpB,qBAAqB,EACrB,IAAI,CAAC,MAAM,CAAC,SAAS,CACtB,CAAC;YACF,MAAM,aAAa,GAAG,KAAK,GAAG,YAAY,CAAC;YAE3C,IAAI,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;gBAC1F,UAAU,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;gBAC9C,YAAY,GAAG,KAAK,CAAC;gBACrB,eAAe,GAAG,CAAC,SAAS,CAAC,CAAC;gBAC9B,qBAAqB,GAAG,SAAS,CAAC;gBAClC,SAAS;YACX,CAAC;YAED,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAChC,qBAAqB,GAAG,SAAS,CAAC;QACpC,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,0BAA0B,CAAC,QAAmB,EAAE,UAAsB;QAC5E,MAAM,UAAU,GAAoB,EAAE,CAAC;QACvC,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;QAC9D,IAAI,cAAc,GAAG,CAAC;YAAE,OAAO,UAAU,CAAC;QAE1C,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,MAAM,GAAkD,EAAE,CAAC;QAE/D,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;YACxD,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;YACpC,IAAI,CAAC,SAAS;gBAAE,SAAS;YAEzB,MAAM,GAAG,CAAC,GAAG,MAAM,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAC1E,IAAI,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU;gBAAE,SAAS;YAErD,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC;YACjD,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,CAAC;YAC9C,MAAM,aAAa,GAAG,MAAM,CAAC,EAAE,CAAC,CAAC,cAAc,CAAC,EAAE,KAAK,IAAI,KAAK,CAAC;YACjE,MAAM,cAAc,GAAG,QAAQ,CAAC,aAAa,CAAC,CAAC;YAC/C,MAAM,gBAAgB,GAAG,UAAU,CAAC,aAAa,CAAC,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;YACjG,MAAM,iBAAiB,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,SAAS,IAAI,IAAI,CAAC;YAE7D,IAAI,CAAC,cAAc;gBAAE,SAAS;YAE9B,MAAM,UAAU,GAAG,iBAAiB,CAClC,gBAAgB,EAChB,cAAc,EACd,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAC3C,iBAAiB,EACjB,IAAI,CAAC,MAAM,CAAC,SAAS,CACtB,CAAC;YACF,MAAM,aAAa,GAAG,aAAa,GAAG,YAAY,CAAC;YAEnD,IAAI,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;gBAC1F,UAAU,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;gBAC7D,YAAY,GAAG,aAAa,CAAC;gBAC7B,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,IAAI,aAAa,CAAC,CAAC;YAChE,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,KAAK,CAAC,qBAAqB,CACjC,UAAkB,EAClB,cAAyB,EACzB,cAAuB;QAEvB,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,GAAG,CAAC;QACjD,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,GAAG,CAAC;QAClD,MAAM,WAAW,GAAG,UAAU,IAAI,YAAY,IAAI,UAAU,IAAI,aAAa,CAAC;QAE9E,IAAI,CAAC,WAAW,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,qBAAqB,IAAI,CAAC,IAAI,CAAC,GAAG;YAAE,OAAO,UAAU,CAAC;QAEvF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;QAC/E,OAAO,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,GAAG,CAAC;IAC7E,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAAC,cAAyB,EAAE,cAAuB;QAClF,MAAM,MAAM,GAAG,sBAAsB,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;QAEtE,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;YAC/E,OAAO,QAAQ,EAAE,QAAQ,CAAC,WAAW,CAAC,IAAI,KAAK,CAAC;QAClD,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAEO,iBAAiB,CAAC,SAAmB,EAAE,aAA0B;QACvE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,SAAS,CAAC;QAClF,OAAO,iBAAiB,CAAC,SAAS,EAAE,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,IAAI,SAAS,CAAC;IAChG,CAAC;CACF"}

package/dist/prompts/factRetrievalPrompt.d.ts

@@ -0,0 +1,4 @@
+import type { MemoryNode, TopicNode } from "../types.js";
+export declare function formatFactBlock(facts: MemoryNode[], parentNode: TopicNode): string;
+export declare function buildFactRetrievalPrompt(blocks: string[]): string;
+//# sourceMappingURL=factRetrievalPrompt.d.ts.map

package/dist/prompts/factRetrievalPrompt.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"factRetrievalPrompt.d.ts","sourceRoot":"","sources":["../../src/prompts/factRetrievalPrompt.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAezD,wBAAgB,eAAe,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE,UAAU,EAAE,SAAS,GAAG,MAAM,CASlF;AAED,wBAAgB,wBAAwB,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,CAOjE"}

package/dist/prompts/factRetrievalPrompt.js

@@ -0,0 +1,25 @@
+const FACT_RETRIEVAL_SUBHEADER = "The following facts were retrieved from prior conversation memory. " +
+    "Use them to inform your response where relevant.";
+function compactSummary(summary) {
+    return summary.replace(/\s+/g, " ").trim();
+}
+function formatFactLine(fact) {
+    return `[${fact.memoryType.toUpperCase()}] ${fact.subject} → ${fact.predicate}: ` +
+        `${fact.value} (conf: ${fact.confidence.toFixed(2)})`;
+}
+export function formatFactBlock(facts, parentNode) {
+    const header = `## ${parentNode.label} (order: ${parentNode.topicOrder})`;
+    const summary = `> ${compactSummary(parentNode.summary)}`;
+    if (facts.length === 0) {
+        return [header, "", summary].join("\n") + "\n";
+    }
+    return [header, ...facts.map(formatFactLine), "", summary].join("\n") + "\n";
+}
+export function buildFactRetrievalPrompt(blocks) {
+    if (blocks.length === 0) {
+        return ["### Retrieved Memory", FACT_RETRIEVAL_SUBHEADER].join("\n") + "\n";
+    }
+    return ["### Retrieved Memory", FACT_RETRIEVAL_SUBHEADER, "", blocks.join("\n---\n")]
+        .join("\n") + "\n";
+}
+//# sourceMappingURL=factRetrievalPrompt.js.map

package/dist/prompts/factRetrievalPrompt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"factRetrievalPrompt.js","sourceRoot":"","sources":["../../src/prompts/factRetrievalPrompt.ts"],"names":[],"mappings":"AAEA,MAAM,wBAAwB,GAC5B,qEAAqE;IACrE,kDAAkD,CAAC;AAErD,SAAS,cAAc,CAAC,OAAe;IACrC,OAAO,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;AAC7C,CAAC;AAED,SAAS,cAAc,CAAC,IAAgB;IACtC,OAAO,IAAI,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,KAAK,IAAI,CAAC,OAAO,MAAM,IAAI,CAAC,SAAS,IAAI;QAC/E,GAAG,IAAI,CAAC,KAAK,WAAW,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;AAC1D,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,KAAmB,EAAE,UAAqB;IACxE,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,YAAY,UAAU,CAAC,UAAU,GAAG,CAAC;IAC1E,MAAM,OAAO,GAAG,KAAK,cAAc,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;IAE1D,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,OAAO,CAAC,MAAM,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACjD,CAAC;IAED,OAAO,CAAC,MAAM,EAAE,GAAG,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;AAC/E,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,MAAgB;IACvD,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,OAAO,CAAC,sBAAsB,EAAE,wBAAwB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC9E,CAAC;IAED,OAAO,CAAC,sBAAsB,EAAE,wBAAwB,EAAE,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;SAClF,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;AACvB,CAAC"}

package/dist/prompts/historyCompressionPrompt.d.ts

@@ -0,0 +1,3 @@
+import type { TopicNode } from "../types.js";
+export declare function formatCompressedTopic(node: TopicNode): string;
+//# sourceMappingURL=historyCompressionPrompt.d.ts.map