@agtlantis/core 0.4.1 → 0.5.1

package/dist/session/simple-session.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"simple-session.d.ts","sourceRoot":"","sources":["../../src/session/simple-session.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EACV,aAAa,EAEb,OAAO,EACR,MAAM,IAAI,CAAC;AAEZ,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AAEzD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAEzE,OAAO,EACL,cAAc,EACd,KAAK,aAAa,EAClB,KAAK,kBAAkB,EACvB,KAAK,uBAAuB,EAC5B,KAAK,iBAAiB,EACtB,KAAK,gBAAgB,EACrB,KAAK,qBAAqB,EAC1B,KAAK,aAAa,EAClB,KAAK,WAAW,EAChB,KAAK,eAAe,EACpB,KAAK,UAAU,EACf,KAAK,cAAc,EACpB,MAAM,YAAY,CAAC;AAGpB;;;GAGG;AACH,KAAK,eAAe,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;AAE/D,MAAM,WAAW,oBAAoB;IACnC,oBAAoB,CAAC,EAAE,aAAa,GAAG,IAAI,CAAC;IAC5C,YAAY,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,aAAa,CAAC;IAClD,YAAY,EAAE,YAAY,CAAC;IAC3B,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,WAAW,EAAE,WAAW,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;;OAGG;IACH,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB;;;OAGG;IACH,sBAAsB,CAAC,EAAE,eAAe,CAAC;IACzC;;;OAGG;IACH,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB;;;OAGG;IACH,wBAAwB,CAAC,EAAE,iBAAiB,CAAC;CAC9C;AAED,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAuB;IAC5D,OAAO,CAAC,QAAQ,CAAC,YAAY,CAA8C;IAC3E,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAe;IAC5C,OAAO,CAAC,QAAQ,CAAC,eAAe,CAA8B;IAC9D,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAA8B;IACrE,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAsB;IACnD,OAAO,CAAC,QAAQ,CAAC,wBAAwB,CAAgC;IACzE,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAc;IAC3C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAChC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;IAC1C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAc;IAEtC,OAAO,CAAC,OAAO,CAAkB;IACjC,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAiD;IAEtF,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAyC;IAEnE,YAAY,OAAO,EAAE,oBAAoB,EAaxC;IAED,OAAO,CAAC,QAAQ;IAoBhB,OAAO,CAAC,cAAc;IAWhB,YAAY,CAChB,KAAK,SAAS,OAAO,GAAG,EAAE,EAC1B,MAAM,SAAS,UAAU,GAAG,aAAa,EAEzC,MAAM,EAAE,kBAAkB,CAAC,KAAK,EAAE,MAAM,CAAC,GACxC,OAAO,CAAC,uBAAuB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CA2EjD;IAED,UAAU,CACR,KAAK,SAAS,OAAO,GAAG,EAAE,EAC1B,MAAM,SAAS,UAAU,GAAG,aAAa,EACzC,MAAM,EAAE,gBAAgB,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,qBAAqB,CAAC,KAAK,EAAE,MAAM,CAAC,CAgE/E;IAED,IAAI,WAAW,IAAI,WAAW,CAE7B;IAED,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAE1C;IAED,cAAc,CAAC,eAAe,EAAE,eAAe,GAAG,IAAI,CAErD;IAED,aAAa,CAAC,MAAM,EAAE,IAAI,CAAC,aAAa,EAAE,MAAM,CAAC,GAAG;QAAE,IAAI,CAAC,EAAE,WAAW,CAAA;KAAE,GAAG,IAAI,CAMhF;IAED,oBAAoB,CAAC,IAAI,EAAE,IAAI,CAAC,cAAc,EAAE,WAAW,CAAC,GAAG,IAAI,CAKlE;IAED,WAAW,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,GAAG,IAAI,CAAC;IAC/C,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IAWjD,OAAO,CAAC,wBAAwB;IAWhC,MAAM,CAAC,EAAE,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,GAAG,IAAI,CAE3C;IAEK,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC,CAUpC;IAEK,UAAU,IAAI,OAAO,CAAC,cAAc,CAAC,CAG1C;IAED;;;OAGG;IACH,oBAAoB,IAAI,IAAI,CAK3B;IAED;;;;;OAKG;IACG,mBAAmB,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAStE;IAED;;;;;;OAMG;IACG,oBAAoB,CAAC,KAAK,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAczE;IAED,SAAS,KAAK,OAAO,IAAI,MAAM,CAE9B;IAED,SAAS,KAAK,UAAU,IAAI,MAAM,CAEjC;IAED,SAAS,KAAK,QAAQ,IAAI,MAAM,CAK/B;CACF"}

package/dist/session/simple-session.js

