@superblocksteam/vite-plugin-file-sync 2.0.59-next.1 → 2.0.59-next.10

package/dist/ai-service/llm/interaction/provider.d.ts

@@ -0,0 +1,173 @@
+/**
+ * High-level provider interface for LLM streaming interactions.
+ *
+ * This interface defines a complete, SDK-agnostic contract for multi-step
+ * LLM streaming, including all lifecycle hooks. Concrete implementations
+ * (adapters) bridge this interface to specific SDKs (Vercel AI, Anthropic, etc.).
+ *
+ * Key features:
+ * - Complete lifecycle: prepareStep → onChunk → onStepFinish → onFinish → onFinally
+ * - onFinally: Guaranteed to run after stream consumption (our addition!)
+ * - SDK-agnostic: Could be implemented with any LLM SDK
+ * - Testable: Easy to mock for testing
+ */
+import type { Tool } from "../../agent/tools2/types.js";
+import type { StopCondition } from "../../util/stop-condition.js";
+import type { ProviderOptions } from "@ai-sdk/provider-utils";
+import type { LanguageModel, TextStreamPart, LanguageModelUsage, ModelMessage, UserModelMessage } from "ai";
+/**
+ * Provider interface for LLM streaming interactions.
+ *
+ * Implementations of this interface wrap specific SDKs and provide
+ * a consistent, high-level API with complete lifecycle management.
+ */
+export interface StreamTextProvider {
+    /**
+     * Streams text from an LLM with multi-step tool calling support.
+     *
+     * @param options - Configuration and lifecycle hooks
+     * @returns Promise resolving to stream result with text/full streams
+     */
+    streamText<TOOLS extends Record<string, Tool> = Record<string, Tool>>(options: StreamTextOptions<TOOLS>): Promise<StreamTextResult<TOOLS>>;
+}
+/**
+ * Options for a streamText call.
+ *
+ * Includes model configuration, messages, tools, and comprehensive
+ * lifecycle hooks including onFinally (runs after stream consumption).
+ */
+export interface StreamTextOptions<TOOLS extends Record<string, Tool>> {
+    /** The language model to use */
+    model: LanguageModel;
+    /**
+     * Conversation messages.
+     * Note: When using context middleware, this will be populated automatically
+     * from context history. You typically provide user instead.
+     */
+    messages?: ModelMessage[];
+    /**
+     * User message for this interaction.
+     * When using context middleware, this is added to the conversation history
+     * and messages are populated automatically from context.
+     */
+    user?: UserModelMessage;
+    /** Available tools the model can call */
+    tools?: TOOLS;
+    /**
+     * System prompt.
+     * When using context middleware, this is set on the context.
+     */
+    system?: string;
+    /** Maximum number of automatic tool call steps */
+    maxSteps?: number;
+    /** Abort signal for cancellation */
+    abortSignal?: AbortSignal;
+    /** Custom headers for the request */
+    headers?: Record<string, string | undefined>;
+    /** Provider-specific options */
+    providerOptions?: ProviderOptions;
+    /**
+     * Vercel AI SDK specific: Condition to stop generation early.
+     * Pass through to the underlying SDK.
+     */
+    stopWhen?: StopCondition<any> | StopCondition<any>[];
+    /**
+     * Vercel AI SDK specific: Transformation stream for the output.
+     * Pass through to the underlying SDK.
+     */
+    experimental_transform?: any;
+    /**
+     * Called before each LLM step.
+     * Can transform the step parameters before execution.
+     */
+    prepareStep?: (step: StepParameters) => Promise<StepParameters> | StepParameters;
+    /**
+     * Called for each chunk in the stream during generation.
+     */
+    onChunk?: (event: {
+        chunk: StreamChunk;
+    }) => Promise<void> | void;
+    /**
+     * Called after each step completes (after tool execution).
+     */
+    onStepFinish?: (step: StepResult) => Promise<void> | void;
+    /**
+     * Called when the entire multi-step interaction finishes.
+     * This fires after generation completes, before stream consumption.
+     */
+    onFinish?: (result: FinalResult) => Promise<void> | void;
+    /**
+     * Called after stream consumption completes.
+     *
+     * This is the key lifecycle hook for cleanup operations:
+     * - Releasing locks (context handles)
+     * - Finishing tracing spans
+     * - Closing database connections
+     * - Saving final artifacts
+     *
+     * Guarantees:
+     * - Runs exactly once, even if multiple stream accessors are used
+     * - Runs after the stream is fully consumed OR abandoned
+     * - Runs in error cases as well (usage may be partial)
+     * - Idempotent and safe to call multiple times
+     *
+     */
+    onFinally?: () => Promise<void> | void;
+    /**
+     * Called if an error occurs during generation or consumption.
+     */
+    onError?: (error: Error) => Promise<void> | void;
+    /**
+     * Called if the operation is aborted via the abortSignal.
+     */
+    onAbort?: () => Promise<void> | void;
+}
+/**
+ * Parameters for a single LLM step.
+ */
+export interface StepParameters {
+    messages: ModelMessage[];
+    stepNumber: number;
+}
+/**
+ * A chunk from the streaming response.
+ * This is a direct re-export of Vercel AI SDK's TextStreamPart.
+ */
+export type StreamChunk = TextStreamPart<Record<string, Tool>>;
+import type { StepResult as VercelStepResult } from "ai";
+/**
+ * Result of a single LLM step.
+ * Re-exported from Vercel AI SDK to avoid type incompatibilities.
+ */
+export type StepResult = VercelStepResult<any>;
+/**
+ * Final result of the entire multi-step interaction.
+ * This is the result passed to onFinish callback.
+ * Re-exported from Vercel AI SDK's internal type.
+ */
+export type FinalResult = StepResult & {
+    readonly steps: StepResult[];
+    readonly totalUsage: LanguageModelUsage;
+};
+import type { AsyncIterableStream } from "ai";
+/**
+ * Result of a streamText call.
+ *
+ * This type defines only the properties we actually use from the Vercel AI SDK's
+ * StreamTextResult. Adapters must use `createStreamResult()` to build this type
+ * from raw SDK results, ensuring all properties are plain enumerable values.
+ *
+ * The brand prevents returning raw SDK results (compile error) and ensures
+ * spread operators work correctly in middlewares.
+ */
+export type StreamTextResult<TOOLS extends Record<string, Tool> = Record<string, Tool>> = {
+    /** Brand to ensure properties are materialized (enumerable) */
+    readonly __materialized: unique symbol;
+    /** Stream of text deltas */
+    textStream: AsyncIterableStream<string>;
+    /** Stream of all events (text, tool calls, etc.) */
+    fullStream: AsyncIterableStream<TextStreamPart<TOOLS>>;
+    /** Promise that resolves with the final result */
+    response: Promise<FinalResult>;
+};
+//# sourceMappingURL=provider.d.ts.map

