@animalabs/context-manager 0.1.0

package/dist/src/blob-manager.d.ts

@@ -0,0 +1,37 @@
+import type { JsStore } from '@animalabs/chronicle';
+import type { ContentBlock } from '@animalabs/membrane';
+import type { StoredContentBlock } from './types/index.js';
+/**
+ * Manages blob storage for media content.
+ * Extracts base64 data from content blocks and stores them in Chronicle's blob storage.
+ * Resolves blob references back to inline content on retrieval.
+ */
+export declare class BlobManager {
+    private store;
+    constructor(store: JsStore);
+    /**
+     * Extract blobs from content blocks and return references.
+     * Replaces inline base64 data with blob references.
+     */
+    extractBlobs(content: ContentBlock[]): StoredContentBlock[];
+    /**
+     * Resolve blob references back to inline content.
+     */
+    resolveBlobs(content: StoredContentBlock[]): ContentBlock[];
+    private extractBlobFromBlock;
+    private extractFromImage;
+    private extractFromDocument;
+    private extractFromAudio;
+    private extractFromVideo;
+    private storeBase64;
+    private resolveBlobInBlock;
+    /**
+     * Check if a content block contains media that would be stored as a blob.
+     */
+    static hasStorableMedia(block: ContentBlock): boolean;
+    /**
+     * Estimate the storage size of media content.
+     */
+    static estimateMediaSize(block: ContentBlock): number;
+}
+//# sourceMappingURL=blob-manager.d.ts.map

package/dist/src/blob-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"blob-manager.d.ts","sourceRoot":"","sources":["../../src/blob-manager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,KAAK,EACV,YAAY,EAMb,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAiB,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AAE1E;;;;GAIG;AACH,qBAAa,WAAW;IACV,OAAO,CAAC,KAAK;gBAAL,KAAK,EAAE,OAAO;IAElC;;;OAGG;IACH,YAAY,CAAC,OAAO,EAAE,YAAY,EAAE,GAAG,kBAAkB,EAAE;IAI3D;;OAEG;IACH,YAAY,CAAC,OAAO,EAAE,kBAAkB,EAAE,GAAG,YAAY,EAAE;IAI3D,OAAO,CAAC,oBAAoB;IAgB5B,OAAO,CAAC,gBAAgB;IAUxB,OAAO,CAAC,mBAAmB;IAK3B,OAAO,CAAC,gBAAgB;IAKxB,OAAO,CAAC,gBAAgB;IAKxB,OAAO,CAAC,WAAW;IAcnB,OAAO,CAAC,kBAAkB;IA+B1B;;OAEG;IACH,MAAM,CAAC,gBAAgB,CAAC,KAAK,EAAE,YAAY,GAAG,OAAO;IAarD;;OAEG;IACH,MAAM,CAAC,iBAAiB,CAAC,KAAK,EAAE,YAAY,GAAG,MAAM;CAetD"}

package/dist/src/blob-manager.js