@@ -0,0 +1,299 @@
+import { generateText as aiGenerateText, streamText as aiStreamText, } from 'ai';
+import { deepMerge } from '../utils/deep-merge.js';
+import { noopLogger } from '../observability/logger.js';
+import { calculateTotalCost } from '../pricing/calculator.js';
+import { SessionSummary, } from './types.js';
+import { mergeUsages, createZeroUsage } from './usage-extractors.js';
+export class SimpleSession {
+    defaultLanguageModel;
+    modelFactory;
+    providerType;
+    providerPricing;
+    defaultProviderOptions;
+    defaultTools;
+    defaultGenerationOptions;
+    _fileManager;
+    logger;
+    sessionStartTime;
+    signal;
+    summary;
+    pendingUsagePromises = [];
+    onDoneFns = [];
+    constructor(options) {
+        this.defaultLanguageModel = options.defaultLanguageModel ?? null;
+        this.modelFactory = options.modelFactory ?? null;
+        this.providerType = options.providerType;
+        this.providerPricing = options.providerPricing;
+        this.defaultProviderOptions = options.defaultProviderOptions;
+        this.defaultTools = options.defaultTools;
+        this.defaultGenerationOptions = options.defaultGenerationOptions;
+        this._fileManager = options.fileManager;
+        this.logger = options.logger ?? noopLogger;
+        this.sessionStartTime = options.startTime ?? Date.now();
+        this.signal = options.signal;
+        this.summary = SessionSummary.empty(this.sessionStartTime);
+    }
+    getModel(requestedModelId) {
+        if (requestedModelId) {
+            if (!this.modelFactory) {
+                throw new Error(`Model '${requestedModelId}' requested but no modelFactory provided. ` +
+                    `Either use the default model or configure the provider with modelFactory.`);
+            }
+            return this.modelFactory(requestedModelId);
+        }
+        if (!this.defaultLanguageModel) {
+            throw new Error('No model specified and no default model set. ' +
+                'Either specify a model in the call or configure the provider with withDefaultModel().');
+        }
+        return this.defaultLanguageModel;
+    }
+    extractModelId(model) {
+        const modelWithId = model;
+        if (!modelWithId.modelId) {
+            console.warn('[SimpleSession] Model does not have modelId property, using "unknown". ' +
+                'This may affect cost tracking accuracy.');
+        }
+        return modelWithId.modelId ?? 'unknown';
+    }
+    async generateText(params) {
+        const callStartTime = Date.now();
+        const { model: requestedModel, providerOptions, tools, ...restParams } = params;
+        const languageModel = this.getModel(requestedModel);
+        const modelId = this.extractModelId(languageModel);
+        const mergedProviderOptions = (this.defaultProviderOptions || providerOptions)
+            ? deepMerge(this.defaultProviderOptions ?? {}, providerOptions ?? {})
+            : undefined;
+        const mergedTools = (this.defaultTools || tools)
+            ? { ...this.defaultTools, ...tools }
+            : undefined;
+        this.logger.onLLMCallStart?.({
+            type: 'llm_call_start',
+            callType: 'generateText',
+            modelId,
+            timestamp: callStartTime,
+            request: { params: restParams },
+        });
+        try {
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const result = await aiGenerateText({
+                ...this.defaultGenerationOptions,
+                ...restParams,
+                tools: mergedTools,
+                providerOptions: mergedProviderOptions,
+                model: languageModel,
+                abortSignal: this.signal,
+            });
+            const callEndTime = Date.now();
+            const call = {
+                startTime: callStartTime,
+                endTime: callEndTime,
+                duration: callEndTime - callStartTime,
+                usage: result.usage ?? createZeroUsage(),
+                type: 'generateText',
+                model: modelId,
+                provider: this.providerType,
+            };
+            this.updateSummaryWithLLMCall(call);
+            this.logger.onLLMCallEnd?.({
+                type: 'llm_call_end',
+                callType: 'generateText',
+                modelId,
+                timestamp: callEndTime,
+                response: {
+                    duration: callEndTime - callStartTime,
+                    usage: result.usage,
+                    raw: result,
+                },
+            });
+            return result;
+        }
+        catch (error) {
+            const callEndTime = Date.now();
+            this.logger.onLLMCallEnd?.({
+                type: 'llm_call_end',
+                callType: 'generateText',
+                modelId,
+                timestamp: callEndTime,
+                response: {
+                    duration: callEndTime - callStartTime,
+                    raw: null,
+                    error: error instanceof Error ? error : new Error(String(error)),
+                },
+            });
+            throw error;
+        }
+    }
+    streamText(params) {
+        const callStartTime = Date.now();
+        const { model: requestedModel, providerOptions, tools, ...restParams } = params;
+        const languageModel = this.getModel(requestedModel);
+        const modelId = this.extractModelId(languageModel);
+        const mergedProviderOptions = (this.defaultProviderOptions || providerOptions)
+            ? deepMerge(this.defaultProviderOptions ?? {}, providerOptions ?? {})
+            : undefined;
+        const mergedTools = (this.defaultTools || tools)
+            ? { ...this.defaultTools, ...tools }
+            : undefined;
+        this.logger.onLLMCallStart?.({
+            type: 'llm_call_start',
+            callType: 'streamText',
+            modelId,
+            timestamp: callStartTime,
+            request: { params: restParams },
+        });
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const result = aiStreamText({
+            ...this.defaultGenerationOptions,
+            ...restParams,
+            tools: mergedTools,
+            providerOptions: mergedProviderOptions,
+            model: languageModel,
+            abortSignal: this.signal,
+        });
+        const usagePromise = Promise.resolve(result.usage).then((usage) => {
+            const callEndTime = Date.now();
+            const call = {
+                startTime: callStartTime,
+                endTime: callEndTime,
+                duration: callEndTime - callStartTime,
+                usage: usage ?? createZeroUsage(),
+                type: 'streamText',
+                model: modelId,
+                provider: this.providerType,
+            };
+            this.updateSummaryWithLLMCall(call);
+            this.logger.onLLMCallEnd?.({
+                type: 'llm_call_end',
+                callType: 'streamText',
+                modelId,
+                timestamp: callEndTime,
+                response: {
+                    duration: callEndTime - callStartTime,
+                    usage,
+                    raw: result,
+                },
+            });
+            return usage;
+        });
+        this.pendingUsagePromises.push(usagePromise);
+        return result;
+    }
+    get fileManager() {
+        return this._fileManager;
+    }
+    record(data) {
+        this.summary = this.summary.withCustomRecord(data);
+    }
+    recordToolCall(toolCallSummary) {
+        this.summary = this.summary.withToolCall(toolCallSummary);
+    }
+    recordLLMCall(record) {
+        const call = {
+            ...record,
+            type: record.type ?? 'manual',
+        };
+        this.updateSummaryWithLLMCall(call);
+    }
+    recordAdditionalCost(cost) {
+        this.summary = this.summary.withAdditionalCost({
+            ...cost,
+            timestamp: Date.now(),
+        });
+    }
+    setMetadata(keyOrData, value) {
+        if (typeof keyOrData === 'string') {
+            this.summary = this.summary.withMetadata(keyOrData, value);
+        }
+        else {
+            for (const [k, v] of Object.entries(keyOrData)) {
+                this.summary = this.summary.withMetadata(k, v);
+            }
+        }
+    }
+    updateSummaryWithLLMCall(call) {
+        const newCalls = [...this.summary.llmCalls, call];
+        const { totalCost: llmCost, costByModel } = calculateTotalCost(newCalls.map((c) => ({ usage: c.usage, model: c.model, provider: c.provider })), this.providerPricing);
+        const newTotalUsage = mergeUsages(newCalls.map((c) => c.usage));
+        this.summary = this.summary.withLLMCall(call, llmCost, costByModel, newTotalUsage);
+    }
+    onDone(fn) {
+        this.onDoneFns.push(fn);
+    }
+    async runOnDoneHooks() {
+        const reversedHooks = [...this.onDoneFns].reverse();
+        for (const fn of reversedHooks) {
+            try {
+                await fn();
+            }
+            catch (error) {
+                console.error('[SimpleSession] onDone hook error:', error);
+            }
+        }
+    }
+    async getSummary() {
+        await Promise.all(this.pendingUsagePromises);
+        return this.summary;
+    }
+    /**
+     * Notifies Logger of execution start.
+     * @internal Called by SimpleExecutionHost - not intended for direct use.
+     */
+    notifyExecutionStart() {
+        this._logger.onExecutionStart?.({
+            type: 'execution_start',
+            timestamp: Date.now(),
+        });
+    }
+    /**
+     * Notifies Logger of execution completion with result data and summary.
+     * @param data - The execution result data
+     * @param startTime - Execution start timestamp for duration calculation
+     * @internal Called by SimpleExecutionHost - not intended for direct use.
+     */
+    async notifyExecutionDone(data, startTime) {
+        const summary = await this.getSummary();
+        this._logger.onExecutionDone?.({
+            type: 'execution_done',
+            timestamp: Date.now(),
+            duration: Date.now() - startTime,
+            data,
+            summary,
+        });
+    }
+    /**
+     * Notifies Logger of execution error with error details and summary (if available).
+     * Gracefully handles getSummary() failures - summary will be undefined if it fails.
+     * @param error - The error that occurred
+     * @param startTime - Execution start timestamp for duration calculation
+     * @internal Called by SimpleExecutionHost - not intended for direct use.
+     */
+    async notifyExecutionError(error, startTime) {
+        let summary;
+        try {
+            summary = await this.getSummary();
+        }
+        catch {
+            // Ignore summary errors on failure path
+        }
+        this._logger.onExecutionError?.({
+            type: 'execution_error',
+            timestamp: Date.now(),
+            duration: Date.now() - startTime,
+            error,
+            summary,
+        });
+    }
+    get _logger() {
+        return this.logger;
+    }
+    get _startTime() {
+        return this.sessionStartTime;
+    }
+    get _modelId() {
+        if (!this.defaultLanguageModel) {
+            return 'unknown';
+        }
+        return this.extractModelId(this.defaultLanguageModel);
+    }
+}
+//# sourceMappingURL=simple-session.js.map