package/dist/ai-service/llm/interaction/provider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/llm/interaction/provider.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAEH,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,6BAA6B,CAAC;AACxD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAClE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAC9D,OAAO,KAAK,EACV,aAAa,EACb,cAAc,EACd,kBAAkB,EAClB,YAAY,EACZ,gBAAgB,EACjB,MAAM,IAAI,CAAC;AAEZ;;;;;GAKG;AACH,MAAM,WAAW,kBAAkB;IACjC;;;;;OAKG;IACH,UAAU,CAAC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,EAClE,OAAO,EAAE,iBAAiB,CAAC,KAAK,CAAC,GAChC,OAAO,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC,CAAC;CACrC;AAED;;;;;GAKG;AACH,MAAM,WAAW,iBAAiB,CAAC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC;IACnE,gCAAgC;IAChC,KAAK,EAAE,aAAa,CAAC;IAErB;;;;OAIG;IACH,QAAQ,CAAC,EAAE,YAAY,EAAE,CAAC;IAE1B;;;;OAIG;IACH,IAAI,CAAC,EAAE,gBAAgB,CAAC;IAExB,yCAAyC;IACzC,KAAK,CAAC,EAAE,KAAK,CAAC;IAEd;;;OAGG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB,kDAAkD;IAClD,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB,oCAAoC;IACpC,WAAW,CAAC,EAAE,WAAW,CAAC;IAE1B,qCAAqC;IACrC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC;IAE7C,gCAAgC;IAChC,eAAe,CAAC,EAAE,eAAe,CAAC;IAElC;;;OAGG;IACH,QAAQ,CAAC,EAAE,aAAa,CAAC,GAAG,CAAC,GAAG,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC;IAErD;;;OAGG;IACH,sBAAsB,CAAC,EAAE,GAAG,CAAC;IAE7B;;;OAGG;IACH,WAAW,CAAC,EAAE,CACZ,IAAI,EAAE,cAAc,KACjB,OAAO,CAAC,cAAc,CAAC,GAAG,cAAc,CAAC;IAE9C;;OAEG;IACH,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE;QAAE,KAAK,EAAE,WAAW,CAAA;KAAE,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAElE;;OAEG;IACH,YAAY,CAAC,EAAE,CAAC,IAAI,EAAE,UAAU,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAE1D;;;OAGG;IACH,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,WAAW,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAEzD;;;;;;;;;;;;;;;OAeG;IACH,SAAS,CAAC,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAEvC;;OAEG;IACH,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAEjD;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;CACtC;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED;;;GAGG;AACH,MAAM,MAAM,WAAW,GAAG,cAAc,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;AAE/D,OAAO,KAAK,EAAE,UAAU,IAAI,gBAAgB,EAAE,MAAM,IAAI,CAAC;AAEzD;;;GAGG;AACH,MAAM,MAAM,UAAU,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC;AAE/C;;;;GAIG;AACH,MAAM,MAAM,WAAW,GAAG,UAAU,GAAG;IACrC,QAAQ,CAAC,KAAK,EAAE,UAAU,EAAE,CAAC;IAC7B,QAAQ,CAAC,UAAU,EAAE,kBAAkB,CAAC;CACzC,CAAC;AAEF,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,IAAI,CAAC;AAE9C;;;;;;;;;GASG;AACH,MAAM,MAAM,gBAAgB,CAC1B,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IACvD;IACF,+DAA+D;IAC/D,QAAQ,CAAC,cAAc,EAAE,OAAO,MAAM,CAAC;IAEvC,4BAA4B;IAC5B,UAAU,EAAE,mBAAmB,CAAC,MAAM,CAAC,CAAC;IAExC,oDAAoD;IACpD,UAAU,EAAE,mBAAmB,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC;IAEvD,kDAAkD;IAClD,QAAQ,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;CAChC,CAAC"}

package/dist/ai-service/llm/interaction/provider.js

@@ -0,0 +1,15 @@
+/**
+ * High-level provider interface for LLM streaming interactions.
+ *
+ * This interface defines a complete, SDK-agnostic contract for multi-step
+ * LLM streaming, including all lifecycle hooks. Concrete implementations
+ * (adapters) bridge this interface to specific SDKs (Vercel AI, Anthropic, etc.).
+ *
+ * Key features:
+ * - Complete lifecycle: prepareStep → onChunk → onStepFinish → onFinish → onFinally
+ * - onFinally: Guaranteed to run after stream consumption (our addition!)
+ * - SDK-agnostic: Could be implemented with any LLM SDK
+ * - Testable: Easy to mock for testing
+ */
+export {};
+//# sourceMappingURL=provider.js.map

package/dist/ai-service/llm/interaction/provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.js","sourceRoot":"","sources":["../../../../src/ai-service/llm/interaction/provider.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG"}

package/dist/ai-service/llm/interaction/result-utils.d.ts

