n8n-nodes-engram 0.2.0

package/dist/extraction/LlmClient.js

@@ -0,0 +1,73 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LlmClient = void 0;
+class LlmClient {
+    constructor(config) {
+        var _a;
+        this.apiKey = config.apiKey;
+        this.baseUrl = config.baseUrl.replace(/\/$/, '');
+        this.model = config.model;
+        this.timeoutMs = (_a = config.timeoutMs) !== null && _a !== void 0 ? _a : 30000;
+    }
+    async chat(messages, options) {
+        var _a, _b, _c, _d;
+        const url = `${this.baseUrl}/chat/completions`;
+        const body = {
+            model: this.model,
+            messages,
+            temperature: (_a = options === null || options === void 0 ? void 0 : options.temperature) !== null && _a !== void 0 ? _a : 0.0,
+        };
+        if (options === null || options === void 0 ? void 0 : options.maxTokens) {
+            body.max_tokens = options.maxTokens;
+        }
+        if (options === null || options === void 0 ? void 0 : options.responseFormat) {
+            body.response_format = options.responseFormat;
+        }
+        const controller = new AbortController();
+        const timeout = setTimeout(() => controller.abort(), this.timeoutMs);
+        let response;
+        try {
+            response = await fetch(url, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    Authorization: `Bearer ${this.apiKey}`,
+                },
+                body: JSON.stringify(body),
+                signal: controller.signal,
+            });
+        }
+        catch (error) {
+            if (error instanceof Error && error.name === 'AbortError') {
+                throw new Error(`LLM API request timed out after ${this.timeoutMs}ms`);
+            }
+            throw error;
+        }
+        finally {
+            clearTimeout(timeout);
+        }
+        if (!response.ok) {
+            const errorText = await response.text();
+            throw new Error(`LLM API error (${response.status}): ${errorText}`);
+        }
+        const data = (await response.json());
+        return {
+            content: (_d = (_c = (_b = data.choices[0]) === null || _b === void 0 ? void 0 : _b.message) === null || _c === void 0 ? void 0 : _c.content) !== null && _d !== void 0 ? _d : '',
+            usage: data.usage,
+        };
+    }
+    async chatJson(messages, options) {
+        const response = await this.chat(messages, {
+            ...options,
+            responseFormat: { type: 'json_object' },
+        });
+        try {
+            return JSON.parse(response.content);
+        }
+        catch {
+            throw new Error(`Failed to parse LLM JSON response: ${response.content.slice(0, 200)}`);
+        }
+    }
+}
+exports.LlmClient = LlmClient;
+//# sourceMappingURL=LlmClient.js.map

package/dist/extraction/LlmClient.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"LlmClient.js","sourceRoot":"","sources":["../../src/extraction/LlmClient.ts"],"names":[],"mappings":";;;AA2BA,MAAa,SAAS;IAMpB,YAAY,MAAuB;;QACjC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACjD,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC1B,IAAI,CAAC,SAAS,GAAG,MAAA,MAAM,CAAC,SAAS,mCAAI,KAAM,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,IAAI,CACR,QAAsB,EACtB,OAIC;;QAED,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,mBAAmB,CAAC;QAE/C,MAAM,IAAI,GAA4B;YACpC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,QAAQ;YACR,WAAW,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW,mCAAI,GAAG;SACzC,CAAC;QAEF,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,SAAS,EAAE,CAAC;YACvB,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC;QACtC,CAAC;QAED,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,cAAc,EAAE,CAAC;YAC5B,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,cAAc,CAAC;QAChD,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAErE,IAAI,QAAkB,CAAC;QACvB,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAC1B,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;iBACvC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;gBAC1B,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAC1D,MAAM,IAAI,KAAK,CAAC,mCAAmC,IAAI,CAAC,SAAS,IAAI,CAAC,CAAC;YACzE,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,OAAO,CAAC,CAAC;QACxB,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACxC,MAAM,IAAI,KAAK,CAAC,kBAAkB,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,CAAC,CAAC;QACtE,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAGlC,CAAC;QAEF,OAAO;YACL,OAAO,EAAE,MAAA,MAAA,MAAA,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,0CAAE,OAAO,0CAAE,OAAO,mCAAI,EAAE;YAChD,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,QAAsB,EACtB,OAAsD;QAEtD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;YACzC,GAAG,OAAO;YACV,cAAc,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE;SACxC,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAM,CAAC;QAC3C,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,KAAK,CAAC,sCAAsC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;QAC1F,CAAC;IACH,CAAC;CACF;AA3FD,8BA2FC"}

package/dist/extraction/RelationshipExtractor.d.ts

@@ -0,0 +1,12 @@
+import type { LlmClient } from './LlmClient';
+export interface ExtractedRelationship {
+    source_entity: string;
+    target_entity: string;
+    name: string;
+    fact: string;
+}
+export declare class RelationshipExtractor {
+    private llm;
+    constructor(llm: LlmClient);
+    extract(humanMessage: string, aiMessage: string, entityNames: string[]): Promise<ExtractedRelationship[]>;
+}

package/dist/extraction/RelationshipExtractor.js