package/dist/session/simple-session.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"simple-session.js","sourceRoot":"","sources":["../../src/session/simple-session.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,YAAY,IAAI,cAAc,EAC9B,UAAU,IAAI,YAAY,GAC3B,MAAM,IAAI,CAAC;AAMZ,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAEnD,OAAO,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAGxD,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EACL,cAAc,GAYf,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAsCrE,MAAM,OAAO,aAAa;IACP,oBAAoB,CAAuB;IAC3C,YAAY,CAA8C;IAC1D,YAAY,CAAe;IAC3B,eAAe,CAA8B;IAC7C,sBAAsB,CAA8B;IACpD,YAAY,CAAsB;IAClC,wBAAwB,CAAgC;IACxD,YAAY,CAAc;IAC1B,MAAM,CAAS;IACf,gBAAgB,CAAS;IACzB,MAAM,CAAe;IAE9B,OAAO,CAAkB;IAChB,oBAAoB,GAA8C,EAAE,CAAC;IAErE,SAAS,GAAsC,EAAE,CAAC;IAEnE,YAAY,OAA6B,EAAE;QACzC,IAAI,CAAC,oBAAoB,GAAG,OAAO,CAAC,oBAAoB,IAAI,IAAI,CAAC;QACjE,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,IAAI,CAAC;QACjD,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;QAC/C,IAAI,CAAC,sBAAsB,GAAG,OAAO,CAAC,sBAAsB,CAAC;QAC7D,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,wBAAwB,GAAG,OAAO,CAAC,wBAAwB,CAAC;QACjE,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;QACxC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,UAAU,CAAC;QAC3C,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;QACxD,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,OAAO,GAAG,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAAA,CAC5D;IAEO,QAAQ,CAAC,gBAAyB,EAAiB;QACzD,IAAI,gBAAgB,EAAE,CAAC;YACrB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;gBACvB,MAAM,IAAI,KAAK,CACb,UAAU,gBAAgB,4CAA4C;oBACtE,2EAA2E,CAC5E,CAAC;YACJ,CAAC;YACD,OAAO,IAAI,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC;QAC7C,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CACb,+CAA+C;gBAC/C,uFAAuF,CACxF,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC,oBAAoB,CAAC;IAAA,CAClC;IAEO,cAAc,CAAC,KAAoB,EAAU;QACnD,MAAM,WAAW,GAAG,KAAwC,CAAC;QAC7D,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO,CAAC,IAAI,CACV,yEAAyE;gBACvE,yCAAyC,CAC5C,CAAC;QACJ,CAAC;QACD,OAAO,WAAW,CAAC,OAAO,IAAI,SAAS,CAAC;IAAA,CACzC;IAED,KAAK,CAAC,YAAY,CAIhB,MAAyC,EACQ;QACjD,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACjC,MAAM,EAAE,KAAK,EAAE,cAAc,EAAE,eAAe,EAAE,KAAK,EAAE,GAAG,UAAU,EAAE,GAAG,MAAM,CAAC;QAChF,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;QACpD,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;QAEnD,MAAM,qBAAqB,GAAG,CAAC,IAAI,CAAC,sBAAsB,IAAI,eAAe,CAAC;YAC5E,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,sBAAsB,IAAI,EAAE,EAAE,eAAe,IAAI,EAAE,CAAoB;YACxF,CAAC,CAAC,SAAS,CAAC;QAEd,MAAM,WAAW,GAAG,CAAC,IAAI,CAAC,YAAY,IAAI,KAAK,CAAC;YAC9C,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,KAAK,EAAE;YACpC,CAAC,CAAC,SAAS,CAAC;QAEd,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YAC3B,IAAI,EAAE,gBAAgB;YACtB,QAAQ,EAAE,cAAc;YACxB,OAAO;YACP,SAAS,EAAE,aAAa;YACxB,OAAO,EAAE,EAAE,MAAM,EAAE,UAAqC,EAAE;SAC3D,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,8DAA8D;YAC9D,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC;gBAClC,GAAG,IAAI,CAAC,wBAAwB;gBAChC,GAAG,UAAU;gBACb,KAAK,EAAE,WAAW;gBAClB,eAAe,EAAE,qBAAqB;gBACtC,KAAK,EAAE,aAAa;gBACpB,WAAW,EAAE,IAAI,CAAC,MAAM;aAClB,CAAC,CAAC;YACV,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAE/B,MAAM,IAAI,GAAkB;gBAC1B,SAAS,EAAE,aAAa;gBACxB,OAAO,EAAE,WAAW;gBACpB,QAAQ,EAAE,WAAW,GAAG,aAAa;gBACrC,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,eAAe,EAAE;gBACxC,IAAI,EAAE,cAAc;gBACpB,KAAK,EAAE,OAAO;gBACd,QAAQ,EAAE,IAAI,CAAC,YAAY;aAC5B,CAAC;YACF,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC;YAEpC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;gBACzB,IAAI,EAAE,cAAc;gBACpB,QAAQ,EAAE,cAAc;gBACxB,OAAO;gBACP,SAAS,EAAE,WAAW;gBACtB,QAAQ,EAAE;oBACR,QAAQ,EAAE,WAAW,GAAG,aAAa;oBACrC,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,GAAG,EAAE,MAAM;iBACZ;aACF,CAAC,CAAC;YAEH,OAAO,MAA2D,CAAC;QACrE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAE/B,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;gBACzB,IAAI,EAAE,cAAc;gBACpB,QAAQ,EAAE,cAAc;gBACxB,OAAO;gBACP,SAAS,EAAE,WAAW;gBACtB,QAAQ,EAAE;oBACR,QAAQ,EAAE,WAAW,GAAG,aAAa;oBACrC,GAAG,EAAE,IAAI;oBACT,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;iBACjE;aACF,CAAC,CAAC;YAEH,MAAM,KAAK,CAAC;QACd,CAAC;IAAA,CACF;IAED,UAAU,CAGR,MAAuC,EAAwC;QAC/E,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACjC,MAAM,EAAE,KAAK,EAAE,cAAc,EAAE,eAAe,EAAE,KAAK,EAAE,GAAG,UAAU,EAAE,GAAG,MAAM,CAAC;QAChF,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;QACpD,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;QAEnD,MAAM,qBAAqB,GAAG,CAAC,IAAI,CAAC,sBAAsB,IAAI,eAAe,CAAC;YAC5E,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,sBAAsB,IAAI,EAAE,EAAE,eAAe,IAAI,EAAE,CAAoB;YACxF,CAAC,CAAC,SAAS,CAAC;QAEd,MAAM,WAAW,GAAG,CAAC,IAAI,CAAC,YAAY,IAAI,KAAK,CAAC;YAC9C,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,YAAY,EAAE,GAAG,KAAK,EAAE;YACpC,CAAC,CAAC,SAAS,CAAC;QAEd,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YAC3B,IAAI,EAAE,gBAAgB;YACtB,QAAQ,EAAE,YAAY;YACtB,OAAO;YACP,SAAS,EAAE,aAAa;YACxB,OAAO,EAAE,EAAE,MAAM,EAAE,UAAqC,EAAE;SAC3D,CAAC,CAAC;QAEH,8DAA8D;QAC9D,MAAM,MAAM,GAAG,YAAY,CAAC;YAC1B,GAAG,IAAI,CAAC,wBAAwB;YAChC,GAAG,UAAU;YACb,KAAK,EAAE,WAAW;YAClB,eAAe,EAAE,qBAAqB;YACtC,KAAK,EAAE,aAAa;YACpB,WAAW,EAAE,IAAI,CAAC,MAAM;SAClB,CAAC,CAAC;QAEV,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC;YACjE,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAE/B,MAAM,IAAI,GAAkB;gBAC1B,SAAS,EAAE,aAAa;gBACxB,OAAO,EAAE,WAAW;gBACpB,QAAQ,EAAE,WAAW,GAAG,aAAa;gBACrC,KAAK,EAAE,KAAK,IAAI,eAAe,EAAE;gBACjC,IAAI,EAAE,YAAY;gBAClB,KAAK,EAAE,OAAO;gBACd,QAAQ,EAAE,IAAI,CAAC,YAAY;aAC5B,CAAC;YACF,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC;YAEpC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;gBACzB,IAAI,EAAE,cAAc;gBACpB,QAAQ,EAAE,YAAY;gBACtB,OAAO;gBACP,SAAS,EAAE,WAAW;gBACtB,QAAQ,EAAE;oBACR,QAAQ,EAAE,WAAW,GAAG,aAAa;oBACrC,KAAK;oBACL,GAAG,EAAE,MAAM;iBACZ;aACF,CAAC,CAAC;YAEH,OAAO,KAAK,CAAC;QAAA,CACd,CAAC,CAAC;QAEH,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAE7C,OAAO,MAAyD,CAAC;IAAA,CAClE;IAED,IAAI,WAAW,GAAgB;QAC7B,OAAO,IAAI,CAAC,YAAY,CAAC;IAAA,CAC1B;IAED,MAAM,CAAC,IAA6B,EAAQ;QAC1C,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;IAAA,CACpD;IAED,cAAc,CAAC,eAAgC,EAAQ;QACrD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC;IAAA,CAC3D;IAED,aAAa,CAAC,MAA4D,EAAQ;QAChF,MAAM,IAAI,GAAkB;YAC1B,GAAG,MAAM;YACT,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,QAAQ;SAC9B,CAAC;QACF,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC;IAAA,CACrC;IAED,oBAAoB,CAAC,IAAuC,EAAQ;QAClE,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC;YAC7C,GAAG,IAAI;YACP,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;IAAA,CACJ;IAID,WAAW,CAAC,SAA2C,EAAE,KAAe,EAAQ;QAC9E,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;YAClC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAC7D,CAAC;aAAM,CAAC;YACN,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC/C,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YACjD,CAAC;QACH,CAAC;IAAA,CACF;IAEO,wBAAwB,CAAC,IAAmB,EAAQ;QAC1D,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAClD,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,kBAAkB,CAC5D,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,EAC/E,IAAI,CAAC,eAAe,CACrB,CAAC;QACF,MAAM,aAAa,GAAG,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;QAEhE,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;IAAA,CACpF;IAED,MAAM,CAAC,EAA8B,EAAQ;QAC3C,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAAA,CACzB;IAED,KAAK,CAAC,cAAc,GAAkB;QACpC,MAAM,aAAa,GAAG,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC;QAEpD,KAAK,MAAM,EAAE,IAAI,aAAa,EAAE,CAAC;YAC/B,IAAI,CAAC;gBACH,MAAM,EAAE,EAAE,CAAC;YACb,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC;IAAA,CACF;IAED,KAAK,CAAC,UAAU,GAA4B;QAC1C,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAC7C,OAAO,IAAI,CAAC,OAAO,CAAC;IAAA,CACrB;IAED;;;OAGG;IACH,oBAAoB,GAAS;QAC3B,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC;YAC9B,IAAI,EAAE,iBAAiB;YACvB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;IAAA,CACJ;IAED;;;;;OAKG;IACH,KAAK,CAAC,mBAAmB,CAAI,IAAO,EAAE,SAAiB,EAAiB;QACtE,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxC,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC;YAC7B,IAAI,EAAE,gBAAgB;YACtB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;YAChC,IAAI;YACJ,OAAO;SACR,CAAC,CAAC;IAAA,CACJ;IAED;;;;;;OAMG;IACH,KAAK,CAAC,oBAAoB,CAAC,KAAY,EAAE,SAAiB,EAAiB;QACzE,IAAI,OAAmC,CAAC;QACxC,IAAI,CAAC;YACH,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACpC,CAAC;QAAC,MAAM,CAAC;YACP,wCAAwC;QAC1C,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC;YAC9B,IAAI,EAAE,iBAAiB;YACvB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;YAChC,KAAK;YACL,OAAO;SACR,CAAC,CAAC;IAAA,CACJ;IAED,IAAc,OAAO,GAAW;QAC9B,OAAO,IAAI,CAAC,MAAM,CAAC;IAAA,CACpB;IAED,IAAc,UAAU,GAAW;QACjC,OAAO,IAAI,CAAC,gBAAgB,CAAC;IAAA,CAC9B;IAED,IAAc,QAAQ,GAAW;QAC/B,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IAAA,CACvD;CACF"}