@@ -0,0 +1,43 @@
+/**
+ * Utilities for creating StreamTextResult objects.
+ *
+ * The Vercel AI SDK's StreamTextResult has non-enumerable getters that are lost
+ * when using spread operators. This utility extracts only the properties we need
+ * into plain enumerable values, allowing spread operators to work correctly.
+ */
+import type { StreamTextResult } from "./provider.js";
+import type { Tool } from "../../agent/tools2/types.js";
+import type { StreamTextResult as VercelStreamTextResult } from "ai";
+/**
+ * Creates our StreamTextResult from a Vercel SDK result.
+ *
+ * Extracts only the properties we actually use (textStream, fullStream, response)
+ * and creates a plain object with enumerable properties. This allows:
+ * 1. Spread operators to work correctly in middlewares
+ * 2. Type safety via the brand (can't return raw SDK results)
+ * 3. Simpler API - we only expose what we need
+ *
+ * **When to use:**
+ * - In adapters: Always call this on raw SDK results before returning
+ * - In middlewares: Never needed - just use spread operators
+ *
+ * @param raw - Raw StreamTextResult from Vercel AI SDK
+ * @returns Our branded StreamTextResult with enumerable properties
+ *
+ * @example
+ * ```typescript
+ * // In Vercel adapter:
+ * async streamText(options) {
+ *   const result = await streamText(...); // Vercel SDK
+ *   return createStreamResult(result); // Extract + brand
+ * }
+ *
+ * // In middleware (no helper needed!):
+ * return {
+ *   ...result, // Spread works because properties are enumerable
+ *   textStream: wrappedStream
+ * };
+ * ```
+ */
+export declare function createStreamResult<TOOLS extends Record<string, Tool>>(raw: VercelStreamTextResult<TOOLS, any>): StreamTextResult<TOOLS>;
+//# sourceMappingURL=result-utils.d.ts.map

package/dist/ai-service/llm/interaction/result-utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"result-utils.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/llm/interaction/result-utils.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACtD,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,6BAA6B,CAAC;AACxD,OAAO,KAAK,EAAE,gBAAgB,IAAI,sBAAsB,EAAE,MAAM,IAAI,CAAC;AAErE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,wBAAgB,kBAAkB,CAAC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,EACnE,GAAG,EAAE,sBAAsB,CAAC,KAAK,EAAE,GAAG,CAAC,GACtC,gBAAgB,CAAC,KAAK,CAAC,CASzB"}

package/dist/ai-service/llm/interaction/result-utils.js

@@ -0,0 +1,49 @@
+/**
+ * Utilities for creating StreamTextResult objects.
+ *
+ * The Vercel AI SDK's StreamTextResult has non-enumerable getters that are lost
+ * when using spread operators. This utility extracts only the properties we need
+ * into plain enumerable values, allowing spread operators to work correctly.
+ */
+/**
+ * Creates our StreamTextResult from a Vercel SDK result.
+ *
+ * Extracts only the properties we actually use (textStream, fullStream, response)
+ * and creates a plain object with enumerable properties. This allows:
+ * 1. Spread operators to work correctly in middlewares
+ * 2. Type safety via the brand (can't return raw SDK results)
+ * 3. Simpler API - we only expose what we need
+ *
+ * **When to use:**
+ * - In adapters: Always call this on raw SDK results before returning
+ * - In middlewares: Never needed - just use spread operators
+ *
+ * @param raw - Raw StreamTextResult from Vercel AI SDK
+ * @returns Our branded StreamTextResult with enumerable properties
+ *
+ * @example
+ * ```typescript
+ * // In Vercel adapter:
+ * async streamText(options) {
+ *   const result = await streamText(...); // Vercel SDK
+ *   return createStreamResult(result); // Extract + brand
+ * }
+ *
+ * // In middleware (no helper needed!):
+ * return {
+ *   ...result, // Spread works because properties are enumerable
+ *   textStream: wrappedStream
+ * };
+ * ```
+ */
+export function createStreamResult(raw) {
+    // Extract only the properties we use
+    // Access them directly to get the getter values
+    // Cast via unknown to satisfy the brand requirement
+    return {
+        textStream: raw.textStream,
+        fullStream: raw.fullStream,
+        response: raw.response,
+    };
+}
+//# sourceMappingURL=result-utils.js.map

package/dist/ai-service/llm/interaction/result-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"result-utils.js","sourceRoot":"","sources":["../../../../src/ai-service/llm/interaction/result-utils.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAMH;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,MAAM,UAAU,kBAAkB,CAChC,GAAuC;IAEvC,qCAAqC;IACrC,gDAAgD;IAChD,oDAAoD;IACpD,OAAO;QACL,UAAU,EAAE,GAAG,CAAC,UAAU;QAC1B,UAAU,EAAE,GAAG,CAAC,UAAU;QAC1B,QAAQ,EAAE,GAAG,CAAC,QAAQ;KACe,CAAC;AAC1C,CAAC"}

package/dist/ai-service/llm/interaction/stream-lifecycle.d.ts