@@ -0,0 +1,38 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RelationshipExtractor = void 0;
+const prompts_1 = require("./prompts");
+class RelationshipExtractor {
+    constructor(llm) {
+        this.llm = llm;
+    }
+    async extract(humanMessage, aiMessage, entityNames) {
+        if (entityNames.length < 2)
+            return [];
+        try {
+            const result = await this.llm.chatJson([
+                { role: 'system', content: prompts_1.RELATIONSHIP_EXTRACTION_SYSTEM },
+                {
+                    role: 'user',
+                    content: (0, prompts_1.relationshipExtractionUser)(humanMessage, aiMessage, entityNames),
+                },
+            ]);
+            if (!result.relationships || !Array.isArray(result.relationships)) {
+                return [];
+            }
+            const entitySet = new Set(entityNames.map((n) => n.toLowerCase()));
+            return result.relationships.filter((r) => typeof r.source_entity === 'string' &&
+                typeof r.target_entity === 'string' &&
+                typeof r.name === 'string' &&
+                typeof r.fact === 'string' &&
+                entitySet.has(r.source_entity.toLowerCase()) &&
+                entitySet.has(r.target_entity.toLowerCase()));
+        }
+        catch (error) {
+            console.warn('Engram: Relationship extraction failed:', error.message);
+            return [];
+        }
+    }
+}
+exports.RelationshipExtractor = RelationshipExtractor;
+//# sourceMappingURL=RelationshipExtractor.js.map

package/dist/extraction/RelationshipExtractor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"RelationshipExtractor.js","sourceRoot":"","sources":["../../src/extraction/RelationshipExtractor.ts"],"names":[],"mappings":";;;AACA,uCAAuF;AAavF,MAAa,qBAAqB;IAGhC,YAAY,GAAc;QACxB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,OAAO,CACX,YAAoB,EACpB,SAAiB,EACjB,WAAqB;QAErB,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC;YAAE,OAAO,EAAE,CAAC;QAEtC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,QAAQ,CAA+B;gBACnE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,wCAA8B,EAAE;gBAC3D;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,IAAA,oCAA0B,EAAC,YAAY,EAAE,SAAS,EAAE,WAAW,CAAC;iBAC1E;aACF,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;gBAClE,OAAO,EAAE,CAAC;YACZ,CAAC;YAGD,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;YACnE,OAAO,MAAM,CAAC,aAAa,CAAC,MAAM,CAChC,CAAC,CAAC,EAAE,EAAE,CACJ,OAAO,CAAC,CAAC,aAAa,KAAK,QAAQ;gBACnC,OAAO,CAAC,CAAC,aAAa,KAAK,QAAQ;gBACnC,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ;gBAC1B,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ;gBAC1B,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,aAAa,CAAC,WAAW,EAAE,CAAC;gBAC5C,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,aAAa,CAAC,WAAW,EAAE,CAAC,CAC/C,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,IAAI,CAAC,yCAAyC,EAAG,KAAe,CAAC,OAAO,CAAC,CAAC;YAClF,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;CACF;AA3CD,sDA2CC"}

package/dist/extraction/index.d.ts

@@ -0,0 +1,6 @@
+export { LlmClient, type LlmClientConfig } from './LlmClient';
+export { EntityExtractor, type ExtractedEntity } from './EntityExtractor';
+export { RelationshipExtractor, type ExtractedRelationship } from './RelationshipExtractor';
+export { EntityDeduplicator } from './EntityDeduplicator';
+export { ContradictionDetector, type ContradictionResolution } from './ContradictionDetector';
+export { ExtractionPipeline, type ExtractionPipelineConfig } from './ExtractionPipeline';

package/dist/extraction/index.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ExtractionPipeline = exports.ContradictionDetector = exports.EntityDeduplicator = exports.RelationshipExtractor = exports.EntityExtractor = exports.LlmClient = void 0;
+var LlmClient_1 = require("./LlmClient");
+Object.defineProperty(exports, "LlmClient", { enumerable: true, get: function () { return LlmClient_1.LlmClient; } });
+var EntityExtractor_1 = require("./EntityExtractor");
+Object.defineProperty(exports, "EntityExtractor", { enumerable: true, get: function () { return EntityExtractor_1.EntityExtractor; } });
+var RelationshipExtractor_1 = require("./RelationshipExtractor");
+Object.defineProperty(exports, "RelationshipExtractor", { enumerable: true, get: function () { return RelationshipExtractor_1.RelationshipExtractor; } });
+var EntityDeduplicator_1 = require("./EntityDeduplicator");
+Object.defineProperty(exports, "EntityDeduplicator", { enumerable: true, get: function () { return EntityDeduplicator_1.EntityDeduplicator; } });
+var ContradictionDetector_1 = require("./ContradictionDetector");
+Object.defineProperty(exports, "ContradictionDetector", { enumerable: true, get: function () { return ContradictionDetector_1.ContradictionDetector; } });
+var ExtractionPipeline_1 = require("./ExtractionPipeline");
+Object.defineProperty(exports, "ExtractionPipeline", { enumerable: true, get: function () { return ExtractionPipeline_1.ExtractionPipeline; } });
+//# sourceMappingURL=index.js.map

package/dist/extraction/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/extraction/index.ts"],"names":[],"mappings":";;;AAAA,yCAA8D;AAArD,sGAAA,SAAS,OAAA;AAClB,qDAA0E;AAAjE,kHAAA,eAAe,OAAA;AACxB,iEAA4F;AAAnF,8HAAA,qBAAqB,OAAA;AAC9B,2DAA0D;AAAjD,wHAAA,kBAAkB,OAAA;AAC3B,iEAA8F;AAArF,8HAAA,qBAAqB,OAAA;AAC9B,2DAAyF;AAAhF,wHAAA,kBAAkB,OAAA"}

package/dist/extraction/prompts.d.ts

