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

package/dist/ai-service/llm/stream/event-bus.js

@@ -0,0 +1,51 @@
+/**
+ * Type-safe event bus for stream lifecycle events.
+ *
+ * Observers subscribe to specific events and receive typed arguments.
+ * Handlers are called in parallel, and errors in one handler don't affect others.
+ *
+ * @example
+ * ```typescript
+ * const events = new StreamEventBus();
+ *
+ * // Subscribe to events
+ * const unsubscribe = events.on("chunk", (chunk, session) => {
+ *   console.log(`Received chunk: ${chunk.type}`);
+ * });
+ *
+ * // Emit events
+ * await events.emit("chunk", myChunk, mySession);
+ *
+ * // Unsubscribe when done
+ * unsubscribe();
+ * ```
+ */
+export class StreamEventBus {
+    listeners = new Map();
+    on(event, handler) {
+        let handlers = this.listeners.get(event);
+        if (!handlers) {
+            handlers = new Set();
+            this.listeners.set(event, handlers);
+        }
+        handlers.add(handler);
+        return () => {
+            handlers?.delete(handler);
+        };
+    }
+    async emit(event, ...args) {
+        const handlers = this.listeners.get(event);
+        if (!handlers || handlers.size === 0)
+            return;
+        const results = await Promise.allSettled([...handlers].map(async (handler) => handler(...args)));
+        for (const result of results) {
+            if (result.status === "rejected") {
+                console.error(`Observer error on "${event}" event:`, result.reason);
+            }
+        }
+    }
+    clear() {
+        this.listeners.clear();
+    }
+}
+//# sourceMappingURL=event-bus.js.map

package/dist/ai-service/llm/stream/event-bus.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"event-bus.js","sourceRoot":"","sources":["../../../../src/ai-service/llm/stream/event-bus.ts"],"names":[],"mappings":"AAiDA;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,OAAO,cAAc;IACR,SAAS,GAAG,IAAI,GAAG,EAGjC,CAAC;IAEJ,EAAE,CACA,KAAQ,EACR,OAA8B;QAE9B,IAAI,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACzC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,QAAQ,GAAG,IAAI,GAAG,EAAE,CAAC;YACrB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QACtC,CAAC;QACD,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAEtB,OAAO,GAAG,EAAE;YACV,QAAQ,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QAC5B,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,IAAI,CACR,KAAQ,EACR,GAAG,IAAqB;QAExB,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC;YAAE,OAAO;QAE7C,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,CACtC,CAAC,GAAG,QAAQ,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,CAAC,CACvD,CAAC;QACF,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,IAAI,MAAM,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;gBACjC,OAAO,CAAC,KAAK,CAAC,sBAAsB,KAAK,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;YACtE,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK;QACH,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;CACF"}

package/dist/ai-service/llm/stream/index.d.ts

@@ -0,0 +1,49 @@
+/**
+ * Handling LLM streaming operations.
+ *
+ * @example
+ * ```typescript
+ * import {
+ *   StreamOrchestrator,
+ *   StreamConfig,
+ *   LoggingObserver,
+ *   LLMObsObserver,
+ * } from "./stream/index.js";
+ *
+ * // Create configuration
+ * const config = StreamConfig.create({
+ *   retry: { maxAttempts: 3, stallTimeoutMs: 60_000 },
+ *   fallback: { enabled: true, initialProvider: "vertex" },
+ *   logging: { enabled: true, conversationId: "abc" },
+ * });
+ *
+ * // Create orchestrator with observers
+ * const orchestrator = new StreamOrchestrator({
+ *   provider: createVercelProvider(),
+ *   config,
+ *   observers: [
+ *     new LoggingObserver(),
+ *     new LLMObsObserver({ tracer, logger }),
+ *   ],
+ *   logger,
+ * });
+ *
+ * // Execute stream with full lifecycle management
+ * const result = await orchestrator.execute({
+ *   model,
+ *   messages,
+ *   tools,
+ * });
+ *
+ * for await (const chunk of result.fullStream) {
+ *   console.log(chunk);
+ * }
+ * ```
+ */
+export { StreamConfig } from "./config.js";
+export type { RetryConfigOptions, ProviderType, CoreModelType, } from "./config.js";
+export type { StreamObserver } from "./observer.js";
+export { StreamOrchestrator } from "./orchestrator.js";
+export { LoggingObserver, LLMObsObserver, ContextObserver, RetryNotificationObserver, wrapToolsForObservability, } from "./observers/index.js";
+export type { LogRef } from "./observers/index.js";
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/llm/stream/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAyCG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,YAAY,EACV,kBAAkB,EAClB,YAAY,EACZ,aAAa,GACd,MAAM,aAAa,CAAC;AAErB,YAAY,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAEpD,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAEvD,OAAO,EACL,eAAe,EACf,cAAc,EACd,eAAe,EACf,yBAAyB,EACzB,yBAAyB,GAC1B,MAAM,sBAAsB,CAAC;AAC9B,YAAY,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAC"}

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