@@ -0,0 +1,48 @@
+/**
+ * Stream lifecycle management utilities.
+ *
+ * Provides utilities for wrapping streams with lifecycle hooks, particularly
+ * `onFinally` which runs after stream consumption completes.
+ *
+ * Key challenge: The `streamText()` function returns immediately, but the
+ * streams are consumed asynchronously later. We need to track when consumption
+ * actually completes to run cleanup logic.
+ *
+ * Solution: Wrap streams with `TransformStream` that tracks consumption via
+ * the `flush()` method, which runs when the input stream closes.
+ */
+import type { StreamTextResult } from "./provider.js";
+import type { Tool } from "../../agent/tools2/types.js";
+/**
+ * Wraps a StreamTextResult to add onFinally lifecycle hook.
+ *
+ * The onFinally callback runs after stream consumption completes, handling:
+ * - Normal completion (stream exhausted)
+ * - Errors during consumption
+ * - Stream cancellation/abandonment
+ *
+ * Guarantees:
+ * - Runs exactly once, even if multiple stream accessors are used
+ * - Idempotent (safe to call multiple times)
+ * - Works with all consumption patterns (iteration, promises, piping)
+ *
+ * @param result - The stream result to wrap
+ * @param onFinally - Callback to run after consumption
+ * @returns Wrapped result with lifecycle tracking
+ *
+ * @example
+ * ```typescript
+ * const result = await streamText({ model, messages });
+ * const wrapped = withStreamLifecycle(result, async () => {
+ *   console.log('Stream consumed!');
+ *   await cleanup();
+ * });
+ *
+ * for await (const text of wrapped.textStream) {
+ *   console.log(text);
+ * }
+ * // onFinally runs here ^
+ * ```
+ */
+export declare function withStreamLifecycle<TOOLS extends Record<string, Tool> = Record<string, Tool>>(result: StreamTextResult<TOOLS>, onFinally: () => Promise<void> | void): StreamTextResult<TOOLS>;
+//# sourceMappingURL=stream-lifecycle.d.ts.map

package/dist/ai-service/llm/interaction/stream-lifecycle.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream-lifecycle.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/llm/interaction/stream-lifecycle.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAEH,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACtD,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,6BAA6B,CAAC;AAGxD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,wBAAgB,mBAAmB,CACjC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,EAEzD,MAAM,EAAE,gBAAgB,CAAC,KAAK,CAAC,EAC/B,SAAS,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,GACpC,gBAAgB,CAAC,KAAK,CAAC,CAiCzB"}

package/dist/ai-service/llm/interaction/stream-lifecycle.js

@@ -0,0 +1,129 @@
+/**
+ * Stream lifecycle management utilities.
+ *
+ * Provides utilities for wrapping streams with lifecycle hooks, particularly
+ * `onFinally` which runs after stream consumption completes.
+ *
+ * Key challenge: The `streamText()` function returns immediately, but the
+ * streams are consumed asynchronously later. We need to track when consumption
+ * actually completes to run cleanup logic.
+ *
+ * Solution: Wrap streams with `TransformStream` that tracks consumption via
+ * the `flush()` method, which runs when the input stream closes.
+ */
+/**
+ * Wraps a StreamTextResult to add onFinally lifecycle hook.
+ *
+ * The onFinally callback runs after stream consumption completes, handling:
+ * - Normal completion (stream exhausted)
+ * - Errors during consumption
+ * - Stream cancellation/abandonment
+ *
+ * Guarantees:
+ * - Runs exactly once, even if multiple stream accessors are used
+ * - Idempotent (safe to call multiple times)
+ * - Works with all consumption patterns (iteration, promises, piping)
+ *
+ * @param result - The stream result to wrap
+ * @param onFinally - Callback to run after consumption
+ * @returns Wrapped result with lifecycle tracking
+ *
+ * @example
+ * ```typescript
+ * const result = await streamText({ model, messages });
+ * const wrapped = withStreamLifecycle(result, async () => {
+ *   console.log('Stream consumed!');
+ *   await cleanup();
+ * });
+ *
+ * for await (const text of wrapped.textStream) {
+ *   console.log(text);
+ * }
+ * // onFinally runs here ^
+ * ```
+ */
+export function withStreamLifecycle(result, onFinally) {
+    // Track whether finalization has run
+    let finalized = false;
+    /**
+     * Ensures onFinally runs exactly once.
+     * Safe to call multiple times (idempotent).
+     */
+    const ensureFinally = async () => {
+        if (finalized)
+            return;
+        finalized = true;
+        try {
+            await onFinally();
+        }
+        catch (error) {
+            // Log but don't throw - we're in cleanup, don't break the stream
+            console.error("[streamLifecycle] Error in onFinally:", error);
+        }
+    };
+    // Wrap textStream to track consumption
+    const wrappedTextStream = wrapAsyncIterable(result.textStream, ensureFinally);
+    // Wrap fullStream to track consumption
+    const wrappedFullStream = wrapAsyncIterable(result.fullStream, ensureFinally);
+    // Return result with wrapped streams
+    // Spread works because adapter materialized getters into enumerable properties
+    return {
+        ...result,
+        textStream: wrappedTextStream,
+        fullStream: wrappedFullStream,
+    };
+}
+/**
+ * Wraps an async iterable with a finalization callback.
+ *
+ * Uses TransformStream to intercept the stream lifecycle:
+ * - `transform()`: Passes chunks through unchanged
+ * - `flush()`: Called when input stream closes (after all chunks processed)
+ * - `cancel()`: Called if stream is cancelled/abandoned
+ *
+ * The finalization callback runs in both flush() and cancel() to ensure
+ * cleanup happens whether the stream completes normally or is abandoned.
+ *
+ * @param source - The async iterable to wrap
+ * @param onFinalize - Callback to run after consumption
+ * @returns Wrapped async iterable
+ */
+function wrapAsyncIterable(source, onFinalize) {
+    // Convert async iterable to ReadableStream if needed
+    let sourceStream;
+    if (source instanceof ReadableStream) {
+        sourceStream = source;
+    }
+    else {
+        // Manually convert async iterable to ReadableStream
+        sourceStream = new ReadableStream({
+            async start(controller) {
+                try {
+                    for await (const chunk of source) {
+                        controller.enqueue(chunk);
+                    }
+                    controller.close();
+                }
+                catch (error) {
+                    controller.error(error);
+                }
+            },
+        });
+    }
+    // Create transform stream that tracks lifecycle
+    const lifecycleTransform = new TransformStream({
+        // Pass through all chunks unchanged
+        transform(chunk, controller) {
+            controller.enqueue(chunk);
+        },
+        // Called when input stream closes (after all chunks processed)
+        async flush() {
+            await onFinalize();
+        },
+    });
+    // Pipe source through lifecycle tracker
+    const wrappedStream = sourceStream.pipeThrough(lifecycleTransform);
+    // Return as AsyncIterableStream (has both stream and iterable interfaces)
+    return wrappedStream;
+}
+//# sourceMappingURL=stream-lifecycle.js.map

