@agent-relay/memory 0.1.0

package/dist/adapters/supermemory.js

@@ -0,0 +1,338 @@
+/**
+ * Supermemory.ai Memory Adapter
+ *
+ * Integration with supermemory.ai for semantic memory storage and retrieval.
+ * Provides AI-optimized search with embedding-based similarity.
+ *
+ * @see https://supermemory.ai/docs
+ */
+/**
+ * Supermemory.ai adapter for semantic memory storage
+ */
+export class SupermemoryAdapter {
+    type = 'supermemory';
+    apiKey;
+    endpoint;
+    container;
+    defaultAgentId;
+    defaultProjectId;
+    timeout;
+    initialized = false;
+    constructor(options) {
+        if (!options.apiKey) {
+            throw new Error('SupermemoryAdapter requires an API key');
+        }
+        this.apiKey = options.apiKey;
+        this.endpoint = options.endpoint ?? 'https://api.supermemory.ai';
+        this.container = options.container;
+        this.defaultAgentId = options.defaultAgentId;
+        this.defaultProjectId = options.defaultProjectId;
+        this.timeout = options.timeout ?? 30000;
+    }
+    async init() {
+        // Verify API key by making a simple request
+        try {
+            const response = await this.fetch('/v3/documents/list', {
+                method: 'POST',
+                body: JSON.stringify({ limit: 1 }),
+            });
+            if (!response.ok && response.status !== 404) {
+                const error = await response.text();
+                throw new Error(`Supermemory API error: ${error}`);
+            }
+            this.initialized = true;
+        }
+        catch (error) {
+            if (error instanceof Error && error.message.includes('fetch')) {
+                throw new Error(`Failed to connect to Supermemory API: ${error.message}`);
+            }
+            throw error;
+        }
+    }
+    async add(content, options) {
+        try {
+            const metadata = {
+                source: options?.source ?? 'agent-relay',
+                agentId: options?.agentId ?? this.defaultAgentId,
+                projectId: options?.projectId ?? this.defaultProjectId,
+                sessionId: options?.sessionId,
+                tags: options?.tags,
+                ...options?.metadata,
+            };
+            // Remove undefined values
+            Object.keys(metadata).forEach(key => {
+                if (metadata[key] === undefined) {
+                    delete metadata[key];
+                }
+            });
+            const body = {
+                content,
+                metadata,
+            };
+            if (this.container) {
+                body.containerTags = [this.container];
+            }
+            const response = await this.fetch('/v3/documents', {
+                method: 'POST',
+                body: JSON.stringify(body),
+            });
+            if (!response.ok) {
+                const error = await response.text();
+                return { success: false, error: `Failed to add memory: ${error}` };
+            }
+            const result = await response.json();
+            return { success: true, id: result.id ?? result.documentId };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : 'Unknown error',
+            };
+        }
+    }
+    async search(query) {
+        try {
+            const filters = {};
+            if (query.agentId) {
+                filters.agentId = query.agentId;
+            }
+            if (query.projectId) {
+                filters.projectId = query.projectId;
+            }
+            if (query.tags && query.tags.length > 0) {
+                filters.tags = query.tags;
+            }
+            const body = {
+                query: query.query,
+                limit: query.limit ?? 10,
+                minScore: query.minScore ?? 0.5,
+            };
+            if (Object.keys(filters).length > 0) {
+                body.filters = filters;
+            }
+            if (this.container) {
+                body.containerTags = [this.container];
+            }
+            // Use v4 search for lower latency
+            const response = await this.fetch('/v4/search', {
+                method: 'POST',
+                body: JSON.stringify(body),
+            });
+            if (!response.ok) {
+                console.error('[supermemory] Search failed:', await response.text());
+                return [];
+            }
+            const result = await response.json();
+            const results = result.results ?? [];
+            return results.map(doc => this.documentToMemoryEntry(doc));
+        }
+        catch (error) {
+            console.error('[supermemory] Search error:', error);
+            return [];
+        }
+    }
+    async get(id) {
+        try {
+            const response = await this.fetch(`/v3/documents/${encodeURIComponent(id)}`, {
+                method: 'GET',
+            });
+            if (!response.ok) {
+                if (response.status === 404) {
+                    return null;
+                }
+                console.error('[supermemory] Get failed:', await response.text());
+                return null;
+            }
+            const doc = await response.json();
+            return this.documentToMemoryEntry(doc);
+        }
+        catch (error) {
+            console.error('[supermemory] Get error:', error);
+            return null;
+        }
+    }
+    async delete(id) {
+        try {
+            const response = await this.fetch(`/v3/documents/${encodeURIComponent(id)}`, {
+                method: 'DELETE',
+            });
+            if (!response.ok && response.status !== 404) {
+                const error = await response.text();
+                return { success: false, error: `Failed to delete: ${error}` };
+            }
+            return { success: true, id };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : 'Unknown error',
+            };
+        }
+    }
+    async update(id, content, options) {
+        try {
+            const body = { content };
+            if (options) {
+                const metadata = {};
+                if (options.tags)
+                    metadata.tags = options.tags;
+                if (options.metadata)
+                    Object.assign(metadata, options.metadata);
+                if (Object.keys(metadata).length > 0) {
+                    body.metadata = metadata;
+                }
+            }
+            const response = await this.fetch(`/v3/documents/${encodeURIComponent(id)}`, {
+                method: 'PATCH',
+                body: JSON.stringify(body),
+            });
+            if (!response.ok) {
+                const error = await response.text();
+                return { success: false, error: `Failed to update: ${error}` };
+            }
+            return { success: true, id };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : 'Unknown error',
+            };
+        }
+    }
+    async list(options) {
+        try {
+            const body = {
+                limit: options?.limit ?? 50,
+                sortBy: 'createdAt',
+                sortOrder: 'desc',
+            };
+            const filters = {};
+            if (options?.agentId)
+                filters.agentId = options.agentId;
+            if (options?.projectId)
+                filters.projectId = options.projectId;
+            if (Object.keys(filters).length > 0) {
+                body.filters = filters;
+            }
+            if (this.container) {
+                body.containerTags = [this.container];
+            }
+            const response = await this.fetch('/v3/documents/list', {
+                method: 'POST',
+                body: JSON.stringify(body),
+            });
+            if (!response.ok) {
+                console.error('[supermemory] List failed:', await response.text());
+                return [];
+            }
+            const result = await response.json();
+            return (result.documents ?? []).map(doc => this.documentToMemoryEntry(doc));
+        }
+        catch (error) {
+            console.error('[supermemory] List error:', error);
+            return [];
+        }
+    }
+    async clear(options) {
+        try {
+            // Supermemory supports bulk delete by container tags
+            // For more specific filtering, we need to list and delete individually
+            if (!options?.agentId && !options?.projectId && this.container) {
+                // Delete by container
+                const response = await this.fetch('/v3/documents/bulk', {
+                    method: 'DELETE',
+                    body: JSON.stringify({ containerTags: [this.container] }),
+                });
+                if (!response.ok) {
+                    return { success: false, error: await response.text() };
+                }
+                return { success: true };
+            }
+            // List and delete matching memories
+            const memories = await this.list({
+                limit: 1000,
+                agentId: options?.agentId,
+                projectId: options?.projectId,
+            });
+            const toDelete = options?.before
+                ? memories.filter(m => m.createdAt < options.before)
+                : memories;
+            for (const memory of toDelete) {
+                await this.delete(memory.id);
+            }
+            return { success: true };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : 'Unknown error',
+            };
+        }
+    }
+    async stats() {
+        // Supermemory doesn't have a stats endpoint, so we approximate
+        const memories = await this.list({ limit: 1000 });
+        const byAgent = {};
+        const byProject = {};
+        for (const memory of memories) {
+            if (memory.agentId) {
+                byAgent[memory.agentId] = (byAgent[memory.agentId] ?? 0) + 1;
+            }
+            if (memory.projectId) {
+                byProject[memory.projectId] = (byProject[memory.projectId] ?? 0) + 1;
+            }
+        }
+        return {
+            totalCount: memories.length,
+            byAgent,
+            byProject,
+        };
+    }
+    async close() {
+        this.initialized = false;
+    }
+    /**
+     * Make a fetch request to the Supermemory API
+     */
+    async fetch(path, options) {
+        const url = `${this.endpoint}${path}`;
+        const controller = new AbortController();
+        const timeoutId = setTimeout(() => controller.abort(), this.timeout);
+        try {
+            const response = await fetch(url, {
+                ...options,
+                headers: {
+                    'Content-Type': 'application/json',
+                    'Authorization': `Bearer ${this.apiKey}`,
+                    ...options.headers,
+                },
+                signal: controller.signal,
+            });
+            return response;
+        }
+        finally {
+            clearTimeout(timeoutId);
+        }
+    }
+    /**
+     * Convert a Supermemory document to a MemoryEntry
+     */
+    documentToMemoryEntry(doc) {
+        const metadata = doc.metadata ?? {};
+        return {
+            id: doc.id,
+            content: doc.content,
+            createdAt: doc.createdAt
+                ? new Date(doc.createdAt).getTime()
+                : Date.now(),
+            tags: metadata.tags,
+            source: metadata.source,
+            agentId: metadata.agentId,
+            projectId: metadata.projectId,
+            sessionId: metadata.sessionId,
+            score: doc.score,
+            metadata,
+        };
+    }
+}
+//# sourceMappingURL=supermemory.js.map