@@ -0,0 +1,46 @@
+/**
+ * Handling LLM streaming operations.
+ *
+ * @example
+ * ```typescript
+ * import {
+ *   StreamOrchestrator,
+ *   StreamConfig,
+ *   LoggingObserver,
+ *   LLMObsObserver,
+ * } from "./stream/index.js";
+ *
+ * // Create configuration
+ * const config = StreamConfig.create({
+ *   retry: { maxAttempts: 3, stallTimeoutMs: 60_000 },
+ *   fallback: { enabled: true, initialProvider: "vertex" },
+ *   logging: { enabled: true, conversationId: "abc" },
+ * });
+ *
+ * // Create orchestrator with observers
+ * const orchestrator = new StreamOrchestrator({
+ *   provider: createVercelProvider(),
+ *   config,
+ *   observers: [
+ *     new LoggingObserver(),
+ *     new LLMObsObserver({ tracer, logger }),
+ *   ],
+ *   logger,
+ * });
+ *
+ * // Execute stream with full lifecycle management
+ * const result = await orchestrator.execute({
+ *   model,
+ *   messages,
+ *   tools,
+ * });
+ *
+ * for await (const chunk of result.fullStream) {
+ *   console.log(chunk);
+ * }
+ * ```
+ */
+export { StreamConfig } from "./config.js";
+export { StreamOrchestrator } from "./orchestrator.js";
+export { LoggingObserver, LLMObsObserver, ContextObserver, RetryNotificationObserver, wrapToolsForObservability, } from "./observers/index.js";
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/ai-service/llm/stream/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAyCG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAS3C,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAEvD,OAAO,EACL,eAAe,EACf,cAAc,EACd,eAAe,EACf,yBAAyB,EACzB,yBAAyB,GAC1B,MAAM,sBAAsB,CAAC"}

package/dist/ai-service/llm/stream/managed-stream.d.ts

@@ -0,0 +1,25 @@
+import type { StreamSession } from "./session.js";
+import type { Tool } from "../../agent/tools2/types.js";
+import type { AsyncIterableStream, TextStreamPart } from "ai";
+/**
+ * Creates a managed stream that wraps the source stream with lifecycle management.
+ *
+ * The managed stream:
+ * - Emits "chunk" events for each chunk
+ * - Throws errors from error chunks
+ * - Finalizes the session on complete or cancel (but NOT on error - see below)
+ *
+ * IMPORTANT: This stream intentionally does NOT finalize the session on errors.
+ * If you call session.finalize("error") here, the session becomes non-pending,
+ * which means new managed streams created during retry will immediately close
+ * (see the `!session.isPending()` check in pull).
+ *
+ * The retry logic in wrapWithRetry is responsible for finalizing the session
+ * when all retries are exhausted. Don't be a hero and add finalization here.
+ *
+ * @param source - The source async iterable stream
+ * @param session - The session to update with events
+ * @returns A managed stream with lifecycle hooks
+ */
+export declare function createManagedStream<T extends TextStreamPart<Record<string, Tool>>>(source: AsyncIterableStream<T>, session: StreamSession): AsyncIterableStream<T>;
+//# sourceMappingURL=managed-stream.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"managed-stream.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/llm/stream/managed-stream.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAClD,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,6BAA6B,CAAC;AACxD,OAAO,KAAK,EAAE,mBAAmB,EAAE,cAAc,EAAE,MAAM,IAAI,CAAC;AAE9D;;;;;;;;;;;;;;;;;;;GAmBG;AACH,wBAAgB,mBAAmB,CACjC,CAAC,SAAS,cAAc,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,EAE9C,MAAM,EAAE,mBAAmB,CAAC,CAAC,CAAC,EAC9B,OAAO,EAAE,aAAa,GACrB,mBAAmB,CAAC,CAAC,CAAC,CAqDxB"}

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