@@ -0,0 +1,16 @@
+export declare const ENTITY_EXTRACTION_SYSTEM: string;
+export declare function entityExtractionUser(humanMessage: string, aiMessage: string, entityTypes: string[], existingEntities: string[]): string;
+export declare const RELATIONSHIP_EXTRACTION_SYSTEM: string;
+export declare function relationshipExtractionUser(humanMessage: string, aiMessage: string, entities: string[]): string;
+export declare const DEDUPLICATION_SYSTEM: string;
+export declare function deduplicationUser(entity1: {
+    name: string;
+    summary: string;
+    entity_type: string;
+}, entity2: {
+    name: string;
+    summary: string;
+    entity_type: string;
+}): string;
+export declare const CONTRADICTION_SYSTEM: string;
+export declare function contradictionUser(existingFact: string, newFact: string, sourceEntity: string, targetEntity: string): string;

package/dist/extraction/prompts.js

@@ -0,0 +1,101 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CONTRADICTION_SYSTEM = exports.DEDUPLICATION_SYSTEM = exports.RELATIONSHIP_EXTRACTION_SYSTEM = exports.ENTITY_EXTRACTION_SYSTEM = void 0;
+exports.entityExtractionUser = entityExtractionUser;
+exports.relationshipExtractionUser = relationshipExtractionUser;
+exports.deduplicationUser = deduplicationUser;
+exports.contradictionUser = contradictionUser;
+exports.ENTITY_EXTRACTION_SYSTEM = [
+    'You are an entity extraction system. Extract entities from the conversation.',
+    '',
+    'Rules:',
+    '- Extract real-world entities mentioned: people, organizations, locations, concepts, events, products, etc.',
+    '- Each entity needs: name (string), entity_type (string), summary (one sentence description)',
+    '- Use consistent naming: proper nouns as-is, concepts in lowercase',
+    '- entity_type should be one of the allowed types provided',
+    '- Return ONLY valid JSON, no extra text',
+    '',
+    'Output format:',
+    '{"entities": [{"name": "Alice", "entity_type": "person", "summary": "A software engineer"}]}',
+].join('\n');
+function entityExtractionUser(humanMessage, aiMessage, entityTypes, existingEntities) {
+    const typesStr = entityTypes.join(', ');
+    const parts = [
+        'Extract entities from this conversation turn.',
+        '',
+        'Allowed entity types: ' + typesStr,
+    ];
+    if (existingEntities.length > 0) {
+        parts.push('Already known entities (avoid duplicates): ' + existingEntities.join(', '));
+    }
+    parts.push('', 'Human: ' + humanMessage, 'AI: ' + aiMessage);
+    return parts.join('\n');
+}
+exports.RELATIONSHIP_EXTRACTION_SYSTEM = [
+    'You are a relationship extraction system. Extract facts and relationships between entities.',
+    '',
+    'Rules:',
+    '- Extract relationships as factual statements connecting two entities',
+    '- source_entity and target_entity must be exact entity names from the provided list',
+    '- name should be SCREAMING_SNAKE_CASE (e.g., WORKS_AT, LIVES_IN, KNOWS)',
+    '- fact should be a natural language statement of the relationship',
+    '- Only extract relationships where both entities exist in the provided list',
+    '- Return ONLY valid JSON, no extra text',
+    '',
+    'Output format:',
+    '{"relationships": [{"source_entity": "Alice", "target_entity": "Acme Corp", "name": "WORKS_AT", "fact": "Alice works at Acme Corp as a senior engineer"}]}',
+].join('\n');
+function relationshipExtractionUser(humanMessage, aiMessage, entities) {
+    const parts = [
+        'Extract relationships from this conversation turn.',
+        '',
+        'Available entities: ' + entities.join(', '),
+        '',
+        'Human: ' + humanMessage,
+        'AI: ' + aiMessage,
+    ];
+    return parts.join('\n');
+}
+exports.DEDUPLICATION_SYSTEM = [
+    'You are an entity deduplication system. Determine if two entities refer to the same real-world thing.',
+    '',
+    'Rules:',
+    '- Consider name variations (Bob/Robert, NYC/New York City)',
+    '- Consider context clues from summaries',
+    '- Return ONLY valid JSON',
+    '',
+    'Output format:',
+    '{"is_duplicate": true, "merged_summary": "Combined summary of both entities"}',
+].join('\n');
+function deduplicationUser(entity1, entity2) {
+    const parts = [
+        'Are these the same entity?',
+        '',
+        'Entity 1: ' + entity1.name + ' (' + entity1.entity_type + ') - ' + entity1.summary,
+        'Entity 2: ' + entity2.name + ' (' + entity2.entity_type + ') - ' + entity2.summary,
+    ];
+    return parts.join('\n');
+}
+exports.CONTRADICTION_SYSTEM = [
+    'You are a contradiction detection system. Determine if a new fact contradicts an existing fact.',
+    '',
+    'Rules:',
+    '- A contradiction means the new fact makes the old fact no longer true',
+    '- Example: "Alice lives in London" contradicts "Alice lives in Tokyo" (can only live in one place)',
+    '- Example: "Alice knows Python" does NOT contradict "Alice knows JavaScript" (can know both)',
+    '- Return ONLY valid JSON',
+    '',
+    'Output format:',
+    '{"is_contradiction": true, "explanation": "The new fact about living location replaces the old one"}',
+].join('\n');
+function contradictionUser(existingFact, newFact, sourceEntity, targetEntity) {
+    const parts = [
+        'Does the new fact contradict the existing fact?',
+        '',
+        'Entities: ' + sourceEntity + ' -> ' + targetEntity,
+        'Existing fact: ' + existingFact,
+        'New fact: ' + newFact,
+    ];
+    return parts.join('\n');
+}
+//# sourceMappingURL=prompts.js.map

