@superblocksteam/vite-plugin-file-sync 2.0.70 → 2.0.71-next.1

package/dist/ai-service/llm/stream/session.js

@@ -0,0 +1,116 @@
+import { StreamEventBus } from "./event-bus.js";
+/**
+ * State container for a single stream operation.
+ *
+ * Contains all mutable state for the session:
+ * - Current attempt number
+ * - Current provider info
+ * - Metrics (chunks, tokens, retries, duration)
+ * - Error and outcome status
+ *
+ * The session also owns the event bus for this operation.
+ *
+ * @example
+ * ```typescript
+ * const session = new StreamSession(config, initialProvider);
+ *
+ * // Record activity
+ * session.recordChunk();
+ * session.recordRetry(500);
+ *
+ * // Finalize when done
+ * session.finalize("complete");
+ * ```
+ */
+export class StreamSession {
+    id;
+    startTime;
+    events;
+    config;
+    attempt = 1;
+    provider;
+    metrics = {
+        chunkCount: 0,
+        totalTokens: 0,
+        retryCount: 0,
+        totalRetryDelayMs: 0,
+        durationMs: 0,
+        timeToFirstTokenMs: undefined,
+    };
+    error = null;
+    outcome = "pending";
+    // tracks when we last received a chunk or started a retry, used for max-time checks.
+    lastActivityTime;
+    constructor(config, initialProvider) {
+        this.id = crypto.randomUUID();
+        this.startTime = Date.now();
+        this.lastActivityTime = this.startTime;
+        this.config = config;
+        this.provider = initialProvider;
+        this.events = new StreamEventBus();
+    }
+    /**
+     * Finalize the session with an outcome.
+     *
+     * Can only be called once. Subsequent calls are ignored.
+     * Emits the appropriate lifecycle event based on outcome.
+     *
+     * @param outcome - The final outcome of the session
+     * @param error - Optional error that caused failure
+     */
+    finalize(outcome, error) {
+        if (this.outcome !== "pending") {
+            return;
+        }
+        this.outcome = outcome;
+        this.error = error ?? null;
+        this.metrics.durationMs = Date.now() - this.startTime;
+        switch (outcome) {
+            case "complete":
+                void this.events.emit("session:complete", this);
+                break;
+            case "error":
+                void this.events.emit("session:error", error, this);
+                break;
+            case "aborted":
+                void this.events.emit("session:abort", this);
+                break;
+        }
+    }
+    recordRetry(delayMs) {
+        this.attempt++;
+        this.metrics.retryCount++;
+        this.metrics.totalRetryDelayMs += delayMs;
+        this.lastActivityTime = Date.now();
+    }
+    // called when falling back to a new provider, each provider gets fresh attempts.
+    // if we do not reset, we burn through all providers in one go.
+    resetAttempts() {
+        this.attempt = 0;
+    }
+    recordChunk() {
+        if (this.metrics.chunkCount === 0) {
+            this.metrics.timeToFirstTokenMs = Date.now() - this.startTime;
+        }
+        this.metrics.chunkCount++;
+        this.lastActivityTime = Date.now();
+    }
+    setTotalTokens(tokens) {
+        this.metrics.totalTokens = tokens;
+    }
+    isPending() {
+        return this.outcome === "pending";
+    }
+    getElapsedMs() {
+        return Date.now() - this.startTime;
+    }
+    // use this for max-time checks, NOT getElapsedMs(). we only want to timeout
+    // if nothing is happening, not if the stream has been running for a while.
+    getTimeSinceLastActivity() {
+        return Date.now() - this.lastActivityTime;
+    }
+    resetActivityTimer() {
+        this.lastActivityTime = Date.now();
+    }
+}
+//# sourceMappingURL=session.js.map