@@ -0,0 +1,68 @@
+/**
+ * Creates a managed stream that wraps the source stream with lifecycle management.
+ *
+ * The managed stream:
+ * - Emits "chunk" events for each chunk
+ * - Throws errors from error chunks
+ * - Finalizes the session on complete or cancel (but NOT on error - see below)
+ *
+ * IMPORTANT: This stream intentionally does NOT finalize the session on errors.
+ * If you call session.finalize("error") here, the session becomes non-pending,
+ * which means new managed streams created during retry will immediately close
+ * (see the `!session.isPending()` check in pull).
+ *
+ * The retry logic in wrapWithRetry is responsible for finalizing the session
+ * when all retries are exhausted. Don't be a hero and add finalization here.
+ *
+ * @param source - The source async iterable stream
+ * @param session - The session to update with events
+ * @returns A managed stream with lifecycle hooks
+ */
+export function createManagedStream(source, session) {
+    const iterator = source[Symbol.asyncIterator]();
+    let closed = false;
+    const readable = new ReadableStream({
+        async pull(controller) {
+            if (closed || !session.isPending()) {
+                controller.close();
+                return;
+            }
+            try {
+                const result = await iterator.next();
+                if (result.done) {
+                    closed = true;
+                    session.finalize("complete");
+                    controller.close();
+                    return;
+                }
+                session.recordChunk();
+                void session.events.emit("chunk", result.value, session);
+                if (result.value.type === "error") {
+                    const chunkError = result.value
+                        .error;
+                    closed = true;
+                    // don't finalize here, wrapWithRetry catches this and decides whether to retry
+                    controller.error(chunkError);
+                    return;
+                }
+                controller.enqueue(result.value);
+            }
+            catch (error) {
+                closed = true;
+                // no finalization, let wrapWithRetry handle it
+                controller.error(error);
+            }
+        },
+        cancel() {
+            if (!closed) {
+                closed = true;
+                if (session.isPending()) {
+                    session.finalize("aborted");
+                }
+            }
+            void iterator.return?.();
+        },
+    });
+    return readable;
+}
+//# sourceMappingURL=managed-stream.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"managed-stream.js","sourceRoot":"","sources":["../../../../src/ai-service/llm/stream/managed-stream.ts"],"names":[],"mappings":"AAIA;;;;;;;;;;;;;;;;;;;GAmBG;AACH,MAAM,UAAU,mBAAmB,CAGjC,MAA8B,EAC9B,OAAsB;IAEtB,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;IAChD,IAAI,MAAM,GAAG,KAAK,CAAC;IAEnB,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAI;QACrC,KAAK,CAAC,IAAI,CAAC,UAAU;YACnB,IAAI,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;gBACnC,UAAU,CAAC,KAAK,EAAE,CAAC;gBACnB,OAAO;YACT,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBAErC,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;oBAChB,MAAM,GAAG,IAAI,CAAC;oBACd,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;oBAC7B,UAAU,CAAC,KAAK,EAAE,CAAC;oBACnB,OAAO;gBACT,CAAC;gBAED,OAAO,CAAC,WAAW,EAAE,CAAC;gBACtB,KAAK,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;gBAEzD,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;oBAClC,MAAM,UAAU,GAAI,MAAM,CAAC,KAAuC;yBAC/D,KAAK,CAAC;oBACT,MAAM,GAAG,IAAI,CAAC;oBACd,+EAA+E;oBAC/E,UAAU,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;oBAC7B,OAAO;gBACT,CAAC;gBAED,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACnC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,GAAG,IAAI,CAAC;gBACd,+CAA+C;gBAC/C,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;QAED,MAAM;YACJ,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,GAAG,IAAI,CAAC;gBACd,IAAI,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;oBACxB,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,KAAK,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;QAC3B,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,QAAkC,CAAC;AAC5C,CAAC"}