package/dist/extraction/prompts.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompts.js","sourceRoot":"","sources":["../../src/extraction/prompts.ts"],"names":[],"mappings":";;;AAmBA,oDAmBC;AAiBD,gEAcC;AAcD,8CAWC;AAeD,8CAcC;AAtHY,QAAA,wBAAwB,GAAG;IACtC,8EAA8E;IAC9E,EAAE;IACF,QAAQ;IACR,6GAA6G;IAC7G,8FAA8F;IAC9F,oEAAoE;IACpE,2DAA2D;IAC3D,yCAAyC;IACzC,EAAE;IACF,gBAAgB;IAChB,8FAA8F;CAC/F,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAEb,SAAgB,oBAAoB,CAClC,YAAoB,EACpB,SAAiB,EACjB,WAAqB,EACrB,gBAA0B;IAE1B,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxC,MAAM,KAAK,GAAG;QACZ,+CAA+C;QAC/C,EAAE;QACF,wBAAwB,GAAG,QAAQ;KACpC,CAAC;IAEF,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChC,KAAK,CAAC,IAAI,CAAC,6CAA6C,GAAG,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IAC1F,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,GAAG,YAAY,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC;IAC7D,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAEY,QAAA,8BAA8B,GAAG;IAC5C,6FAA6F;IAC7F,EAAE;IACF,QAAQ;IACR,uEAAuE;IACvE,qFAAqF;IACrF,yEAAyE;IACzE,mEAAmE;IACnE,6EAA6E;IAC7E,yCAAyC;IACzC,EAAE;IACF,gBAAgB;IAChB,4JAA4J;CAC7J,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAEb,SAAgB,0BAA0B,CACxC,YAAoB,EACpB,SAAiB,EACjB,QAAkB;IAElB,MAAM,KAAK,GAAG;QACZ,oDAAoD;QACpD,EAAE;QACF,sBAAsB,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC;QAC5C,EAAE;QACF,SAAS,GAAG,YAAY;QACxB,MAAM,GAAG,SAAS;KACnB,CAAC;IACF,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAEY,QAAA,oBAAoB,GAAG;IAClC,uGAAuG;IACvG,EAAE;IACF,QAAQ;IACR,4DAA4D;IAC5D,yCAAyC;IACzC,0BAA0B;IAC1B,EAAE;IACF,gBAAgB;IAChB,+EAA+E;CAChF,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAEb,SAAgB,iBAAiB,CAC/B,OAA+D,EAC/D,OAA+D;IAE/D,MAAM,KAAK,GAAG;QACZ,4BAA4B;QAC5B,EAAE;QACF,YAAY,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,GAAG,OAAO,CAAC,WAAW,GAAG,MAAM,GAAG,OAAO,CAAC,OAAO;QACnF,YAAY,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,GAAG,OAAO,CAAC,WAAW,GAAG,MAAM,GAAG,OAAO,CAAC,OAAO;KACpF,CAAC;IACF,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAEY,QAAA,oBAAoB,GAAG;IAClC,iGAAiG;IACjG,EAAE;IACF,QAAQ;IACR,wEAAwE;IACxE,oGAAoG;IACpG,8FAA8F;IAC9F,0BAA0B;IAC1B,EAAE;IACF,gBAAgB;IAChB,sGAAsG;CACvG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAEb,SAAgB,iBAAiB,CAC/B,YAAoB,EACpB,OAAe,EACf,YAAoB,EACpB,YAAoB;IAEpB,MAAM,KAAK,GAAG;QACZ,iDAAiD;QACjD,EAAE;QACF,YAAY,GAAG,YAAY,GAAG,MAAM,GAAG,YAAY;QACnD,iBAAiB,GAAG,YAAY;QAChC,YAAY,GAAG,OAAO;KACvB,CAAC;IACF,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC"}

package/dist/memory/EngramChatMemory.d.ts

@@ -0,0 +1,42 @@
+import { BaseChatMemory, type BaseChatMemoryInput } from 'langchain/memory';
+import type { InputValues, MemoryVariables, OutputValues } from '@langchain/core/memory';
+import type { IGraphStorage } from '../storage/IGraphStorage';
+import { type EmbeddingConfig } from '../embeddings';
+import type { LlmClientConfig } from '../extraction/LlmClient';
+export interface EngramChatMemoryInput extends BaseChatMemoryInput {
+    storage: IGraphStorage;
+    groupId: string;
+    memoryKey?: string;
+    contextWindow?: number;
+    maxFactsPerQuery?: number;
+    minRelevanceScore?: number;
+    enableExtraction?: boolean;
+    llmConfig?: LlmClientConfig;
+    entityTypes?: string[];
+    embeddingConfig?: EmbeddingConfig;
+    enableTraversal?: boolean;
+    traversalHops?: number;
+}
+export declare class EngramChatMemory extends BaseChatMemory {
+    storage: IGraphStorage;
+    groupId: string;
+    memoryKey: string;
+    contextWindow: number;
+    maxFactsPerQuery: number;
+    minRelevanceScore: number;
+    enableExtraction: boolean;
+    private extractionPipeline;
+    private searchEngine;
+    private engramHistory;
+    private enableTraversal;
+    private traversalHops;
+    constructor(fields: EngramChatMemoryInput);
+    get memoryKeys(): string[];
+    loadMemoryVariables(values: InputValues): Promise<MemoryVariables>;
+    saveContext(inputValues: InputValues, outputValues: OutputValues): Promise<void>;
+    clear(): Promise<void>;
+    private getRelevantFacts;
+    private extractInputText;
+    private runExtraction;
+    private extractOutputText;
+}