package/dist/session/streaming-session.d.ts

@@ -0,0 +1,90 @@
+import type { LanguageModel, ToolSet } from 'ai';
+import type { Logger } from '../observability/logger.js';
+import type { FileManager } from '../provider/types.js';
+import type { ProviderType, ProviderPricing } from '../pricing/types.js';
+import type { GenerationOptions, SessionSummary } from './types.js';
+import { SimpleSession } from './simple-session.js';
+import type { SessionEvent, EmittableEventInput, ExtractResult } from '../execution/types.js';
+type ProviderOptions = Record<string, Record<string, unknown>>;
+export interface StreamingSessionOptions {
+    defaultLanguageModel?: LanguageModel | null;
+    modelFactory?: (modelId: string) => LanguageModel;
+    providerType: ProviderType;
+    providerPricing?: ProviderPricing;
+    fileManager: FileManager;
+    logger?: Logger;
+    startTime?: number;
+    signal?: AbortSignal;
+    defaultProviderOptions?: ProviderOptions;
+    defaultTools?: ToolSet;
+    defaultGenerationOptions?: GenerationOptions;
+}
+export declare class StreamingSession<TEvent extends {
+    type: string;
+}> extends SimpleSession {
+    private lastEventTime;
+    private _terminated;
+    constructor(options: StreamingSessionOptions);
+    /**
+     * Emits a streaming event with automatically attached metrics.
+     *
+     * Reserved types ('complete', 'error') throw at runtime - use session.done()
+     * or session.fail() instead.
+     *
+     * @param event - The event to emit (metrics will be added automatically)
+     * @returns The complete event with metrics attached
+     * @throws Error when attempting to emit reserved types ('complete', 'error')
+     */
+    emit(event: EmittableEventInput<TEvent>): SessionEvent<TEvent>;
+    /**
+     * Internal emit method - bypasses reserved type check.
+     * Used by done() and fail() to emit terminal events.
+     */
+    private emitInternal;
+    /**
+     * Signals successful completion of the streaming execution.
+     * Emits a 'complete' event with the result data and session summary.
+     * Also triggers Logger.onExecutionDone for observability.
+     * @param data - The final result data
+     * @returns The complete event with data and summary
+     */
+    done(data: ExtractResult<TEvent>): Promise<SessionEvent<TEvent>>;
+    /**
+     * Signals that the streaming execution failed with an error.
+     * Emits an 'error' event and triggers Logger.onExecutionError for observability.
+     * Gracefully handles getSummary() failures - summary will be undefined if it fails.
+     * @param error - The error that caused the failure
+     * @param data - Optional partial result data (if any was produced before failure)
+     * @returns The error event
+     */
+    fail(error: Error, data?: ExtractResult<TEvent>): Promise<SessionEvent<TEvent>>;
+    private createMetrics;
+}
+export interface StreamingSessionInternal<TEvent extends {
+    type: string;
+}> {
+    generateText: SimpleSession['generateText'];
+    streamText: SimpleSession['streamText'];
+    fileManager: FileManager;
+    onDone: SimpleSession['onDone'];
+    record: SimpleSession['record'];
+    recordToolCall: SimpleSession['recordToolCall'];
+    emit(event: EmittableEventInput<TEvent>): SessionEvent<TEvent>;
+    done(data: ExtractResult<TEvent>): Promise<SessionEvent<TEvent>>;
+    fail(error: Error, data?: ExtractResult<TEvent>): Promise<SessionEvent<TEvent>>;
+    runOnDoneHooks(): Promise<void>;
+    getSummary(): Promise<SessionSummary>;
+}
+export interface CreateStreamingSessionOptions {
+    defaultLanguageModel: LanguageModel;
+    providerType: ProviderType;
+    fileManager: FileManager;
+    logger?: Logger;
+    startTime?: number;
+    signal?: AbortSignal;
+}
+export declare function createStreamingSession<TEvent extends {
+    type: string;
+}>(options: CreateStreamingSessionOptions): StreamingSessionInternal<TEvent>;
+export {};
+//# sourceMappingURL=streaming-session.d.ts.map