package/dist/ai-service/llm/stream/session.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session.js","sourceRoot":"","sources":["../../../../src/ai-service/llm/stream/session.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAIhD;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,MAAM,OAAO,aAAa;IACf,EAAE,CAAS;IACX,SAAS,CAAS;IAClB,MAAM,CAAiB;IACvB,MAAM,CAAe;IAE9B,OAAO,GAAG,CAAC,CAAC;IACZ,QAAQ,CAAe;IACvB,OAAO,GAAkB;QACvB,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,CAAC;QACd,UAAU,EAAE,CAAC;QACb,iBAAiB,EAAE,CAAC;QACpB,UAAU,EAAE,CAAC;QACb,kBAAkB,EAAE,SAAS;KAC9B,CAAC;IACF,KAAK,GAAY,IAAI,CAAC;IACtB,OAAO,GAAmB,SAAS,CAAC;IAEpC,qFAAqF;IAC7E,gBAAgB,CAAS;IAEjC,YAAY,MAAoB,EAAE,eAA6B;QAC7D,IAAI,CAAC,EAAE,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;QAC9B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC5B,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,SAAS,CAAC;QACvC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,eAAe,CAAC;QAChC,IAAI,CAAC,MAAM,GAAG,IAAI,cAAc,EAAE,CAAC;IACrC,CAAC;IAED;;;;;;;;OAQG;IACH,QAAQ,CAAC,OAA2C,EAAE,KAAe;QACnE,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QAED,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,KAAK,GAAG,KAAK,IAAI,IAAI,CAAC;QAC3B,IAAI,CAAC,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC;QAEtD,QAAQ,OAAO,EAAE,CAAC;YAChB,KAAK,UAAU;gBACb,KAAK,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC;gBAChD,MAAM;YACR,KAAK,OAAO;gBACV,KAAK,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;gBACpD,MAAM;YACR,KAAK,SAAS;gBACZ,KAAK,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;gBAC7C,MAAM;QACV,CAAC;IACH,CAAC;IAED,WAAW,CAAC,OAAe;QACzB,IAAI,CAAC,OAAO,EAAE,CAAC;QACf,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;QAC1B,IAAI,CAAC,OAAO,CAAC,iBAAiB,IAAI,OAAO,CAAC;QAC1C,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACrC,CAAC;IAED,iFAAiF;IACjF,+DAA+D;IAC/D,aAAa;QACX,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC;IACnB,CAAC;IAED,WAAW;QACT,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,KAAK,CAAC,EAAE,CAAC;YAClC,IAAI,CAAC,OAAO,CAAC,kBAAkB,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC;QAChE,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;QAC1B,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACrC,CAAC;IAED,cAAc,CAAC,MAAc;QAC3B,IAAI,CAAC,OAAO,CAAC,WAAW,GAAG,MAAM,CAAC;IACpC,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,OAAO,KAAK,SAAS,CAAC;IACpC,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC;IACrC,CAAC;IAED,4EAA4E;IAC5E,2EAA2E;IAC3E,wBAAwB;QACtB,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC;IAC5C,CAAC;IAED,kBAAkB;QAChB,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACrC,CAAC;CACF"}

package/dist/ai-service/llm/stream/tests/helpers/mock-provider.d.ts