package/dist/ai-service/llm/stream/observer.d.ts

@@ -0,0 +1,34 @@
+import type { StreamConfig } from "./config.js";
+import type { StreamEventBus } from "./event-bus.js";
+/**
+ * Interface for stream event observers.
+ *
+ * Observers subscribe to session events via the attach() method.
+ * They can perform side effects (logging, tracing, etc.) but cannot
+ * modify the stream flow.
+ *
+ * @example
+ * ```typescript
+ * class LoggingObserver implements StreamObserver {
+ *   attach(events: StreamEventBus, config: StreamConfig): void {
+ *     events.on("session:start", (session) => {
+ *       console.log(`Session ${session.id} started`);
+ *     });
+ *
+ *     events.on("session:complete", (session) => {
+ *       console.log(`Session ${session.id} completed`);
+ *     });
+ *   }
+ * }
+ * ```
+ */
+export interface StreamObserver {
+    /**
+     * Subscribe to session events.
+     *
+     * Called once when the observer is registered with an orchestrator.
+     * The observer should set up all its event subscriptions here.
+     */
+    attach(events: StreamEventBus, config: StreamConfig): void;
+}
+//# sourceMappingURL=observer.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"observer.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/llm/stream/observer.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAChD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAErD;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,WAAW,cAAc;IAC7B;;;;;OAKG;IACH,MAAM,CAAC,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,YAAY,GAAG,IAAI,CAAC;CAC5D"}

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

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=observer.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"observer.js","sourceRoot":"","sources":["../../../../src/ai-service/llm/stream/observer.ts"],"names":[],"mappings":""}

package/dist/ai-service/llm/stream/observers/context.d.ts

@@ -0,0 +1,40 @@
+import type { StreamConfig } from "../config.js";
+import type { StreamEventBus } from "../event-bus.js";
+import type { StreamObserver } from "../observer.js";
+export interface ContextHandle {
+    context: ConversationContext;
+    release(): void;
+}
+export interface ConversationContext {
+    startStep(): void;
+    endStep(messages: unknown[], usage?: unknown): void;
+    endTurn(usage?: unknown): void;
+}
+export interface ContextObserverOptions {
+    contextHandle: ContextHandle;
+}
+/**
+ * Observer that manages conversation context during LLM interactions.
+ *
+ * Handles:
+ * - Starting and ending steps in the context
+ * - Ending the turn on completion
+ * - Releasing the context handle on completion/error/abort
+ *
+ * @example
+ * ```typescript
+ * const observer = new ContextObserver({
+ *   contextHandle: await contextManager.acquireContext(contextId, ownerId),
+ * });
+ * ```
+ */
+export declare class ContextObserver implements StreamObserver {
+    private readonly contextHandle;
+    private released;
+    constructor(options: ContextObserverOptions);
+    attach(events: StreamEventBus, _config: StreamConfig): void;
+    private onStepComplete;
+    private onStreamFinish;
+    private release;
+}
+//# sourceMappingURL=context.d.ts.map