package/dist/session/streaming-session.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming-session.d.ts","sourceRoot":"","sources":["../../src/session/streaming-session.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AAEjD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AAEzD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACzE,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AACpE,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,KAAK,EAAE,YAAY,EAAE,mBAAmB,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAE9F,KAAK,eAAe,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;AAE/D,MAAM,WAAW,uBAAuB;IACtC,oBAAoB,CAAC,EAAE,aAAa,GAAG,IAAI,CAAC;IAC5C,YAAY,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,aAAa,CAAC;IAClD,YAAY,EAAE,YAAY,CAAC;IAC3B,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,WAAW,EAAE,WAAW,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,sBAAsB,CAAC,EAAE,eAAe,CAAC;IACzC,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,wBAAwB,CAAC,EAAE,iBAAiB,CAAC;CAC9C;AAED,qBAAa,gBAAgB,CAAC,MAAM,SAAS;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,CAAE,SAAQ,aAAa;IAClF,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,WAAW,CAAS;IAE5B,YAAY,OAAO,EAAE,uBAAuB,EAqB3C;IAED;;;;;;;;;OASG;IACH,IAAI,CAAC,KAAK,EAAE,mBAAmB,CAAC,MAAM,CAAC,GAAG,YAAY,CAAC,MAAM,CAAC,CAe7D;IAED;;;OAGG;IACH,OAAO,CAAC,YAAY;IAYpB;;;;;;OAMG;IACG,IAAI,CAAC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAsBrE;IAED;;;;;;;OAOG;IACG,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,CAAC,EAAE,aAAa,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAqCpF;IAED,OAAO,CAAC,aAAa;CAUtB;AAED,MAAM,WAAW,wBAAwB,CAAC,MAAM,SAAS;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE;IACvE,YAAY,EAAE,aAAa,CAAC,cAAc,CAAC,CAAC;IAC5C,UAAU,EAAE,aAAa,CAAC,YAAY,CAAC,CAAC;IACxC,WAAW,EAAE,WAAW,CAAC;IACzB,MAAM,EAAE,aAAa,CAAC,QAAQ,CAAC,CAAC;IAChC,MAAM,EAAE,aAAa,CAAC,QAAQ,CAAC,CAAC;IAChC,cAAc,EAAE,aAAa,CAAC,gBAAgB,CAAC,CAAC;IAEhD,IAAI,CAAC,KAAK,EAAE,mBAAmB,CAAC,MAAM,CAAC,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;IAC/D,IAAI,CAAC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC;IACjE,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,CAAC,EAAE,aAAa,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC;IAEhF,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAChC,UAAU,IAAI,OAAO,CAAC,cAAc,CAAC,CAAC;CACvC;AAED,MAAM,WAAW,6BAA6B;IAC5C,oBAAoB,EAAE,aAAa,CAAC;IACpC,YAAY,EAAE,YAAY,CAAC;IAC3B,WAAW,EAAE,WAAW,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAED,wBAAgB,sBAAsB,CAAC,MAAM,SAAS;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,EACpE,OAAO,EAAE,6BAA6B,GACrC,wBAAwB,CAAC,MAAM,CAAC,CAWlC"}