package/dist/ai-service/llm/interaction/stream-lifecycle.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream-lifecycle.js","sourceRoot":"","sources":["../../../../src/ai-service/llm/interaction/stream-lifecycle.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAMH;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,MAAM,UAAU,mBAAmB,CAGjC,MAA+B,EAC/B,SAAqC;IAErC,qCAAqC;IACrC,IAAI,SAAS,GAAG,KAAK,CAAC;IAEtB;;;OAGG;IACH,MAAM,aAAa,GAAG,KAAK,IAAI,EAAE;QAC/B,IAAI,SAAS;YAAE,OAAO;QACtB,SAAS,GAAG,IAAI,CAAC;QAEjB,IAAI,CAAC;YACH,MAAM,SAAS,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,iEAAiE;YACjE,OAAO,CAAC,KAAK,CAAC,uCAAuC,EAAE,KAAK,CAAC,CAAC;QAChE,CAAC;IACH,CAAC,CAAC;IAEF,uCAAuC;IACvC,MAAM,iBAAiB,GAAG,iBAAiB,CAAC,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;IAE9E,uCAAuC;IACvC,MAAM,iBAAiB,GAAG,iBAAiB,CAAC,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;IAE9E,qCAAqC;IACrC,+EAA+E;IAC/E,OAAO;QACL,GAAG,MAAM;QACT,UAAU,EAAE,iBAAiB;QAC7B,UAAU,EAAE,iBAAiB;KAC9B,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;GAcG;AACH,SAAS,iBAAiB,CACxB,MAAwB,EACxB,UAA+B;IAE/B,qDAAqD;IACrD,IAAI,YAA+B,CAAC;IACpC,IAAI,MAAM,YAAY,cAAc,EAAE,CAAC;QACrC,YAAY,GAAG,MAAM,CAAC;IACxB,CAAC;SAAM,CAAC;QACN,oDAAoD;QACpD,YAAY,GAAG,IAAI,cAAc,CAAI;YACnC,KAAK,CAAC,KAAK,CAAC,UAAU;gBACpB,IAAI,CAAC;oBACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;wBACjC,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;oBAC5B,CAAC;oBACD,UAAU,CAAC,KAAK,EAAE,CAAC;gBACrB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBAC1B,CAAC;YACH,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAED,gDAAgD;IAChD,MAAM,kBAAkB,GAAG,IAAI,eAAe,CAAO;QACnD,oCAAoC;QACpC,SAAS,CAAC,KAAK,EAAE,UAAU;YACzB,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC5B,CAAC;QAED,+DAA+D;QAC/D,KAAK,CAAC,KAAK;YACT,MAAM,UAAU,EAAE,CAAC;QACrB,CAAC;KACF,CAAC,CAAC;IAEH,wCAAwC;IACxC,MAAM,aAAa,GAAG,YAAY,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAC;IAEnE,0EAA0E;IAC1E,OAAO,aAAuC,CAAC;AACjD,CAAC"}

package/dist/ai-service/llm/provider.d.ts

@@ -1,5 +1,4 @@
-import { type RetryOptions } from "./middleware/retry.js";
 import type { LLMProvider, LLMProviderSettings } from "./types.js";
 import type { LLMProviderConfig } from "@superblocksteam/library-shared/types";
-export declare const createLLMProvider: (settings: LLMProviderSettings, getJwt?: () => string | undefined, llmConfig?: LLMProviderConfig, retryOptions?: RetryOptions) => LLMProvider;
+export declare const createLLMProvider: (settings: LLMProviderSettings, getJwt?: () => string | undefined, llmConfig?: LLMProviderConfig) => LLMProvider;
 //# sourceMappingURL=provider.d.ts.map

