@superblocksteam/vite-plugin-file-sync 2.0.43-next.8 → 2.0.48-next.0

package/dist/ai-service/llm/context/manager.d.ts

@@ -5,9 +5,12 @@
  * - Load/save context state from/to database
  * - Cache active contexts in memory
  * - Create ephemeral contexts for one-off operations
+ * - Ensure thread-safe exclusive context ownership
  *
  */
 import { type CacheStrategyType } from "./caching/index.js";
+import { ContextHandle, type AcquireContextOptions } from "./context-handle.js";
+import { ContextLockManager } from "./context-lock.js";
 import { Context } from "./context.js";
 import type { ContextStorage } from "./storage/index.js";
 import type { ContextId, ContextOptions, LoggingOptions } from "./types.js";
@@ -23,7 +26,7 @@ export interface ContextManagerOptions {
     logging?: LoggingOptions;
 }
 /**
- * Manager for creating and managing conversation contexts.
+ * Manager for creating and managing conversation contexts with thread-safe ownership.
  */
 export declare class ContextManager {
     private storage;
@@ -31,24 +34,70 @@ export declare class ContextManager {
     private contextOptions?;
     private loggingOptions?;
     private contexts;
+    private lockManager;
     private static readonly logPrefix;
     constructor(options: ContextManagerOptions);
     /**
-     * Gets or creates a persistent context.
+     * Creates a unique key for a context ID.
+     */
+    private getContextKey;
+    /**
+     * Acquires exclusive access to a context.
      *
-     * Contexts are:
-     * 1. Checked in memory cache first
-     * 2. Loaded from storage if not cached
-     * 3. Created fresh if not found in storage
-     * 4. Cached for subsequent requests
-     * 5. Automatically persisted on changes
+     * This method ensures thread-safe access to contexts by requiring
+     * an owner ID and returning a handle that must be released when done.
      *
      * @param id - Unique context identifier
-     * @param options - Context configuration options
-     * @returns Context instance (cached, loaded, or new)
+     * @param ownerId - Unique identifier of the entity acquiring the context
+     * @param options - Context and acquisition options
+     * @returns ContextHandle with exclusive access to the context
+     * @throws {ContextLockError} if the context is locked and acquisition fails
+     *
+     * @example
+     * ```typescript
+     * const handle = await manager.acquireContext(contextId, "api-subagent-1");
+     * try {
+     *   handle.context.startTurn(userMessage);
+     *   // ... use the context
+     * } finally {
+     *   handle.release();
+     * }
+     * ```
      */
-    getContext(id: ContextId, options?: ContextOptions): Promise<Context>;
+    acquireContext(id: ContextId, ownerId: string, options?: {
+        contextOptions?: ContextOptions;
+        acquireOptions?: AcquireContextOptions;
+    }): Promise<ContextHandle>;
     private get logger();
+    /**
+     * Checks if a context is currently locked.
+     *
+     * @param id - Context identifier
+     * @returns true if the context is locked
+     */
+    isContextLocked(id: ContextId): boolean;
+    /**
+     * Gets the current owner of a context.
+     *
+     * @param id - Context identifier
+     * @returns The owner ID if locked, undefined otherwise
+     */
+    getContextOwner(id: ContextId): string | undefined;
+    /**
+     * Gets information about all active context locks.
+     * Useful for debugging and monitoring.
+     *
+     * @returns Array of active lock information
+     */
+    getActiveLocks(): ReturnType<ContextLockManager["getActiveLocks"]>;
+    /**
+     * Releases a context lock if owned by the specified owner.
+     * This is useful for cleanup in error scenarios.
+     *
+     * @param id - Context identifier
+     * @param ownerId - The owner releasing the context
+     */
+    releaseContext(id: ContextId, ownerId: string): void;
     /**
      * Creates an ephemeral context (no persistence or caching).
      *

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

@@ -1 +1 @@
-{"version":3,"file":"manager.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/llm/context/manager.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAEL,KAAK,iBAAiB,EACvB,MAAM,oBAAoB,CAAC;AAE5B,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAGvC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,KAAK,EAAE,SAAS,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAE5E;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,mDAAmD;IACnD,OAAO,EAAE,cAAc,CAAC;IACxB,oEAAoE;IACpE,aAAa,CAAC,EAAE,iBAAiB,CAAC;IAClC,6CAA6C;IAC7C,OAAO,CAAC,EAAE,cAAc,CAAC;CAC1B;AAED;;GAEG;AACH,qBAAa,cAAc;IACzB,OAAO,CAAC,OAAO,CAAiB;IAChC,OAAO,CAAC,iBAAiB,CAAC,CAAoB;IAC9C,OAAO,CAAC,cAAc,CAAC,CAAiB;IACxC,OAAO,CAAC,cAAc,CAAC,CAAiB;IACxC,OAAO,CAAC,QAAQ,CAAiC;IACjD,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAuB;gBAE5C,OAAO,EAAE,qBAAqB;IAkB1C;;;;;;;;;;;;;OAaG;IACG,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,cAAc,GAAG,OAAO,CAAC,OAAO,CAAC;IAkC3E,OAAO,KAAK,MAAM,GAEjB;IAED;;;;;;;;;;OAUG;IACH,aAAa,CAAC,OAAO,CAAC,EAAE,cAAc,GAAG,OAAO;IAShD;;;;;OAKG;YACW,WAAW;IAoBzB;;;;;OAKG;YACW,WAAW;CAW1B"}
+{"version":3,"file":"manager.d.ts","sourceRoot":"","sources":["../../../../src/ai-service/llm/context/manager.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAGH,OAAO,EAEL,KAAK,iBAAiB,EACvB,MAAM,oBAAoB,CAAC;AAE5B,OAAO,EACL,aAAa,EAEb,KAAK,qBAAqB,EAC3B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAoB,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AACzE,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAEvC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,KAAK,EAAE,SAAS,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAE5E;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,mDAAmD;IACnD,OAAO,EAAE,cAAc,CAAC;IACxB,oEAAoE;IACpE,aAAa,CAAC,EAAE,iBAAiB,CAAC;IAClC,6CAA6C;IAC7C,OAAO,CAAC,EAAE,cAAc,CAAC;CAC1B;AAED;;GAEG;AACH,qBAAa,cAAc;IACzB,OAAO,CAAC,OAAO,CAAiB;IAChC,OAAO,CAAC,iBAAiB,CAAC,CAAoB;IAC9C,OAAO,CAAC,cAAc,CAAC,CAAiB;IACxC,OAAO,CAAC,cAAc,CAAC,CAAiB;IACxC,OAAO,CAAC,QAAQ,CAA8B;IAC9C,OAAO,CAAC,WAAW,CAA4B;IAC/C,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAuB;gBAE5C,OAAO,EAAE,qBAAqB;IAkB1C;;OAEG;IACH,OAAO,CAAC,aAAa;IAIrB;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACG,cAAc,CAClB,EAAE,EAAE,SAAS,EACb,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE;QACR,cAAc,CAAC,EAAE,cAAc,CAAC;QAChC,cAAc,CAAC,EAAE,qBAAqB,CAAC;KACxC,GACA,OAAO,CAAC,aAAa,CAAC;IA0DzB,OAAO,KAAK,MAAM,GAEjB;IAED;;;;;OAKG;IACH,eAAe,CAAC,EAAE,EAAE,SAAS,GAAG,OAAO;IAIvC;;;;;OAKG;IACH,eAAe,CAAC,EAAE,EAAE,SAAS,GAAG,MAAM,GAAG,SAAS;IAIlD;;;;;OAKG;IACH,cAAc,IAAI,UAAU,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC;IAIlE;;;;;;OAMG;IACH,cAAc,CAAC,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAWpD;;;;;;;;;;OAUG;IACH,aAAa,CAAC,OAAO,CAAC,EAAE,cAAc,GAAG,OAAO;IAShD;;;;;OAKG;YACW,WAAW;IAoBzB;;;;;OAKG;YACW,WAAW;CAW1B"}

package/dist/ai-service/llm/context/manager.js

@@ -5,14 +5,17 @@
  * - Load/save context state from/to database
  * - Cache active contexts in memory
  * - Create ephemeral contexts for one-off operations
+ * - Ensure thread-safe exclusive context ownership
  *
  */
+import { getPrefixedLogger } from "../../../util/logger.js";
 import { CacheStrategyFactory, } from "./caching/index.js";
 import { loadContextConfigFromEnv } from "./config.js";
+import { ContextHandle, tryAcquireWithRetries, } from "./context-handle.js";
+import { ContextLockError, ContextLockManager } from "./context-lock.js";
 import { Context } from "./context.js";
-import { createPrefixedLogger } from "./logger.js";
 /**
- * Manager for creating and managing conversation contexts.
+ * Manager for creating and managing conversation contexts with thread-safe ownership.
  */
 export class ContextManager {
     storage;
@@ -20,6 +23,7 @@ export class ContextManager {
     contextOptions;
     loggingOptions;
     contexts = new Map();
+    lockManager = new ContextLockManager();
     static logPrefix = "[context-manager]";
     constructor(options) {
         this.storage = options.storage;
@@ -34,48 +38,120 @@ export class ContextManager {
         }
     }
     /**
-     * Gets or creates a persistent context.
+     * Creates a unique key for a context ID.
+     */
+    getContextKey(id) {
+        return `${id.appId}::${id.userId}::${id.name}`;
+    }
+    /**
+     * Acquires exclusive access to a context.
      *
-     * Contexts are:
-     * 1. Checked in memory cache first
-     * 2. Loaded from storage if not cached
-     * 3. Created fresh if not found in storage
-     * 4. Cached for subsequent requests
-     * 5. Automatically persisted on changes
+     * This method ensures thread-safe access to contexts by requiring
+     * an owner ID and returning a handle that must be released when done.
      *
      * @param id - Unique context identifier
-     * @param options - Context configuration options
-     * @returns Context instance (cached, loaded, or new)
+     * @param ownerId - Unique identifier of the entity acquiring the context
+     * @param options - Context and acquisition options
+     * @returns ContextHandle with exclusive access to the context
+     * @throws {ContextLockError} if the context is locked and acquisition fails
+     *
+     * @example
+     * ```typescript
+     * const handle = await manager.acquireContext(contextId, "api-subagent-1");
+     * try {
+     *   handle.context.startTurn(userMessage);
+     *   // ... use the context
+     * } finally {
+     *   handle.release();
+     * }
+     * ```
      */
-    async getContext(id, options) {
-        // Check cache first
-        if (this.contexts.has(id)) {
-            return this.contexts.get(id);
+    async acquireContext(id, ownerId, options) {
+        const key = this.getContextKey(id);
+        // Try to acquire the lock
+        const acquired = await tryAcquireWithRetries(this.lockManager, id, ownerId, options?.acquireOptions);
+        if (!acquired) {
+            const currentOwner = this.lockManager.getOwner(id);
+            throw new ContextLockError(id, currentOwner ?? "unknown", ownerId);
+        }
+        try {
+            // Get or create the context while holding the lock
+            let context = this.contexts.get(key);
+            if (!context) {
+                const onPersist = async (record) => {
+                    await this.saveContext(id, record);
+                };
+                // Create cache strategy
+                const cacheStrategy = CacheStrategyFactory.create(this.cacheStrategyType);
+                // Load from storage
+                const record = await this.loadContext(id);
+                // Create context with persistence callback, cache strategy, and logging
+                const params = {
+                    id,
+                    options: { ...options?.contextOptions, ...this.contextOptions },
+                    logging: this.loggingOptions,
+                    onPersist,
+                    cacheStrategy,
+                };
+                context = record
+                    ? Context.fromRecord(record, params)
+                    : new Context(params);
+                // Cache for future requests
+                this.contexts.set(key, context);
+            }
+            return new ContextHandle(context, id, ownerId, this.lockManager);
+        }
+        catch (error) {
+            // Release the lock if context creation fails
+            this.lockManager.release(id, ownerId);
+            throw error;
         }
-        const onPersist = async (record) => {
-            await this.saveContext(id, record);
-        };
-        // Create cache strategy
-        const cacheStrategy = CacheStrategyFactory.create(this.cacheStrategyType);
-        // Load from storage
-        const record = await this.loadContext(id);
-        // Create context with persistence callback, cache strategy, and logging
-        const params = {
-            id,
-            options: { ...options, ...this.contextOptions },
-            logging: this.loggingOptions,
-            onPersist,
-            cacheStrategy,
-        };
-        const context = record
-            ? Context.fromRecord(record, params)
-            : new Context(params);
-        // Cache for future requests
-        this.contexts.set(id, context);
-        return context;
     }
     get logger() {
-        return createPrefixedLogger(ContextManager.logPrefix);
+        return getPrefixedLogger(ContextManager.logPrefix);
+    }
+    /**
+     * Checks if a context is currently locked.
+     *
+     * @param id - Context identifier
+     * @returns true if the context is locked
+     */
+    isContextLocked(id) {
+        return this.lockManager.isLocked(id);
+    }
+    /**
+     * Gets the current owner of a context.
+     *
+     * @param id - Context identifier
+     * @returns The owner ID if locked, undefined otherwise
+     */
+    getContextOwner(id) {
+        return this.lockManager.getOwner(id);
+    }
+    /**
+     * Gets information about all active context locks.
+     * Useful for debugging and monitoring.
+     *
+     * @returns Array of active lock information
+     */
+    getActiveLocks() {
+        return this.lockManager.getActiveLocks();
+    }
+    /**
+     * Releases a context lock if owned by the specified owner.
+     * This is useful for cleanup in error scenarios.
+     *
+     * @param id - Context identifier
+     * @param ownerId - The owner releasing the context
+     */
+    releaseContext(id, ownerId) {
+        try {
+            this.lockManager.release(id, ownerId);
+            this.logger.debug(`Released context '${id.name}' for owner '${ownerId}'`);
+        }
+        catch (error) {
+            this.logger.warn(`Failed to release context '${id.name}' for owner '${ownerId}': ${error}`);
+        }
     }
     /**
      * Creates an ephemeral context (no persistence or caching).

package/dist/ai-service/llm/context/manager.js.map

@@ -1 +1 @@
-{"version":3,"file":"manager.js","sourceRoot":"","sources":["../../../../src/ai-service/llm/context/manager.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EACL,oBAAoB,GAErB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AACvC,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAiBnD;;GAEG;AACH,MAAM,OAAO,cAAc;IACjB,OAAO,CAAiB;IACxB,iBAAiB,CAAqB;IACtC,cAAc,CAAkB;IAChC,cAAc,CAAkB;IAChC,QAAQ,GAAG,IAAI,GAAG,EAAsB,CAAC;IACzC,MAAM,CAAU,SAAS,GAAG,mBAAmB,CAAC;IAExD,YAAY,OAA8B;QACxC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAE/B,yDAAyD;QACzD,MAAM,SAAS,GAAG,wBAAwB,EAAE,CAAC;QAE7C,2EAA2E;QAC3E,IAAI,CAAC,iBAAiB,GAAG,SAAS,EAAE,aAAa,IAAI,OAAO,CAAC,aAAa,CAAC;QAC3E,IAAI,CAAC,cAAc,GAAG,SAAS,EAAE,OAAO,CAAC;QACzC,IAAI,CAAC,cAAc,GAAG,SAAS,EAAE,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC;QAE5D,IAAI,SAAS,EAAE,CAAC;YACd,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,oFAAoF,IAAI,CAAC,iBAAiB,aAAa,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,EAAE,CAC5K,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;;;;;;;;;;;OAaG;IACH,KAAK,CAAC,UAAU,CAAC,EAAa,EAAE,OAAwB;QACtD,oBAAoB;QACpB,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAE,CAAC;QAChC,CAAC;QAED,MAAM,SAAS,GAAG,KAAK,EAAE,MAAqB,EAAE,EAAE;YAChD,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QACrC,CAAC,CAAC;QAEF,wBAAwB;QACxB,MAAM,aAAa,GAAG,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAE1E,oBAAoB;QACpB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAE1C,wEAAwE;QACxE,MAAM,MAAM,GAAG;YACb,EAAE;YACF,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,GAAG,IAAI,CAAC,cAAc,EAAE;YAC/C,OAAO,EAAE,IAAI,CAAC,cAAc;YAC5B,SAAS;YACT,aAAa;SACd,CAAC;QACF,MAAM,OAAO,GAAG,MAAM;YACpB,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC;YACpC,CAAC,CAAC,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC;QAExB,4BAA4B;QAC5B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAE/B,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAY,MAAM;QAChB,OAAO,oBAAoB,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;IACxD,CAAC;IAED;;;;;;;;;;OAUG;IACH,aAAa,CAAC,OAAwB;QACpC,MAAM,aAAa,GAAG,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC1E,OAAO,IAAI,OAAO,CAAC;YACjB,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,GAAG,IAAI,CAAC,cAAc,EAAE;YAC/C,OAAO,EAAE,IAAI,CAAC,cAAc;YAC5B,aAAa;SACd,CAAC,CAAC;IACL,CAAC;IAED;;;;;OAKG;IACK,KAAK,CAAC,WAAW,CAAC,EAAa;QACrC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;YAClD,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;gBACpB,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,kCAAkC,EAAE,CAAC,IAAI,oBAAoB,CAC9D,CAAC;gBACF,OAAO,IAAI,CAAC;YACd,CAAC;YAED,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,gCAAgC,EAAE,CAAC,IAAI,iBAAiB,CACzD,CAAC;YACF,OAAO,MAAuB,CAAC;QACjC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,+BAA+B,EAAE,CAAC,IAAI,MAAM,GAAG,EAAE,CAAC,CAAC;YACrE,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACK,KAAK,CAAC,WAAW,CACvB,EAAa,EACb,MAAqB;QAErB,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;YAC3C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sBAAsB,EAAE,CAAC,IAAI,GAAG,CAAC,CAAC;QACtD,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,+BAA+B,EAAE,MAAM,GAAG,EAAE,CAAC,CAAC;QAClE,CAAC;IACH,CAAC"}
+{"version":3,"file":"manager.js","sourceRoot":"","sources":["../../../../src/ai-service/llm/context/manager.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EACL,oBAAoB,GAErB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EACL,aAAa,EACb,qBAAqB,GAEtB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AACzE,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAiBvC;;GAEG;AACH,MAAM,OAAO,cAAc;IACjB,OAAO,CAAiB;IACxB,iBAAiB,CAAqB;IACtC,cAAc,CAAkB;IAChC,cAAc,CAAkB;IAChC,QAAQ,GAAG,IAAI,GAAG,EAAmB,CAAC;IACtC,WAAW,GAAG,IAAI,kBAAkB,EAAE,CAAC;IACvC,MAAM,CAAU,SAAS,GAAG,mBAAmB,CAAC;IAExD,YAAY,OAA8B;QACxC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAE/B,yDAAyD;QACzD,MAAM,SAAS,GAAG,wBAAwB,EAAE,CAAC;QAE7C,2EAA2E;QAC3E,IAAI,CAAC,iBAAiB,GAAG,SAAS,EAAE,aAAa,IAAI,OAAO,CAAC,aAAa,CAAC;QAC3E,IAAI,CAAC,cAAc,GAAG,SAAS,EAAE,OAAO,CAAC;QACzC,IAAI,CAAC,cAAc,GAAG,SAAS,EAAE,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC;QAE5D,IAAI,SAAS,EAAE,CAAC;YACd,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,oFAAoF,IAAI,CAAC,iBAAiB,aAAa,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,EAAE,CAC5K,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,EAAa;QACjC,OAAO,GAAG,EAAE,CAAC,KAAK,KAAK,EAAE,CAAC,MAAM,KAAK,EAAE,CAAC,IAAI,EAAE,CAAC;IACjD,CAAC;IAED;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACH,KAAK,CAAC,cAAc,CAClB,EAAa,EACb,OAAe,EACf,OAGC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;QAEnC,0BAA0B;QAC1B,MAAM,QAAQ,GAAG,MAAM,qBAAqB,CAC1C,IAAI,CAAC,WAAW,EAChB,EAAE,EACF,OAAO,EACP,OAAO,EAAE,cAAc,CACxB,CAAC;QAEF,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,YAAY,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YACnD,MAAM,IAAI,gBAAgB,CAAC,EAAE,EAAE,YAAY,IAAI,SAAS,EAAE,OAAO,CAAC,CAAC;QACrE,CAAC;QAED,IAAI,CAAC;YACH,mDAAmD;YACnD,IAAI,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAErC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,SAAS,GAAG,KAAK,EAAE,MAAqB,EAAE,EAAE;oBAChD,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;gBACrC,CAAC,CAAC;gBAEF,wBAAwB;gBACxB,MAAM,aAAa,GAAG,oBAAoB,CAAC,MAAM,CAC/C,IAAI,CAAC,iBAAiB,CACvB,CAAC;gBAEF,oBAAoB;gBACpB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;gBAE1C,wEAAwE;gBACxE,MAAM,MAAM,GAAG;oBACb,EAAE;oBACF,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,cAAc,EAAE,GAAG,IAAI,CAAC,cAAc,EAAE;oBAC/D,OAAO,EAAE,IAAI,CAAC,cAAc;oBAC5B,SAAS;oBACT,aAAa;iBACd,CAAC;gBAEF,OAAO,GAAG,MAAM;oBACd,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC;oBACpC,CAAC,CAAC,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC;gBAExB,4BAA4B;gBAC5B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;YAClC,CAAC;YAED,OAAO,IAAI,aAAa,CAAC,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QACnE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,6CAA6C;YAC7C,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACtC,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,IAAY,MAAM;QAChB,OAAO,iBAAiB,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;IACrD,CAAC;IAED;;;;;OAKG;IACH,eAAe,CAAC,EAAa;QAC3B,OAAO,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACvC,CAAC;IAED;;;;;OAKG;IACH,eAAe,CAAC,EAAa;QAC3B,OAAO,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACvC,CAAC;IAED;;;;;OAKG;IACH,cAAc;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC,cAAc,EAAE,CAAC;IAC3C,CAAC;IAED;;;;;;OAMG;IACH,cAAc,CAAC,EAAa,EAAE,OAAe;QAC3C,IAAI,CAAC;YACH,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACtC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qBAAqB,EAAE,CAAC,IAAI,gBAAgB,OAAO,GAAG,CAAC,CAAC;QAC5E,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,8BAA8B,EAAE,CAAC,IAAI,gBAAgB,OAAO,MAAM,KAAK,EAAE,CAC1E,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;;;;;;;;OAUG;IACH,aAAa,CAAC,OAAwB;QACpC,MAAM,aAAa,GAAG,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC1E,OAAO,IAAI,OAAO,CAAC;YACjB,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,GAAG,IAAI,CAAC,cAAc,EAAE;YAC/C,OAAO,EAAE,IAAI,CAAC,cAAc;YAC5B,aAAa;SACd,CAAC,CAAC;IACL,CAAC;IAED;;;;;OAKG;IACK,KAAK,CAAC,WAAW,CAAC,EAAa;QACrC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;YAClD,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;gBACpB,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,kCAAkC,EAAE,CAAC,IAAI,oBAAoB,CAC9D,CAAC;gBACF,OAAO,IAAI,CAAC;YACd,CAAC;YAED,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,gCAAgC,EAAE,CAAC,IAAI,iBAAiB,CACzD,CAAC;YACF,OAAO,MAAuB,CAAC;QACjC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,+BAA+B,EAAE,CAAC,IAAI,MAAM,GAAG,EAAE,CAAC,CAAC;YACrE,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACK,KAAK,CAAC,WAAW,CACvB,EAAa,EACb,MAAqB;QAErB,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;YAC3C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sBAAsB,EAAE,CAAC,IAAI,GAAG,CAAC,CAAC;QACtD,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,+BAA+B,EAAE,MAAM,GAAG,EAAE,CAAC,CAAC;QAClE,CAAC;IACH,CAAC"}

package/dist/ai-service/llm/context/utils/content-compaction.d.ts

@@ -8,7 +8,7 @@ import type { ContentPart } from "../internal-types.js";
 import type { ReasoningPart } from "@ai-sdk/provider-utils";
 import type { TextPart, ToolCallPart, ToolResultPart } from "ai";
 /**
- * Compacts content parts by removing tool interaction details.
+ * Compacts content by trimming tool calls and multimedia.
  *
  * Used when demoting turns to L2 (moderate compression):
  * - Text parts: Preserved unchanged
@@ -22,5 +22,5 @@ import type { TextPart, ToolCallPart, ToolResultPart } from "ai";
  * @param parts - Content parts to compact
  * @returns Compacted content parts
  */
-export declare function compactContentParts(parts: ContentPart[]): (TextPart | ReasoningPart | ToolCallPart | ToolResultPart)[];
+export declare function compactContent(content: string | ContentPart[]): (TextPart | ReasoningPart | ToolCallPart | ToolResultPart)[];
 //# sourceMappingURL=content-compaction.d.ts.map

package/dist/ai-service/llm/context/utils/content-compaction.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"content-compaction.d.ts","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/utils/content-compaction.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,KAAK,EAAE,QAAQ,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,IAAI,CAAC;AAEjE;;;;;;;;;;;;;;GAcG;AACH,wBAAgB,mBAAmB,CACjC,KAAK,EAAE,WAAW,EAAE,GACnB,CAAC,QAAQ,GAAG,aAAa,GAAG,YAAY,GAAG,cAAc,CAAC,EAAE,CA0B9D"}
+{"version":3,"file":"content-compaction.d.ts","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/utils/content-compaction.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,KAAK,EAAE,QAAQ,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,IAAI,CAAC;AAEjE;;;;;;;;;;;;;;GAcG;AACH,wBAAgB,cAAc,CAC5B,OAAO,EAAE,MAAM,GAAG,WAAW,EAAE,GAC9B,CAAC,QAAQ,GAAG,aAAa,GAAG,YAAY,GAAG,cAAc,CAAC,EAAE,CA6B9D"}

package/dist/ai-service/llm/context/utils/content-compaction.js

@@ -5,7 +5,7 @@
  * removing tool call inputs and results while preserving the conversation structure.
  */
 /**
- * Compacts content parts by removing tool interaction details.
+ * Compacts content by trimming tool calls and multimedia.
  *
  * Used when demoting turns to L2 (moderate compression):
  * - Text parts: Preserved unchanged
@@ -19,8 +19,11 @@
  * @param parts - Content parts to compact
  * @returns Compacted content parts
  */
-export function compactContentParts(parts) {
-    return parts.flatMap((part) => {
+export function compactContent(content) {
+    if (typeof content === "string") {
+        return [{ type: "text", text: content }];
+    }
+    return content.flatMap((part) => {
         if (part.type === "text" || part.type === "reasoning") {
             return part;
         }

package/dist/ai-service/llm/context/utils/content-compaction.js.map

@@ -1 +1 @@
-{"version":3,"file":"content-compaction.js","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/utils/content-compaction.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAMH;;;;;;;;;;;;;;GAcG;AACH,MAAM,UAAU,mBAAmB,CACjC,KAAoB;IAEpB,OAAO,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QAC5B,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACtD,OAAO,IAAI,CAAC;QACd,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACrC,OAAO;gBACL,IAAI,EAAE,WAAW;gBACjB,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,KAAK,EAAE,EAAE;aACa,CAAC;QAC3B,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;YACvC,OAAO;gBACL,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,MAAM,EAAE;oBACN,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,uBAAuB;iBAC/B;aACuB,CAAC;QAC7B,CAAC;aAAM,CAAC;YACN,yBAAyB;YACzB,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC"}
+{"version":3,"file":"content-compaction.js","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/utils/content-compaction.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAMH;;;;;;;;;;;;;;GAcG;AACH,MAAM,UAAU,cAAc,CAC5B,OAA+B;IAE/B,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IAC3C,CAAC;IACD,OAAO,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QAC9B,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACtD,OAAO,IAAI,CAAC;QACd,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACrC,OAAO;gBACL,IAAI,EAAE,WAAW;gBACjB,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,KAAK,EAAE,EAAE;aACa,CAAC;QAC3B,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;YACvC,OAAO;gBACL,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,MAAM,EAAE;oBACN,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,uBAAuB;iBAC/B;aACuB,CAAC;QAC7B,CAAC;aAAM,CAAC;YACN,yBAAyB;YACzB,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/ai-service/llm/context/utils/index.d.ts

@@ -4,7 +4,7 @@
  * Internal utilities used by Context and Turn classes.
  */
 export { estimateTokenCount, countChars as countChars, } from "./token-estimation.js";
-export { compactContentParts } from "./content-compaction.js";
+export { compactContent } from "./content-compaction.js";
 export { normalizeToolResultOutput } from "./message-utils.js";
 export { renderContext } from "./visualization.js";
 //# sourceMappingURL=index.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/utils/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,kBAAkB,EAClB,UAAU,IAAI,UAAU,GACzB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAC9D,OAAO,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/utils/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,kBAAkB,EAClB,UAAU,IAAI,UAAU,GACzB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC"}

package/dist/ai-service/llm/context/utils/index.js

@@ -4,7 +4,7 @@
  * Internal utilities used by Context and Turn classes.
  */
 export { estimateTokenCount, countChars as countChars, } from "./token-estimation.js";
-export { compactContentParts } from "./content-compaction.js";
+export { compactContent } from "./content-compaction.js";
 export { normalizeToolResultOutput } from "./message-utils.js";
 export { renderContext } from "./visualization.js";
 //# sourceMappingURL=index.js.map

package/dist/ai-service/llm/context/utils/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/utils/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,kBAAkB,EAClB,UAAU,IAAI,UAAU,GACzB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAC9D,OAAO,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/utils/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,kBAAkB,EAClB,UAAU,IAAI,UAAU,GACzB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC"}

package/dist/ai-service/llm/context/utils/message-utils.d.ts

@@ -4,8 +4,9 @@
  * Ensures messages conform to expected formats, particularly for tool results
  * which must match the LanguageModelV2ToolResultOutput schema.
  */
+import type { TokenizedAssistantMessage } from "../internal-types.js";
 import type { LanguageModelV2ToolResultOutput } from "@ai-sdk/provider";
-import type { AssistantModelMessage } from "ai";
+import type { ModelMessage } from "ai";
 /**
  * Validates and normalizes a tool result output to ensure it conforms to
  * LanguageModelV2ToolResultOutput.
@@ -21,14 +22,23 @@ import type { AssistantModelMessage } from "ai";
  */
 export declare function normalizeToolResultOutput(output: unknown): LanguageModelV2ToolResultOutput;
 /**
- * Ensures an assistant message's content starts with a reasoning block.
- * If the first content part is not a reasoning block, one is prepended.
+ * Ensures an assistant message's content starts with reasoning blocks.
+ * Reorders content so all reasoning blocks are first, as required by AWS Bedrock.
+ * If no reasoning blocks exist, prepends an empty one.
  *
- * Avoids issues that can arise when dropping messages, as some models
- * expect reasoning blocks in specific positions in the turn sequence.
+ * AWS Bedrock requirement: "Thinking blocks must be the first content blocks
+ * in an assistant message" when using extended thinking with tool use.
  *
  * @param message - Assistant message to check and modify if needed
- * @returns Assistant message guaranteed to start with a reasoning block
  */
-export declare function prependReasoningBlockIfNeeded(message: AssistantModelMessage): AssistantModelMessage;
+export declare function prependReasoningBlockIfNeeded(message: TokenizedAssistantMessage): void;
+/** Checks if a message is empty.
+ * - For string content: empty if trimmed string length is 0.
+ * - For array content: empty if all parts are empty (textual parts
+ *   must have non-empty trimmed text; other part types are non-empty).
+ *
+ * @param message - Message to check
+ * @returns True if message is empty, false otherwise
+ */
+export declare function isEmpty(message: ModelMessage): boolean;
 //# sourceMappingURL=message-utils.d.ts.map

package/dist/ai-service/llm/context/utils/message-utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"message-utils.d.ts","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/utils/message-utils.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,KAAK,EAAE,+BAA+B,EAAE,MAAM,kBAAkB,CAAC;AACxE,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,IAAI,CAAC;AAEhD;;;;;;;;;;;;GAYG;AACH,wBAAgB,yBAAyB,CACvC,MAAM,EAAE,OAAO,GACd,+BAA+B,CAuBjC;AAED;;;;;;;;;GASG;AACH,wBAAgB,6BAA6B,CAC3C,OAAO,EAAE,qBAAqB,GAC7B,qBAAqB,CAuBvB"}
+{"version":3,"file":"message-utils.d.ts","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/utils/message-utils.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,sBAAsB,CAAC;AACtE,OAAO,KAAK,EAAE,+BAA+B,EAAE,MAAM,kBAAkB,CAAC;AACxE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAEvC;;;;;;;;;;;;GAYG;AACH,wBAAgB,yBAAyB,CACvC,MAAM,EAAE,OAAO,GACd,+BAA+B,CAuBjC;AAED;;;;;;;;;GASG;AACH,wBAAgB,6BAA6B,CAC3C,OAAO,EAAE,yBAAyB,QAuBnC;AAED;;;;;;;GAOG;AACH,wBAAgB,OAAO,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAYtD"}

package/dist/ai-service/llm/context/utils/message-utils.js

@@ -37,14 +37,14 @@ export function normalizeToolResultOutput(output) {
     return output;
 }
 /**
- * Ensures an assistant message's content starts with a reasoning block.
- * If the first content part is not a reasoning block, one is prepended.
+ * Ensures an assistant message's content starts with reasoning blocks.
+ * Reorders content so all reasoning blocks are first, as required by AWS Bedrock.
+ * If no reasoning blocks exist, prepends an empty one.
  *
- * Avoids issues that can arise when dropping messages, as some models
- * expect reasoning blocks in specific positions in the turn sequence.
+ * AWS Bedrock requirement: "Thinking blocks must be the first content blocks
+ * in an assistant message" when using extended thinking with tool use.
  *
  * @param message - Assistant message to check and modify if needed
- * @returns Assistant message guaranteed to start with a reasoning block
  */
 export function prependReasoningBlockIfNeeded(message) {
     if (typeof message.content == "string") {
@@ -55,19 +55,32 @@ export function prependReasoningBlockIfNeeded(message) {
             },
         ];
     }
-    const firstPart = message.content.at(0);
-    if (!firstPart || firstPart.type === "reasoning") {
-        return message;
+    const reasoningBlocks = message.content.filter((part) => part.type === "reasoning");
+    const otherBlocks = message.content.filter((part) => part.type !== "reasoning");
+    if (reasoningBlocks.length === 0) {
+        reasoningBlocks.push({ type: "reasoning", text: "" });
     }
-    return {
-        ...message,
-        content: [
-            {
-                type: "reasoning",
-                text: "",
-            },
-            ...message.content,
-        ],
-    };
+    message.content = [...reasoningBlocks, ...otherBlocks];
+}
+/** Checks if a message is empty.
+ * - For string content: empty if trimmed string length is 0.
+ * - For array content: empty if all parts are empty (textual parts
+ *   must have non-empty trimmed text; other part types are non-empty).
+ *
+ * @param message - Message to check
+ * @returns True if message is empty, false otherwise
+ */
+export function isEmpty(message) {
+    if (typeof message.content === "string") {
+        return message.content.trim().length === 0;
+    }
+    const nonEmptyParts = message.content.filter((part) => {
+        if ("text" in part) {
+            return part.text.trim().length > 0;
+        }
+        // non-textual content parts are never considered empty
+        return true;
+    });
+    return nonEmptyParts.length === 0;
 }
 //# sourceMappingURL=message-utils.js.map

package/dist/ai-service/llm/context/utils/message-utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"message-utils.js","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/utils/message-utils.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,4BAA4B,CAAC;AAIvD;;;;;;;;;;;;GAYG;AACH,MAAM,UAAU,yBAAyB,CACvC,MAAe;IAEf,IAAI,MAAM,KAAK,IAAI,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QAC5C,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAqC,CAAC;IAC1E,CAAC;IAED,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAqC,CAAC;IAC5E,CAAC;IAED,MAAM,SAAS,GAAG,MAAiC,CAAC;IACpD,IAAI,CAAC,CAAC,MAAM,IAAI,SAAS,CAAC,IAAI,OAAO,SAAS,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QACjE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAqC,CAAC;IAC5E,CAAC;IAED,MAAM,UAAU,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,YAAY,CAAC,CAAC;IAChE,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;QACzC,SAAS,EAAE,CAAC,IAAI,CACd,8CAA8C,SAAS,CAAC,IAAI,6BAA6B,CAC1F,CAAC;QACF,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAqC,CAAC;IAC5E,CAAC;IAED,OAAO,MAAyC,CAAC;AACnD,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,UAAU,6BAA6B,CAC3C,OAA8B;IAE9B,IAAI,OAAO,OAAO,CAAC,OAAO,IAAI,QAAQ,EAAE,CAAC;QACvC,OAAO,CAAC,OAAO,GAAG;YAChB;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,OAAO,CAAC,OAAO;aACtB;SACF,CAAC;IACJ,CAAC;IACD,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACxC,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;QACjD,OAAO,OAAO,CAAC;IACjB,CAAC;IACD,OAAO;QACL,GAAG,OAAO;QACV,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,EAAE;aACT;YACD,GAAG,OAAO,CAAC,OAAO;SACnB;KACF,CAAC;AACJ,CAAC"}
+{"version":3,"file":"message-utils.js","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/utils/message-utils.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,4BAA4B,CAAC;AAKvD;;;;;;;;;;;;GAYG;AACH,MAAM,UAAU,yBAAyB,CACvC,MAAe;IAEf,IAAI,MAAM,KAAK,IAAI,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QAC5C,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAqC,CAAC;IAC1E,CAAC;IAED,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAqC,CAAC;IAC5E,CAAC;IAED,MAAM,SAAS,GAAG,MAAiC,CAAC;IACpD,IAAI,CAAC,CAAC,MAAM,IAAI,SAAS,CAAC,IAAI,OAAO,SAAS,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QACjE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAqC,CAAC;IAC5E,CAAC;IAED,MAAM,UAAU,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,YAAY,CAAC,CAAC;IAChE,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;QACzC,SAAS,EAAE,CAAC,IAAI,CACd,8CAA8C,SAAS,CAAC,IAAI,6BAA6B,CAC1F,CAAC;QACF,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAqC,CAAC;IAC5E,CAAC;IAED,OAAO,MAAyC,CAAC;AACnD,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,UAAU,6BAA6B,CAC3C,OAAkC;IAElC,IAAI,OAAO,OAAO,CAAC,OAAO,IAAI,QAAQ,EAAE,CAAC;QACvC,OAAO,CAAC,OAAO,GAAG;YAChB;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,OAAO,CAAC,OAAO;aACtB;SACF,CAAC;IACJ,CAAC;IAED,MAAM,eAAe,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAC5C,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,CACpC,CAAC;IACF,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CACxC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,CACpC,CAAC;IAEF,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACjC,eAAe,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;IACxD,CAAC;IAED,OAAO,CAAC,OAAO,GAAG,CAAC,GAAG,eAAe,EAAE,GAAG,WAAW,CAAC,CAAC;AACzD,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,OAAO,CAAC,OAAqB;IAC3C,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;QACxC,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,CAAC;IAC7C,CAAC;IACD,MAAM,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;QACpD,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;YACnB,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC;QACrC,CAAC;QACD,uDAAuD;QACvD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IACH,OAAO,aAAa,CAAC,MAAM,KAAK,CAAC,CAAC;AACpC,CAAC"}

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

@@ -2,17 +2,24 @@ import { streamText, type ToolSet } from "ai";
 import { type Logger } from "../../util/logger.js";
 import { LLMObsTracer } from "./tracer.js";
 import type { LLMObsSpan } from "./types.js";
+import type { RetryOptions } from "../util/retry-on-timeout.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>;
 export declare function getJwtTraceTags(jwt: string | undefined): Record<string, string>;
 export declare function getCommonTraceTags(jwt: string | undefined, applicationId: string, organizationId: string, additionalTags?: Record<string, string | boolean>): Record<string, string | boolean>;
 /**
- * Instruments ai.streamText with LLMObs tracing.
+ * Instruments ai.streamText with LLMObs tracing and retry logic.
  * Wraps user-provided callbacks to ensure spans are finished correctly.
  * Captures input/output data, metrics, errors, and metadata for each step and the overall call.
+ * Automatically retries on timeout errors with exponential backoff.
+ *
+ * @param params - Parameters to pass to streamText
+ * @param llmobs - LLMObs tracer instance
+ * @param logger - Logger instance
+ * @param retryOptions - Optional retry configuration (defaults to 3 retries with exponential backoff)
  */
-export declare function tracedStreamText<T extends ToolSet>(params: Parameters<typeof streamText<T>>[0], llmobs?: LLMObsTracer, logger?: Logger): import("ai").StreamTextResult<T, never>;
+export declare function tracedStreamText<T extends ToolSet>(params: Parameters<typeof streamText<T>>[0], llmobs?: LLMObsTracer, logger?: Logger, retryOptions?: RetryOptions): 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;

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;;;;GAIG;AAEH,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,2CAI7B;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,CA8CF,CAAC"}
+{"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;AAK9D,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAGhE,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;;;;;;;;;;GAUG;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,EAC5B,YAAY,GAAE,YAAiB,2CAYhC;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,CA8CF,CAAC"}

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

@@ -2,6 +2,7 @@ import { streamText } from "ai";
 import { getLogger } from "../../util/logger.js";
 import { parseJwt } from "../util/parse-jwt.js";
 import { safeJsonStringify } from "../util/safe-stringify.js";
+import { RetryMiddleware } from "./middleware/retry.js";
 import { LLMObsStreamTextMiddleware } from "./middleware/stream-text.js";
 import { LLMObsTracer } from "./tracer.js";
 // Helper function for workflow tracing with common patterns
@@ -69,13 +70,25 @@ export function getCommonTraceTags(jwt, applicationId, organizationId, additiona
     return result;
 }
 /**
- * Instruments ai.streamText with LLMObs tracing.
+ * Instruments ai.streamText with LLMObs tracing and retry logic.
  * Wraps user-provided callbacks to ensure spans are finished correctly.
  * Captures input/output data, metrics, errors, and metadata for each step and the overall call.
+ * Automatically retries on timeout errors with exponential backoff.
+ *
+ * @param params - Parameters to pass to streamText
+ * @param llmobs - LLMObs tracer instance
+ * @param logger - Logger instance
+ * @param retryOptions - Optional retry configuration (defaults to 3 retries with exponential backoff)
  */
-export function tracedStreamText(params, llmobs = LLMObsTracer.instance, logger = getLogger()) {
-    const middleware = new LLMObsStreamTextMiddleware(llmobs, logger);
-    return middleware.wrap(streamText)(params);
+export function tracedStreamText(params, llmobs = LLMObsTracer.instance, logger = getLogger(), retryOptions = {}) {
+    // Compose middlewares: retry first (inner), then LLMObs tracing (outer)
+    // This ensures retries happen before tracing sees the error
+    const retryMiddleware = new RetryMiddleware(retryOptions, logger);
+    const llmobsMiddleware = new LLMObsStreamTextMiddleware(llmobs, logger);
+    // Apply retry middleware first, then LLMObs tracing
+    const retriedStreamText = retryMiddleware.wrap(streamText);
+    const tracedStreamText = llmobsMiddleware.wrap(retriedStreamText);
+    return tracedStreamText(params);
 }
 export const traceToolSet = (toolset, llmobs = LLMObsTracer.instance, hooks) => {
     const tracedToolset = {};