@@ -0,0 +1,128 @@
+/**
+ * Manages blob storage for media content.
+ * Extracts base64 data from content blocks and stores them in Chronicle's blob storage.
+ * Resolves blob references back to inline content on retrieval.
+ */
+export class BlobManager {
+    store;
+    constructor(store) {
+        this.store = store;
+    }
+    /**
+     * Extract blobs from content blocks and return references.
+     * Replaces inline base64 data with blob references.
+     */
+    extractBlobs(content) {
+        return content.map((block) => this.extractBlobFromBlock(block));
+    }
+    /**
+     * Resolve blob references back to inline content.
+     */
+    resolveBlobs(content) {
+        return content.map((block) => this.resolveBlobInBlock(block));
+    }
+    extractBlobFromBlock(block) {
+        switch (block.type) {
+            case 'image':
+                return this.extractFromImage(block);
+            case 'document':
+                return this.extractFromDocument(block);
+            case 'audio':
+                return this.extractFromAudio(block);
+            case 'video':
+                return this.extractFromVideo(block);
+            default:
+                // Other block types pass through unchanged
+                return block;
+        }
+    }
+    extractFromImage(block) {
+        if (block.source.type === 'url') {
+            // URL sources don't need blob storage - pass through unchanged
+            return block;
+        }
+        const ref = this.storeBase64(block.source, 'image');
+        return { type: 'blob_ref', ref };
+    }
+    extractFromDocument(block) {
+        const ref = this.storeBase64(block.source, 'document');
+        return { type: 'blob_ref', ref };
+    }
+    extractFromAudio(block) {
+        const ref = this.storeBase64(block.source, 'audio');
+        return { type: 'blob_ref', ref };
+    }
+    extractFromVideo(block) {
+        const ref = this.storeBase64(block.source, 'video');
+        return { type: 'blob_ref', ref };
+    }
+    storeBase64(source, originalType) {
+        const buffer = Buffer.from(source.data, 'base64');
+        const hash = this.store.storeBlob(buffer, source.mediaType);
+        return {
+            hash,
+            mediaType: source.mediaType,
+            originalType,
+        };
+    }
+    resolveBlobInBlock(block) {
+        if (block.type !== 'blob_ref') {
+            return block;
+        }
+        const { ref } = block;
+        const buffer = this.store.getBlob(ref.hash);
+        if (!buffer) {
+            throw new Error(`Blob not found: ${ref.hash}`);
+        }
+        const data = buffer.toString('base64');
+        const source = {
+            type: 'base64',
+            data,
+            mediaType: ref.mediaType,
+        };
+        switch (ref.originalType) {
+            case 'image':
+                return { type: 'image', source };
+            case 'document':
+                return { type: 'document', source };
+            case 'audio':
+                return { type: 'audio', source };
+            case 'video':
+                return { type: 'video', source };
+        }
+    }
+    /**
+     * Check if a content block contains media that would be stored as a blob.
+     */
+    static hasStorableMedia(block) {
+        switch (block.type) {
+            case 'image':
+                return block.source.type === 'base64';
+            case 'document':
+            case 'audio':
+            case 'video':
+                return true;
+            default:
+                return false;
+        }
+    }
+    /**
+     * Estimate the storage size of media content.
+     */
+    static estimateMediaSize(block) {
+        switch (block.type) {
+            case 'image':
+                if (block.source.type === 'base64') {
+                    return Math.ceil(block.source.data.length * 0.75); // base64 -> bytes
+                }
+                return 0;
+            case 'document':
+            case 'audio':
+            case 'video':
+                return Math.ceil(block.source.data.length * 0.75);
+            default:
+                return 0;
+        }
+    }
+}
+//# sourceMappingURL=blob-manager.js.map