@@ -0,0 +1,129 @@
+/**
+ * Mock provider helpers for testing the stream architecture.
+ *
+ * Provides declarative mock builders that make tests readable.
+ */
+import type { Tool } from "../../../../agent/tools2/types.js";
+import type { StreamTextProvider, StreamTextResult, StreamTextOptions } from "../../../interaction/provider.js";
+import type { TextStreamPart, AsyncIterableStream } from "ai";
+/**
+ * Default chunks to return from mock streams.
+ */
+export declare const defaultChunks: TextStreamPart<Record<string, Tool>>[];
+/** Default usage stats for sentence-based providers. */
+export declare const sentenceUsage: {
+    inputTokens: number;
+    outputTokens: number;
+    totalTokens: number;
+};
+/**
+ * Create full stream chunks from an array of text strings.
+ */
+export declare function chunksFromText<TOOLS extends Record<string, Tool>>(textChunks: string[], usage?: {
+    inputTokens: number;
+    outputTokens: number;
+    totalTokens: number;
+}): TextStreamPart<TOOLS>[];
+/**
+ * Split a sentence into word-based text chunks (space-separated).
+ * Each word after the first is prefixed with a space.
+ */
+export declare function textToChunks(text: string): string[];
+/**
+ * Create a mock StreamTextResult from text chunks.
+ */
+export declare function createResultFromText<TOOLS extends Record<string, Tool>>(textChunks: string[], usage?: {
+    inputTokens: number;
+    outputTokens: number;
+    totalTokens: number;
+}): StreamTextResult<TOOLS>;
+/**
+ * Create a mock StreamTextResult from a sentence (splits by spaces).
+ */
+export declare function createResultFromSentence<TOOLS extends Record<string, Tool>>(text: string): StreamTextResult<TOOLS>;
+/**
+ * Create an AsyncIterableStream from an array of items.
+ * Each call to [Symbol.asyncIterator]() creates a fresh iterator.
+ */
+export declare function asyncIterableFrom<T>(items: T[]): AsyncIterableStream<T>;
+export declare const MockProvider: {
+    /**
+     * Provider that always succeeds with the given chunks.
+     */
+    success<TOOLS extends Record<string, Tool> = Record<string, Tool<any, any>>>(chunks?: TextStreamPart<TOOLS>[]): StreamTextProvider;
+    /**
+     * Provider that always succeeds with the given text (splits by spaces).
+     */
+    successWithText(text: string): StreamTextProvider;
+    /**
+     * Provider that fails on the first N attempts then succeeds.
+     * Optionally tracks calls by pushing to the provided `calls` array.
+     */
+    failsThenSucceeds(failCount: number, options?: {
+        error?: Error;
+        text?: string;
+        textChunks?: string[];
+        calls?: Array<{
+            messages: any[];
+            timestamp: number;
+        }>;
+    }): StreamTextProvider;
+    /**
+     * Provider where the stream fails after N chunks (sends error chunk).
+     */
+    streamFailsAfter(chunkCount: number, error?: Error): StreamTextProvider;
+    /**
+     * Provider where the iterator throws after N chunks (simulates network disconnect).
+     * Unlike streamFailsAfter which sends an error chunk, this throws from iterator.next().
+     */
+    iteratorThrowsAfter(chunkCount: number, error?: Error): StreamTextProvider;
+    /**
+     * Provider where iterator throws on first N attempts, then succeeds.
+     * Simulates network issues that recover after retries.
+     */
+    iteratorThrowsThenSucceeds(failCount: number, options?: {
+        error?: Error;
+        chunksBeforeError?: number;
+        text?: string;
+    }): StreamTextProvider;
+    /**
+     * Provider that stalls after N chunks.
+     */
+    streamStalls(chunkCount: number): StreamTextProvider;
+    /**
+     * Provider that tracks calls for verification.
+     */
+    tracked(text?: string): {
+        provider: StreamTextProvider;
+        calls: Array<{
+            options: StreamTextOptions<any>;
+            timestamp: number;
+        }>;
+    };
+    /**
+     * Provider that always fails with the given error.
+     */
+    alwaysFails(error?: Error): StreamTextProvider;
+    /**
+     * Provider that waits for a delay then fails.
+     */
+    slowAlwaysFails(delayMs: number, error?: Error): StreamTextProvider;
+    /**
+     * Provider that stalls for a specified duration then continues producing chunks.
+     * This simulates the scenario where stall detection fires but the underlying
+     * stream continues to produce data (e.g., tool calls appear after the stall error).
+     *
+     * @param stallDurationMs - How long to stall before continuing
+     * @param chunksBeforeStall - Number of chunks to emit before stalling
+     * @param chunksAfterStall - Optional chunks to emit after stall (defaults to success chunks)
+     */
+    stallsThenContinues(stallDurationMs: number, chunksBeforeStall?: number, chunksAfterStall?: TextStreamPart<Record<string, Tool>>[]): StreamTextProvider;
+    /**
+     * Provider with slow steps on first N attempts, then succeeds.
+     * Simulates slow LLM responses that trigger stall detection.
+     */
+    slowStepsThenSucceeds(slowCount: number, stepDelayMs: number, options?: {
+        text?: string;
+    }): StreamTextProvider;
+};
+//# sourceMappingURL=mock-provider.d.ts.map