package/dist/adapters/supermemory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"supermemory.js","sourceRoot":"","sources":["../../src/adapters/supermemory.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAoDH;;GAEG;AACH,MAAM,OAAO,kBAAkB;IACpB,IAAI,GAAG,aAAa,CAAC;IAEtB,MAAM,CAAS;IACf,QAAQ,CAAS;IACjB,SAAS,CAAU;IACnB,cAAc,CAAU;IACxB,gBAAgB,CAAU;IAC1B,OAAO,CAAS;IAChB,WAAW,GAAG,KAAK,CAAC;IAE5B,YAAY,OAAkC;QAC5C,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC5D,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,4BAA4B,CAAC;QACjE,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;QAC7C,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC;QACjD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,KAAK,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,IAAI;QACR,4CAA4C;QAC5C,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,oBAAoB,EAAE;gBACtD,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;aACnC,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC5C,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACpC,MAAM,IAAI,KAAK,CAAC,0BAA0B,KAAK,EAAE,CAAC,CAAC;YACrD,CAAC;YAED,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAC1B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC9D,MAAM,IAAI,KAAK,CAAC,yCAAyC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YAC5E,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,OAAe,EAAE,OAA0B;QACnD,IAAI,CAAC;YACH,MAAM,QAAQ,GAA4B;gBACxC,MAAM,EAAE,OAAO,EAAE,MAAM,IAAI,aAAa;gBACxC,OAAO,EAAE,OAAO,EAAE,OAAO,IAAI,IAAI,CAAC,cAAc;gBAChD,SAAS,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI,CAAC,gBAAgB;gBACtD,SAAS,EAAE,OAAO,EAAE,SAAS;gBAC7B,IAAI,EAAE,OAAO,EAAE,IAAI;gBACnB,GAAG,OAAO,EAAE,QAAQ;aACrB,CAAC;YAEF,0BAA0B;YAC1B,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;gBAClC,IAAI,QAAQ,CAAC,GAAG,CAAC,KAAK,SAAS,EAAE,CAAC;oBAChC,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC;gBACvB,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,MAAM,IAAI,GAA4B;gBACpC,OAAO;gBACP,QAAQ;aACT,CAAC;YAEF,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnB,IAAI,CAAC,aAAa,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACxC,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,eAAe,EAAE;gBACjD,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACpC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,yBAAyB,KAAK,EAAE,EAAE,CAAC;YACrE,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAA0C,CAAC;YAC7E,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;QAC/D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;aAChE,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAwB;QACnC,IAAI,CAAC;YACH,MAAM,OAAO,GAA4B,EAAE,CAAC;YAE5C,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;gBAClB,OAAO,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;YAClC,CAAC;YACD,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;gBACpB,OAAO,CAAC,SAAS,GAAG,KAAK,CAAC,SAAS,CAAC;YACtC,CAAC;YACD,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxC,OAAO,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;YAC5B,CAAC;YAED,MAAM,IAAI,GAA4B;gBACpC,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,KAAK,EAAE,KAAK,CAAC,KAAK,IAAI,EAAE;gBACxB,QAAQ,EAAE,KAAK,CAAC,QAAQ,IAAI,GAAG;aAChC,CAAC;YAEF,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACpC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;YACzB,CAAC;YAED,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnB,IAAI,CAAC,aAAa,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACxC,CAAC;YAED,kCAAkC;YAClC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE;gBAC9C,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;gBACrE,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAA6C,CAAC;YAChF,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;YAErC,OAAO,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC,CAAC;QAC7D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;YACpD,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,EAAU;QAClB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,iBAAiB,kBAAkB,CAAC,EAAE,CAAC,EAAE,EAAE;gBAC3E,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;oBAC5B,OAAO,IAAI,CAAC;gBACd,CAAC;gBACD,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;gBAClE,OAAO,IAAI,CAAC;YACd,CAAC;YAED,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAyB,CAAC;YACzD,OAAO,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC;QACzC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;YACjD,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,iBAAiB,kBAAkB,CAAC,EAAE,CAAC,EAAE,EAAE;gBAC3E,MAAM,EAAE,QAAQ;aACjB,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC5C,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACpC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,qBAAqB,KAAK,EAAE,EAAE,CAAC;YACjE,CAAC;YAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;aAChE,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CACV,EAAU,EACV,OAAe,EACf,OAAmC;QAEnC,IAAI,CAAC;YACH,MAAM,IAAI,GAA4B,EAAE,OAAO,EAAE,CAAC;YAElD,IAAI,OAAO,EAAE,CAAC;gBACZ,MAAM,QAAQ,GAA4B,EAAE,CAAC;gBAC7C,IAAI,OAAO,CAAC,IAAI;oBAAE,QAAQ,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;gBAC/C,IAAI,OAAO,CAAC,QAAQ;oBAAE,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;gBAChE,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACrC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;gBAC3B,CAAC;YACH,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,iBAAiB,kBAAkB,CAAC,EAAE,CAAC,EAAE,EAAE;gBAC3E,MAAM,EAAE,OAAO;gBACf,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACpC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,qBAAqB,KAAK,EAAE,EAAE,CAAC;YACjE,CAAC;YAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;aAChE,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAIV;QACC,IAAI,CAAC;YACH,MAAM,IAAI,GAA4B;gBACpC,KAAK,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE;gBAC3B,MAAM,EAAE,WAAW;gBACnB,SAAS,EAAE,MAAM;aAClB,CAAC;YAEF,MAAM,OAAO,GAA4B,EAAE,CAAC;YAC5C,IAAI,OAAO,EAAE,OAAO;gBAAE,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;YACxD,IAAI,OAAO,EAAE,SAAS;gBAAE,OAAO,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;YAE9D,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACpC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;YACzB,CAAC;YAED,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnB,IAAI,CAAC,aAAa,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACxC,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,oBAAoB,EAAE;gBACtD,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,OAAO,CAAC,KAAK,CAAC,4BAA4B,EAAE,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;gBACnE,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAA6B,CAAC;YAChE,OAAO,CAAC,MAAM,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC,CAAC;QAC9E,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,CAAC;YAClD,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,OAIX;QACC,IAAI,CAAC;YACH,qDAAqD;YACrD,uEAAuE;YACvE,IAAI,CAAC,OAAO,EAAE,OAAO,IAAI,CAAC,OAAO,EAAE,SAAS,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBAC/D,sBAAsB;gBACtB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,oBAAoB,EAAE;oBACtD,MAAM,EAAE,QAAQ;oBAChB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,aAAa,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;iBAC1D,CAAC,CAAC;gBAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACjB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC;gBAC1D,CAAC;gBACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YAC3B,CAAC;YAED,oCAAoC;YACpC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;gBAC/B,KAAK,EAAE,IAAI;gBACX,OAAO,EAAE,OAAO,EAAE,OAAO;gBACzB,SAAS,EAAE,OAAO,EAAE,SAAS;aAC9B,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,OAAO,EAAE,MAAM;gBAC9B,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,OAAO,CAAC,MAAO,CAAC;gBACrD,CAAC,CAAC,QAAQ,CAAC;YAEb,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;gBAC9B,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC/B,CAAC;YAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;aAChE,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QAKT,+DAA+D;QAC/D,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAElD,MAAM,OAAO,GAA2B,EAAE,CAAC;QAC3C,MAAM,SAAS,GAA2B,EAAE,CAAC;QAE7C,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;YAC9B,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;YAC/D,CAAC;YACD,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;gBACrB,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;YACvE,CAAC;QACH,CAAC;QAED,OAAO;YACL,UAAU,EAAE,QAAQ,CAAC,MAAM;YAC3B,OAAO;YACP,SAAS;SACV,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;IAC3B,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,KAAK,CAAC,IAAY,EAAE,OAAoB;QACpD,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,QAAQ,GAAG,IAAI,EAAE,CAAC;QACtC,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAErE,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAChC,GAAG,OAAO;gBACV,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;oBACxC,GAAG,OAAO,CAAC,OAAO;iBACnB;gBACD,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;YAEH,OAAO,QAAQ,CAAC;QAClB,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,SAAS,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAED;;OAEG;IACK,qBAAqB,CAC3B,GAAkD;QAElD,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,IAAI,EAAE,CAAC;QAEpC,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,SAAS,EAAG,GAA2B,CAAC,SAAS;gBAC/C,CAAC,CAAC,IAAI,IAAI,CAAE,GAA2B,CAAC,SAAU,CAAC,CAAC,OAAO,EAAE;gBAC7D,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE;YACd,IAAI,EAAE,QAAQ,CAAC,IAA4B;YAC3C,MAAM,EAAE,QAAQ,CAAC,MAA4B;YAC7C,OAAO,EAAE,QAAQ,CAAC,OAA6B;YAC/C,SAAS,EAAE,QAAQ,CAAC,SAA+B;YACnD,SAAS,EAAE,QAAQ,CAAC,SAA+B;YACnD,KAAK,EAAG,GAA+B,CAAC,KAAK;YAC7C,QAAQ;SACT,CAAC;IACJ,CAAC;CACF"}