package/dist/src/blob-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"blob-manager.js","sourceRoot":"","sources":["../../src/blob-manager.ts"],"names":[],"mappings":"AAWA;;;;GAIG;AACH,MAAM,OAAO,WAAW;IACF;IAApB,YAAoB,KAAc;QAAd,UAAK,GAAL,KAAK,CAAS;IAAG,CAAC;IAEtC;;;OAGG;IACH,YAAY,CAAC,OAAuB;QAClC,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC;IAClE,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,OAA6B;QACxC,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC;IAChE,CAAC;IAEO,oBAAoB,CAAC,KAAmB;QAC9C,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,OAAO;gBACV,OAAO,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;YACtC,KAAK,UAAU;gBACb,OAAO,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;YACzC,KAAK,OAAO;gBACV,OAAO,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;YACtC,KAAK,OAAO;gBACV,OAAO,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;YACtC;gBACE,2CAA2C;gBAC3C,OAAO,KAA2B,CAAC;QACvC,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,KAAmB;QAC1C,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;YAChC,+DAA+D;YAC/D,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACpD,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC;IACnC,CAAC;IAEO,mBAAmB,CAAC,KAAsB;QAChD,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QACvD,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC;IACnC,CAAC;IAEO,gBAAgB,CAAC,KAAmB;QAC1C,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACpD,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC;IACnC,CAAC;IAEO,gBAAgB,CAAC,KAAmB;QAC1C,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACpD,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC;IACnC,CAAC;IAEO,WAAW,CACjB,MAAoB,EACpB,YAA2C;QAE3C,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QAClD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;QAE5D,OAAO;YACL,IAAI;YACJ,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,YAAY;SACb,CAAC;IACJ,CAAC;IAEO,kBAAkB,CAAC,KAAyB;QAClD,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC9B,OAAO,KAAqB,CAAC;QAC/B,CAAC;QAED,MAAM,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC;QACtB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAE5C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,mBAAmB,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;QACjD,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACvC,MAAM,MAAM,GAAiB;YAC3B,IAAI,EAAE,QAAQ;YACd,IAAI;YACJ,SAAS,EAAE,GAAG,CAAC,SAAS;SACzB,CAAC;QAEF,QAAQ,GAAG,CAAC,YAAY,EAAE,CAAC;YACzB,KAAK,OAAO;gBACV,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAkB,CAAC;YACnD,KAAK,UAAU;gBACb,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAqB,CAAC;YACzD,KAAK,OAAO;gBACV,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAkB,CAAC;YACnD,KAAK,OAAO;gBACV,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAkB,CAAC;QACrD,CAAC;IACH,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,gBAAgB,CAAC,KAAmB;QACzC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,OAAO;gBACV,OAAO,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC;YACxC,KAAK,UAAU,CAAC;YAChB,KAAK,OAAO,CAAC;YACb,KAAK,OAAO;gBACV,OAAO,IAAI,CAAC;YACd;gBACE,OAAO,KAAK,CAAC;QACjB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,iBAAiB,CAAC,KAAmB;QAC1C,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,OAAO;gBACV,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;oBACnC,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,kBAAkB;gBACvE,CAAC;gBACD,OAAO,CAAC,CAAC;YACX,KAAK,UAAU,CAAC;YAChB,KAAK,OAAO,CAAC;YACb,KAAK,OAAO;gBACV,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;YACpD;gBACE,OAAO,CAAC,CAAC;QACb,CAAC;IACH,CAAC;CACF"}

package/dist/src/context-log.d.ts

@@ -0,0 +1,99 @@
+import type { JsStore } from '@animalabs/chronicle';
+import type { ContentBlock } from '@animalabs/membrane';
+import type { MessageId, ContextEntry, SourceRelation, ContextLogView } from './types/index.js';
+/**
+ * Wrapper around Chronicle append_log state for context log storage.
+ * The context log is a materialized, editable working set derived from the message store.
+ *
+ * Supports namespacing for multi-agent scenarios where each agent has its own context log
+ * but shares the same message store.
+ */
+export declare class ContextLog {
+    private store;
+    private blobManager;
+    private sourceToIndices;
+    private tokenEstimator;
+    private stateId;
+    constructor(store: JsStore, options?: {
+        estimator?: (text: string) => number;
+        /** Namespace for multi-agent support. Creates state ID: `{namespace}/context` */
+        namespace?: string;
+    });
+    /**
+     * Register the context log state in Chronicle.
+     * Should be called once when setting up the store.
+     *
+     * @param store The Chronicle store
+     * @param namespace Optional namespace for multi-agent support
+     */
+    static register(store: JsStore, namespace?: string): void;
+    private rebuildSourceIndex;
+    /**
+     * Append a new entry to the context log.
+     */
+    append(participant: string, content: ContentBlock[], sourceMessageId?: MessageId, sourceRelation?: SourceRelation, cacheMarker?: boolean): ContextEntry;
+    /**
+     * Edit an entry's content at a specific index.
+     */
+    edit(index: number, content: ContentBlock[]): void;
+    /**
+     * Remove an entry at a specific index.
+     */
+    remove(index: number): void;
+    /**
+     * Remove a range of entries.
+     */
+    removeRange(start: number, end: number): void;
+    /**
+     * Replace the entire context log with new entries.
+     * Useful for strategies that rebuild the context.
+     */
+    replaceAll(entries: Array<{
+        participant: string;
+        content: ContentBlock[];
+        sourceMessageId?: MessageId;
+        sourceRelation?: SourceRelation;
+        cacheMarker?: boolean;
+    }>): ContextEntry[];
+    /**
+     * Get an entry by index.
+     */
+    get(index: number): ContextEntry | null;
+    /**
+     * Get all entries.
+     */
+    getAll(): ContextEntry[];
+    /**
+     * Get entries from a specific index.
+     */
+    getFrom(index: number): ContextEntry[];
+    /**
+     * Get the last N entries.
+     */
+    getTail(count: number): ContextEntry[];
+    /**
+     * Get the total number of entries.
+     */
+    length(): number;
+    /**
+     * Find all entries that reference a specific source message.
+     */
+    findBySource(sourceMessageId: MessageId): ContextEntry[];
+    /**
+     * Get the source relation for entries referencing a message.
+     */
+    getSourceRelation(sourceMessageId: MessageId): Map<number, SourceRelation | undefined>;
+    /**
+     * Estimate tokens for an entry.
+     */
+    estimateTokens(entry: ContextEntry): number;
+    private estimateBlockTokens;
+    /**
+     * Create a read-only view of the log for strategies.
+     */
+    createView(): ContextLogView;
+    private getAllInternal;
+    private getInternal;
+    private internalToEntry;
+}
+//# sourceMappingURL=context-log.d.ts.map