package/dist/ai-service/llm/stream/tests/helpers/mock-provider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock-provider.d.ts","sourceRoot":"","sources":["../../../../../../src/ai-service/llm/stream/tests/helpers/mock-provider.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,mCAAmC,CAAC;AAC9D,OAAO,KAAK,EACV,kBAAkB,EAClB,gBAAgB,EAChB,iBAAiB,EAClB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,KAAK,EAAE,cAAc,EAAE,mBAAmB,EAAE,MAAM,IAAI,CAAC;AAE9D;;GAEG;AACH,eAAO,MAAM,aAAa,EAAE,cAAc,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,EAW/D,CAAC;AAgBF,wDAAwD;AACxD,eAAO,MAAM,aAAa;;;;CAIzB,CAAC;AAEF;;GAEG;AACH,wBAAgB,cAAc,CAAC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,EAC/D,UAAU,EAAE,MAAM,EAAE,EACpB,KAAK;;;;CAAwD,GAC5D,cAAc,CAAC,KAAK,CAAC,EAAE,CAezB;AAED;;;GAGG;AACH,wBAAgB,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE,CAGnD;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,EACrE,UAAU,EAAE,MAAM,EAAE,EACpB,KAAK;;;;CAAwD,GAC5D,gBAAgB,CAAC,KAAK,CAAC,CAazB;AAED;;GAEG;AACH,wBAAgB,wBAAwB,CAAC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,EACzE,IAAI,EAAE,MAAM,GACX,gBAAgB,CAAC,KAAK,CAAC,CAEzB;AAED;;;GAGG;AACH,wBAAgB,iBAAiB,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,GAAG,mBAAmB,CAAC,CAAC,CAAC,CA8BvE;AAED,eAAO,MAAM,YAAY;IACvB;;OAEG;YACK,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,4CAChC,cAAc,CAAC,KAAK,CAAC,EAAE,GAC9B,kBAAkB;IAuBrB;;OAEG;0BACmB,MAAM,GAAG,kBAAkB;IAUjD;;;OAGG;iCAEU,MAAM,YACR;QACP,KAAK,CAAC,EAAE,KAAK,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;QACtB,KAAK,CAAC,EAAE,KAAK,CAAC;YAAE,QAAQ,EAAE,GAAG,EAAE,CAAC;YAAC,SAAS,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KACvD,GACA,kBAAkB;IAmCrB;;OAEG;iCAEW,MAAM,kBAEjB,kBAAkB;IAoBrB;;;OAGG;oCAEW,MAAM,kBAEjB,kBAAkB;IA4CrB;;;OAGG;0CAEU,MAAM,YACR;QACP,KAAK,CAAC,EAAE,KAAK,CAAC;QACd,iBAAiB,CAAC,EAAE,MAAM,CAAC;QAC3B,IAAI,CAAC,EAAE,MAAM,CAAC;KACf,GACA,kBAAkB;IA8DrB;;OAEG;6BACsB,MAAM,GAAG,kBAAkB;IA6CpD;;OAEG;mBACY,MAAM,GAAG;QACtB,QAAQ,EAAE,kBAAkB,CAAC;QAC7B,KAAK,EAAE,KAAK,CAAC;YAAE,OAAO,EAAE,iBAAiB,CAAC,GAAG,CAAC,CAAC;YAAC,SAAS,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KACtE;IAsBD;;OAEG;gCAC6C,kBAAkB;IAUlE;;OAEG;6BAEQ,MAAM,kBAEd,kBAAkB;IAWrB;;;;;;;;OAQG;yCAEgB,MAAM,sBACJ,MAAM,qBACN,cAAc,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,EAAE,GACxD,kBAAkB;IA0ErB;;;OAGG;qCAEU,MAAM,eACJ,MAAM,YACV;QAAE,IAAI,CAAC,EAAE,MAAM,CAAA;KAAE,GACzB,kBAAkB;CAwEtB,CAAC"}