package/dist/ai-service/llm/stream/observers/context.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.d.ts","sourceRoot":"","sources":["../../../../../src/ai-service/llm/stream/observers/context.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AACjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAGrD,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,mBAAmB,CAAC;IAC7B,OAAO,IAAI,IAAI,CAAC;CACjB;AAED,MAAM,WAAW,mBAAmB;IAClC,SAAS,IAAI,IAAI,CAAC;IAClB,OAAO,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,KAAK,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;IACpD,OAAO,CAAC,KAAK,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;CAChC;AAED,MAAM,WAAW,sBAAsB;IACrC,aAAa,EAAE,aAAa,CAAC;CAC9B;AAED;;;;;;;;;;;;;;GAcG;AACH,qBAAa,eAAgB,YAAW,cAAc;IACpD,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAgB;IAC9C,OAAO,CAAC,QAAQ,CAAS;gBAEb,OAAO,EAAE,sBAAsB;IAI3C,MAAM,CAAC,MAAM,EAAE,cAAc,EAAE,OAAO,EAAE,YAAY,GAAG,IAAI;IAsB3D,OAAO,CAAC,cAAc;IAUtB,OAAO,CAAC,cAAc;IAOtB,OAAO,CAAC,OAAO;CAMhB"}

package/dist/ai-service/llm/stream/observers/context.js

@@ -0,0 +1,55 @@
+/**
+ * Observer that manages conversation context during LLM interactions.
+ *
+ * Handles:
+ * - Starting and ending steps in the context
+ * - Ending the turn on completion
+ * - Releasing the context handle on completion/error/abort
+ *
+ * @example
+ * ```typescript
+ * const observer = new ContextObserver({
+ *   contextHandle: await contextManager.acquireContext(contextId, ownerId),
+ * });
+ * ```
+ */
+export class ContextObserver {
+    contextHandle;
+    released = false;
+    constructor(options) {
+        this.contextHandle = options.contextHandle;
+    }
+    attach(events, _config) {
+        events.on("step:complete", (step, session) => {
+            this.onStepComplete(step, session);
+        });
+        events.on("stream:finish", (result, session) => {
+            this.onStreamFinish(result, session);
+        });
+        events.on("session:complete", () => {
+            this.release();
+        });
+        events.on("session:error", () => {
+            this.release();
+        });
+        events.on("session:abort", () => {
+            this.release();
+        });
+    }
+    onStepComplete(step, _session) {
+        if (step.response?.messages) {
+            this.contextHandle.context.endStep(step.response.messages, step.usage);
+        }
+        this.contextHandle.context.startStep();
+    }
+    onStreamFinish(result, _session) {
+        this.contextHandle.context.endTurn(result.totalUsage);
+    }
+    release() {
+        if (!this.released) {
+            this.released = true;
+            this.contextHandle.release();
+        }
+    }
+}
+//# sourceMappingURL=context.js.map

package/dist/ai-service/llm/stream/observers/context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.js","sourceRoot":"","sources":["../../../../../src/ai-service/llm/stream/observers/context.ts"],"names":[],"mappings":"AAsBA;;;;;;;;;;;;;;GAcG;AACH,MAAM,OAAO,eAAe;IACT,aAAa,CAAgB;IACtC,QAAQ,GAAG,KAAK,CAAC;IAEzB,YAAY,OAA+B;QACzC,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;IAC7C,CAAC;IAED,MAAM,CAAC,MAAsB,EAAE,OAAqB;QAClD,MAAM,CAAC,EAAE,CAAC,eAAe,EAAE,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE;YAC3C,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,eAAe,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;YAC7C,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,kBAAkB,EAAE,GAAG,EAAE;YACjC,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,eAAe,EAAE,GAAG,EAAE;YAC9B,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,eAAe,EAAE,GAAG,EAAE;YAC9B,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,cAAc,CACpB,IAAsC,EACtC,QAAuB;QAEvB,IAAI,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,CAAC;YAC5B,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QACzE,CAAC;QACD,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;IACzC,CAAC;IAEO,cAAc,CACpB,MAAyC,EACzC,QAAuB;QAEvB,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IACxD,CAAC;IAEO,OAAO;QACb,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;YACrB,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;QAC/B,CAAC;IACH,CAAC;CACF"}

package/dist/ai-service/llm/stream/observers/index.d.ts