package/dist/context-compaction.d.ts

@@ -0,0 +1,156 @@
+/**
+ * Context Compaction for Long Agent Sessions
+ *
+ * Manages conversation context to prevent token limit exhaustion.
+ * Provides token counting, message summarization, and context pruning.
+ *
+ * Inspired by russian-code-ts context management targets:
+ * - Token estimation: <20ms
+ * - Embeddings-based semantic search for relevant context
+ *
+ * Strategies:
+ * 1. Fast token estimation (character-based heuristic)
+ * 2. Importance-weighted message retention
+ * 3. Sliding window with summary injection
+ * 4. Semantic deduplication of similar messages
+ */
+export interface Message {
+    id: string;
+    role: 'user' | 'assistant' | 'system';
+    content: string;
+    timestamp: number;
+    /** Message importance (0-100, higher = more important) */
+    importance?: number;
+    /** Whether this is a summary message */
+    isSummary?: boolean;
+    /** Original message IDs if this is a summary */
+    summarizes?: string[];
+    /** Thread ID for grouping */
+    thread?: string;
+    /** Token count (cached) */
+    tokenCount?: number;
+}
+export interface ContextWindow {
+    messages: Message[];
+    totalTokens: number;
+    maxTokens: number;
+    usagePercent: number;
+}
+export interface CompactionResult {
+    /** Messages after compaction */
+    messages: Message[];
+    /** Number of messages removed */
+    messagesRemoved: number;
+    /** Tokens saved */
+    tokensSaved: number;
+    /** Summary message added (if any) */
+    summaryAdded?: Message;
+    /** Compaction strategy used */
+    strategy: CompactionStrategy;
+}
+export type CompactionStrategy = 'none' | 'trim_old' | 'trim_low_importance' | 'summarize' | 'deduplicate' | 'aggressive';
+export interface CompactionConfig {
+    /** Maximum tokens for context window */
+    maxTokens: number;
+    /** Target token usage after compaction (e.g., 0.7 = 70%) */
+    targetUsage: number;
+    /** Threshold to trigger compaction (e.g., 0.9 = 90%) */
+    compactionThreshold: number;
+    /** Minimum importance to retain during compaction */
+    minImportanceRetain: number;
+    /** Number of recent messages to always keep */
+    keepRecentCount: number;
+    /** Enable summarization */
+    enableSummarization: boolean;
+    /** Enable semantic deduplication */
+    enableDeduplication: boolean;
+    /** Similarity threshold for deduplication (0-1) */
+    deduplicationThreshold: number;
+}
+/**
+ * Fast token estimation using character-based heuristic.
+ * Targets <20ms latency for large texts.
+ *
+ * Heuristic: ~4 characters per token for English text.
+ * Adjusts for code (more tokens per char) and whitespace.
+ */
+export declare function estimateTokens(text: string): number;
+/**
+ * Estimate tokens for a message (uses caching).
+ */
+export declare function estimateMessageTokens(message: Message): number;
+/**
+ * Estimate tokens for entire context.
+ */
+export declare function estimateContextTokens(messages: Message[]): number;
+/**
+ * Calculate importance score for a message.
+ * Higher scores = more important to retain.
+ */
+export declare function calculateImportance(message: Message, index: number, total: number): number;
+/**
+ * Simple similarity score between two strings (Jaccard on word set).
+ * Returns 0-1 where 1 = identical.
+ */
+export declare function calculateSimilarity(a: string, b: string): number;
+/**
+ * Find duplicate/similar messages.
+ */
+export declare function findDuplicates(messages: Message[], threshold?: number): Map<string, string[]>;
+/**
+ * Create a summary of multiple messages.
+ * This is a simple extractive summary - in production, use an LLM.
+ */
+export declare function createSummary(messages: Message[]): Message;
+/**
+ * Context compaction manager.
+ */
+export declare class ContextCompactor {
+    private config;
+    constructor(config?: Partial<CompactionConfig>);
+    /**
+     * Get current context window status.
+     */
+    getContextWindow(messages: Message[]): ContextWindow;
+    /**
+     * Check if compaction is needed.
+     */
+    needsCompaction(messages: Message[]): boolean;
+    /**
+     * Perform context compaction.
+     */
+    compact(messages: Message[]): CompactionResult;
+    /**
+     * Add a message to context with automatic compaction if needed.
+     */
+    addMessage(messages: Message[], newMessage: Message): {
+        messages: Message[];
+        compacted: boolean;
+        result?: CompactionResult;
+    };
+    /**
+     * Get token budget remaining.
+     */
+    getTokenBudget(messages: Message[]): {
+        used: number;
+        remaining: number;
+        percentUsed: number;
+    };
+}
+/**
+ * Create a context compactor with the given configuration.
+ */
+export declare function createContextCompactor(config?: Partial<CompactionConfig>): ContextCompactor;
+/**
+ * Format token count for display.
+ */
+export declare function formatTokenCount(tokens: number): string;
+/**
+ * Benchmark token estimation performance.
+ */
+export declare function benchmarkTokenEstimation(iterations?: number): {
+    avgNs: number;
+    maxNs: number;
+    tokensPerMs: number;
+};
+//# sourceMappingURL=context-compaction.d.ts.map