package/dist/ai-service/llm/provider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"provider.d.ts","sourceRoot":"","sources":["../../../src/ai-service/llm/provider.ts"],"names":[],"mappings":"AAIA,OAAO,EAAmB,KAAK,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAE3E,OAAO,KAAK,EACV,WAAW,EACX,mBAAmB,EAGpB,MAAM,YAAY,CAAC;AAEpB,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,uCAAuC,CAAC;AAE/E,eAAO,MAAM,iBAAiB,GAC5B,UAAU,mBAAmB,EAC7B,SAAS,MAAM,MAAM,GAAG,SAAS,EACjC,YAAY,iBAAiB,EAC7B,eAAe,YAAY,KAC1B,WA6DF,CAAC"}
+{"version":3,"file":"provider.d.ts","sourceRoot":"","sources":["../../../src/ai-service/llm/provider.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EACV,WAAW,EACX,mBAAmB,EAGpB,MAAM,YAAY,CAAC;AAEpB,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,uCAAuC,CAAC;AAE/E,eAAO,MAAM,iBAAiB,GAC5B,UAAU,mBAAmB,EAC7B,SAAS,MAAM,MAAM,GAAG,SAAS,EACjC,YAAY,iBAAiB,KAC5B,WAuDF,CAAC"}

package/dist/ai-service/llm/provider.js

@@ -2,18 +2,15 @@ import { customProvider, wrapLanguageModel } from "ai";
 import { getLogger } from "../../util/logger.js";
 import { anthropicImpl } from "./impl/anthropic.js";
 import { clarkImpl } from "./impl/clark.js";
-import { retryMiddleware } from "./middleware/retry.js";
 import { jwtMiddleware, modelFromClassification } from "./utils.js";
-export const createLLMProvider = (settings, getJwt, llmConfig, retryOptions) => {
+export const createLLMProvider = (settings, getJwt, llmConfig) => {
     const providerType = llmConfig?.provider || "bedrock";
     const wrapDefaults = (model) => wrapLanguageModel({
         model,
         middleware: [
             jwtMiddleware(getJwt),
-            retryMiddleware(providerType, {
-                maxTotalRetryTimeMs: 2 * 60 * 1000, // 2 minutes
-                ...retryOptions,
-            }, getLogger()),
+            // Note: Retry is now handled at the StreamText level via createStreamRetryMiddleware
+            // to ensure fresh Vercel SDK state and correct message sequences on retry
         ],
     });
     getLogger().info(`Creating LLM provider: type=${providerType}, config=${JSON.stringify(llmConfig || {})}`);

package/dist/ai-service/llm/provider.js.map

@@ -1 +1 @@
-{"version":3,"file":"provider.js","sourceRoot":"","sources":["../../../src/ai-service/llm/provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,IAAI,CAAC;AACvD,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,eAAe,EAAqB,MAAM,uBAAuB,CAAC;AAC3E,OAAO,EAAE,aAAa,EAAE,uBAAuB,EAAE,MAAM,YAAY,CAAC;AAUpE,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAC/B,QAA6B,EAC7B,MAAiC,EACjC,SAA6B,EAC7B,YAA2B,EACd,EAAE;IACf,MAAM,YAAY,GAAG,SAAS,EAAE,QAAQ,IAAI,SAAS,CAAC;IAEtD,MAAM,YAAY,GAAG,CAAC,KAAsB,EAAE,EAAE,CAC9C,iBAAiB,CAAC;QAChB,KAAK;QACL,UAAU,EAAE;YACV,aAAa,CAAC,MAAM,CAAC;YACrB,eAAe,CACb,YAAY,EACZ;gBACE,mBAAmB,EAAE,CAAC,GAAG,EAAE,GAAG,IAAI,EAAE,YAAY;gBAChD,GAAG,YAAY;aAChB,EACD,SAAS,EAAE,CACZ;SACF;KACF,CAAC,CAAC;IACL,SAAS,EAAE,CAAC,IAAI,CACd,+BAA+B,YAAY,YAAY,IAAI,CAAC,SAAS,CAAC,SAAS,IAAI,EAAE,CAAC,EAAE,CACzF,CAAC;IACF,IAAI,QAAQ,CAAC;IACb,QAAQ,YAAY,EAAE,CAAC;QACrB,KAAK,SAAS;YACZ,QAAQ,GAAG,SAAS,CAAC;YACrB,MAAM;QACR,KAAK,QAAQ;YACX,QAAQ,GAAG,SAAS,CAAC;YACrB,MAAM;QACR,KAAK,WAAW,CAAC;QACjB;YACE,QAAQ,GAAG,aAAa,CAAC;YACzB,MAAM;IACV,CAAC;IAED,MAAM,IAAI,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAEhC,MAAM,cAAc,GAAkB;QACpC,cAAc,EAAE,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC;QACjD,aAAa,EAAE,YAAY,CAAC,IAAI,CAAC,aAAa,CAAC;QAC/C,OAAO,EAAE,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC;QACnC,oBAAoB,EAAE,YAAY,CAAC,IAAI,CAAC,oBAAoB,CAAC;QAC7D,iBAAiB,EAAE,YAAY,CAAC,IAAI,CAAC,iBAAiB,CAAC;QACvD,kBAAkB,EAAE,YAAY,CAAC,IAAI,CAAC,kBAAkB,CAAC;QACzD,iBAAiB,EAAE,YAAY,CAAC,IAAI,CAAC,iBAAiB,CAAC;KACxD,CAAC;IAEF,MAAM,WAAW,GAAG,cAAc,CAAC;QACjC,cAAc;KACf,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,CAAC,OAAgB,EAAE,EAAE;QACxC,OAAO,WAAW,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;IAC5C,CAAC,CAAC;IAEF,OAAO;QACL,IAAI,EAAE,YAAY;QAClB,YAAY;QACZ,sBAAsB,EAAE,CAAC,cAAc,EAAE,EAAE,CACzC,uBAAuB,CAAC,YAAY,EAAE,cAAc,CAAC;KACxD,CAAC;AACJ,CAAC,CAAC"}
+{"version":3,"file":"provider.js","sourceRoot":"","sources":["../../../src/ai-service/llm/provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,IAAI,CAAC;AACvD,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,uBAAuB,EAAE,MAAM,YAAY,CAAC;AAUpE,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAC/B,QAA6B,EAC7B,MAAiC,EACjC,SAA6B,EAChB,EAAE;IACf,MAAM,YAAY,GAAG,SAAS,EAAE,QAAQ,IAAI,SAAS,CAAC;IAEtD,MAAM,YAAY,GAAG,CAAC,KAAsB,EAAE,EAAE,CAC9C,iBAAiB,CAAC;QAChB,KAAK;QACL,UAAU,EAAE;YACV,aAAa,CAAC,MAAM,CAAC;YACrB,qFAAqF;YACrF,0EAA0E;SAC3E;KACF,CAAC,CAAC;IACL,SAAS,EAAE,CAAC,IAAI,CACd,+BAA+B,YAAY,YAAY,IAAI,CAAC,SAAS,CAAC,SAAS,IAAI,EAAE,CAAC,EAAE,CACzF,CAAC;IACF,IAAI,QAAQ,CAAC;IACb,QAAQ,YAAY,EAAE,CAAC;QACrB,KAAK,SAAS;YACZ,QAAQ,GAAG,SAAS,CAAC;YACrB,MAAM;QACR,KAAK,QAAQ;YACX,QAAQ,GAAG,SAAS,CAAC;YACrB,MAAM;QACR,KAAK,WAAW,CAAC;QACjB;YACE,QAAQ,GAAG,aAAa,CAAC;YACzB,MAAM;IACV,CAAC;IAED,MAAM,IAAI,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAEhC,MAAM,cAAc,GAAkB;QACpC,cAAc,EAAE,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC;QACjD,aAAa,EAAE,YAAY,CAAC,IAAI,CAAC,aAAa,CAAC;QAC/C,OAAO,EAAE,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC;QACnC,oBAAoB,EAAE,YAAY,CAAC,IAAI,CAAC,oBAAoB,CAAC;QAC7D,iBAAiB,EAAE,YAAY,CAAC,IAAI,CAAC,iBAAiB,CAAC;QACvD,kBAAkB,EAAE,YAAY,CAAC,IAAI,CAAC,kBAAkB,CAAC;QACzD,iBAAiB,EAAE,YAAY,CAAC,IAAI,CAAC,iBAAiB,CAAC;KACxD,CAAC;IAEF,MAAM,WAAW,GAAG,cAAc,CAAC;QACjC,cAAc;KACf,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,CAAC,OAAgB,EAAE,EAAE;QACxC,OAAO,WAAW,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;IAC5C,CAAC,CAAC;IAEF,OAAO;QACL,IAAI,EAAE,YAAY;QAClB,YAAY;QACZ,sBAAsB,EAAE,CAAC,cAAc,EAAE,EAAE,CACzC,uBAAuB,CAAC,YAAY,EAAE,cAAc,CAAC;KACxD,CAAC;AACJ,CAAC,CAAC"}

package/dist/ai-service/llmobs/helpers.d.ts

@@ -1,7 +1,8 @@
-import { streamText, type ToolSet } from "ai";
 import { type Logger } from "../../util/logger.js";
+import { type StreamTextOptions } from "../llm/interaction/index.js";
 import { LLMObsTracer } from "./tracer.js";
 import type { LLMObsSpan } from "./types.js";
+import type { Tool } from "../agent/tools2/types.js";
 export declare function traceWorkflow<T>(name: string, operation: (span: any) => Promise<T>, tags?: Record<string, string | boolean>, llmobs?: LLMObsTracer): Promise<T>;
 export declare function traceTask<T>(name: string, operation: (span: any) => Promise<T>, tags?: Record<string, string | boolean>, llmobs?: LLMObsTracer): Promise<T>;
 export declare function traceLLM<T>(name: string, modelName: string, modelProvider: string, operation: (span: LLMObsSpan) => Promise<T>, tags?: Record<string, string | boolean>, llmobs?: LLMObsTracer): Promise<T>;
@@ -16,13 +17,11 @@ export declare function getCommonTraceTags(jwt: string | undefined, applicationI
  * and chaos monkey error injection is handled at the fetch level. This function only applies
  * LLMObs tracing.
  *
- * @param clark - Clark state machine instance
- * @param services - Clark state handler parameters
  * @param params - Parameters to pass to streamText
+ * @param tracer - Optional LLMObs tracer
+ * @param logger - Optional logger
+ *
+ * @deprecated use the LLMClient with LLMObs middleware instead (Clark.streamText in most cases)
  */
-export declare function tracedStreamText<T extends ToolSet>(params: Parameters<typeof streamText<T>>[0], tracer?: LLMObsTracer, logger?: Logger): import("ai").StreamTextResult<T, never>;
-export declare const traceToolSet: <T extends ToolSet>(toolset: T, llmobs?: LLMObsTracer, hooks?: {
-    beforeExecute?: (name: string) => void;
-    afterExecute?: (name: string, span: LLMObsSpan) => void;
-}) => T;
+export declare function tracedStreamText<T extends Record<string, Tool>>(params: StreamTextOptions<T>, tracer?: LLMObsTracer, logger?: Logger): Promise<import("../llm/interaction/provider.js").StreamTextResult<T>>;
 //# sourceMappingURL=helpers.d.ts.map

package/dist/ai-service/llmobs/helpers.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"helpers.d.ts","sourceRoot":"","sources":["../../../src/ai-service/llmobs/helpers.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,KAAK,OAAO,EAAE,MAAM,IAAI,CAAC;AAC9C,OAAO,EAAE,KAAK,MAAM,EAAa,MAAM,sBAAsB,CAAC;AAI9D,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAG7C,wBAAgB,aAAa,CAAC,CAAC,EAC7B,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,CAAC,EACpC,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,EACvC,MAAM,GAAE,YAAoC,GAC3C,OAAO,CAAC,CAAC,CAAC,CAaZ;AAGD,wBAAgB,SAAS,CAAC,CAAC,EACzB,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,CAAC,EACpC,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,EACvC,MAAM,GAAE,YAAoC,GAC3C,OAAO,CAAC,CAAC,CAAC,CAaZ;AAGD,wBAAgB,QAAQ,CAAC,CAAC,EACxB,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,MAAM,EACjB,aAAa,EAAE,MAAM,EACrB,SAAS,EAAE,CAAC,IAAI,EAAE,UAAU,KAAK,OAAO,CAAC,CAAC,CAAC,EAC3C,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,EACvC,MAAM,GAAE,YAAoC,GAC3C,OAAO,CAAC,CAAC,CAAC,CAeZ;AAGD,wBAAgB,eAAe,CAC7B,GAAG,EAAE,MAAM,GAAG,SAAS,GACtB,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAUxB;AAGD,wBAAgB,kBAAkB,CAChC,GAAG,EAAE,MAAM,GAAG,SAAS,EACvB,aAAa,EAAE,MAAM,EACrB,cAAc,EAAE,MAAM,EACtB,cAAc,GAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAM,GACpD,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,CAclC;AAED;;;;;;;;;;;;GAYG;AACH,wBAAgB,gBAAgB,CAAC,CAAC,SAAS,OAAO,EAChD,MAAM,EAAE,UAAU,CAAC,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAC3C,MAAM,GAAE,YAAoC,EAC5C,MAAM,GAAE,MAAoB,2CAa7B;AAED,eAAO,MAAM,YAAY,GAAI,CAAC,SAAS,OAAO,EAC5C,SAAS,CAAC,EACV,SAAQ,YAAoC,EAC5C,QAAQ;IACN,aAAa,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;IACvC,YAAY,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,KAAK,IAAI,CAAC;CACzD,KACA,CAyCF,CAAC"}
+{"version":3,"file":"helpers.d.ts","sourceRoot":"","sources":["../../../src/ai-service/llmobs/helpers.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,MAAM,EAAa,MAAM,sBAAsB,CAAC;AAE9D,OAAO,EAGL,KAAK,iBAAiB,EACvB,MAAM,6BAA6B,CAAC;AAErC,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,0BAA0B,CAAC;AAGrD,wBAAgB,aAAa,CAAC,CAAC,EAC7B,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,CAAC,EACpC,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,EACvC,MAAM,GAAE,YAAoC,GAC3C,OAAO,CAAC,CAAC,CAAC,CAaZ;AAGD,wBAAgB,SAAS,CAAC,CAAC,EACzB,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,CAAC,EACpC,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,EACvC,MAAM,GAAE,YAAoC,GAC3C,OAAO,CAAC,CAAC,CAAC,CAaZ;AAGD,wBAAgB,QAAQ,CAAC,CAAC,EACxB,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,MAAM,EACjB,aAAa,EAAE,MAAM,EACrB,SAAS,EAAE,CAAC,IAAI,EAAE,UAAU,KAAK,OAAO,CAAC,CAAC,CAAC,EAC3C,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,EACvC,MAAM,GAAE,YAAoC,GAC3C,OAAO,CAAC,CAAC,CAAC,CAeZ;AAGD,wBAAgB,eAAe,CAC7B,GAAG,EAAE,MAAM,GAAG,SAAS,GACtB,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAUxB;AAGD,wBAAgB,kBAAkB,CAChC,GAAG,EAAE,MAAM,GAAG,SAAS,EACvB,aAAa,EAAE,MAAM,EACrB,cAAc,EAAE,MAAM,EACtB,cAAc,GAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAM,GACpD,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,CAclC;AAED;;;;;;;;;;;;;;GAcG;AACH,wBAAgB,gBAAgB,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,EAC7D,MAAM,EAAE,iBAAiB,CAAC,CAAC,CAAC,EAC5B,MAAM,GAAE,YAAoC,EAC5C,MAAM,GAAE,MAAoB,yEAQ7B"}

package/dist/ai-service/llmobs/helpers.js

@@ -1,8 +1,7 @@
-import { streamText } from "ai";
 import { getLogger } from "../../util/logger.js";
+import { applyMiddleware } from "../llm/interaction/compose.js";
+import { createLLMObsMiddleware, VercelStreamTextAdapter, } from "../llm/interaction/index.js";
 import { parseJwt } from "../util/parse-jwt.js";
-import { safeJsonStringify } from "../util/safe-stringify.js";
-import { LLMObsStreamTextMiddleware } from "./middleware/stream-text.js";
 import { LLMObsTracer } from "./tracer.js";
 // Helper function for workflow tracing with common patterns
 export function traceWorkflow(name, operation, tags, llmobs = LLMObsTracer.instance) {
@@ -77,54 +76,15 @@ export function getCommonTraceTags(jwt, applicationId, organizationId, additiona
  * and chaos monkey error injection is handled at the fetch level. This function only applies
  * LLMObs tracing.
  *
- * @param clark - Clark state machine instance
- * @param services - Clark state handler parameters
  * @param params - Parameters to pass to streamText
+ * @param tracer - Optional LLMObs tracer
+ * @param logger - Optional logger
+ *
+ * @deprecated use the LLMClient with LLMObs middleware instead (Clark.streamText in most cases)
  */
 export function tracedStreamText(params, tracer = LLMObsTracer.instance, logger = getLogger()) {
-    // Disable built-in retries - we handle retries at the LanguageModelV2 middleware level
-    params = {
-        ...params,
-        maxRetries: 0,
-    };
     // Apply LLMObs tracing middleware
-    const llmobsMiddleware = new LLMObsStreamTextMiddleware(tracer, logger);
-    const tracedStreamText = llmobsMiddleware.wrap(streamText);
-    return tracedStreamText(params);
+    const traced = applyMiddleware(new VercelStreamTextAdapter(), createLLMObsMiddleware({ tracer, logger }));
+    return traced.streamText(params);
 }
-export const traceToolSet = (toolset, llmobs = LLMObsTracer.instance, hooks) => {
-    const tracedToolset = {};
-    for (const [name, tool] of Object.entries(toolset)) {
-        tracedToolset[name] = {
-            ...tool,
-            execute: async (input, options) => {
-                hooks?.beforeExecute?.(name);
-                return await llmobs.trace({
-                    kind: "tool",
-                    name: `tool.${name}`,
-                }, async (span) => {
-                    if (!tool.execute) {
-                        throw new Error(`Tool ${name} has no execute method`);
-                    }
-                    const result = await tool.execute(input, options);
-                    hooks?.afterExecute?.(name, span);
-                    const metadata = {
-                        toolCallid: options?.toolCallId,
-                        ...Object.fromEntries(Object.entries(tool).map(([key, value]) => [
-                            key,
-                            safeJsonStringify(value),
-                        ])),
-                    };
-                    llmobs.annotate(span, {
-                        inputData: safeJsonStringify(input),
-                        outputData: safeJsonStringify(result),
-                        metadata,
-                    });
-                    return result;
-                });
-            },
-        };
-    }
-    return tracedToolset;
-};
 //# sourceMappingURL=helpers.js.map