package/dist/memory/EngramChatMemory.js

@@ -0,0 +1,162 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.EngramChatMemory = void 0;
+const memory_1 = require("langchain/memory");
+const messages_1 = require("@langchain/core/messages");
+const EngramChatMessageHistory_1 = require("./EngramChatMessageHistory");
+const ExtractionPipeline_1 = require("../extraction/ExtractionPipeline");
+const HybridSearchEngine_1 = require("../search/HybridSearchEngine");
+const embeddings_1 = require("../embeddings");
+const GraphTraverser_1 = require("../traversal/GraphTraverser");
+class EngramChatMemory extends memory_1.BaseChatMemory {
+    constructor(fields) {
+        var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m;
+        const history = new EngramChatMessageHistory_1.EngramChatMessageHistory({
+            storage: fields.storage,
+            groupId: fields.groupId,
+            contextWindow: (_a = fields.contextWindow) !== null && _a !== void 0 ? _a : 10,
+        });
+        super({
+            chatHistory: history,
+            returnMessages: (_b = fields.returnMessages) !== null && _b !== void 0 ? _b : true,
+            inputKey: (_c = fields.inputKey) !== null && _c !== void 0 ? _c : 'input',
+            outputKey: (_d = fields.outputKey) !== null && _d !== void 0 ? _d : 'output',
+        });
+        this.extractionPipeline = null;
+        this.storage = fields.storage;
+        this.groupId = fields.groupId;
+        this.memoryKey = (_e = fields.memoryKey) !== null && _e !== void 0 ? _e : 'chat_history';
+        this.contextWindow = (_f = fields.contextWindow) !== null && _f !== void 0 ? _f : 10;
+        this.maxFactsPerQuery = (_g = fields.maxFactsPerQuery) !== null && _g !== void 0 ? _g : 10;
+        this.minRelevanceScore = (_h = fields.minRelevanceScore) !== null && _h !== void 0 ? _h : 0.5;
+        this.enableExtraction = (_j = fields.enableExtraction) !== null && _j !== void 0 ? _j : false;
+        this.engramHistory = history;
+        this.enableTraversal = (_k = fields.enableTraversal) !== null && _k !== void 0 ? _k : false;
+        this.traversalHops = (_l = fields.traversalHops) !== null && _l !== void 0 ? _l : 1;
+        const embeddingService = fields.embeddingConfig
+            ? new embeddings_1.EmbeddingService(fields.embeddingConfig)
+            : undefined;
+        this.searchEngine = new HybridSearchEngine_1.HybridSearchEngine(fields.storage, embeddingService);
+        if (this.enableExtraction && fields.llmConfig) {
+            this.extractionPipeline = new ExtractionPipeline_1.ExtractionPipeline(fields.storage, {
+                llmConfig: fields.llmConfig,
+                entityTypes: (_m = fields.entityTypes) !== null && _m !== void 0 ? _m : [
+                    'person',
+                    'organization',
+                    'location',
+                    'concept',
+                    'event',
+                ],
+                groupId: fields.groupId,
+                embeddingConfig: fields.embeddingConfig,
+            });
+        }
+    }
+    get memoryKeys() {
+        return [this.memoryKey];
+    }
+    async loadMemoryVariables(values) {
+        const messages = await this.chatHistory.getMessages();
+        const currentInput = this.extractInputText(values);
+        const factMessages = await this.getRelevantFacts(currentInput);
+        const allMessages = [...factMessages, ...messages];
+        if (this.returnMessages) {
+            return { [this.memoryKey]: allMessages };
+        }
+        const formatted = allMessages
+            .map((m) => {
+            const type = m._getType();
+            const prefix = type === 'human' ? 'Human' : type === 'ai' ? 'AI' : 'System';
+            return `${prefix}: ${m.content}`;
+        })
+            .join('\n');
+        return { [this.memoryKey]: formatted };
+    }
+    async saveContext(inputValues, outputValues) {
+        await super.saveContext(inputValues, outputValues);
+        if (this.enableExtraction) {
+            await this.runExtraction(inputValues, outputValues);
+        }
+    }
+    async clear() {
+        await this.engramHistory.clear();
+    }
+    async getRelevantFacts(query) {
+        if (!query.trim())
+            return [];
+        try {
+            const results = await this.searchEngine.search(query, this.groupId, {
+                limit: this.maxFactsPerQuery,
+                minScore: this.minRelevanceScore,
+            });
+            if (results.edges.length === 0 && results.entities.length === 0)
+                return [];
+            let contextText = this.searchEngine.formatAsContext(results);
+            if (this.enableTraversal && results.entities.length > 0) {
+                try {
+                    const seedUuids = results.entities.map((r) => r.entity.uuid);
+                    const traverser = new GraphTraverser_1.GraphTraverser();
+                    const traversal = await traverser.traverse(this.storage, seedUuids, {
+                        maxHops: this.traversalHops,
+                        maxEntities: 20,
+                    });
+                    if (traversal.context) {
+                        contextText += '\n\n' + traversal.context;
+                    }
+                }
+                catch (err) {
+                    console.warn('Engram: Traversal enrichment failed:', err.message);
+                }
+            }
+            if (!contextText)
+                return [];
+            return [new messages_1.SystemMessage(`Relevant knowledge from memory:\n${contextText}`)];
+        }
+        catch (error) {
+            console.warn('Engram: Failed to retrieve relevant facts:', error.message);
+            return [];
+        }
+    }
+    extractInputText(values) {
+        if (this.inputKey && values[this.inputKey]) {
+            const val = values[this.inputKey];
+            return typeof val === 'string' ? val : JSON.stringify(val);
+        }
+        for (const key of ['input', 'question', 'query', 'text']) {
+            if (values[key]) {
+                const val = values[key];
+                return typeof val === 'string' ? val : JSON.stringify(val);
+            }
+        }
+        return '';
+    }
+    async runExtraction(inputValues, outputValues) {
+        if (!this.extractionPipeline)
+            return;
+        const humanText = this.extractInputText(inputValues);
+        const aiText = this.extractOutputText(outputValues);
+        if (!humanText && !aiText)
+            return;
+        try {
+            await this.extractionPipeline.process(humanText, aiText);
+        }
+        catch (error) {
+            console.warn('Engram: Extraction pipeline failed:', error.message);
+        }
+    }
+    extractOutputText(values) {
+        if (this.outputKey && values[this.outputKey]) {
+            const val = values[this.outputKey];
+            return typeof val === 'string' ? val : JSON.stringify(val);
+        }
+        for (const key of ['output', 'response', 'text', 'answer']) {
+            if (values[key]) {
+                const val = values[key];
+                return typeof val === 'string' ? val : JSON.stringify(val);
+            }
+        }
+        return '';
+    }
+}
+exports.EngramChatMemory = EngramChatMemory;
+//# sourceMappingURL=EngramChatMemory.js.map