package/dist/context-compaction.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-compaction.d.ts","sourceRoot":"","sources":["../src/context-compaction.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAMH,MAAM,WAAW,OAAO;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,0DAA0D;IAC1D,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,wCAAwC;IACxC,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,gDAAgD;IAChD,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACtB,6BAA6B;IAC7B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,2BAA2B;IAC3B,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,aAAa;IAC5B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,gBAAgB;IAC/B,gCAAgC;IAChC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,iCAAiC;IACjC,eAAe,EAAE,MAAM,CAAC;IACxB,mBAAmB;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,qCAAqC;IACrC,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,+BAA+B;IAC/B,QAAQ,EAAE,kBAAkB,CAAC;CAC9B;AAED,MAAM,MAAM,kBAAkB,GAC1B,MAAM,GACN,UAAU,GACV,qBAAqB,GACrB,WAAW,GACX,aAAa,GACb,YAAY,CAAC;AAEjB,MAAM,WAAW,gBAAgB;IAC/B,wCAAwC;IACxC,SAAS,EAAE,MAAM,CAAC;IAClB,4DAA4D;IAC5D,WAAW,EAAE,MAAM,CAAC;IACpB,wDAAwD;IACxD,mBAAmB,EAAE,MAAM,CAAC;IAC5B,qDAAqD;IACrD,mBAAmB,EAAE,MAAM,CAAC;IAC5B,+CAA+C;IAC/C,eAAe,EAAE,MAAM,CAAC;IACxB,2BAA2B;IAC3B,mBAAmB,EAAE,OAAO,CAAC;IAC7B,oCAAoC;IACpC,mBAAmB,EAAE,OAAO,CAAC;IAC7B,mDAAmD;IACnD,sBAAsB,EAAE,MAAM,CAAC;CAChC;AAqBD;;;;;;GAMG;AACH,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CA6CnD;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM,CAW9D;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,MAAM,CAQjE;AAMD;;;GAGG;AACH,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,CAmD1F;AAMD;;;GAGG;AACH,wBAAgB,mBAAmB,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,GAAG,MAAM,CAiBhE;AAED;;GAEG;AACH,wBAAgB,cAAc,CAC5B,QAAQ,EAAE,OAAO,EAAE,EACnB,SAAS,GAAE,MAAa,GACvB,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAoBvB;AAMD;;;GAGG;AACH,wBAAgB,aAAa,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,CAgC1D;AAMD;;GAEG;AACH,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,MAAM,CAAmB;gBAErB,MAAM,GAAE,OAAO,CAAC,gBAAgB,CAAM;IAIlD;;OAEG;IACH,gBAAgB,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,aAAa;IAUpD;;OAEG;IACH,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO;IAK7C;;OAEG;IACH,OAAO,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,gBAAgB;IA0H9C;;OAEG;IACH,UAAU,CACR,QAAQ,EAAE,OAAO,EAAE,EACnB,UAAU,EAAE,OAAO,GAClB;QAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;QAAC,SAAS,EAAE,OAAO,CAAC;QAAC,MAAM,CAAC,EAAE,gBAAgB,CAAA;KAAE;IAkBzE;;OAEG;IACH,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG;QACnC,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC;QAClB,WAAW,EAAE,MAAM,CAAC;KACrB;CAQF;AAMD;;GAEG;AACH,wBAAgB,sBAAsB,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,GAAG,gBAAgB,CAE3F;AAMD;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAQvD;AAED;;GAEG;AACH,wBAAgB,wBAAwB,CAAC,UAAU,GAAE,MAAc,GAAG;IACpE,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;CACrB,CA8BA"}