package/dist/session/streaming-session.js

@@ -0,0 +1,151 @@
+import { noopLogger } from '../observability/logger.js';
+import { SimpleSession } from './simple-session.js';
+export class StreamingSession extends SimpleSession {
+    lastEventTime;
+    _terminated = false;
+    constructor(options) {
+        super({
+            defaultLanguageModel: options.defaultLanguageModel,
+            modelFactory: options.modelFactory,
+            providerType: options.providerType,
+            providerPricing: options.providerPricing,
+            fileManager: options.fileManager,
+            logger: options.logger,
+            startTime: options.startTime,
+            signal: options.signal,
+            defaultProviderOptions: options.defaultProviderOptions,
+            defaultTools: options.defaultTools,
+            defaultGenerationOptions: options.defaultGenerationOptions,
+        });
+        this.lastEventTime = this._startTime;
+        this._logger.onExecutionStart?.({
+            type: 'execution_start',
+            timestamp: Date.now(),
+        });
+    }
+    /**
+     * Emits a streaming event with automatically attached metrics.
+     *
+     * Reserved types ('complete', 'error') throw at runtime - use session.done()
+     * or session.fail() instead.
+     *
+     * @param event - The event to emit (metrics will be added automatically)
+     * @returns The complete event with metrics attached
+     * @throws Error when attempting to emit reserved types ('complete', 'error')
+     */
+    emit(event) {
+        if (this._terminated) {
+            throw new Error('Session already terminated. Cannot call emit() after a terminal operation.');
+        }
+        // Runtime check: prevent reserved types even when TypeScript is bypassed
+        const eventType = event.type;
+        if (eventType === 'complete' || eventType === 'error') {
+            throw new Error(`Cannot emit reserved type "${eventType}". ` +
+                'Use session.done() for completion or session.fail() for errors.');
+        }
+        return this.emitInternal(event);
+    }
+    /**
+     * Internal emit method - bypasses reserved type check.
+     * Used by done() and fail() to emit terminal events.
+     */
+    emitInternal(event) {
+        const metrics = this.createMetrics();
+        const fullEvent = { ...event, metrics };
+        this._logger.onExecutionEmit?.({
+            type: 'execution_emit',
+            event: fullEvent,
+        });
+        return fullEvent;
+    }
+    /**
+     * Signals successful completion of the streaming execution.
+     * Emits a 'complete' event with the result data and session summary.
+     * Also triggers Logger.onExecutionDone for observability.
+     * @param data - The final result data
+     * @returns The complete event with data and summary
+     */
+    async done(data) {
+        if (this._terminated) {
+            throw new Error('Session already terminated. Cannot call done() after a terminal operation.');
+        }
+        this._terminated = true;
+        const summary = await this.getSummary();
+        this._logger.onExecutionDone?.({
+            type: 'execution_done',
+            timestamp: Date.now(),
+            duration: summary.totalDuration,
+            data,
+            summary,
+        });
+        // Use emitInternal to bypass reserved type check (internal only)
+        return this.emitInternal({
+            type: 'complete',
+            data,
+            summary,
+        });
+    }
+    /**
+     * Signals that the streaming execution failed with an error.
+     * Emits an 'error' event and triggers Logger.onExecutionError for observability.
+     * Gracefully handles getSummary() failures - summary will be undefined if it fails.
+     * @param error - The error that caused the failure
+     * @param data - Optional partial result data (if any was produced before failure)
+     * @returns The error event
+     */
+    async fail(error, data) {
+        if (this._terminated) {
+            throw new Error('Session already terminated. Cannot call fail() after a terminal operation.');
+        }
+        this._terminated = true;
+        let summary;
+        try {
+            summary = await this.getSummary();
+        }
+        catch {
+            // Ignore summary errors on failure path
+        }
+        this._logger.onExecutionError?.({
+            type: 'execution_error',
+            timestamp: Date.now(),
+            duration: summary?.totalDuration ?? (Date.now() - this._startTime),
+            error,
+            data,
+            summary,
+        });
+        const errorEvent = {
+            type: 'error',
+            error,
+        };
+        if (summary) {
+            errorEvent.summary = summary;
+        }
+        if (data !== undefined) {
+            errorEvent.data = data;
+        }
+        // Use emitInternal to bypass reserved type check (internal only)
+        return this.emitInternal(errorEvent);
+    }
+    createMetrics() {
+        const now = Date.now();
+        const metrics = {
+            timestamp: now,
+            elapsedMs: now - this._startTime,
+            deltaMs: now - this.lastEventTime,
+        };
+        this.lastEventTime = now;
+        return metrics;
+    }
+}
+export function createStreamingSession(options) {
+    const session = new StreamingSession({
+        defaultLanguageModel: options.defaultLanguageModel,
+        providerType: options.providerType,
+        fileManager: options.fileManager,
+        logger: options.logger ?? noopLogger,
+        startTime: options.startTime,
+        signal: options.signal,
+    });
+    return session;
+}
+//# sourceMappingURL=streaming-session.js.map