package/dist/memory/EngramChatMemory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"EngramChatMemory.js","sourceRoot":"","sources":["../../src/memory/EngramChatMemory.ts"],"names":[],"mappings":";;;AAAA,6CAA4E;AAE5E,uDAAyD;AAEzD,yEAAsE;AACtE,yEAG0C;AAC1C,qEAAkE;AAClE,8CAAuE;AAEvE,gEAA6D;AA+B7D,MAAa,gBAAiB,SAAQ,uBAAc;IAelD,YAAY,MAA6B;;QACvC,MAAM,OAAO,GAAG,IAAI,mDAAwB,CAAC;YAC3C,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,aAAa,EAAE,MAAA,MAAM,CAAC,aAAa,mCAAI,EAAE;SAC1C,CAAC,CAAC;QAEH,KAAK,CAAC;YACJ,WAAW,EAAE,OAAO;YACpB,cAAc,EAAE,MAAA,MAAM,CAAC,cAAc,mCAAI,IAAI;YAC7C,QAAQ,EAAE,MAAA,MAAM,CAAC,QAAQ,mCAAI,OAAO;YACpC,SAAS,EAAE,MAAA,MAAM,CAAC,SAAS,mCAAI,QAAQ;SACxC,CAAC,CAAC;QAnBG,uBAAkB,GAA8B,IAAI,CAAC;QAqB3D,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAC9B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAC9B,IAAI,CAAC,SAAS,GAAG,MAAA,MAAM,CAAC,SAAS,mCAAI,cAAc,CAAC;QACpD,IAAI,CAAC,aAAa,GAAG,MAAA,MAAM,CAAC,aAAa,mCAAI,EAAE,CAAC;QAChD,IAAI,CAAC,gBAAgB,GAAG,MAAA,MAAM,CAAC,gBAAgB,mCAAI,EAAE,CAAC;QACtD,IAAI,CAAC,iBAAiB,GAAG,MAAA,MAAM,CAAC,iBAAiB,mCAAI,GAAG,CAAC;QACzD,IAAI,CAAC,gBAAgB,GAAG,MAAA,MAAM,CAAC,gBAAgB,mCAAI,KAAK,CAAC;QACzD,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC;QAC7B,IAAI,CAAC,eAAe,GAAG,MAAA,MAAM,CAAC,eAAe,mCAAI,KAAK,CAAC;QACvD,IAAI,CAAC,aAAa,GAAG,MAAA,MAAM,CAAC,aAAa,mCAAI,CAAC,CAAC;QAG/C,MAAM,gBAAgB,GAAG,MAAM,CAAC,eAAe;YAC7C,CAAC,CAAC,IAAI,6BAAgB,CAAC,MAAM,CAAC,eAAe,CAAC;YAC9C,CAAC,CAAC,SAAS,CAAC;QAGd,IAAI,CAAC,YAAY,GAAG,IAAI,uCAAkB,CAAC,MAAM,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;QAG7E,IAAI,IAAI,CAAC,gBAAgB,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;YAC9C,IAAI,CAAC,kBAAkB,GAAG,IAAI,uCAAkB,CAAC,MAAM,CAAC,OAAO,EAAE;gBAC/D,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,WAAW,EAAE,MAAA,MAAM,CAAC,WAAW,mCAAI;oBACjC,QAAQ;oBACR,cAAc;oBACd,UAAU;oBACV,SAAS;oBACT,OAAO;iBACR;gBACD,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,eAAe,EAAE,MAAM,CAAC,eAAe;aACxC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,IAAI,UAAU;QACZ,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,MAAmB;QAE3C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC;QAGtD,MAAM,YAAY,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACnD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;QAG/D,MAAM,WAAW,GAAG,CAAC,GAAG,YAAY,EAAE,GAAG,QAAQ,CAAC,CAAC;QAEnD,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,OAAO,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,CAAC;QAC3C,CAAC;QAGD,MAAM,SAAS,GAAG,WAAW;aAC1B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACT,MAAM,IAAI,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;YAC1B,MAAM,MAAM,GAAG,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC;YAC5E,OAAO,GAAG,MAAM,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC;QACnC,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,OAAO,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,WAAwB,EAAE,YAA0B;QAGpE,MAAM,KAAK,CAAC,WAAW,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;QAInD,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,MAAM,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;IACnC,CAAC;IAMO,KAAK,CAAC,gBAAgB,CAAC,KAAa;QAC1C,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE;YAAE,OAAO,EAAE,CAAC;QAE7B,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,OAAO,EAAE;gBAClE,KAAK,EAAE,IAAI,CAAC,gBAAgB;gBAC5B,QAAQ,EAAE,IAAI,CAAC,iBAAiB;aACjC,CAAC,CAAC;YAEH,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,EAAE,CAAC;YAE3E,IAAI,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;YAG7D,IAAI,IAAI,CAAC,eAAe,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxD,IAAI,CAAC;oBACH,MAAM,SAAS,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;oBAC7D,MAAM,SAAS,GAAG,IAAI,+BAAc,EAAE,CAAC;oBACvC,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE;wBAClE,OAAO,EAAE,IAAI,CAAC,aAAa;wBAC3B,WAAW,EAAE,EAAE;qBAChB,CAAC,CAAC;oBACH,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;wBACtB,WAAW,IAAI,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC;oBAC5C,CAAC;gBACH,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,OAAO,CAAC,IAAI,CAAC,sCAAsC,EAAG,GAAa,CAAC,OAAO,CAAC,CAAC;gBAC/E,CAAC;YACH,CAAC;YAED,IAAI,CAAC,WAAW;gBAAE,OAAO,EAAE,CAAC;YAE5B,OAAO,CAAC,IAAI,wBAAa,CAAC,oCAAoC,WAAW,EAAE,CAAC,CAAC,CAAC;QAChF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAEf,OAAO,CAAC,IAAI,CAAC,4CAA4C,EAAG,KAAe,CAAC,OAAO,CAAC,CAAC;YACrF,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,MAAmB;QAC1C,IAAI,IAAI,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3C,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAClC,OAAO,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAC7D,CAAC;QAGD,KAAK,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,CAAC,EAAE,CAAC;YACzD,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;gBAChB,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;gBACxB,OAAO,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,WAAwB,EAAE,YAA0B;QAC9E,IAAI,CAAC,IAAI,CAAC,kBAAkB;YAAE,OAAO;QAErC,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAEpD,IAAI,CAAC,SAAS,IAAI,CAAC,MAAM;YAAE,OAAO;QAElC,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QAC3D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAEf,OAAO,CAAC,IAAI,CAAC,qCAAqC,EAAG,KAAe,CAAC,OAAO,CAAC,CAAC;QAChF,CAAC;IACH,CAAC;IAEO,iBAAiB,CAAC,MAAoB;QAC5C,IAAI,IAAI,CAAC,SAAS,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;YAC7C,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACnC,OAAO,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAC7D,CAAC;QAED,KAAK,MAAM,GAAG,IAAI,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,CAAC;YAC3D,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;gBAChB,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;gBACxB,OAAO,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;CACF;AA5MD,4CA4MC"}