@@ -0,0 +1,8 @@
+/**
+ * Stream observers for handling side effects during LLM interactions.
+ */
+export { LoggingObserver, type LoggingObserverOptions, type LogRef, } from "./logging.js";
+export { LLMObsObserver, wrapToolsForObservability, type LLMObsObserverOptions, } from "./llmobs.js";
+export { ContextObserver, type ContextObserverOptions, type ContextHandle, type ConversationContext, } from "./context.js";
+export { RetryNotificationObserver, type RetryNotificationOptions, } from "./retry-notification.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/ai-service/llm/stream/observers/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../src/ai-service/llm/stream/observers/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EACL,eAAe,EACf,KAAK,sBAAsB,EAC3B,KAAK,MAAM,GACZ,MAAM,cAAc,CAAC;AACtB,OAAO,EACL,cAAc,EACd,yBAAyB,EACzB,KAAK,qBAAqB,GAC3B,MAAM,aAAa,CAAC;AACrB,OAAO,EACL,eAAe,EACf,KAAK,sBAAsB,EAC3B,KAAK,aAAa,EAClB,KAAK,mBAAmB,GACzB,MAAM,cAAc,CAAC;AACtB,OAAO,EACL,yBAAyB,EACzB,KAAK,wBAAwB,GAC9B,MAAM,yBAAyB,CAAC"}

package/dist/ai-service/llm/stream/observers/index.js

@@ -0,0 +1,8 @@
+/**
+ * Stream observers for handling side effects during LLM interactions.
+ */
+export { LoggingObserver, } from "./logging.js";
+export { LLMObsObserver, wrapToolsForObservability, } from "./llmobs.js";
+export { ContextObserver, } from "./context.js";
+export { RetryNotificationObserver, } from "./retry-notification.js";
+//# sourceMappingURL=index.js.map

package/dist/ai-service/llm/stream/observers/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/ai-service/llm/stream/observers/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EACL,eAAe,GAGhB,MAAM,cAAc,CAAC;AACtB,OAAO,EACL,cAAc,EACd,yBAAyB,GAE1B,MAAM,aAAa,CAAC;AACrB,OAAO,EACL,eAAe,GAIhB,MAAM,cAAc,CAAC;AACtB,OAAO,EACL,yBAAyB,GAE1B,MAAM,yBAAyB,CAAC"}

package/dist/ai-service/llm/stream/observers/llmobs.d.ts

@@ -0,0 +1,78 @@
+import { type ErrorCategory } from "../../error.js";
+import type { Logger } from "../../../../util/logger.js";
+import type { Tool } from "../../../agent/tools2/types.js";
+import type { LLMObsTracer } from "../../../llmobs/index.js";
+import type { StreamConfig } from "../config.js";
+import type { StreamEventBus } from "../event-bus.js";
+import type { StreamObserver } from "../observer.js";
+export interface ClassifiedError {
+    category: ErrorCategory;
+    code?: string | number;
+    isRetryable: boolean;
+    message: string;
+}
+export declare function classifyError(error: unknown): ClassifiedError;
+export interface LLMObsObserverOptions {
+    tracer: LLMObsTracer;
+    logger: Logger;
+}
+/**
+ * Observer that manages LLM Observability tracing spans.
+ *
+ * Creates and manages:
+ * - A full workflow span (`ai.streamText`) for the entire session
+ * - Step spans (`ai.streamText.step`) for each LLM step
+ * - Segment spans (`ai.streamText.{type}`) for thinking/reasoning/text/tool-input
+ * - Annotations with usage metrics, model info, and input/output data
+ *
+ * @example
+ * ```typescript
+ * const observer = new LLMObsObserver({
+ *   tracer: LLMObsTracer.instance,
+ *   logger: getLogger(),
+ * });
+ * ```
+ */
+export declare class LLMObsObserver implements StreamObserver {
+    private readonly tracer;
+    private readonly logger;
+    private fullSpan?;
+    private stepSpan?;
+    private currentSegment?;
+    private stepNumber;
+    constructor(options: LLMObsObserverOptions);
+    attach(events: StreamEventBus, _config: StreamConfig): void;
+    private onSessionStart;
+    private onChunk;
+    private onStepComplete;
+    private onStreamFinish;
+    private onRetry;
+    private onFallback;
+    private onSessionComplete;
+    private onSessionError;
+    private onSessionAbort;
+    private startSegment;
+    private finishSegment;
+    private ensureSegment;
+    private appendTextToCurrentSegment;
+    private appendToolInputDelta;
+    private isTextualSegment;
+    private getSegmentTypeFromChunk;
+    private startStepSpan;
+    private finishStepSpan;
+    private finishFullSpan;
+    private compileMetrics;
+}
+/**
+ * Wraps tools with LLMObs tracing for real-time tool execution observability.
+ *
+ * This utility intercepts each tool's `execute` function and wraps it with
+ * a traced span, capturing:
+ * - Tool name (`tool.{name}`)
+ * - Input arguments
+ * - Output result
+ * - Execution duration
+ * - Error status
+ */
+export declare function wrapToolsForObservability<TOOLS extends Record<string, Tool>>(tools: TOOLS, tracer: LLMObsTracer): TOOLS;
+//# sourceMappingURL=llmobs.d.ts.map