package/dist/src/context-log.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-log.d.ts","sourceRoot":"","sources":["../../src/context-log.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,KAAK,EACV,SAAS,EACT,YAAY,EAEZ,cAAc,EACd,cAAc,EACf,MAAM,kBAAkB,CAAC;AAK1B;;;;;;GAMG;AACH,qBAAa,UAAU;IAOnB,OAAO,CAAC,KAAK;IANf,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,eAAe,CAA0C;IACjE,OAAO,CAAC,cAAc,CAA2B;IACjD,OAAO,CAAC,OAAO,CAAS;gBAGd,KAAK,EAAE,OAAO,EACtB,OAAO,GAAE;QACP,SAAS,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,MAAM,CAAC;QACrC,iFAAiF;QACjF,SAAS,CAAC,EAAE,MAAM,CAAC;KACf;IAUR;;;;;;OAMG;IACH,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI;IAUzD,OAAO,CAAC,kBAAkB;IAa1B;;OAEG;IACH,MAAM,CACJ,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,YAAY,EAAE,EACvB,eAAe,CAAC,EAAE,SAAS,EAC3B,cAAc,CAAC,EAAE,cAAc,EAC/B,WAAW,CAAC,EAAE,OAAO,GACpB,YAAY;IAgCf;;OAEG;IACH,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE,GAAG,IAAI;IAelD;;OAEG;IACH,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAU3B;;OAEG;IACH,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,IAAI;IAK7C;;;OAGG;IACH,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC;QACxB,WAAW,EAAE,MAAM,CAAC;QACpB,OAAO,EAAE,YAAY,EAAE,CAAC;QACxB,eAAe,CAAC,EAAE,SAAS,CAAC;QAC5B,cAAc,CAAC,EAAE,cAAc,CAAC;QAChC,WAAW,CAAC,EAAE,OAAO,CAAC;KACvB,CAAC,GAAG,YAAY,EAAE;IAsBnB;;OAEG;IACH,GAAG,CAAC,KAAK,EAAE,MAAM,GAAG,YAAY,GAAG,IAAI;IAQvC;;OAEG;IACH,MAAM,IAAI,YAAY,EAAE;IAIxB;;OAEG;IACH,OAAO,CAAC,KAAK,EAAE,MAAM,GAAG,YAAY,EAAE;IAItC;;OAEG;IACH,OAAO,CAAC,KAAK,EAAE,MAAM,GAAG,YAAY,EAAE;IAKtC;;OAEG;IACH,MAAM,IAAI,MAAM;IAIhB;;OAEG;IACH,YAAY,CAAC,eAAe,EAAE,SAAS,GAAG,YAAY,EAAE;IAgBxD;;OAEG;IACH,iBAAiB,CAAC,eAAe,EAAE,SAAS,GAAG,GAAG,CAAC,MAAM,EAAE,cAAc,GAAG,SAAS,CAAC;IAgBtF;;OAEG;IACH,cAAc,CAAC,KAAK,EAAE,YAAY,GAAG,MAAM;IAQ3C,OAAO,CAAC,mBAAmB;IAyB3B;;OAEG;IACH,UAAU,IAAI,cAAc;IAU5B,OAAO,CAAC,cAAc;IAQtB,OAAO,CAAC,WAAW;IAKnB,OAAO,CAAC,eAAe;CAUxB"}