package/dist/session/streaming-session.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming-session.js","sourceRoot":"","sources":["../../src/session/streaming-session.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAIxD,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAmBpD,MAAM,OAAO,gBAAkD,SAAQ,aAAa;IAC1E,aAAa,CAAS;IACtB,WAAW,GAAG,KAAK,CAAC;IAE5B,YAAY,OAAgC,EAAE;QAC5C,KAAK,CAAC;YACJ,oBAAoB,EAAE,OAAO,CAAC,oBAAoB;YAClD,YAAY,EAAE,OAAO,CAAC,YAAY;YAClC,YAAY,EAAE,OAAO,CAAC,YAAY;YAClC,eAAe,EAAE,OAAO,CAAC,eAAe;YACxC,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,sBAAsB,EAAE,OAAO,CAAC,sBAAsB;YACtD,YAAY,EAAE,OAAO,CAAC,YAAY;YAClC,wBAAwB,EAAE,OAAO,CAAC,wBAAwB;SAC3D,CAAC,CAAC;QAEH,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,UAAU,CAAC;QAErC,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC;YAC9B,IAAI,EAAE,iBAAiB;YACvB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;IAAA,CACJ;IAED;;;;;;;;;OASG;IACH,IAAI,CAAC,KAAkC,EAAwB;QAC7D,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,4EAA4E,CAAC,CAAC;QAChG,CAAC;QAED,yEAAyE;QACzE,MAAM,SAAS,GAAI,KAA0B,CAAC,IAAI,CAAC;QACnD,IAAI,SAAS,KAAK,UAAU,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;YACtD,MAAM,IAAI,KAAK,CACb,8BAA8B,SAAS,KAAK;gBAC5C,iEAAiE,CAClE,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC,YAAY,CAAC,KAAe,CAAC,CAAC;IAAA,CAC3C;IAED;;;OAGG;IACK,YAAY,CAAC,KAAa,EAAwB;QACxD,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;QACrC,MAAM,SAAS,GAAG,EAAE,GAAG,KAAK,EAAE,OAAO,EAA0B,CAAC;QAEhE,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC;YAC7B,IAAI,EAAE,gBAAgB;YACtB,KAAK,EAAE,SAAS;SACjB,CAAC,CAAC;QAEH,OAAO,SAAS,CAAC;IAAA,CAClB;IAED;;;;;;OAMG;IACH,KAAK,CAAC,IAAI,CAAC,IAA2B,EAAiC;QACrE,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,4EAA4E,CAAC,CAAC;QAChG,CAAC;QACD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAExB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAExC,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC;YAC7B,IAAI,EAAE,gBAAgB;YACtB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,QAAQ,EAAE,OAAO,CAAC,aAAa;YAC/B,IAAI;YACJ,OAAO;SACR,CAAC,CAAC;QAEH,iEAAiE;QACjE,OAAO,IAAI,CAAC,YAAY,CAAC;YACvB,IAAI,EAAE,UAAU;YAChB,IAAI;YACJ,OAAO;SAC4C,CAAC,CAAC;IAAA,CACxD;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,IAAI,CAAC,KAAY,EAAE,IAA4B,EAAiC;QACpF,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,4EAA4E,CAAC,CAAC;QAChG,CAAC;QACD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAExB,IAAI,OAAmC,CAAC;QACxC,IAAI,CAAC;YACH,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACpC,CAAC;QAAC,MAAM,CAAC;YACP,wCAAwC;QAC1C,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC;YAC9B,IAAI,EAAE,iBAAiB;YACvB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,QAAQ,EAAE,OAAO,EAAE,aAAa,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,UAAU,CAAC;YAClE,KAAK;YACL,IAAI;YACJ,OAAO;SACR,CAAC,CAAC;QAEH,MAAM,UAAU,GAA4B;YAC1C,IAAI,EAAE,OAAO;YACb,KAAK;SACN,CAAC;QAEF,IAAI,OAAO,EAAE,CAAC;YACZ,UAAU,CAAC,OAAO,GAAG,OAAO,CAAC;QAC/B,CAAC;QAED,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,UAAU,CAAC,IAAI,GAAG,IAAI,CAAC;QACzB,CAAC;QAED,iEAAiE;QACjE,OAAO,IAAI,CAAC,YAAY,CAAC,UAA2D,CAAC,CAAC;IAAA,CACvF;IAEO,aAAa,GAAiB;QACpC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,OAAO,GAAiB;YAC5B,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG,GAAG,IAAI,CAAC,UAAU;YAChC,OAAO,EAAE,GAAG,GAAG,IAAI,CAAC,aAAa;SAClC,CAAC;QACF,IAAI,CAAC,aAAa,GAAG,GAAG,CAAC;QACzB,OAAO,OAAO,CAAC;IAAA,CAChB;CACF;AA2BD,MAAM,UAAU,sBAAsB,CACpC,OAAsC,EACJ;IAClC,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAS;QAC3C,oBAAoB,EAAE,OAAO,CAAC,oBAAoB;QAClD,YAAY,EAAE,OAAO,CAAC,YAAY;QAClC,WAAW,EAAE,OAAO,CAAC,WAAW;QAChC,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,UAAU;QACpC,SAAS,EAAE,OAAO,CAAC,SAAS;QAC5B,MAAM,EAAE,OAAO,CAAC,MAAM;KACvB,CAAC,CAAC;IAEH,OAAO,OAAsD,CAAC;AAAA,CAC/D"}