package/dist/ai-service/llm/stream/observers/llmobs.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"llmobs.d.ts","sourceRoot":"","sources":["../../../../../src/ai-service/llm/stream/observers/llmobs.ts"],"names":[],"mappings":"AACA,OAAO,EAAoB,KAAK,aAAa,EAAE,MAAM,gBAAgB,CAAC;AACtE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,gCAAgC,CAAC;AAC3D,OAAO,KAAK,EACV,YAAY,EAGb,MAAM,0BAA0B,CAAC;AAMlC,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AACjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAGrD,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,aAAa,CAAC;IACxB,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IACvB,WAAW,EAAE,OAAO,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,wBAAgB,aAAa,CAAC,KAAK,EAAE,OAAO,GAAG,eAAe,CAqD7D;AAYD,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,YAAY,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;CAChB;AAkBD;;;;;;;;;;;;;;;;GAgBG;AACH,qBAAa,cAAe,YAAW,cAAc;IACnD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAe;IACtC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAEhC,OAAO,CAAC,QAAQ,CAAC,CAAa;IAC9B,OAAO,CAAC,QAAQ,CAAC,CAAa;IAC9B,OAAO,CAAC,cAAc,CAAC,CAAU;IACjC,OAAO,CAAC,UAAU,CAAK;gBAEX,OAAO,EAAE,qBAAqB;IAK1C,MAAM,CAAC,MAAM,EAAE,cAAc,EAAE,OAAO,EAAE,YAAY,GAAG,IAAI;IAsC3D,OAAO,CAAC,cAAc;IAsBtB,OAAO,CAAC,OAAO;IAwBf,OAAO,CAAC,cAAc;IAiBtB,OAAO,CAAC,cAAc;IAsBtB,OAAO,CAAC,OAAO;IA8Bf,OAAO,CAAC,UAAU;IA4BlB,OAAO,CAAC,iBAAiB;IAIzB,OAAO,CAAC,cAAc;IAsBtB,OAAO,CAAC,cAAc;IAWtB,OAAO,CAAC,YAAY;IAcpB,OAAO,CAAC,aAAa;IAqBrB,OAAO,CAAC,aAAa;IAMrB,OAAO,CAAC,0BAA0B;IASlC,OAAO,CAAC,oBAAoB;IAO5B,OAAO,CAAC,gBAAgB;IAUxB,OAAO,CAAC,uBAAuB;IAkB/B,OAAO,CAAC,aAAa;IAYrB,OAAO,CAAC,cAAc;IAsCtB,OAAO,CAAC,cAAc;IAuCtB,OAAO,CAAC,cAAc;CAuCvB;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,yBAAyB,CAAC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,EAC1E,KAAK,EAAE,KAAK,EACZ,MAAM,EAAE,YAAY,GACnB,KAAK,CAuDP"}