package/dist/src/context-log.js

@@ -0,0 +1,277 @@
+import { BlobManager } from './blob-manager.js';
+const DEFAULT_CONTEXT_STATE_ID = 'context';
+/**
+ * Wrapper around Chronicle append_log state for context log storage.
+ * The context log is a materialized, editable working set derived from the message store.
+ *
+ * Supports namespacing for multi-agent scenarios where each agent has its own context log
+ * but shares the same message store.
+ */
+export class ContextLog {
+    store;
+    blobManager;
+    sourceToIndices = new Map();
+    tokenEstimator;
+    stateId;
+    constructor(store, options = {}) {
+        this.store = store;
+        this.stateId = options.namespace
+            ? `${options.namespace}/context`
+            : DEFAULT_CONTEXT_STATE_ID;
+        this.blobManager = new BlobManager(store);
+        this.tokenEstimator = options.estimator ?? defaultTokenEstimator;
+        this.rebuildSourceIndex();
+    }
+    /**
+     * Register the context log state in Chronicle.
+     * Should be called once when setting up the store.
+     *
+     * @param store The Chronicle store
+     * @param namespace Optional namespace for multi-agent support
+     */
+    static register(store, namespace) {
+        const stateId = namespace ? `${namespace}/context` : DEFAULT_CONTEXT_STATE_ID;
+        store.registerState({
+            id: stateId,
+            strategy: 'append_log',
+            deltaSnapshotEvery: 50,
+            fullSnapshotEvery: 10,
+        });
+    }
+    rebuildSourceIndex() {
+        this.sourceToIndices.clear();
+        const entries = this.getAllInternal();
+        for (let i = 0; i < entries.length; i++) {
+            const entry = entries[i];
+            if (entry.sourceMessageId) {
+                const set = this.sourceToIndices.get(entry.sourceMessageId) ?? new Set();
+                set.add(i);
+                this.sourceToIndices.set(entry.sourceMessageId, set);
+            }
+        }
+    }
+    /**
+     * Append a new entry to the context log.
+     */
+    append(participant, content, sourceMessageId, sourceRelation, cacheMarker) {
+        const storedContent = this.blobManager.extractBlobs(content);
+        const index = this.length();
+        const internal = {
+            index,
+            sourceMessageId,
+            sourceRelation,
+            participant,
+            content: storedContent,
+            cacheMarker,
+        };
+        this.store.appendToStateJson(this.stateId, internal);
+        // Update source index
+        if (sourceMessageId) {
+            const set = this.sourceToIndices.get(sourceMessageId) ?? new Set();
+            set.add(index);
+            this.sourceToIndices.set(sourceMessageId, set);
+        }
+        return {
+            index,
+            sourceMessageId,
+            sourceRelation,
+            participant,
+            content,
+            cacheMarker,
+        };
+    }
+    /**
+     * Edit an entry's content at a specific index.
+     */
+    edit(index, content) {
+        const internal = this.getInternal(index);
+        if (!internal) {
+            throw new Error(`Context entry not found at index: ${index}`);
+        }
+        const storedContent = this.blobManager.extractBlobs(content);
+        const updated = {
+            ...internal,
+            content: storedContent,
+        };
+        this.store.editStateItem(this.stateId, index, Buffer.from(JSON.stringify(updated)));
+    }
+    /**
+     * Remove an entry at a specific index.
+     */
+    remove(index) {
+        const internal = this.getInternal(index);
+        if (!internal) {
+            throw new Error(`Context entry not found at index: ${index}`);
+        }
+        this.store.redactStateItems(this.stateId, index, index + 1);
+        this.rebuildSourceIndex();
+    }
+    /**
+     * Remove a range of entries.
+     */
+    removeRange(start, end) {
+        this.store.redactStateItems(this.stateId, start, end);
+        this.rebuildSourceIndex();
+    }
+    /**
+     * Replace the entire context log with new entries.
+     * Useful for strategies that rebuild the context.
+     */
+    replaceAll(entries) {
+        // Clear existing entries
+        const len = this.length();
+        if (len > 0) {
+            this.store.redactStateItems(this.stateId, 0, len);
+        }
+        // Add new entries
+        const result = [];
+        for (const entry of entries) {
+            result.push(this.append(entry.participant, entry.content, entry.sourceMessageId, entry.sourceRelation, entry.cacheMarker));
+        }
+        return result;
+    }
+    /**
+     * Get an entry by index.
+     */
+    get(index) {
+        const internal = this.getInternal(index);
+        if (!internal) {
+            return null;
+        }
+        return this.internalToEntry(internal);
+    }
+    /**
+     * Get all entries.
+     */
+    getAll() {
+        return this.getAllInternal().map((internal) => this.internalToEntry(internal));
+    }
+    /**
+     * Get entries from a specific index.
+     */
+    getFrom(index) {
+        return this.getAll().slice(index);
+    }
+    /**
+     * Get the last N entries.
+     */
+    getTail(count) {
+        const all = this.getAll();
+        return all.slice(Math.max(0, all.length - count));
+    }
+    /**
+     * Get the total number of entries.
+     */
+    length() {
+        return this.store.getStateLen(this.stateId) ?? 0;
+    }
+    /**
+     * Find all entries that reference a specific source message.
+     */
+    findBySource(sourceMessageId) {
+        const indices = this.sourceToIndices.get(sourceMessageId);
+        if (!indices) {
+            return [];
+        }
+        const entries = [];
+        for (const index of indices) {
+            const entry = this.get(index);
+            if (entry) {
+                entries.push(entry);
+            }
+        }
+        return entries;
+    }
+    /**
+     * Get the source relation for entries referencing a message.
+     */
+    getSourceRelation(sourceMessageId) {
+        const result = new Map();
+        const indices = this.sourceToIndices.get(sourceMessageId);
+        if (!indices) {
+            return result;
+        }
+        for (const index of indices) {
+            const entry = this.get(index);
+            if (entry) {
+                result.set(index, entry.sourceRelation);
+            }
+        }
+        return result;
+    }
+    /**
+     * Estimate tokens for an entry.
+     */
+    estimateTokens(entry) {
+        let tokens = 0;
+        for (const block of entry.content) {
+            tokens += this.estimateBlockTokens(block);
+        }
+        return tokens;
+    }
+    estimateBlockTokens(block) {
+        switch (block.type) {
+            case 'text':
+                return this.tokenEstimator(block.text);
+            case 'thinking':
+                return this.tokenEstimator(block.thinking);
+            case 'tool_use':
+                return this.tokenEstimator(JSON.stringify(block.input)) + 20;
+            case 'tool_result':
+                if (!block.content)
+                    return 0;
+                if (typeof block.content === 'string') {
+                    return this.tokenEstimator(block.content);
+                }
+                return block.content.reduce((sum, b) => sum + this.estimateBlockTokens(b), 0);
+            case 'image':
+                return block.tokenEstimate ?? 1000;
+            case 'document':
+            case 'audio':
+            case 'video':
+                return 1000;
+            default:
+                return 0;
+        }
+    }
+    /**
+     * Create a read-only view of the log for strategies.
+     */
+    createView() {
+        return {
+            getAll: () => this.getAll(),
+            getFrom: (index) => this.getFrom(index),
+            getTail: (count) => this.getTail(count),
+            length: () => this.length(),
+            estimateTokens: (entry) => this.estimateTokens(entry),
+        };
+    }
+    getAllInternal() {
+        const state = this.store.getStateJson(this.stateId);
+        if (!state || !Array.isArray(state)) {
+            return [];
+        }
+        return state;
+    }
+    getInternal(index) {
+        const all = this.getAllInternal();
+        return all[index] ?? null;
+    }
+    internalToEntry(internal) {
+        return {
+            index: internal.index,
+            sourceMessageId: internal.sourceMessageId,
+            sourceRelation: internal.sourceRelation,
+            participant: internal.participant,
+            content: this.blobManager.resolveBlobs(internal.content),
+            cacheMarker: internal.cacheMarker,
+        };
+    }
+}
+/**
+ * Default token estimator: chars / 4
+ */
+function defaultTokenEstimator(text) {
+    return Math.ceil(text.length / 4);
+}
+//# sourceMappingURL=context-log.js.map