package/dist/memory/EngramChatMessageHistory.d.ts

@@ -0,0 +1,22 @@
+import { BaseChatMessageHistory } from '@langchain/core/chat_history';
+import { BaseMessage } from '@langchain/core/messages';
+import type { IGraphStorage } from '../storage/IGraphStorage';
+export declare class EngramChatMessageHistory extends BaseChatMessageHistory {
+    lc_namespace: string[];
+    private storage;
+    private groupId;
+    private lastEpisodeUuid;
+    private contextWindow;
+    constructor(params: {
+        storage: IGraphStorage;
+        groupId: string;
+        contextWindow?: number;
+    });
+    getMessages(): Promise<BaseMessage[]>;
+    addMessage(message: BaseMessage): Promise<void>;
+    addUserMessage(message: string): Promise<void>;
+    addAIChatMessage(message: string): Promise<void>;
+    clear(): Promise<void>;
+    private episodeToMessage;
+    private messageToRole;
+}

package/dist/memory/EngramChatMessageHistory.js

@@ -0,0 +1,72 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.EngramChatMessageHistory = void 0;
+const chat_history_1 = require("@langchain/core/chat_history");
+const messages_1 = require("@langchain/core/messages");
+class EngramChatMessageHistory extends chat_history_1.BaseChatMessageHistory {
+    constructor(params) {
+        var _a;
+        super();
+        this.lc_namespace = ['engram', 'memory'];
+        this.lastEpisodeUuid = null;
+        this.storage = params.storage;
+        this.groupId = params.groupId;
+        this.contextWindow = (_a = params.contextWindow) !== null && _a !== void 0 ? _a : 10;
+    }
+    async getMessages() {
+        const episodes = await this.storage.getRecentEpisodes(this.groupId, this.contextWindow);
+        if (episodes.length > 0) {
+            this.lastEpisodeUuid = episodes[episodes.length - 1].uuid;
+        }
+        return episodes.map((ep) => this.episodeToMessage(ep));
+    }
+    async addMessage(message) {
+        const role = this.messageToRole(message);
+        const content = typeof message.content === 'string' ? message.content : JSON.stringify(message.content);
+        const episode = await this.storage.addEpisode({
+            group_id: this.groupId,
+            content,
+            role,
+            reference_time: new Date().toISOString(),
+            previous_episode_uuid: this.lastEpisodeUuid,
+        });
+        this.lastEpisodeUuid = episode.uuid;
+    }
+    async addUserMessage(message) {
+        await this.addMessage(new messages_1.HumanMessage(message));
+    }
+    async addAIChatMessage(message) {
+        await this.addMessage(new messages_1.AIMessage(message));
+    }
+    async clear() {
+        await this.storage.clearGroup(this.groupId);
+        this.lastEpisodeUuid = null;
+    }
+    episodeToMessage(episode) {
+        switch (episode.role) {
+            case 'human':
+                return new messages_1.HumanMessage(episode.content);
+            case 'ai':
+                return new messages_1.AIMessage(episode.content);
+            case 'system':
+                return new messages_1.SystemMessage(episode.content);
+            default:
+                return new messages_1.HumanMessage(episode.content);
+        }
+    }
+    messageToRole(message) {
+        const type = message._getType();
+        switch (type) {
+            case 'human':
+                return 'human';
+            case 'ai':
+                return 'ai';
+            case 'system':
+                return 'system';
+            default:
+                return 'human';
+        }
+    }
+}
+exports.EngramChatMessageHistory = EngramChatMessageHistory;
+//# sourceMappingURL=EngramChatMessageHistory.js.map