package/dist/session/test-utils.d.ts

@@ -0,0 +1,19 @@
+import { vi } from 'vitest';
+import type { LanguageModel, LanguageModelUsage } from 'ai';
+import type { Logger } from '../observability/logger.js';
+import type { FileManager } from '../provider/types.js';
+export declare const TEST_PROVIDER_TYPE: "google";
+export declare function createMockModel(): LanguageModel;
+export declare function createMockFileManager(): FileManager;
+export type MockLogger = Logger & {
+    onLLMCallStart: ReturnType<typeof vi.fn>;
+    onLLMCallEnd: ReturnType<typeof vi.fn>;
+    onExecutionStart: ReturnType<typeof vi.fn>;
+    onExecutionEmit: ReturnType<typeof vi.fn>;
+    onExecutionDone: ReturnType<typeof vi.fn>;
+    onExecutionError: ReturnType<typeof vi.fn>;
+};
+export declare function createMockLogger(): MockLogger;
+export declare function createMockUsage(overrides?: Partial<LanguageModelUsage>): LanguageModelUsage;
+export declare function createTestUsage(partial: Partial<LanguageModelUsage>): LanguageModelUsage;
+//# sourceMappingURL=test-utils.d.ts.map

package/dist/session/test-utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-utils.d.ts","sourceRoot":"","sources":["../../src/session/test-utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC5B,OAAO,KAAK,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,IAAI,CAAC;AAC5D,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAExD,eAAO,MAAM,kBAAkB,UAAoB,CAAC;AAEpD,wBAAgB,eAAe,IAAI,aAAa,CAS/C;AAED,wBAAgB,qBAAqB,IAAI,WAAW,CAOnD;AAED,MAAM,MAAM,UAAU,GAAG,MAAM,GAAG;IAChC,cAAc,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IACzC,YAAY,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IACvC,gBAAgB,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IAC3C,eAAe,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IAC1C,eAAe,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IAC1C,gBAAgB,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;CAC5C,CAAC;AAEF,wBAAgB,gBAAgB,IAAI,UAAU,CAS7C;AAED,wBAAgB,eAAe,CAAC,SAAS,GAAE,OAAO,CAAC,kBAAkB,CAAM,GAAG,kBAAkB,CAgB/F;AAED,wBAAgB,eAAe,CAAC,OAAO,EAAE,OAAO,CAAC,kBAAkB,CAAC,GAAG,kBAAkB,CAgBxF"}

package/dist/session/test-utils.js

@@ -0,0 +1,65 @@
+import { vi } from 'vitest';
+export const TEST_PROVIDER_TYPE = 'google';
+export function createMockModel() {
+    return {
+        specificationVersion: 'v1',
+        provider: 'test-provider',
+        modelId: 'test-model',
+        defaultObjectGenerationMode: 'json',
+        doGenerate: vi.fn(),
+        doStream: vi.fn(),
+    };
+}
+export function createMockFileManager() {
+    return {
+        upload: vi.fn().mockResolvedValue([]),
+        delete: vi.fn().mockResolvedValue(undefined),
+        clear: vi.fn().mockResolvedValue(undefined),
+        getUploadedFiles: vi.fn().mockReturnValue([]),
+    };
+}
+export function createMockLogger() {
+    return {
+        onLLMCallStart: vi.fn(),
+        onLLMCallEnd: vi.fn(),
+        onExecutionStart: vi.fn(),
+        onExecutionEmit: vi.fn(),
+        onExecutionDone: vi.fn(),
+        onExecutionError: vi.fn(),
+    };
+}
+export function createMockUsage(overrides = {}) {
+    return {
+        inputTokens: 100,
+        outputTokens: 50,
+        totalTokens: 150,
+        inputTokenDetails: {
+            noCacheTokens: 100,
+            cacheReadTokens: 0,
+            cacheWriteTokens: 0,
+        },
+        outputTokenDetails: {
+            textTokens: 50,
+            reasoningTokens: 0,
+        },
+        ...overrides,
+    };
+}
+export function createTestUsage(partial) {
+    return {
+        inputTokens: 0,
+        outputTokens: 0,
+        totalTokens: 0,
+        inputTokenDetails: {
+            noCacheTokens: 0,
+            cacheReadTokens: 0,
+            cacheWriteTokens: 0,
+        },
+        outputTokenDetails: {
+            textTokens: 0,
+            reasoningTokens: 0,
+        },
+        ...partial,
+    };
+}
+//# sourceMappingURL=test-utils.js.map