package/dist/src/context-log.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-log.js","sourceRoot":"","sources":["../../src/context-log.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAEhD,MAAM,wBAAwB,GAAG,SAAS,CAAC;AAE3C;;;;;;GAMG;AACH,MAAM,OAAO,UAAU;IAOX;IANF,WAAW,CAAc;IACzB,eAAe,GAAgC,IAAI,GAAG,EAAE,CAAC;IACzD,cAAc,CAA2B;IACzC,OAAO,CAAS;IAExB,YACU,KAAc,EACtB,UAII,EAAE;QALE,UAAK,GAAL,KAAK,CAAS;QAOtB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,SAAS;YAC9B,CAAC,CAAC,GAAG,OAAO,CAAC,SAAS,UAAU;YAChC,CAAC,CAAC,wBAAwB,CAAC;QAC7B,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC;QAC1C,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,SAAS,IAAI,qBAAqB,CAAC;QACjE,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC5B,CAAC;IAED;;;;;;OAMG;IACH,MAAM,CAAC,QAAQ,CAAC,KAAc,EAAE,SAAkB;QAChD,MAAM,OAAO,GAAG,SAAS,CAAC,CAAC,CAAC,GAAG,SAAS,UAAU,CAAC,CAAC,CAAC,wBAAwB,CAAC;QAC9E,KAAK,CAAC,aAAa,CAAC;YAClB,EAAE,EAAE,OAAO;YACX,QAAQ,EAAE,YAAY;YACtB,kBAAkB,EAAE,EAAE;YACtB,iBAAiB,EAAE,EAAE;SACtB,CAAC,CAAC;IACL,CAAC;IAEO,kBAAkB;QACxB,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QACtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACxC,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACzB,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;gBAC1B,MAAM,GAAG,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,IAAI,GAAG,EAAE,CAAC;gBACzE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBACX,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,MAAM,CACJ,WAAmB,EACnB,OAAuB,EACvB,eAA2B,EAC3B,cAA+B,EAC/B,WAAqB;QAErB,MAAM,aAAa,GAAG,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC7D,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAE5B,MAAM,QAAQ,GAAyB;YACrC,KAAK;YACL,eAAe;YACf,cAAc;YACd,WAAW;YACX,OAAO,EAAE,aAAa;YACtB,WAAW;SACZ,CAAC;QAEF,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAErD,sBAAsB;QACtB,IAAI,eAAe,EAAE,CAAC;YACpB,MAAM,GAAG,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,IAAI,GAAG,EAAE,CAAC;YACnE,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACf,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC;QACjD,CAAC;QAED,OAAO;YACL,KAAK;YACL,eAAe;YACf,cAAc;YACd,WAAW;YACX,OAAO;YACP,WAAW;SACZ,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,IAAI,CAAC,KAAa,EAAE,OAAuB;QACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QACzC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,qCAAqC,KAAK,EAAE,CAAC,CAAC;QAChE,CAAC;QAED,MAAM,aAAa,GAAG,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC7D,MAAM,OAAO,GAAyB;YACpC,GAAG,QAAQ;YACX,OAAO,EAAE,aAAa;SACvB,CAAC;QAEF,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IACtF,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAa;QAClB,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QACzC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,qCAAqC,KAAK,EAAE,CAAC,CAAC;QAChE,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;QAC5D,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,KAAa,EAAE,GAAW;QACpC,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC;QACtD,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC5B,CAAC;IAED;;;OAGG;IACH,UAAU,CAAC,OAMT;QACA,yBAAyB;QACzB,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAC1B,IAAI,GAAG,GAAG,CAAC,EAAE,CAAC;YACZ,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,GAAG,CAAC,CAAC;QACpD,CAAC;QAED,kBAAkB;QAClB,MAAM,MAAM,GAAmB,EAAE,CAAC;QAClC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CACrB,KAAK,CAAC,WAAW,EACjB,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,eAAe,EACrB,KAAK,CAAC,cAAc,EACpB,KAAK,CAAC,WAAW,CAClB,CAAC,CAAC;QACL,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,GAAG,CAAC,KAAa;QACf,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QACzC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,OAAO,IAAI,CAAC,cAAc,EAAE,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC;IACjF,CAAC;IAED;;OAEG;IACH,OAAO,CAAC,KAAa;QACnB,OAAO,IAAI,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC;IAED;;OAEG;IACH,OAAO,CAAC,KAAa;QACnB,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAC1B,OAAO,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC,CAAC;IACpD,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,OAAO,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACnD,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,eAA0B;QACrC,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QAC1D,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,OAAO,GAAmB,EAAE,CAAC;QACnC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAC9B,IAAI,KAAK,EAAE,CAAC;gBACV,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACtB,CAAC;QACH,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,eAA0B;QAC1C,MAAM,MAAM,GAAG,IAAI,GAAG,EAAsC,CAAC;QAC7D,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QAC1D,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAC9B,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,cAAc,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,KAAmB;QAChC,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClC,MAAM,IAAI,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;QAC5C,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,mBAAmB,CAAC,KAAmB;QAC7C,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,MAAM;gBACT,OAAO,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACzC,KAAK,UAAU;gBACb,OAAO,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC7C,KAAK,UAAU;gBACb,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,CAAC;YAC/D,KAAK,aAAa;gBAChB,IAAI,CAAC,KAAK,CAAC,OAAO;oBAAE,OAAO,CAAC,CAAC;gBAC7B,IAAI,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;oBACtC,OAAO,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBAC5C,CAAC;gBACD,OAAO,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAChF,KAAK,OAAO;gBACV,OAAO,KAAK,CAAC,aAAa,IAAI,IAAI,CAAC;YACrC,KAAK,UAAU,CAAC;YAChB,KAAK,OAAO,CAAC;YACb,KAAK,OAAO;gBACV,OAAO,IAAI,CAAC;YACd;gBACE,OAAO,CAAC,CAAC;QACb,CAAC;IACH,CAAC;IAED;;OAEG;IACH,UAAU;QACR,OAAO;YACL,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE;YAC3B,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;YACvC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;YACvC,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE;YAC3B,cAAc,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC;SACtD,CAAC;IACJ,CAAC;IAEO,cAAc;QACpB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACpD,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACpC,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,OAAO,KAA+B,CAAC;IACzC,CAAC;IAEO,WAAW,CAAC,KAAa;QAC/B,MAAM,GAAG,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QAClC,OAAO,GAAG,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC;IAC5B,CAAC;IAEO,eAAe,CAAC,QAA8B;QACpD,OAAO;YACL,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,eAAe,EAAE,QAAQ,CAAC,eAAe;YACzC,cAAc,EAAE,QAAQ,CAAC,cAAc;YACvC,WAAW,EAAE,QAAQ,CAAC,WAAW;YACjC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC;YACxD,WAAW,EAAE,QAAQ,CAAC,WAAW;SAClC,CAAC;IACJ,CAAC;CACF;AAED;;GAEG;AACH,SAAS,qBAAqB,CAAC,IAAY;IACzC,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AACpC,CAAC"}