package/dist/memory/EngramChatMessageHistory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"EngramChatMessageHistory.js","sourceRoot":"","sources":["../../src/memory/EngramChatMessageHistory.ts"],"names":[],"mappings":";;;AAAA,+DAAsE;AACtE,uDAA+F;AAQ/F,MAAa,wBAAyB,SAAQ,qCAAsB;IAQlE,YAAY,MAA2E;;QACrF,KAAK,EAAE,CAAC;QARV,iBAAY,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAI5B,oBAAe,GAAkB,IAAI,CAAC;QAK5C,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAC9B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAC9B,IAAI,CAAC,aAAa,GAAG,MAAA,MAAM,CAAC,aAAa,mCAAI,EAAE,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,WAAW;QACf,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QAGxF,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,IAAI,CAAC,eAAe,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;QAC5D,CAAC;QAED,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC,CAAC;IACzD,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,OAAoB;QACnC,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QACzC,MAAM,OAAO,GACX,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAE1F,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;YAC5C,QAAQ,EAAE,IAAI,CAAC,OAAO;YACtB,OAAO;YACP,IAAI;YACJ,cAAc,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACxC,qBAAqB,EAAE,IAAI,CAAC,eAAe;SAC5C,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,OAAe;QAClC,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,uBAAY,CAAC,OAAO,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,OAAe;QACpC,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,oBAAS,CAAC,OAAO,CAAC,CAAC,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5C,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;IAC9B,CAAC;IAEO,gBAAgB,CAAC,OAAqB;QAC5C,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACrB,KAAK,OAAO;gBACV,OAAO,IAAI,uBAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC3C,KAAK,IAAI;gBACP,OAAO,IAAI,oBAAS,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YACxC,KAAK,QAAQ;gBACX,OAAO,IAAI,wBAAa,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC5C;gBACE,OAAO,IAAI,uBAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,OAAoB;QACxC,MAAM,IAAI,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;QAChC,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,OAAO;gBACV,OAAO,OAAO,CAAC;YACjB,KAAK,IAAI;gBACP,OAAO,IAAI,CAAC;YACd,KAAK,QAAQ;gBACX,OAAO,QAAQ,CAAC;YAClB;gBACE,OAAO,OAAO,CAAC;QACnB,CAAC;IACH,CAAC;CACF;AAjFD,4DAiFC"}

package/dist/memory/index.d.ts

@@ -0,0 +1,2 @@
+export { EngramChatMessageHistory } from './EngramChatMessageHistory';
+export { EngramChatMemory, type EngramChatMemoryInput } from './EngramChatMemory';

package/dist/memory/index.js

@@ -0,0 +1,8 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.EngramChatMemory = exports.EngramChatMessageHistory = void 0;
+var EngramChatMessageHistory_1 = require("./EngramChatMessageHistory");
+Object.defineProperty(exports, "EngramChatMessageHistory", { enumerable: true, get: function () { return EngramChatMessageHistory_1.EngramChatMessageHistory; } });
+var EngramChatMemory_1 = require("./EngramChatMemory");
+Object.defineProperty(exports, "EngramChatMemory", { enumerable: true, get: function () { return EngramChatMemory_1.EngramChatMemory; } });
+//# sourceMappingURL=index.js.map

package/dist/memory/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/memory/index.ts"],"names":[],"mappings":";;;AAAA,uEAAsE;AAA7D,oIAAA,wBAAwB,OAAA;AACjC,uDAAkF;AAAzE,oHAAA,gBAAgB,OAAA"}

package/dist/nodes/EngramAdmin/EngramAdmin.node.d.ts

@@ -0,0 +1,5 @@
+import { type IExecuteFunctions, type INodeExecutionData, type INodeType, type INodeTypeDescription } from 'n8n-workflow';
+export declare class EngramAdmin implements INodeType {
+    description: INodeTypeDescription;
+    execute(this: IExecuteFunctions): Promise<INodeExecutionData[][]>;
+}