@vybestack/llxprt-code-core 0.11.0 → 0.12.0-nightly.260910.2aac6841e

package/dist/src/storage/request-media-resolver.d.ts

@@ -0,0 +1,67 @@
+/**
+ * @license
+ * Copyright 2026 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type IContent } from '../services/history/IContent.js';
+import type { LocalMediaStore } from './local-media-store.js';
+export declare const DEFAULT_REQUEST_MEDIA_BUDGET_BYTES: number;
+export interface RequestMediaResolutionInput {
+    readonly contents: readonly IContent[];
+    readonly requestId: string;
+    readonly turnId: string;
+    readonly aggregateBudgetBytes: number;
+    readonly signal?: AbortSignal;
+}
+export interface RequestMediaAccounting {
+    readonly selectedReferenceCount: number;
+    readonly uniqueContentCount: number;
+    readonly selectedNormalizedBytes: number;
+    readonly materializedNormalizedBytes: number;
+    readonly storeReadCount: number;
+    readonly reservedContentCount: number;
+    readonly released: boolean;
+}
+export interface RequestMediaResolverAccounting {
+    readonly activeRequestCount: number;
+    readonly reservedContentCount: number;
+    readonly materializedNormalizedBytes: number;
+    readonly storeReadCount: number;
+}
+export interface PendingMediaReleaseRecovery {
+    readonly attempted: number;
+    readonly recovered: number;
+    readonly remaining: number;
+}
+export interface ResolvedMediaRequest {
+    withContents<T>(consume: (contents: IContent[]) => T): T;
+    registerCleanup(cleanup: () => void | Promise<void>): void;
+    accounting(): RequestMediaAccounting;
+    release(): Promise<void>;
+}
+export interface RequestMediaResolutionService {
+    resolve(input: RequestMediaResolutionInput): Promise<ResolvedMediaRequest>;
+}
+export declare class RequestMediaResolutionError extends Error {
+    readonly contentId: string;
+    readonly turnId: string;
+    constructor(contentId: string, turnId: string, operation: string, options?: ErrorOptions);
+}
+export declare function assertRequestMediaBudget(contents: readonly IContent[], turnId: string, aggregateBudgetBytes: number): void;
+export declare class RequestMediaResolver {
+    private readonly store;
+    private activeRequestCount;
+    private reservedContentCount;
+    private materializedNormalizedBytes;
+    private storeReadCount;
+    private readonly pendingReleases;
+    constructor(store: LocalMediaStore);
+    accounting(): RequestMediaResolverAccounting;
+    pendingReleaseCount(): number;
+    recoverPendingReleases(): Promise<PendingMediaReleaseRecovery>;
+    private dispose;
+    private release;
+    private materializeReferences;
+    private resolvedRequest;
+    resolve(input: RequestMediaResolutionInput): Promise<ResolvedMediaRequest>;
+}

package/dist/src/storage/request-media-resolver.js

@@ -0,0 +1,325 @@
+/**
+ * @license
+ * Copyright 2026 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { randomUUID } from 'node:crypto';
+import { isMediaReferenceBlock, } from '../services/history/IContent.js';
+import { DEFAULT_IMAGE_PAYLOAD_BUDGET_BYTES } from '../config/configTypes.js';
+export const DEFAULT_REQUEST_MEDIA_BUDGET_BYTES = DEFAULT_IMAGE_PAYLOAD_BUDGET_BYTES;
+export class RequestMediaResolutionError extends Error {
+    contentId;
+    turnId;
+    constructor(contentId, turnId, operation, options) {
+        super(`Media request ${operation} failed for content ${contentId} in turn ${turnId}`, options);
+        this.contentId = contentId;
+        this.turnId = turnId;
+        this.name = 'RequestMediaResolutionError';
+    }
+}
+function abortError() {
+    const error = new Error('Media request resolution was aborted');
+    error.name = 'AbortError';
+    return error;
+}
+function throwIfAborted(signal) {
+    if (signal?.aborted === true)
+        throw abortError();
+}
+function validateBudget(budget) {
+    if (!Number.isSafeInteger(budget) || budget < 0) {
+        throw new Error('Aggregate request media budget must be a non-negative safe integer');
+    }
+}
+function contentTurnId(content, fallbackTurnId) {
+    return (content.metadata?.turnId ??
+        content.metadata?.id ??
+        (content.metadata?.chronology === undefined
+            ? fallbackTurnId
+            : `chronology:${content.metadata.chronology.seq}`));
+}
+function selectReferences(input) {
+    const selected = [];
+    for (const content of input.contents) {
+        const turnId = contentTurnId(content, input.turnId);
+        for (const block of content.blocks) {
+            if (block.type !== 'media' || block.encoding !== 'reference')
+                continue;
+            const candidate = block;
+            if (!isMediaReferenceBlock(candidate)) {
+                throw new RequestMediaResolutionError(block.contentId, turnId, 'metadata validation');
+            }
+            selected.push({ reference: candidate, turnId });
+        }
+    }
+    return selected;
+}
+function selectedNormalizedBytes(input) {
+    let total = 0;
+    for (const content of input.contents) {
+        const turnId = contentTurnId(content, input.turnId);
+        for (const block of content.blocks) {
+            if (block.type !== 'media' || block.encoding === 'url')
+                continue;
+            const normalizedLength = block.encoding === 'reference'
+                ? block.normalizedBase64Length
+                : block.data.length;
+            const nextTotal = total + normalizedLength;
+            if (!Number.isSafeInteger(nextTotal) ||
+                nextTotal > input.aggregateBudgetBytes) {
+                throw new RequestMediaResolutionError(block.encoding === 'reference' ? block.contentId : 'inline-media', turnId, `aggregate budget (${nextTotal} > ${input.aggregateBudgetBytes})`);
+            }
+            total = nextTotal;
+        }
+    }
+    return total;
+}
+export function assertRequestMediaBudget(contents, turnId, aggregateBudgetBytes) {
+    validateBudget(aggregateBudgetBytes);
+    selectedNormalizedBytes({
+        contents,
+        requestId: 'request-media-budget-validation',
+        turnId,
+        aggregateBudgetBytes,
+    });
+}
+function inlineBlock(reference, data) {
+    const block = {
+        type: 'media',
+        encoding: 'base64',
+        data,
+        mimeType: reference.mimeType,
+        ...(reference.caption === undefined ? {} : { caption: reference.caption }),
+        ...(reference.filename === undefined
+            ? {}
+            : { filename: reference.filename }),
+        ...(reference.providerMetadata === undefined
+            ? {}
+            : { providerMetadata: reference.providerMetadata }),
+        ...(reference.dimensions === undefined
+            ? {}
+            : { dimensions: reference.dimensions }),
+        semanticMetadata: reference.semanticMetadata,
+        ...(reference.providerFileIds === undefined
+            ? {}
+            : { providerFileIds: reference.providerFileIds }),
+        ...(reference.providerFiles === undefined
+            ? {}
+            : { providerFiles: reference.providerFiles }),
+    };
+    Object.defineProperty(block, 'sourceContentId', {
+        value: reference.contentId,
+        enumerable: false,
+        configurable: false,
+        writable: false,
+    });
+    return block;
+}
+function materializeContents(contents, encodedByContentId) {
+    return contents.map((content) => {
+        const blocks = content.blocks.map((block) => {
+            if (block.type !== 'media' || block.encoding !== 'reference')
+                return block;
+            const data = encodedByContentId.get(block.contentId);
+            if (data === undefined) {
+                throw new RequestMediaResolutionError(block.contentId, contentTurnId(content, 'unknown'), 'materialization');
+            }
+            return inlineBlock(block, data);
+        });
+        const changed = blocks.some((block, index) => block !== content.blocks[index]);
+        return changed ? { ...content, blocks } : content;
+    });
+}
+function uniqueReferencesByContentId(selected) {
+    const unique = new Map();
+    for (const entry of selected) {
+        if (!unique.has(entry.reference.contentId)) {
+            unique.set(entry.reference.contentId, entry);
+        }
+    }
+    return unique;
+}
+export class RequestMediaResolver {
+    store;
+    activeRequestCount = 0;
+    reservedContentCount = 0;
+    materializedNormalizedBytes = 0;
+    storeReadCount = 0;
+    pendingReleases = new Set();
+    constructor(store) {
+        this.store = store;
+    }
+    accounting() {
+        return {
+            activeRequestCount: this.activeRequestCount,
+            reservedContentCount: this.reservedContentCount,
+            materializedNormalizedBytes: this.materializedNormalizedBytes,
+            storeReadCount: this.storeReadCount,
+        };
+    }
+    pendingReleaseCount() {
+        return this.pendingReleases.size;
+    }
+    async recoverPendingReleases() {
+        const pending = [...this.pendingReleases];
+        const failures = [];
+        let recovered = 0;
+        for (const resources of pending) {
+            try {
+                await this.release(resources);
+                recovered += 1;
+            }
+            catch (error) {
+                failures.push(error);
+            }
+        }
+        const result = {
+            attempted: pending.length,
+            recovered,
+            remaining: this.pendingReleases.size,
+        };
+        if (failures.length > 0) {
+            throw new AggregateError(failures, 'Pending media release recovery failed', {
+                cause: result,
+            });
+        }
+        return result;
+    }
+    async dispose(resources) {
+        const failures = [];
+        if (!resources.disposed) {
+            resources.disposed = true;
+            for (let index = resources.registeredCleanups.length - 1; index >= 0; index -= 1) {
+                try {
+                    await resources.registeredCleanups[index]();
+                }
+                catch (error) {
+                    failures.push(error);
+                }
+            }
+            resources.registeredCleanups.splice(0);
+            resources.resolvedContents.splice(0);
+            resources.encodedByContentId.clear();
+            this.materializedNormalizedBytes -= resources.materializedBytes;
+            resources.materializedBytes = 0;
+        }
+        let releasedReservationCount = 0;
+        for (let index = resources.reservedContentIds.length - 1; index >= 0; index -= 1) {
+            try {
+                await this.store.release(resources.reservedContentIds[index], resources.ownerId);
+                resources.reservedContentIds.splice(index, 1);
+                releasedReservationCount += 1;
+            }
+            catch (error) {
+                failures.push(error);
+            }
+        }
+        this.reservedContentCount -= releasedReservationCount;
+        if (resources.reservedContentIds.length === 0) {
+            this.activeRequestCount -= 1;
+            resources.released = true;
+            this.pendingReleases.delete(resources);
+        }
+        else {
+            this.pendingReleases.add(resources);
+        }
+        if (failures.length > 0) {
+            throw new AggregateError(failures, 'Failed to release media request');
+        }
+    }
+    release(resources) {
+        if (resources.released)
+            return Promise.resolve();
+        resources.releasePromise ??= this.dispose(resources).finally(() => {
+            if (!resources.released)
+                resources.releasePromise = undefined;
+        });
+        return resources.releasePromise;
+    }
+    async materializeReferences(input, unique, resources) {
+        for (const entry of unique.values()) {
+            throwIfAborted(input.signal);
+            this.storeReadCount += 1;
+            resources.readCount += 1;
+            let bytes;
+            try {
+                bytes = await this.store.reserveAndReadVerified(entry.reference, resources.ownerId);
+            }
+            catch (error) {
+                throw new RequestMediaResolutionError(entry.reference.contentId, entry.turnId, 'blob verification', { cause: error });
+            }
+            resources.reservedContentIds.push(entry.reference.contentId);
+            this.reservedContentCount += 1;
+            throwIfAborted(input.signal);
+            const encoded = Buffer.from(bytes).toString('base64');
+            if (encoded.length !== entry.reference.normalizedBase64Length) {
+                throw new RequestMediaResolutionError(entry.reference.contentId, entry.turnId, 'normalized size verification');
+            }
+            resources.encodedByContentId.set(entry.reference.contentId, encoded);
+            resources.materializedBytes += encoded.length;
+            this.materializedNormalizedBytes += encoded.length;
+        }
+    }
+    resolvedRequest(selectedReferenceCount, uniqueContentCount, selectedBytes, resources) {
+        return {
+            withContents: (consume) => {
+                if (resources.disposed) {
+                    throw new Error('Cannot consume media request contents after release');
+                }
+                return consume(resources.resolvedContents);
+            },
+            registerCleanup: (cleanup) => {
+                if (resources.disposed) {
+                    throw new Error('Cannot register materialized request cleanup after disposal');
+                }
+                resources.registeredCleanups.push(cleanup);
+            },
+            accounting: () => ({
+                selectedReferenceCount,
+                uniqueContentCount,
+                selectedNormalizedBytes: selectedBytes,
+                materializedNormalizedBytes: resources.materializedBytes,
+                storeReadCount: resources.readCount,
+                reservedContentCount: resources.reservedContentIds.length,
+                released: resources.released,
+            }),
+            release: () => this.release(resources),
+        };
+    }
+    async resolve(input) {
+        validateBudget(input.aggregateBudgetBytes);
+        throwIfAborted(input.signal);
+        const selected = selectReferences(input);
+        const selectedBytes = selectedNormalizedBytes(input);
+        throwIfAborted(input.signal);
+        const unique = uniqueReferencesByContentId(selected);
+        const resources = {
+            ownerId: `${input.requestId}:${randomUUID()}`,
+            reservedContentIds: [],
+            encodedByContentId: new Map(),
+            registeredCleanups: [],
+            resolvedContents: [],
+            materializedBytes: 0,
+            readCount: 0,
+            disposed: false,
+            released: false,
+            releasePromise: undefined,
+        };
+        this.activeRequestCount += 1;
+        try {
+            await this.materializeReferences(input, unique, resources);
+            resources.resolvedContents = materializeContents(input.contents, resources.encodedByContentId);
+            return this.resolvedRequest(selected.length, unique.size, selectedBytes, resources);
+        }
+        catch (error) {
+            try {
+                await this.release(resources);
+            }
+            catch (releaseError) {
+                throw new AggregateError([error, releaseError], 'Media request resolution and cleanup failed');
+            }
+            throw error;
+        }
+    }
+}
+//# sourceMappingURL=request-media-resolver.js.map

package/dist/src/storage/request-media-resolver.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"request-media-resolver.js","sourceRoot":"","sources":["../../../src/storage/request-media-resolver.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EACL,qBAAqB,GAItB,MAAM,iCAAiC,CAAC;AAEzC,OAAO,EAAE,kCAAkC,EAAE,MAAM,0BAA0B,CAAC;AAE9E,MAAM,CAAC,MAAM,kCAAkC,GAC7C,kCAAkC,CAAC;AAiDrC,MAAM,OAAO,2BAA4B,SAAQ,KAAK;IAEzC;IACA;IAFX,YACW,SAAiB,EACjB,MAAc,EACvB,SAAiB,EACjB,OAAsB;QAEtB,KAAK,CACH,iBAAiB,SAAS,uBAAuB,SAAS,YAAY,MAAM,EAAE,EAC9E,OAAO,CACR,CAAC;QARO,cAAS,GAAT,SAAS,CAAQ;QACjB,WAAM,GAAN,MAAM,CAAQ;QAQvB,IAAI,CAAC,IAAI,GAAG,6BAA6B,CAAC;IAC5C,CAAC;CACF;AAED,SAAS,UAAU;IACjB,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;IAChE,KAAK,CAAC,IAAI,GAAG,YAAY,CAAC;IAC1B,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,cAAc,CAAC,MAA+B;IACrD,IAAI,MAAM,EAAE,OAAO,KAAK,IAAI;QAAE,MAAM,UAAU,EAAE,CAAC;AACnD,CAAC;AAED,SAAS,cAAc,CAAC,MAAc;IACpC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,MAAM,GAAG,CAAC,EAAE,CAAC;QAChD,MAAM,IAAI,KAAK,CACb,oEAAoE,CACrE,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,aAAa,CAAC,OAAiB,EAAE,cAAsB;IAC9D,OAAO,CACL,OAAO,CAAC,QAAQ,EAAE,MAAM;QACxB,OAAO,CAAC,QAAQ,EAAE,EAAE;QACpB,CAAC,OAAO,CAAC,QAAQ,EAAE,UAAU,KAAK,SAAS;YACzC,CAAC,CAAC,cAAc;YAChB,CAAC,CAAC,cAAc,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,CACrD,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB,CACvB,KAAkC;IAElC,MAAM,QAAQ,GAAwB,EAAE,CAAC;IACzC,KAAK,MAAM,OAAO,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;QACrC,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;QACpD,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnC,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,QAAQ,KAAK,WAAW;gBAAE,SAAS;YACvE,MAAM,SAAS,GAAY,KAAK,CAAC;YACjC,IAAI,CAAC,qBAAqB,CAAC,SAAS,CAAC,EAAE,CAAC;gBACtC,MAAM,IAAI,2BAA2B,CACnC,KAAK,CAAC,SAAS,EACf,MAAM,EACN,qBAAqB,CACtB,CAAC;YACJ,CAAC;YACD,QAAQ,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,uBAAuB,CAAC,KAAkC;IACjE,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,OAAO,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;QACrC,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;QACpD,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnC,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,QAAQ,KAAK,KAAK;gBAAE,SAAS;YACjE,MAAM,gBAAgB,GACpB,KAAK,CAAC,QAAQ,KAAK,WAAW;gBAC5B,CAAC,CAAC,KAAK,CAAC,sBAAsB;gBAC9B,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC;YACxB,MAAM,SAAS,GAAG,KAAK,GAAG,gBAAgB,CAAC;YAC3C,IACE,CAAC,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC;gBAChC,SAAS,GAAG,KAAK,CAAC,oBAAoB,EACtC,CAAC;gBACD,MAAM,IAAI,2BAA2B,CACnC,KAAK,CAAC,QAAQ,KAAK,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc,EACjE,MAAM,EACN,qBAAqB,SAAS,MAAM,KAAK,CAAC,oBAAoB,GAAG,CAClE,CAAC;YACJ,CAAC;YACD,KAAK,GAAG,SAAS,CAAC;QACpB,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,QAA6B,EAC7B,MAAc,EACd,oBAA4B;IAE5B,cAAc,CAAC,oBAAoB,CAAC,CAAC;IACrC,uBAAuB,CAAC;QACtB,QAAQ;QACR,SAAS,EAAE,iCAAiC;QAC5C,MAAM;QACN,oBAAoB;KACrB,CAAC,CAAC;AACL,CAAC;AAED,SAAS,WAAW,CAClB,SAA8B,EAC9B,IAAY;IAEZ,MAAM,KAAK,GAAqB;QAC9B,IAAI,EAAE,OAAO;QACb,QAAQ,EAAE,QAAQ;QAClB,IAAI;QACJ,QAAQ,EAAE,SAAS,CAAC,QAAQ;QAC5B,GAAG,CAAC,SAAS,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,SAAS,CAAC,OAAO,EAAE,CAAC;QAC1E,GAAG,CAAC,SAAS,CAAC,QAAQ,KAAK,SAAS;YAClC,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,EAAE,QAAQ,EAAE,SAAS,CAAC,QAAQ,EAAE,CAAC;QACrC,GAAG,CAAC,SAAS,CAAC,gBAAgB,KAAK,SAAS;YAC1C,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,EAAE,gBAAgB,EAAE,SAAS,CAAC,gBAAgB,EAAE,CAAC;QACrD,GAAG,CAAC,SAAS,CAAC,UAAU,KAAK,SAAS;YACpC,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,EAAE,UAAU,EAAE,SAAS,CAAC,UAAU,EAAE,CAAC;QACzC,gBAAgB,EAAE,SAAS,CAAC,gBAAgB;QAC5C,GAAG,CAAC,SAAS,CAAC,eAAe,KAAK,SAAS;YACzC,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,EAAE,eAAe,EAAE,SAAS,CAAC,eAAe,EAAE,CAAC;QACnD,GAAG,CAAC,SAAS,CAAC,aAAa,KAAK,SAAS;YACvC,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,EAAE,aAAa,EAAE,SAAS,CAAC,aAAa,EAAE,CAAC;KAChD,CAAC;IACF,MAAM,CAAC,cAAc,CAAC,KAAK,EAAE,iBAAiB,EAAE;QAC9C,KAAK,EAAE,SAAS,CAAC,SAAS;QAC1B,UAAU,EAAE,KAAK;QACjB,YAAY,EAAE,KAAK;QACnB,QAAQ,EAAE,KAAK;KAChB,CAAC,CAAC;IACH,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,mBAAmB,CAC1B,QAA6B,EAC7B,kBAA+C;IAE/C,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;QAC9B,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YAC1C,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,QAAQ,KAAK,WAAW;gBAC1D,OAAO,KAAK,CAAC;YACf,MAAM,IAAI,GAAG,kBAAkB,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YACrD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;gBACvB,MAAM,IAAI,2BAA2B,CACnC,KAAK,CAAC,SAAS,EACf,aAAa,CAAC,OAAO,EAAE,SAAS,CAAC,EACjC,iBAAiB,CAClB,CAAC;YACJ,CAAC;YACD,OAAO,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CACzB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,KAAK,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAClD,CAAC;QACF,OAAO,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC;AAeD,SAAS,2BAA2B,CAClC,QAAsC;IAEtC,MAAM,MAAM,GAAG,IAAI,GAAG,EAA6B,CAAC;IACpD,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;QAC7B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3C,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,OAAO,oBAAoB;IAOF;IANrB,kBAAkB,GAAG,CAAC,CAAC;IACvB,oBAAoB,GAAG,CAAC,CAAC;IACzB,2BAA2B,GAAG,CAAC,CAAC;IAChC,cAAc,GAAG,CAAC,CAAC;IACV,eAAe,GAAG,IAAI,GAAG,EAA8B,CAAC;IAEzE,YAA6B,KAAsB;QAAtB,UAAK,GAAL,KAAK,CAAiB;IAAG,CAAC;IAEvD,UAAU;QACR,OAAO;YACL,kBAAkB,EAAE,IAAI,CAAC,kBAAkB;YAC3C,oBAAoB,EAAE,IAAI,CAAC,oBAAoB;YAC/C,2BAA2B,EAAE,IAAI,CAAC,2BAA2B;YAC7D,cAAc,EAAE,IAAI,CAAC,cAAc;SACpC,CAAC;IACJ,CAAC;IAED,mBAAmB;QACjB,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,sBAAsB;QAC1B,MAAM,OAAO,GAAG,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC;QAC1C,MAAM,QAAQ,GAAc,EAAE,CAAC;QAC/B,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,KAAK,MAAM,SAAS,IAAI,OAAO,EAAE,CAAC;YAChC,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBAC9B,SAAS,IAAI,CAAC,CAAC;YACjB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACvB,CAAC;QACH,CAAC;QACD,MAAM,MAAM,GAAG;YACb,SAAS,EAAE,OAAO,CAAC,MAAM;YACzB,SAAS;YACT,SAAS,EAAE,IAAI,CAAC,eAAe,CAAC,IAAI;SACrC,CAAC;QACF,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,cAAc,CACtB,QAAQ,EACR,uCAAuC,EACvC;gBACE,KAAK,EAAE,MAAM;aACd,CACF,CAAC;QACJ,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,KAAK,CAAC,OAAO,CAAC,SAAqC;QACzD,MAAM,QAAQ,GAAc,EAAE,CAAC;QAC/B,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC;YACxB,SAAS,CAAC,QAAQ,GAAG,IAAI,CAAC;YAC1B,KACE,IAAI,KAAK,GAAG,SAAS,CAAC,kBAAkB,CAAC,MAAM,GAAG,CAAC,EACnD,KAAK,IAAI,CAAC,EACV,KAAK,IAAI,CAAC,EACV,CAAC;gBACD,IAAI,CAAC;oBACH,MAAM,SAAS,CAAC,kBAAkB,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC9C,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACvB,CAAC;YACH,CAAC;YACD,SAAS,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YACvC,SAAS,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YACrC,SAAS,CAAC,kBAAkB,CAAC,KAAK,EAAE,CAAC;YACrC,IAAI,CAAC,2BAA2B,IAAI,SAAS,CAAC,iBAAiB,CAAC;YAChE,SAAS,CAAC,iBAAiB,GAAG,CAAC,CAAC;QAClC,CAAC;QACD,IAAI,wBAAwB,GAAG,CAAC,CAAC;QACjC,KACE,IAAI,KAAK,GAAG,SAAS,CAAC,kBAAkB,CAAC,MAAM,GAAG,CAAC,EACnD,KAAK,IAAI,CAAC,EACV,KAAK,IAAI,CAAC,EACV,CAAC;YACD,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CACtB,SAAS,CAAC,kBAAkB,CAAC,KAAK,CAAC,EACnC,SAAS,CAAC,OAAO,CAClB,CAAC;gBACF,SAAS,CAAC,kBAAkB,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;gBAC9C,wBAAwB,IAAI,CAAC,CAAC;YAChC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACvB,CAAC;QACH,CAAC;QACD,IAAI,CAAC,oBAAoB,IAAI,wBAAwB,CAAC;QACtD,IAAI,SAAS,CAAC,kBAAkB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9C,IAAI,CAAC,kBAAkB,IAAI,CAAC,CAAC;YAC7B,SAAS,CAAC,QAAQ,GAAG,IAAI,CAAC;YAC1B,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACzC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACtC,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,cAAc,CAAC,QAAQ,EAAE,iCAAiC,CAAC,CAAC;QACxE,CAAC;IACH,CAAC;IAEO,OAAO,CAAC,SAAqC;QACnD,IAAI,SAAS,CAAC,QAAQ;YAAE,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;QACjD,SAAS,CAAC,cAAc,KAAK,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;YAChE,IAAI,CAAC,SAAS,CAAC,QAAQ;gBAAE,SAAS,CAAC,cAAc,GAAG,SAAS,CAAC;QAChE,CAAC,CAAC,CAAC;QACH,OAAO,SAAS,CAAC,cAAc,CAAC;IAClC,CAAC;IAEO,KAAK,CAAC,qBAAqB,CACjC,KAAkC,EAClC,MAA8C,EAC9C,SAAqC;QAErC,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YACpC,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC7B,IAAI,CAAC,cAAc,IAAI,CAAC,CAAC;YACzB,SAAS,CAAC,SAAS,IAAI,CAAC,CAAC;YACzB,IAAI,KAAiB,CAAC;YACtB,IAAI,CAAC;gBACH,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,sBAAsB,CAC7C,KAAK,CAAC,SAAS,EACf,SAAS,CAAC,OAAO,CAClB,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,IAAI,2BAA2B,CACnC,KAAK,CAAC,SAAS,CAAC,SAAS,EACzB,KAAK,CAAC,MAAM,EACZ,mBAAmB,EACnB,EAAE,KAAK,EAAE,KAAK,EAAE,CACjB,CAAC;YACJ,CAAC;YACD,SAAS,CAAC,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YAC7D,IAAI,CAAC,oBAAoB,IAAI,CAAC,CAAC;YAC/B,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC7B,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACtD,IAAI,OAAO,CAAC,MAAM,KAAK,KAAK,CAAC,SAAS,CAAC,sBAAsB,EAAE,CAAC;gBAC9D,MAAM,IAAI,2BAA2B,CACnC,KAAK,CAAC,SAAS,CAAC,SAAS,EACzB,KAAK,CAAC,MAAM,EACZ,8BAA8B,CAC/B,CAAC;YACJ,CAAC;YACD,SAAS,CAAC,kBAAkB,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACrE,SAAS,CAAC,iBAAiB,IAAI,OAAO,CAAC,MAAM,CAAC;YAC9C,IAAI,CAAC,2BAA2B,IAAI,OAAO,CAAC,MAAM,CAAC;QACrD,CAAC;IACH,CAAC;IAEO,eAAe,CACrB,sBAA8B,EAC9B,kBAA0B,EAC1B,aAAqB,EACrB,SAAqC;QAErC,OAAO;YACL,YAAY,EAAE,CAAI,OAAoC,EAAK,EAAE;gBAC3D,IAAI,SAAS,CAAC,QAAQ,EAAE,CAAC;oBACvB,MAAM,IAAI,KAAK,CACb,qDAAqD,CACtD,CAAC;gBACJ,CAAC;gBACD,OAAO,OAAO,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAC7C,CAAC;YACD,eAAe,EAAE,CAAC,OAAmC,EAAQ,EAAE;gBAC7D,IAAI,SAAS,CAAC,QAAQ,EAAE,CAAC;oBACvB,MAAM,IAAI,KAAK,CACb,6DAA6D,CAC9D,CAAC;gBACJ,CAAC;gBACD,SAAS,CAAC,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC7C,CAAC;YACD,UAAU,EAAE,GAA2B,EAAE,CAAC,CAAC;gBACzC,sBAAsB;gBACtB,kBAAkB;gBAClB,uBAAuB,EAAE,aAAa;gBACtC,2BAA2B,EAAE,SAAS,CAAC,iBAAiB;gBACxD,cAAc,EAAE,SAAS,CAAC,SAAS;gBACnC,oBAAoB,EAAE,SAAS,CAAC,kBAAkB,CAAC,MAAM;gBACzD,QAAQ,EAAE,SAAS,CAAC,QAAQ;aAC7B,CAAC;YACF,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC;SACvC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,OAAO,CACX,KAAkC;QAElC,cAAc,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;QAC3C,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAC7B,MAAM,QAAQ,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,aAAa,GAAG,uBAAuB,CAAC,KAAK,CAAC,CAAC;QACrD,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAC7B,MAAM,MAAM,GAAG,2BAA2B,CAAC,QAAQ,CAAC,CAAC;QACrD,MAAM,SAAS,GAA+B;YAC5C,OAAO,EAAE,GAAG,KAAK,CAAC,SAAS,IAAI,UAAU,EAAE,EAAE;YAC7C,kBAAkB,EAAE,EAAE;YACtB,kBAAkB,EAAE,IAAI,GAAG,EAAE;YAC7B,kBAAkB,EAAE,EAAE;YACtB,gBAAgB,EAAE,EAAE;YACpB,iBAAiB,EAAE,CAAC;YACpB,SAAS,EAAE,CAAC;YACZ,QAAQ,EAAE,KAAK;YACf,QAAQ,EAAE,KAAK;YACf,cAAc,EAAE,SAAS;SAC1B,CAAC;QACF,IAAI,CAAC,kBAAkB,IAAI,CAAC,CAAC;QAC7B,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,qBAAqB,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YAC3D,SAAS,CAAC,gBAAgB,GAAG,mBAAmB,CAC9C,KAAK,CAAC,QAAQ,EACd,SAAS,CAAC,kBAAkB,CAC7B,CAAC;YACF,OAAO,IAAI,CAAC,eAAe,CACzB,QAAQ,CAAC,MAAM,EACf,MAAM,CAAC,IAAI,EACX,aAAa,EACb,SAAS,CACV,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAChC,CAAC;YAAC,OAAO,YAAY,EAAE,CAAC;gBACtB,MAAM,IAAI,cAAc,CACtB,CAAC,KAAK,EAAE,YAAY,CAAC,EACrB,6CAA6C,CAC9C,CAAC;YACJ,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF"}

package/dist/src/storage/session-media-service-factories.d.ts

@@ -0,0 +1,19 @@
+/**
+ * @license
+ * Copyright 2026 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Storage } from '@vybestack/llxprt-code-settings';
+import { LocalMediaStore } from './local-media-store.js';
+import { SessionPersistenceService } from './SessionPersistenceService.js';
+interface SessionMediaServiceHost {
+    readonly storage: Storage;
+    getMediaStoreQuotaByteLimit(): number;
+    getSessionPersistenceQueueByteLimit(): number;
+}
+interface SessionMediaServices {
+    readonly store: () => LocalMediaStore;
+    readonly persistence: (sessionId: string) => SessionPersistenceService;
+}
+export declare function sessionMediaServices(host: SessionMediaServiceHost): SessionMediaServices;
+export {};

package/dist/src/storage/session-media-service-factories.js

@@ -0,0 +1,34 @@
+/**
+ * @license
+ * Copyright 2026 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { join } from 'node:path';
+import { LocalMediaStore } from './local-media-store.js';
+import { SessionPersistenceService } from './SessionPersistenceService.js';
+export function sessionMediaServices(host) {
+    let mediaStore;
+    const persistenceServices = new Map();
+    const store = () => {
+        mediaStore ??= new LocalMediaStore({
+            rootDirectory: join(host.storage.getProjectTempDir(), 'media'),
+            quotaBytes: host.getMediaStoreQuotaByteLimit(),
+        });
+        return mediaStore;
+    };
+    return {
+        store,
+        persistence: (sessionId) => {
+            const existing = persistenceServices.get(sessionId);
+            if (existing !== undefined)
+                return existing;
+            const service = new SessionPersistenceService(host.storage, sessionId, {
+                mediaStore: store(),
+                maxQueueBytes: host.getSessionPersistenceQueueByteLimit(),
+            });
+            persistenceServices.set(sessionId, service);
+            return service;
+        },
+    };
+}
+//# sourceMappingURL=session-media-service-factories.js.map

package/dist/src/storage/session-media-service-factories.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-media-service-factories.js","sourceRoot":"","sources":["../../../src/storage/session-media-service-factories.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAa3E,MAAM,UAAU,oBAAoB,CAClC,IAA6B;IAE7B,IAAI,UAAuC,CAAC;IAC5C,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAqC,CAAC;IACzE,MAAM,KAAK,GAAG,GAAoB,EAAE;QAClC,UAAU,KAAK,IAAI,eAAe,CAAC;YACjC,aAAa,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,OAAO,CAAC;YAC9D,UAAU,EAAE,IAAI,CAAC,2BAA2B,EAAE;SAC/C,CAAC,CAAC;QACH,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;IACF,OAAO;QACL,KAAK;QACL,WAAW,EAAE,CAAC,SAAS,EAAE,EAAE;YACzB,MAAM,QAAQ,GAAG,mBAAmB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACpD,IAAI,QAAQ,KAAK,SAAS;gBAAE,OAAO,QAAQ,CAAC;YAC5C,MAAM,OAAO,GAAG,IAAI,yBAAyB,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE;gBACrE,UAAU,EAAE,KAAK,EAAE;gBACnB,aAAa,EAAE,IAAI,CAAC,mCAAmC,EAAE;aAC1D,CAAC,CAAC;YACH,mBAAmB,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YAC5C,OAAO,OAAO,CAAC;QACjB,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/src/storage/session-persistence-helpers.d.ts

@@ -0,0 +1,10 @@
+/**
+ * @license
+ * Copyright 2026 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { IContent } from '../services/history/IContent.js';
+import type { PersistedSession, PersistedUIHistoryItem } from './SessionPersistenceService.js';
+export declare function errorCode(error: unknown): unknown;
+export declare function containsMediaDiagnostic(error: unknown, seen?: Set<object>, depth?: number): boolean;
+export declare function persistenceRequestLowerBound(history: readonly IContent[], metadata: PersistedSession['metadata'] | undefined, uiHistory: readonly PersistedUIHistoryItem[] | undefined): number;

package/dist/src/storage/session-persistence-helpers.js

@@ -0,0 +1,46 @@
+/**
+ * @license
+ * Copyright 2026 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { MediaAdmissionError } from './media-admission-service.js';
+import { MediaReferenceValidationError } from './media-reference-lifecycle.js';
+const MAX_MEDIA_DIAGNOSTIC_DEPTH = 32;
+function isUnknownArray(value) {
+    return Array.isArray(value);
+}
+export function errorCode(error) {
+    return typeof error === 'object' && error !== null
+        ? Reflect.get(error, 'code')
+        : undefined;
+}
+export function containsMediaDiagnostic(error, seen = new Set(), depth = 0) {
+    if (error instanceof MediaReferenceValidationError ||
+        error instanceof MediaAdmissionError) {
+        return true;
+    }
+    if (depth >= MAX_MEDIA_DIAGNOSTIC_DEPTH ||
+        typeof error !== 'object' ||
+        error === null ||
+        seen.has(error)) {
+        return false;
+    }
+    seen.add(error);
+    const nested = Reflect.get(error, 'errors');
+    return ((isUnknownArray(nested) &&
+        nested.some((entry) => containsMediaDiagnostic(entry, seen, depth + 1))) ||
+        containsMediaDiagnostic(Reflect.get(error, 'cause'), seen, depth + 1));
+}
+function serializedStringContentLowerBound(value, visited) {
+    if (typeof value === 'string')
+        return Buffer.byteLength(value, 'utf8');
+    if (typeof value !== 'object' || value === null || visited.has(value)) {
+        return 0;
+    }
+    visited.add(value);
+    return Object.values(value).reduce((bytes, nested) => Math.min(Number.MAX_SAFE_INTEGER, bytes + serializedStringContentLowerBound(nested, visited)), 0);
+}
+export function persistenceRequestLowerBound(history, metadata, uiHistory) {
+    return serializedStringContentLowerBound([history, metadata, uiHistory], new Set());
+}
+//# sourceMappingURL=session-persistence-helpers.js.map

package/dist/src/storage/session-persistence-helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-persistence-helpers.js","sourceRoot":"","sources":["../../../src/storage/session-persistence-helpers.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,mBAAmB,EAAE,MAAM,8BAA8B,CAAC;AACnE,OAAO,EAAE,6BAA6B,EAAE,MAAM,gCAAgC,CAAC;AAM/E,MAAM,0BAA0B,GAAG,EAAE,CAAC;AAEtC,SAAS,cAAc,CAAC,KAAc;IACpC,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,KAAc;IACtC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI;QAChD,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC;QAC5B,CAAC,CAAC,SAAS,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,KAAc,EACd,OAAoB,IAAI,GAAG,EAAU,EACrC,KAAK,GAAG,CAAC;IAET,IACE,KAAK,YAAY,6BAA6B;QAC9C,KAAK,YAAY,mBAAmB,EACpC,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IACE,KAAK,IAAI,0BAA0B;QACnC,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,KAAK,IAAI;QACd,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EACf,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAChB,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IAC5C,OAAO,CACL,CAAC,cAAc,CAAC,MAAM,CAAC;QACrB,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CACpB,uBAAuB,CAAC,KAAK,EAAE,IAAI,EAAE,KAAK,GAAG,CAAC,CAAC,CAChD,CAAC;QACJ,uBAAuB,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,CAAC,EAAE,IAAI,EAAE,KAAK,GAAG,CAAC,CAAC,CACtE,CAAC;AACJ,CAAC;AAED,SAAS,iCAAiC,CACxC,KAAc,EACd,OAAoB;IAEpB,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACvE,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;QACtE,OAAO,CAAC,CAAC;IACX,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACnB,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,CAChC,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,CAChB,IAAI,CAAC,GAAG,CACN,MAAM,CAAC,gBAAgB,EACvB,KAAK,GAAG,iCAAiC,CAAC,MAAM,EAAE,OAAO,CAAC,CAC3D,EACH,CAAC,CACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,4BAA4B,CAC1C,OAA4B,EAC5B,QAAkD,EAClD,SAAwD;IAExD,OAAO,iCAAiC,CACtC,CAAC,OAAO,EAAE,QAAQ,EAAE,SAAS,CAAC,EAC9B,IAAI,GAAG,EAAU,CAClB,CAAC;AACJ,CAAC"}

package/dist/src/test-utils/config.d.ts

@@ -4,7 +4,9 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import { Config } from '../config/config.js';
+import type { AgentClientContract } from '../core/clientContract.js';
 import type { MessageBus as MessageBusType } from '../confirmation-bus/message-bus.js';
+export declare function createTestAgentClient(overrides?: Partial<AgentClientContract>): AgentClientContract;
 /**
  * Test-only helper that returns a session-scoped MessageBus for the given Config.
  *

package/dist/src/test-utils/config.js

@@ -29,7 +29,7 @@ function createTestAgentChat() {
         sendMessageStream: async () => emptyChatStream(),
         generateDirectMessage: async () => emptyModelOutput(),
         getHistory: () => [],
-        setHistory: () => { },
+        setHistory: async () => { },
         clearHistory: () => { },
         getHistoryService: () => null,
         wasRecentlyCompressed: () => false,
@@ -37,7 +37,7 @@ function createTestAgentChat() {
         recordCompletedToolCalls: () => { },
     };
 }
-function createTestAgentClient() {
+export function createTestAgentClient(overrides) {
     const chat = createTestAgentChat();
     return {
         initialize: async () => { },
@@ -47,8 +47,8 @@ function createTestAgentClient() {
         getHistory: async () => [],
         getHistoryService: () => null,
         storeHistoryServiceForReuse: () => { },
-        storeHistoryForLaterUse: () => { },
-        dispose: () => { },
+        storeHistoryForLaterUse: async () => { },
+        dispose: async () => { },
         setTools: async () => { },
         clearTools: () => { },
         updateSystemInstruction: async () => { },
@@ -65,8 +65,8 @@ function createTestAgentClient() {
         generateContent: async () => emptyModelOutput(),
         generateEmbedding: async (texts) => texts.map(() => []),
         sendMessageStream: () => emptyServerAgentStream(),
-        getUserTier: () => undefined,
         getCurrentSequenceModel: () => null,
+        ...overrides,
     };
 }
 const createTestToolScheduler = () => ({

package/dist/src/test-utils/config.js.map

@@ -1 +1 @@
-{"version":3,"file":"config.js","sourceRoot":"","sources":["../../../src/test-utils/config.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,MAAM,EAAyB,MAAM,qBAAqB,CAAC;AAOpE,OAAO,EACL,wBAAwB,GAEzB,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,UAAU,EAAE,MAAM,oCAAoC,CAAC;AAEhE,OAAO,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AAEjE;;;GAGG;AACH,MAAM,YAAY,GAAG,IAAI,OAAO,EAA0B,CAAC;AAE3D,KAAK,SAAS,CAAC,CAAC,cAAc,CAAI,KAAU;IAC1C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,IAAI,CAAC;IACb,CAAC;AACH,CAAC;AAED,SAAS,sBAAsB;IAI7B,OAAO,cAAc,CAAyB,EAAE,CAAC,CAAC;AACpD,CAAC;AAED,SAAS,eAAe;IACtB,OAAO,cAAc,CAAQ,EAAE,CAAC,CAAC;AACnC,CAAC;AAED,SAAS,mBAAmB;IAC1B,OAAO;QACL,WAAW,EAAE,KAAK,IAAI,EAAE,CAAC,gBAAgB,EAAE;QAC3C,iBAAiB,EAAE,KAAK,IAAI,EAAE,CAAC,eAAe,EAAE;QAChD,qBAAqB,EAAE,KAAK,IAAI,EAAE,CAAC,gBAAgB,EAAE;QACrD,UAAU,EAAE,GAAG,EAAE,CAAC,EAAE;QACpB,UAAU,EAAE,GAAG,EAAE,GAAE,CAAC;QACpB,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;QACtB,iBAAiB,EAAE,GAAG,EAAE,CAAC,IAAI;QAC7B,qBAAqB,EAAE,GAAG,EAAE,CAAC,KAAK;QAClC,kBAAkB,EAAE,KAAK,IAAI,EAAE,CAAC,wBAAwB,CAAC,aAAa;QACtE,wBAAwB,EAAE,GAAG,EAAE,GAAE,CAAC;KACnC,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB;IAC5B,MAAM,IAAI,GAAG,mBAAmB,EAAE,CAAC;IACnC,OAAO;QACL,UAAU,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QAC1B,aAAa,EAAE,GAAG,EAAE,CAAC,IAAI;QACzB,kBAAkB,EAAE,GAAG,EAAE,CAAC,IAAI;QAC9B,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI;QACnB,UAAU,EAAE,KAAK,IAAI,EAAE,CAAC,EAAE;QAC1B,iBAAiB,EAAE,GAAG,EAAE,CAAC,IAAI;QAC7B,2BAA2B,EAAE,GAAG,EAAE,GAAE,CAAC;QACrC,uBAAuB,EAAE,GAAG,EAAE,GAAE,CAAC;QACjC,OAAO,EAAE,GAAG,EAAE,GAAE,CAAC;QACjB,QAAQ,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QACxB,UAAU,EAAE,GAAG,EAAE,GAAE,CAAC;QACpB,uBAAuB,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QACvC,UAAU,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QAC1B,SAAS,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QACzB,UAAU,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QAC1B,UAAU,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QAC1B,cAAc,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QAC9B,mBAAmB,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QACnC,mBAAmB,EAAE,GAAG,EAAE,CAAC,SAAkB;QAC7C,SAAS,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI;QAC3B,qBAAqB,EAAE,KAAK,IAAI,EAAE,CAAC,gBAAgB,EAAE;QACrD,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;QAC9B,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,gBAAgB,EAAE;QAC/C,iBAAiB,EAAE,KAAK,EAAE,KAAe,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QACjE,iBAAiB,EAAE,GAAG,EAAE,CAAC,sBAAsB,EAAE;QACjD,WAAW,EAAE,GAAG,EAAE,CAAC,SAAS;QAC5B,uBAAuB,EAAE,GAAG,EAAE,CAAC,IAAI;KACpC,CAAC;AACJ,CAAC;AAED,MAAM,uBAAuB,GAAyB,GAAG,EAAE,CAAC,CAAC;IAC3D,QAAQ,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;IACxB,SAAS,EAAE,GAAG,EAAE,GAAE,CAAC;IACnB,OAAO,EAAE,GAAG,EAAE,GAAE,CAAC;IACjB,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;IACtB,0BAA0B,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;CAC3C,CAAC,CAAC;AAEH;;;;;GAKG;AACH,MAAM,UAAU,wBAAwB,CAAC,MAAc;IACrD,IAAI,GAAG,GAAG,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACnC,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,GAAG,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,eAAe,EAAE,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;QACtE,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAChC,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,MAAc;IACrD,MAAM,MAAM,GAAG,MAGd,CAAC;IACF,MAAM,WAAW,GAA0B,EAAE,CAAC;IAC9C,IAAI,MAAM,CAAC,kBAAkB,KAAK,SAAS,EAAE,CAAC;QAC5C,WAAW,CAAC,kBAAkB,GAAG;YAC/B,KAAK,EAAE,GAAG,EAAE,CAAC,qBAAqB,EAAE;YACpC,YAAY,EAAE,IAAI;SACnB,CAAC;IACJ,CAAC;IACD,IAAI,MAAM,CAAC,oBAAoB,KAAK,SAAS,EAAE,CAAC;QAC9C,WAAW,CAAC,oBAAoB,GAAG;YACjC,KAAK,EAAE,uBAAuB;YAC9B,YAAY,EAAE,IAAI;SACnB,CAAC;IACJ,CAAC;IACD,IAAI,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxC,MAAM,CAAC,gBAAgB,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IAC/C,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,MAAc;IACvD,wBAAwB,CAAC,MAAM,CAAC,CAAC;IACjC,MAAM,iBAAiB,GAAG,wBAAwB,CAAC,MAAM,CAAC,CAAC;IAC3D,MACE,MAGD,CAAC,UAAU,CAAC,EAAE,UAAU,EAAE,iBAAiB,EAAE,CAAC,CAAC;AAClD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,OAE9B;IACC,+CAA+C;IAC/C,MAAM,MAAM,GAAqB;QAC/B,SAAS,EAAE,cAAc;QACzB,SAAS,EAAE,WAAW;QACtB,SAAS,EAAE,KAAK;QAChB,GAAG,EAAE,WAAW;QAChB,KAAK,EAAE,sBAAsB;KAC9B,CAAC;IAEF,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,MAAM,CAAC,CAAC;IAClC,wBAAwB,CAAC,MAAM,CAAC,CAAC;IAEjC,2CAA2C;IAC3C,MAAM,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC;IAExC,qCAAqC;IACrC,IAAI,OAAO,EAAE,iBAAiB,EAAE,CAAC;QAC/B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACrE,MAAM,CAAC,mBAAmB,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED,oDAAoD;IACpD,gGAAgG;IAChG,MAAM,0BAA0B,GAA2B;QACzD,KAAK,EAAE,sBAAsB;QAC7B,MAAM,EAAE,cAAc;KACvB,CAAC;IAEF,gEAAgE;IAChE,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,wBAAwB,EAAE;QACtD,KAAK,EAAE,0BAA0B;QACjC,QAAQ,EAAE,IAAI;QACd,YAAY,EAAE,IAAI;KACnB,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC"}
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../../../src/test-utils/config.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,MAAM,EAAyB,MAAM,qBAAqB,CAAC;AAOpE,OAAO,EACL,wBAAwB,GAEzB,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,UAAU,EAAE,MAAM,oCAAoC,CAAC;AAEhE,OAAO,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AAEjE;;;GAGG;AACH,MAAM,YAAY,GAAG,IAAI,OAAO,EAA0B,CAAC;AAE3D,KAAK,SAAS,CAAC,CAAC,cAAc,CAAI,KAAU;IAC1C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,IAAI,CAAC;IACb,CAAC;AACH,CAAC;AAED,SAAS,sBAAsB;IAI7B,OAAO,cAAc,CAAyB,EAAE,CAAC,CAAC;AACpD,CAAC;AAED,SAAS,eAAe;IACtB,OAAO,cAAc,CAAQ,EAAE,CAAC,CAAC;AACnC,CAAC;AAED,SAAS,mBAAmB;IAC1B,OAAO;QACL,WAAW,EAAE,KAAK,IAAI,EAAE,CAAC,gBAAgB,EAAE;QAC3C,iBAAiB,EAAE,KAAK,IAAI,EAAE,CAAC,eAAe,EAAE;QAChD,qBAAqB,EAAE,KAAK,IAAI,EAAE,CAAC,gBAAgB,EAAE;QACrD,UAAU,EAAE,GAAG,EAAE,CAAC,EAAE;QACpB,UAAU,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QAC1B,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;QACtB,iBAAiB,EAAE,GAAG,EAAE,CAAC,IAAI;QAC7B,qBAAqB,EAAE,GAAG,EAAE,CAAC,KAAK;QAClC,kBAAkB,EAAE,KAAK,IAAI,EAAE,CAAC,wBAAwB,CAAC,aAAa;QACtE,wBAAwB,EAAE,GAAG,EAAE,GAAE,CAAC;KACnC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,SAAwC;IAExC,MAAM,IAAI,GAAG,mBAAmB,EAAE,CAAC;IACnC,OAAO;QACL,UAAU,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QAC1B,aAAa,EAAE,GAAG,EAAE,CAAC,IAAI;QACzB,kBAAkB,EAAE,GAAG,EAAE,CAAC,IAAI;QAC9B,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI;QACnB,UAAU,EAAE,KAAK,IAAI,EAAE,CAAC,EAAE;QAC1B,iBAAiB,EAAE,GAAG,EAAE,CAAC,IAAI;QAC7B,2BAA2B,EAAE,GAAG,EAAE,GAAE,CAAC;QACrC,uBAAuB,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QACvC,OAAO,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QACvB,QAAQ,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QACxB,UAAU,EAAE,GAAG,EAAE,GAAE,CAAC;QACpB,uBAAuB,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QACvC,UAAU,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QAC1B,SAAS,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QACzB,UAAU,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QAC1B,UAAU,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QAC1B,cAAc,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QAC9B,mBAAmB,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QACnC,mBAAmB,EAAE,GAAG,EAAE,CAAC,SAAkB;QAC7C,SAAS,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI;QAC3B,qBAAqB,EAAE,KAAK,IAAI,EAAE,CAAC,gBAAgB,EAAE;QACrD,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;QAC9B,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,gBAAgB,EAAE;QAC/C,iBAAiB,EAAE,KAAK,EAAE,KAAe,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QACjE,iBAAiB,EAAE,GAAG,EAAE,CAAC,sBAAsB,EAAE;QACjD,uBAAuB,EAAE,GAAG,EAAE,CAAC,IAAI;QACnC,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,MAAM,uBAAuB,GAAyB,GAAG,EAAE,CAAC,CAAC;IAC3D,QAAQ,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;IACxB,SAAS,EAAE,GAAG,EAAE,GAAE,CAAC;IACnB,OAAO,EAAE,GAAG,EAAE,GAAE,CAAC;IACjB,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;IACtB,0BAA0B,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;CAC3C,CAAC,CAAC;AAEH;;;;;GAKG;AACH,MAAM,UAAU,wBAAwB,CAAC,MAAc;IACrD,IAAI,GAAG,GAAG,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACnC,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,GAAG,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,eAAe,EAAE,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;QACtE,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAChC,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,MAAc;IACrD,MAAM,MAAM,GAAG,MAGd,CAAC;IACF,MAAM,WAAW,GAA0B,EAAE,CAAC;IAC9C,IAAI,MAAM,CAAC,kBAAkB,KAAK,SAAS,EAAE,CAAC;QAC5C,WAAW,CAAC,kBAAkB,GAAG;YAC/B,KAAK,EAAE,GAAG,EAAE,CAAC,qBAAqB,EAAE;YACpC,YAAY,EAAE,IAAI;SACnB,CAAC;IACJ,CAAC;IACD,IAAI,MAAM,CAAC,oBAAoB,KAAK,SAAS,EAAE,CAAC;QAC9C,WAAW,CAAC,oBAAoB,GAAG;YACjC,KAAK,EAAE,uBAAuB;YAC9B,YAAY,EAAE,IAAI;SACnB,CAAC;IACJ,CAAC;IACD,IAAI,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxC,MAAM,CAAC,gBAAgB,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IAC/C,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,MAAc;IACvD,wBAAwB,CAAC,MAAM,CAAC,CAAC;IACjC,MAAM,iBAAiB,GAAG,wBAAwB,CAAC,MAAM,CAAC,CAAC;IAC3D,MACE,MAGD,CAAC,UAAU,CAAC,EAAE,UAAU,EAAE,iBAAiB,EAAE,CAAC,CAAC;AAClD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,OAE9B;IACC,+CAA+C;IAC/C,MAAM,MAAM,GAAqB;QAC/B,SAAS,EAAE,cAAc;QACzB,SAAS,EAAE,WAAW;QACtB,SAAS,EAAE,KAAK;QAChB,GAAG,EAAE,WAAW;QAChB,KAAK,EAAE,sBAAsB;KAC9B,CAAC;IAEF,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,MAAM,CAAC,CAAC;IAClC,wBAAwB,CAAC,MAAM,CAAC,CAAC;IAEjC,2CAA2C;IAC3C,MAAM,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC;IAExC,qCAAqC;IACrC,IAAI,OAAO,EAAE,iBAAiB,EAAE,CAAC;QAC/B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACrE,MAAM,CAAC,mBAAmB,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED,oDAAoD;IACpD,gGAAgG;IAChG,MAAM,0BAA0B,GAA2B;QACzD,KAAK,EAAE,sBAAsB;QAC7B,MAAM,EAAE,cAAc;KACvB,CAAC;IAEF,gEAAgE;IAChE,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,wBAAwB,EAAE;QACtD,KAAK,EAAE,0BAA0B;QACjC,QAAQ,EAAE,IAAI;QACd,YAAY,EAAE,IAAI;KACnB,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/src/test-utils/runtime.d.ts

@@ -7,6 +7,7 @@ import type { Config } from '../config/config.js';
 import type { RuntimeProvider as IProvider } from '../runtime/contracts/RuntimeProvider.js';
 import type { RuntimeProviderManager } from '../runtime/contracts/RuntimeProviderManager.js';
 import type { RuntimeTokenizerFactory } from '../runtime/contracts/RuntimeTokenizerFactory.js';
+import type { SessionRecordingService } from '../recording/SessionRecordingService.js';
 import type { vi as ViNamespace } from 'bun:test';
 import { type ProviderRuntimeContext } from '../runtime/providerRuntimeContext.js';
 import { SettingsService } from '@vybestack/llxprt-code-settings';
@@ -70,6 +71,7 @@ interface ChatSessionConfigShape {
     getProviderManager: ReturnType<ReturnType<typeof requireVi>['fn']>;
     getSettingsService: ReturnType<ReturnType<typeof requireVi>['fn']>;
     getTokenizerFactory: () => RuntimeTokenizerFactory | undefined;
+    getSessionRecordingService: () => SessionRecordingService | undefined;
 }
 interface ChatSessionRuntimeOptions {
     provider?: IProvider;

package/dist/src/test-utils/runtime.js

@@ -162,8 +162,6 @@ function createDefaultProvider() {
         generateChatCompletion: vi.fn(async function* () {
             yield { speaker: 'ai', blocks: [] };
         }),
-        getServerTools: () => [],
-        invokeServerTool: vi.fn(),
     };
 }
 /**
@@ -227,6 +225,7 @@ export function createChatSessionRuntime(options = {}) {
         getProviderManager: vi.fn().mockReturnValue(providerManager),
         getSettingsService: vi.fn().mockReturnValue(settingsService),
         getTokenizerFactory: () => tokenizerFactory,
+        getSessionRecordingService: () => undefined,
     };
     const config = {
         ...baseConfig,