llmist 2.5.0 → 2.6.0

package/dist/testing/index.cjs

@@ -134,6 +134,12 @@ function isAudioPart(part) {
 function text(content) {
   return { type: "text", text: content };
 }
+function imageFromBase64(data, mediaType) {
+  return {
+    type: "image",
+    source: { type: "base64", mediaType, data }
+  };
+}
 function imageFromUrl(url) {
   return {
     type: "image",
@@ -194,6 +200,12 @@ function toBase64(data) {
   }
   return Buffer.from(data).toString("base64");
 }
+function audioFromBase64(data, mediaType) {
+  return {
+    type: "audio",
+    source: { type: "base64", mediaType, data }
+  };
+}
 function audioFromBuffer(buffer, mediaType) {
   const detectedType = mediaType ?? detectAudioMimeType(buffer);
   if (!detectedType) {
@@ -240,7 +252,9 @@ var init_input_content = __esm({
       // WAV (RIFF)
       { bytes: [82, 73, 70, 70], mimeType: "audio/wav" },
       // WebM
-      { bytes: [26, 69, 223, 163], mimeType: "audio/webm" }
+      { bytes: [26, 69, 223, 163], mimeType: "audio/webm" },
+      // FLAC (fLaC)
+      { bytes: [102, 76, 97, 67], mimeType: "audio/flac" }
     ];
   }
 });
@@ -653,7 +667,7 @@ Produces: { "items": ["first", "second"] }`);
         this.messages.push({ role: "user", content: parts });
         return this;
       }
-      addGadgetCall(gadget, parameters, result) {
+      addGadgetCall(gadget, parameters, result, media, mediaIds) {
         const paramStr = this.formatBlockParameters(parameters, "");
         this.messages.push({
           role: "assistant",
@@ -661,10 +675,25 @@ Produces: { "items": ["first", "second"] }`);
 ${paramStr}
 ${this.endPrefix}`
         });
-        this.messages.push({
-          role: "user",
-          content: `Result: ${result}`
-        });
+        if (media && media.length > 0 && mediaIds && mediaIds.length > 0) {
+          const idRefs = media.map((m, i) => `[Media: ${mediaIds[i]} (${m.kind})]`).join("\n");
+          const textWithIds = `Result: ${result}
+${idRefs}`;
+          const parts = [text(textWithIds)];
+          for (const item of media) {
+            if (item.kind === "image") {
+              parts.push(imageFromBase64(item.data, item.mimeType));
+            } else if (item.kind === "audio") {
+              parts.push(audioFromBase64(item.data, item.mimeType));
+            }
+          }
+          this.messages.push({ role: "user", content: parts });
+        } else {
+          this.messages.push({
+            role: "user",
+            content: `Result: ${result}`
+          });
+        }
         return this;
       }
       /**
@@ -998,6 +1027,210 @@ var init_registry = __esm({
   }
 });
 
+// src/gadgets/media-store.ts
+function getLlmistTmpDir() {
+  return (0, import_node_path2.join)((0, import_node_os.homedir)(), ".llmist", "tmp");
+}
+var import_node_crypto, import_promises, import_node_os, import_node_path2, MIME_TO_EXTENSION, MediaStore;
+var init_media_store = __esm({
+  "src/gadgets/media-store.ts"() {
+    "use strict";
+    import_node_crypto = require("crypto");
+    import_promises = require("fs/promises");
+    import_node_os = require("os");
+    import_node_path2 = require("path");
+    MIME_TO_EXTENSION = {
+      // Images
+      "image/png": ".png",
+      "image/jpeg": ".jpg",
+      "image/gif": ".gif",
+      "image/webp": ".webp",
+      "image/svg+xml": ".svg",
+      "image/bmp": ".bmp",
+      "image/tiff": ".tiff",
+      // Audio
+      "audio/mp3": ".mp3",
+      "audio/mpeg": ".mp3",
+      "audio/wav": ".wav",
+      "audio/webm": ".webm",
+      "audio/ogg": ".ogg",
+      "audio/flac": ".flac",
+      "audio/aac": ".aac",
+      // Video
+      "video/mp4": ".mp4",
+      "video/webm": ".webm",
+      "video/ogg": ".ogv",
+      "video/quicktime": ".mov",
+      "video/x-msvideo": ".avi",
+      // Documents
+      "application/pdf": ".pdf",
+      "application/json": ".json",
+      "text/plain": ".txt",
+      "text/html": ".html",
+      "text/css": ".css",
+      "text/javascript": ".js"
+    };
+    MediaStore = class {
+      items = /* @__PURE__ */ new Map();
+      outputDir;
+      counter = 0;
+      initialized = false;
+      /**
+       * Create a new MediaStore.
+       *
+       * @param sessionId - Optional session ID for the output directory.
+       *                    If not provided, a random ID is generated.
+       */
+      constructor(sessionId) {
+        const id = sessionId ?? (0, import_node_crypto.randomBytes)(8).toString("hex");
+        this.outputDir = (0, import_node_path2.join)(getLlmistTmpDir(), `media-${id}`);
+      }
+      /**
+       * Get the output directory path.
+       */
+      getOutputDir() {
+        return this.outputDir;
+      }
+      /**
+       * Ensure the output directory exists.
+       * @throws Error if directory creation fails
+       */
+      async ensureDir() {
+        if (this.initialized) return;
+        try {
+          await (0, import_promises.mkdir)(this.outputDir, { recursive: true });
+          this.initialized = true;
+        } catch (error) {
+          throw new Error(
+            `MediaStore: Failed to create directory ${this.outputDir}: ${error instanceof Error ? error.message : String(error)}`
+          );
+        }
+      }
+      /**
+       * Generate a unique media ID.
+       * Format: "media_" + 6 random alphanumeric characters
+       */
+      generateId() {
+        const chars = "abcdefghijklmnopqrstuvwxyz0123456789";
+        let id = "media_";
+        const bytes = (0, import_node_crypto.randomBytes)(6);
+        for (let i = 0; i < 6; i++) {
+          id += chars[bytes[i] % chars.length];
+        }
+        return id;
+      }
+      /**
+       * Get file extension from MIME type.
+       */
+      getExtension(mimeType) {
+        return MIME_TO_EXTENSION[mimeType] ?? ".bin";
+      }
+      /**
+       * Store media and return stored metadata with ID.
+       *
+       * @param media - The media output from a gadget
+       * @param gadgetName - Name of the gadget that created this media
+       * @returns Stored media information including generated ID
+       * @throws Error if file write fails
+       */
+      async store(media, gadgetName) {
+        await this.ensureDir();
+        const id = this.generateId();
+        const ext = this.getExtension(media.mimeType);
+        const filename = media.fileName ?? `${gadgetName}_${String(++this.counter).padStart(3, "0")}${ext}`;
+        const filePath = (0, import_node_path2.join)(this.outputDir, filename);
+        const buffer = Buffer.from(media.data, "base64");
+        try {
+          await (0, import_promises.writeFile)(filePath, buffer);
+        } catch (error) {
+          throw new Error(
+            `MediaStore: Failed to write media file ${filePath}: ${error instanceof Error ? error.message : String(error)}`
+          );
+        }
+        const stored = {
+          id,
+          kind: media.kind,
+          path: filePath,
+          mimeType: media.mimeType,
+          sizeBytes: buffer.length,
+          description: media.description,
+          metadata: media.metadata,
+          gadgetName,
+          createdAt: /* @__PURE__ */ new Date()
+        };
+        this.items.set(id, stored);
+        return stored;
+      }
+      /**
+       * Get stored media by ID.
+       *
+       * @param id - The media ID (e.g., "media_a1b2c3")
+       * @returns The stored media or undefined if not found
+       */
+      get(id) {
+        return this.items.get(id);
+      }
+      /**
+       * Get the actual file path for a media ID.
+       * Convenience method for gadgets that need the raw path.
+       *
+       * @param id - The media ID
+       * @returns The file path or undefined if not found
+       */
+      getPath(id) {
+        return this.items.get(id)?.path;
+      }
+      /**
+       * List all stored media, optionally filtered by kind.
+       *
+       * @param kind - Optional media kind to filter by
+       * @returns Array of stored media items
+       */
+      list(kind) {
+        const all = Array.from(this.items.values());
+        if (kind) {
+          return all.filter((item) => item.kind === kind);
+        }
+        return all;
+      }
+      /**
+       * Get the count of stored media items.
+       */
+      get size() {
+        return this.items.size;
+      }
+      /**
+       * Check if a media ID exists.
+       */
+      has(id) {
+        return this.items.has(id);
+      }
+      /**
+       * Clear in-memory store without deleting files.
+       * Resets the counter but leaves files on disk.
+       */
+      clear() {
+        this.items.clear();
+        this.counter = 0;
+      }
+      /**
+       * Delete all stored files and clear memory.
+       * Removes the entire session directory.
+       */
+      async cleanup() {
+        if (this.initialized) {
+          try {
+            await (0, import_promises.rm)(this.outputDir, { recursive: true, force: true });
+          } catch {
+          }
+          this.initialized = false;
+        }
+        this.clear();
+      }
+    };
+  }
+});
+
 // src/gadgets/exceptions.ts
 var BreakLoopException, HumanInputException, TimeoutException, AbortError;
 var init_exceptions = __esm({
@@ -1433,7 +1666,9 @@ var init_gadget = __esm({
               parts.push(`# ${example.comment}`);
             }
             parts.push(`${effectiveStartPrefix}${gadgetName}`);
-            parts.push(formatParamsAsBlock(example.params, "", effectiveArgPrefix));
+            parts.push(
+              formatParamsAsBlock(example.params, "", effectiveArgPrefix)
+            );
             parts.push(effectiveEndPrefix);
             if (example.output !== void 0) {
               parts.push("");
@@ -1635,6 +1870,18 @@ var init_output_viewer = __esm({
   }
 });
 
+// src/agent/agent-internal-key.ts
+function isValidAgentKey(key) {
+  return key === AGENT_INTERNAL_KEY;
+}
+var AGENT_INTERNAL_KEY;
+var init_agent_internal_key = __esm({
+  "src/agent/agent-internal-key.ts"() {
+    "use strict";
+    AGENT_INTERNAL_KEY = Symbol("AGENT_INTERNAL_KEY");
+  }
+});
+
 // src/agent/compaction/config.ts
 function resolveCompactionConfig(config = {}) {
   const trigger = config.triggerThresholdPercent ?? DEFAULT_COMPACTION_CONFIG.triggerThresholdPercent;
@@ -1882,9 +2129,9 @@ var init_hybrid = __esm({
 var init_strategies = __esm({
   "src/agent/compaction/strategies/index.ts"() {
     "use strict";
+    init_hybrid();
     init_sliding_window();
     init_summarization();
-    init_hybrid();
   }
 });
 
@@ -2046,98 +2293,6 @@ var init_manager = __esm({
   }
 });
 
-// src/agent/gadget-output-store.ts
-var import_node_crypto, GadgetOutputStore;
-var init_gadget_output_store = __esm({
-  "src/agent/gadget-output-store.ts"() {
-    "use strict";
-    import_node_crypto = require("crypto");
-    GadgetOutputStore = class {
-      outputs = /* @__PURE__ */ new Map();
-      /**
-       * Store a gadget output and return its ID.
-       *
-       * @param gadgetName - Name of the gadget that produced the output
-       * @param content - Full output content to store
-       * @returns Generated ID for retrieving the output later
-       */
-      store(gadgetName, content) {
-        const id = this.generateId(gadgetName);
-        const encoder = new TextEncoder();
-        const stored = {
-          id,
-          gadgetName,
-          content,
-          byteSize: encoder.encode(content).length,
-          lineCount: content.split("\n").length,
-          timestamp: /* @__PURE__ */ new Date()
-        };
-        this.outputs.set(id, stored);
-        return id;
-      }
-      /**
-       * Retrieve a stored output by ID.
-       *
-       * @param id - The output ID (e.g., "Search_d34db33f")
-       * @returns The stored output or undefined if not found
-       */
-      get(id) {
-        return this.outputs.get(id);
-      }
-      /**
-       * Check if an output exists.
-       *
-       * @param id - The output ID to check
-       * @returns True if the output exists
-       */
-      has(id) {
-        return this.outputs.has(id);
-      }
-      /**
-       * Get all stored output IDs.
-       *
-       * @returns Array of output IDs
-       */
-      getIds() {
-        return Array.from(this.outputs.keys());
-      }
-      /**
-       * Get the number of stored outputs.
-       */
-      get size() {
-        return this.outputs.size;
-      }
-      /**
-       * Clear all stored outputs.
-       * Called when the agent run completes.
-       */
-      clear() {
-        this.outputs.clear();
-      }
-      /**
-       * Generate a unique ID for a stored output.
-       * Format: {GadgetName}_{8 hex chars}
-       */
-      generateId(gadgetName) {
-        const hex = (0, import_node_crypto.randomBytes)(4).toString("hex");
-        return `${gadgetName}_${hex}`;
-      }
-    };
-  }
-});
-
-// src/agent/agent-internal-key.ts
-function isValidAgentKey(key) {
-  return key === AGENT_INTERNAL_KEY;
-}
-var AGENT_INTERNAL_KEY;
-var init_agent_internal_key = __esm({
-  "src/agent/agent-internal-key.ts"() {
-    "use strict";
-    AGENT_INTERNAL_KEY = Symbol("AGENT_INTERNAL_KEY");
-  }
-});
-
 // src/agent/conversation-manager.ts
 var ConversationManager;
 var init_conversation_manager = __esm({
@@ -2168,8 +2323,8 @@ var init_conversation_manager = __esm({
       addAssistantMessage(content) {
         this.historyBuilder.addAssistant(content);
       }
-      addGadgetCall(gadgetName, parameters, result) {
-        this.historyBuilder.addGadgetCall(gadgetName, parameters, result);
+      addGadgetCall(gadgetName, parameters, result, media, mediaIds) {
+        this.historyBuilder.addGadgetCall(gadgetName, parameters, result, media, mediaIds);
       }
       getMessages() {
         return [...this.baseMessages, ...this.initialMessages, ...this.historyBuilder.build()];
@@ -2253,62 +2408,142 @@ var init_event_handlers = __esm({
   }
 });
 
-// src/agent/hook-validators.ts
-function validateBeforeLLMCallAction(action) {
-  if (!action || typeof action !== "object" || !("action" in action)) {
-    throw new HookValidationError(
-      "beforeLLMCall",
-      "Must return an action object with an 'action' field"
-    );
-  }
-  const actionType = action.action;
-  if (actionType !== "proceed" && actionType !== "skip") {
-    throw new HookValidationError(
-      "beforeLLMCall",
-      `Invalid action type: ${actionType}. Must be 'proceed' or 'skip'`
-    );
-  }
-  if (actionType === "skip" && !action.syntheticResponse) {
-    throw new HookValidationError(
-      "beforeLLMCall",
-      "When action is 'skip', syntheticResponse is required"
-    );
-  }
-}
-function validateAfterLLMCallAction(action) {
-  if (!action || typeof action !== "object" || !("action" in action)) {
-    throw new HookValidationError(
-      "afterLLMCall",
-      "Must return an action object with an 'action' field"
-    );
-  }
-  const actionType = action.action;
-  const validActions = ["continue", "append_messages", "modify_and_continue", "append_and_modify"];
-  if (!validActions.includes(actionType)) {
-    throw new HookValidationError(
-      "afterLLMCall",
-      `Invalid action type: ${actionType}. Must be one of: ${validActions.join(", ")}`
-    );
-  }
-  if (actionType === "append_messages" || actionType === "append_and_modify") {
-    if (!("messages" in action) || !action.messages || !Array.isArray(action.messages)) {
-      throw new HookValidationError(
-        "afterLLMCall",
-        `When action is '${actionType}', messages array is required`
-      );
-    }
-    if (action.messages.length === 0) {
-      throw new HookValidationError(
-        "afterLLMCall",
-        `When action is '${actionType}', messages array must not be empty`
-      );
-    }
-    for (let i = 0; i < action.messages.length; i++) {
-      const msg = action.messages[i];
-      if (!msg || typeof msg !== "object") {
-        throw new HookValidationError("afterLLMCall", `Message at index ${i} must be an object`);
-      }
-      if (!msg.role || !msg.content) {
+// src/agent/gadget-output-store.ts
+var import_node_crypto2, GadgetOutputStore;
+var init_gadget_output_store = __esm({
+  "src/agent/gadget-output-store.ts"() {
+    "use strict";
+    import_node_crypto2 = require("crypto");
+    GadgetOutputStore = class {
+      outputs = /* @__PURE__ */ new Map();
+      /**
+       * Store a gadget output and return its ID.
+       *
+       * @param gadgetName - Name of the gadget that produced the output
+       * @param content - Full output content to store
+       * @returns Generated ID for retrieving the output later
+       */
+      store(gadgetName, content) {
+        const id = this.generateId(gadgetName);
+        const encoder = new TextEncoder();
+        const stored = {
+          id,
+          gadgetName,
+          content,
+          byteSize: encoder.encode(content).length,
+          lineCount: content.split("\n").length,
+          timestamp: /* @__PURE__ */ new Date()
+        };
+        this.outputs.set(id, stored);
+        return id;
+      }
+      /**
+       * Retrieve a stored output by ID.
+       *
+       * @param id - The output ID (e.g., "Search_d34db33f")
+       * @returns The stored output or undefined if not found
+       */
+      get(id) {
+        return this.outputs.get(id);
+      }
+      /**
+       * Check if an output exists.
+       *
+       * @param id - The output ID to check
+       * @returns True if the output exists
+       */
+      has(id) {
+        return this.outputs.has(id);
+      }
+      /**
+       * Get all stored output IDs.
+       *
+       * @returns Array of output IDs
+       */
+      getIds() {
+        return Array.from(this.outputs.keys());
+      }
+      /**
+       * Get the number of stored outputs.
+       */
+      get size() {
+        return this.outputs.size;
+      }
+      /**
+       * Clear all stored outputs.
+       * Called when the agent run completes.
+       */
+      clear() {
+        this.outputs.clear();
+      }
+      /**
+       * Generate a unique ID for a stored output.
+       * Format: {GadgetName}_{8 hex chars}
+       */
+      generateId(gadgetName) {
+        const hex = (0, import_node_crypto2.randomBytes)(4).toString("hex");
+        return `${gadgetName}_${hex}`;
+      }
+    };
+  }
+});
+
+// src/agent/hook-validators.ts
+function validateBeforeLLMCallAction(action) {
+  if (!action || typeof action !== "object" || !("action" in action)) {
+    throw new HookValidationError(
+      "beforeLLMCall",
+      "Must return an action object with an 'action' field"
+    );
+  }
+  const actionType = action.action;
+  if (actionType !== "proceed" && actionType !== "skip") {
+    throw new HookValidationError(
+      "beforeLLMCall",
+      `Invalid action type: ${actionType}. Must be 'proceed' or 'skip'`
+    );
+  }
+  if (actionType === "skip" && !action.syntheticResponse) {
+    throw new HookValidationError(
+      "beforeLLMCall",
+      "When action is 'skip', syntheticResponse is required"
+    );
+  }
+}
+function validateAfterLLMCallAction(action) {
+  if (!action || typeof action !== "object" || !("action" in action)) {
+    throw new HookValidationError(
+      "afterLLMCall",
+      "Must return an action object with an 'action' field"
+    );
+  }
+  const actionType = action.action;
+  const validActions = ["continue", "append_messages", "modify_and_continue", "append_and_modify"];
+  if (!validActions.includes(actionType)) {
+    throw new HookValidationError(
+      "afterLLMCall",
+      `Invalid action type: ${actionType}. Must be one of: ${validActions.join(", ")}`
+    );
+  }
+  if (actionType === "append_messages" || actionType === "append_and_modify") {
+    if (!("messages" in action) || !action.messages || !Array.isArray(action.messages)) {
+      throw new HookValidationError(
+        "afterLLMCall",
+        `When action is '${actionType}', messages array is required`
+      );
+    }
+    if (action.messages.length === 0) {
+      throw new HookValidationError(
+        "afterLLMCall",
+        `When action is '${actionType}', messages array must not be empty`
+      );
+    }
+    for (let i = 0; i < action.messages.length; i++) {
+      const msg = action.messages[i];
+      if (!msg || typeof msg !== "object") {
+        throw new HookValidationError("afterLLMCall", `Message at index ${i} must be an object`);
+      }
+      if (!msg.role || !msg.content) {
         throw new HookValidationError(
           "afterLLMCall",
           `Message at index ${i} must have 'role' and 'content' fields`
@@ -2563,8 +2798,7 @@ var init_schema_introspector = __esm({
             const values = def?.values;
             const value = values?.[0] ?? def?.value;
             if (typeof value === "string") return "string";
-            if (typeof value === "number" || typeof value === "bigint")
-              return "number";
+            if (typeof value === "number" || typeof value === "bigint") return "number";
             if (typeof value === "boolean") return "boolean";
             return "unknown";
           }
@@ -2836,7 +3070,13 @@ var init_cost_reporting_client = __esm({
             cacheCreationInputTokens = chunk.usage.cacheCreationInputTokens ?? 0;
           }
         }
-        this.reportCostFromUsage(model, inputTokens, outputTokens, cachedInputTokens, cacheCreationInputTokens);
+        this.reportCostFromUsage(
+          model,
+          inputTokens,
+          outputTokens,
+          cachedInputTokens,
+          cacheCreationInputTokens
+        );
         return result;
       }
       /**
@@ -2876,7 +3116,13 @@ var init_cost_reporting_client = __esm({
             }
           }
         } finally {
-          this.reportCostFromUsage(model, inputTokens, outputTokens, cachedInputTokens, cacheCreationInputTokens);
+          this.reportCostFromUsage(
+            model,
+            inputTokens,
+            outputTokens,
+            cachedInputTokens,
+            cacheCreationInputTokens
+          );
         }
       }
       /**
@@ -2914,7 +3160,13 @@ var init_cost_reporting_client = __esm({
             }
           } finally {
             if (inputTokens > 0 || outputTokens > 0) {
-              reportCostFromUsage(model, inputTokens, outputTokens, cachedInputTokens, cacheCreationInputTokens);
+              reportCostFromUsage(
+                model,
+                inputTokens,
+                outputTokens,
+                cachedInputTokens,
+                cacheCreationInputTokens
+              );
             }
           }
         }
@@ -3126,7 +3378,11 @@ var init_parser = __esm({
           const metadataEndIndex = this.buffer.indexOf("\n", metadataStartIndex);
           if (metadataEndIndex === -1) break;
           const gadgetName = this.buffer.substring(metadataStartIndex, metadataEndIndex).trim();
-          const { actualName: actualGadgetName, invocationId, dependencies } = this.parseGadgetName(gadgetName);
+          const {
+            actualName: actualGadgetName,
+            invocationId,
+            dependencies
+          } = this.parseGadgetName(gadgetName);
           const contentStartIndex = metadataEndIndex + 1;
           let partEndIndex;
           let endMarkerLength = 0;
@@ -3174,7 +3430,11 @@ var init_parser = __esm({
           const metadataEndIndex = this.buffer.indexOf("\n", metadataStartIndex);
           if (metadataEndIndex !== -1) {
             const gadgetName = this.buffer.substring(metadataStartIndex, metadataEndIndex).trim();
-            const { actualName: actualGadgetName, invocationId, dependencies } = this.parseGadgetName(gadgetName);
+            const {
+              actualName: actualGadgetName,
+              invocationId,
+              dependencies
+            } = this.parseGadgetName(gadgetName);
             const contentStartIndex = metadataEndIndex + 1;
             const parametersRaw = this.buffer.substring(contentStartIndex).trim();
             const { parameters, parseError } = this.parseParameters(parametersRaw);
@@ -3219,11 +3479,12 @@ var init_executor = __esm({
     init_exceptions();
     init_parser();
     GadgetExecutor = class {
-      constructor(registry, onHumanInputRequired, logger, defaultGadgetTimeoutMs, errorFormatterOptions, client) {
+      constructor(registry, onHumanInputRequired, logger, defaultGadgetTimeoutMs, errorFormatterOptions, client, mediaStore) {
         this.registry = registry;
         this.onHumanInputRequired = onHumanInputRequired;
         this.defaultGadgetTimeoutMs = defaultGadgetTimeoutMs;
         this.client = client;
+        this.mediaStore = mediaStore;
         this.logger = logger ?? createLogger({ name: "llmist:executor" });
         this.errorFormatter = new GadgetErrorFormatter(errorFormatterOptions);
         this.argPrefix = errorFormatterOptions?.argPrefix ?? GADGET_ARG_PREFIX;
@@ -3246,12 +3507,16 @@ var init_executor = __esm({
       }
       /**
        * Normalizes gadget execute result to consistent format.
-       * Handles both string returns (backwards compat) and object returns with cost.
+       * Handles string returns (backwards compat), object returns with cost,
+       * and object returns with media.
        */
       normalizeExecuteResult(raw) {
         if (typeof raw === "string") {
           return { result: raw, cost: 0 };
         }
+        if ("media" in raw && raw.media) {
+          return { result: raw.result, media: raw.media, cost: raw.cost ?? 0 };
+        }
         return { result: raw.result, cost: raw.cost ?? 0 };
       }
       // Execute a gadget call asynchronously
@@ -3378,8 +3643,21 @@ var init_executor = __esm({
           } else {
             rawResult = await Promise.resolve(gadget.execute(validatedParameters, ctx));
           }
-          const { result, cost: returnCost } = this.normalizeExecuteResult(rawResult);
+          const { result, media, cost: returnCost } = this.normalizeExecuteResult(rawResult);
           const totalCost = callbackCost + returnCost;
+          let mediaIds;
+          let storedMedia;
+          if (media && media.length > 0 && this.mediaStore) {
+            storedMedia = await Promise.all(
+              media.map((item) => this.mediaStore.store(item, call.gadgetName))
+            );
+            mediaIds = storedMedia.map((m) => m.id);
+            this.logger.debug("Stored media outputs", {
+              gadgetName: call.gadgetName,
+              mediaIds,
+              count: media.length
+            });
+          }
           const executionTimeMs = Date.now() - startTime;
           this.logger.info("Gadget executed successfully", {
             gadgetName: call.gadgetName,
@@ -3387,7 +3665,8 @@ var init_executor = __esm({
             executionTimeMs,
             cost: totalCost > 0 ? totalCost : void 0,
             callbackCost: callbackCost > 0 ? callbackCost : void 0,
-            returnCost: returnCost > 0 ? returnCost : void 0
+            returnCost: returnCost > 0 ? returnCost : void 0,
+            mediaCount: media?.length
           });
           this.logger.debug("Gadget result", {
             gadgetName: call.gadgetName,
@@ -3395,7 +3674,8 @@ var init_executor = __esm({
             parameters: validatedParameters,
             result,
             cost: totalCost,
-            executionTimeMs
+            executionTimeMs,
+            mediaIds
           });
           return {
             gadgetName: call.gadgetName,
@@ -3403,7 +3683,10 @@ var init_executor = __esm({
             parameters: validatedParameters,
             result,
             executionTimeMs,
-            cost: totalCost
+            cost: totalCost,
+            media,
+            mediaIds,
+            storedMedia
           };
         } catch (error) {
           if (error instanceof BreakLoopException) {
@@ -3581,7 +3864,8 @@ var init_stream_processor = __esm({
           this.logger.getSubLogger({ name: "executor" }),
           options.defaultGadgetTimeoutMs,
           { argPrefix: options.gadgetArgPrefix },
-          options.client
+          options.client,
+          options.mediaStore
         );
       }
       /**
@@ -4170,13 +4454,14 @@ var init_agent = __esm({
     init_constants();
     init_messages();
     init_model_shortcuts();
+    init_media_store();
     init_output_viewer();
     init_logger();
-    init_manager();
-    init_gadget_output_store();
     init_agent_internal_key();
+    init_manager();
     init_conversation_manager();
     init_event_handlers();
+    init_gadget_output_store();
     init_hook_validators();
     init_stream_processor();
     Agent = class {
@@ -4205,6 +4490,8 @@ var init_agent = __esm({
       outputLimitCharLimit;
       // Context compaction
       compactionManager;
+      // Media storage (for gadgets returning images, audio, etc.)
+      mediaStore;
       // Cancellation
       signal;
       /**
@@ -4235,6 +4522,7 @@ var init_agent = __esm({
         this.defaultMaxTokens = this.resolveMaxTokensFromCatalog(options.model);
         this.outputLimitEnabled = options.gadgetOutputLimit ?? DEFAULT_GADGET_OUTPUT_LIMIT;
         this.outputStore = new GadgetOutputStore();
+        this.mediaStore = new MediaStore();
         const limitPercent = options.gadgetOutputLimitPercent ?? DEFAULT_GADGET_OUTPUT_LIMIT_PERCENT;
         const limits = this.client.modelRegistry.getModelLimits(this.model);
         const contextWindow = limits?.contextWindow ?? FALLBACK_CONTEXT_WINDOW;
@@ -4300,6 +4588,36 @@ var init_agent = __esm({
       getRegistry() {
         return this.registry;
       }
+      /**
+       * Get the media store for this agent session.
+       *
+       * The media store holds all media outputs (images, audio, etc.) produced by gadgets
+       * during this agent's execution. Use this to:
+       * - Access stored media files by ID
+       * - List all stored media
+       * - Clean up temporary files after execution
+       *
+       * @returns The MediaStore instance for this agent
+       *
+       * @example
+       * ```typescript
+       * const agent = new AgentBuilder()
+       *   .withModel("sonnet")
+       *   .build();
+       *
+       * // After execution, access stored media
+       * const store = agent.getMediaStore();
+       * for (const media of store.list()) {
+       *   console.log(`${media.id}: ${media.path}`);
+       * }
+       *
+       * // Clean up when done
+       * await store.cleanup();
+       * ```
+       */
+      getMediaStore() {
+        return this.mediaStore;
+      }
       /**
        * Manually trigger context compaction.
        *
@@ -4475,7 +4793,8 @@ var init_agent = __esm({
               stopOnGadgetError: this.stopOnGadgetError,
               shouldContinueAfterError: this.shouldContinueAfterError,
               defaultGadgetTimeoutMs: this.defaultGadgetTimeoutMs,
-              client: this.client
+              client: this.client,
+              mediaStore: this.mediaStore
             });
             const result = await processor.process(stream2);
             for (const output of result.outputs) {
@@ -4537,7 +4856,9 @@ var init_agent = __esm({
             }
             if (result.didExecuteGadgets) {
               if (this.textWithGadgetsHandler) {
-                const textContent = result.outputs.filter((output) => output.type === "text").map((output) => output.content).join("");
+                const textContent = result.outputs.filter(
+                  (output) => output.type === "text"
+                ).map((output) => output.content).join("");
                 if (textContent.trim()) {
                   const { gadgetName, parameterMapping, resultMapping } = this.textWithGadgetsHandler;
                   this.conversation.addGadgetCall(
@@ -4553,7 +4874,9 @@ var init_agent = __esm({
                   this.conversation.addGadgetCall(
                     gadgetResult.gadgetName,
                     gadgetResult.parameters,
-                    gadgetResult.error ?? gadgetResult.result ?? ""
+                    gadgetResult.error ?? gadgetResult.result ?? "",
+                    gadgetResult.media,
+                    gadgetResult.mediaIds
                   );
                 }
               }
@@ -7645,14 +7968,7 @@ var OPENAI_TTS_VOICES, OPENAI_TTS_EXTENDED_VOICES, OPENAI_TTS_FORMATS, openaiSpe
 var init_openai_speech_models = __esm({
   "src/providers/openai-speech-models.ts"() {
     "use strict";
-    OPENAI_TTS_VOICES = [
-      "alloy",
-      "echo",
-      "fable",
-      "onyx",
-      "nova",
-      "shimmer"
-    ];
+    OPENAI_TTS_VOICES = ["alloy", "echo", "fable", "onyx", "nova", "shimmer"];
     OPENAI_TTS_EXTENDED_VOICES = [
       ...OPENAI_TTS_VOICES,
       "ash",
@@ -8350,9 +8666,7 @@ var init_image = __esm({
         return this.findImageAdapter(modelId) !== void 0;
       }
       findImageAdapter(modelId) {
-        return this.adapters.find(
-          (adapter) => adapter.supportsImageGeneration?.(modelId) ?? false
-        );
+        return this.adapters.find((adapter) => adapter.supportsImageGeneration?.(modelId) ?? false);
       }
     };
   }
@@ -8404,9 +8718,7 @@ var init_speech = __esm({
         return this.findSpeechAdapter(modelId) !== void 0;
       }
       findSpeechAdapter(modelId) {
-        return this.adapters.find(
-          (adapter) => adapter.supportsSpeechGeneration?.(modelId) ?? false
-        );
+        return this.adapters.find((adapter) => adapter.supportsSpeechGeneration?.(modelId) ?? false);
       }
     };
   }
@@ -8517,11 +8829,7 @@ var init_vision = __esm({
             if (!parsed) {
               throw new Error("Invalid data URL format");
             }
-            builder.addUserWithImage(
-              options.prompt,
-              parsed.data,
-              parsed.mimeType
-            );
+            builder.addUserWithImage(options.prompt, parsed.data, parsed.mimeType);
           } else {
             const buffer = Buffer.from(options.image, "base64");
             builder.addUserWithImage(options.prompt, buffer, options.mimeType);
@@ -8952,52 +9260,277 @@ __export(testing_exports, {
 });
 module.exports = __toCommonJS(testing_exports);
 
-// src/gadgets/validation.ts
-function validateAndApplyDefaults(schema, params) {
-  const result = schema.safeParse(params);
-  if (result.success) {
-    return {
-      success: true,
-      data: result.data
-    };
-  }
-  const issues = result.error.issues.map((issue) => ({
-    path: issue.path.join(".") || "root",
-    message: issue.message
-  }));
-  const formattedError = `Invalid parameters: ${issues.map((i) => `${i.path}: ${i.message}`).join("; ")}`;
+// src/testing/cli-helpers.ts
+var import_node_stream = require("stream");
+function createTestEnvironment(options = {}) {
+  const stdin = createMockReadable(options.stdin);
+  const stdout = new import_node_stream.PassThrough();
+  const stderr = new import_node_stream.PassThrough();
+  let exitCode;
   return {
-    success: false,
-    error: formattedError,
-    issues
+    stdin,
+    stdout,
+    stderr,
+    isTTY: options.isTTY ?? false,
+    argv: options.argv ?? ["node", "llmist"],
+    env: { ...filterDefinedEnv(process.env), ...options.env },
+    get exitCode() {
+      return exitCode;
+    },
+    setExitCode: (code) => {
+      exitCode = code;
+    }
   };
 }
-function validateGadgetParams(gadget, params) {
-  if (!gadget.parameterSchema) {
-    return {
-      success: true,
-      data: params
-    };
+function createMockReadable(input) {
+  if (!input) {
+    const stream3 = new import_node_stream.Readable({ read() {
+    } });
+    stream3.push(null);
+    return stream3;
   }
-  return validateAndApplyDefaults(gadget.parameterSchema, params);
+  const content = Array.isArray(input) ? `${input.join("\n")}
+` : input;
+  const stream2 = new import_node_stream.Readable({ read() {
+  } });
+  stream2.push(content);
+  stream2.push(null);
+  return stream2;
 }
-
-// src/testing/gadget-testing.ts
-async function testGadget(gadget, params, options) {
-  let validatedParams = params;
-  if (!options?.skipValidation) {
-    const validationResult = validateGadgetParams(gadget, params);
-    if (!validationResult.success) {
-      return {
-        error: validationResult.error,
-        validatedParams: params
-      };
+function createMockWritable() {
+  const chunks = [];
+  const stream2 = new import_node_stream.Writable({
+    write(chunk, _encoding, callback) {
+      chunks.push(Buffer.from(chunk));
+      callback();
     }
-    validatedParams = validationResult.data;
-  }
-  try {
-    const rawResult = await Promise.resolve(gadget.execute(validatedParams));
-    if (typeof rawResult === "string") {
+  });
+  stream2.getData = () => Buffer.concat(chunks).toString("utf8");
+  return stream2;
+}
+async function collectOutput(stream2, timeout = 5e3) {
+  return new Promise((resolve, reject) => {
+    const chunks = [];
+    const timeoutId = setTimeout(() => {
+      resolve(Buffer.concat(chunks).toString("utf8"));
+    }, timeout);
+    stream2.on("data", (chunk) => {
+      chunks.push(Buffer.from(chunk));
+    });
+    stream2.on("end", () => {
+      clearTimeout(timeoutId);
+      resolve(Buffer.concat(chunks).toString("utf8"));
+    });
+    stream2.on("error", (err) => {
+      clearTimeout(timeoutId);
+      reject(err);
+    });
+  });
+}
+function getBufferedOutput(stream2) {
+  const chunks = [];
+  for (; ; ) {
+    const chunk = stream2.read();
+    if (chunk === null) break;
+    chunks.push(chunk);
+  }
+  return Buffer.concat(chunks).toString("utf8");
+}
+function createMockPrompt(responses) {
+  let index = 0;
+  return async (_question) => {
+    if (index >= responses.length) {
+      throw new Error(`Mock prompt exhausted: no response for question ${index + 1}`);
+    }
+    return responses[index++];
+  };
+}
+var MockPromptRecorder = class {
+  responses;
+  index = 0;
+  questions = [];
+  constructor(responses) {
+    this.responses = responses;
+  }
+  /**
+   * The prompt function to use in tests.
+   */
+  prompt = async (question) => {
+    this.questions.push(question);
+    if (this.index >= this.responses.length) {
+      throw new Error(`Mock prompt exhausted after ${this.index} questions`);
+    }
+    return this.responses[this.index++];
+  };
+  /**
+   * Get all questions that were asked.
+   */
+  getQuestions() {
+    return [...this.questions];
+  }
+  /**
+   * Get the number of questions asked.
+   */
+  getQuestionCount() {
+    return this.questions.length;
+  }
+  /**
+   * Reset the recorder state.
+   */
+  reset(newResponses) {
+    this.index = 0;
+    this.questions = [];
+    if (newResponses) {
+      this.responses = newResponses;
+    }
+  }
+};
+async function waitFor(condition, timeout = 5e3, interval = 50) {
+  const startTime = Date.now();
+  while (!condition()) {
+    if (Date.now() - startTime > timeout) {
+      throw new Error(`waitFor timed out after ${timeout}ms`);
+    }
+    await sleep(interval);
+  }
+}
+function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+function filterDefinedEnv(env) {
+  const result = {};
+  for (const [key, value] of Object.entries(env)) {
+    if (value !== void 0) {
+      result[key] = value;
+    }
+  }
+  return result;
+}
+
+// src/testing/conversation-fixtures.ts
+function createConversation(turnCount, options) {
+  const messages = [];
+  const userPrefix = options?.userPrefix ?? "User message";
+  const assistantPrefix = options?.assistantPrefix ?? "Assistant response";
+  const contentLength = options?.contentLength ?? 100;
+  for (let i = 0; i < turnCount; i++) {
+    const padding = " ".repeat(Math.max(0, contentLength - 30));
+    messages.push({
+      role: "user",
+      content: `${userPrefix} ${i + 1}: This is turn ${i + 1} of the conversation.${padding}`
+    });
+    messages.push({
+      role: "assistant",
+      content: `${assistantPrefix} ${i + 1}: I acknowledge turn ${i + 1}.${padding}`
+    });
+  }
+  return messages;
+}
+function createConversationWithGadgets(turnCount, gadgetCallsPerTurn = 1, options) {
+  const messages = [];
+  const gadgetNames = options?.gadgetNames ?? ["search", "calculate", "read"];
+  const contentLength = options?.contentLength ?? 50;
+  let gadgetIndex = 0;
+  for (let turn = 0; turn < turnCount; turn++) {
+    messages.push({
+      role: "user",
+      content: `User request ${turn + 1}${"x".repeat(contentLength)}`
+    });
+    for (let g = 0; g < gadgetCallsPerTurn; g++) {
+      const gadgetName = gadgetNames[gadgetIndex % gadgetNames.length];
+      gadgetIndex++;
+      messages.push({
+        role: "assistant",
+        content: `!!!GADGET_START:${gadgetName}
+!!!ARG:query
+test query ${turn}-${g}
+!!!GADGET_END`
+      });
+      messages.push({
+        role: "user",
+        content: `Result: Gadget ${gadgetName} returned result for query ${turn}-${g}`
+      });
+    }
+    messages.push({
+      role: "assistant",
+      content: `Final response for turn ${turn + 1}${"y".repeat(contentLength)}`
+    });
+  }
+  return messages;
+}
+function estimateTokens(messages) {
+  return Math.ceil(messages.reduce((sum, msg) => sum + (msg.content?.length ?? 0), 0) / 4);
+}
+function createUserMessage(content) {
+  return { role: "user", content };
+}
+function createAssistantMessage(content) {
+  return { role: "assistant", content };
+}
+function createSystemMessage(content) {
+  return { role: "system", content };
+}
+function createMinimalConversation() {
+  return [
+    { role: "user", content: "Hello" },
+    { role: "assistant", content: "Hi there!" }
+  ];
+}
+function createLargeConversation(targetTokens, options) {
+  const tokensPerTurn = options?.tokensPerTurn ?? 200;
+  const turnsNeeded = Math.ceil(targetTokens / tokensPerTurn);
+  const charsPerMessage = Math.floor(tokensPerTurn * 4 / 2);
+  return createConversation(turnsNeeded, {
+    contentLength: charsPerMessage
+  });
+}
+
+// src/gadgets/validation.ts
+function validateAndApplyDefaults(schema, params) {
+  const result = schema.safeParse(params);
+  if (result.success) {
+    return {
+      success: true,
+      data: result.data
+    };
+  }
+  const issues = result.error.issues.map((issue) => ({
+    path: issue.path.join(".") || "root",
+    message: issue.message
+  }));
+  const formattedError = `Invalid parameters: ${issues.map((i) => `${i.path}: ${i.message}`).join("; ")}`;
+  return {
+    success: false,
+    error: formattedError,
+    issues
+  };
+}
+function validateGadgetParams(gadget, params) {
+  if (!gadget.parameterSchema) {
+    return {
+      success: true,
+      data: params
+    };
+  }
+  return validateAndApplyDefaults(gadget.parameterSchema, params);
+}
+
+// src/testing/gadget-testing.ts
+async function testGadget(gadget, params, options) {
+  let validatedParams = params;
+  if (!options?.skipValidation) {
+    const validationResult = validateGadgetParams(gadget, params);
+    if (!validationResult.success) {
+      return {
+        error: validationResult.error,
+        validatedParams: params
+      };
+    }
+    validatedParams = validationResult.data;
+  }
+  try {
+    const rawResult = await Promise.resolve(gadget.execute(validatedParams));
+    if (typeof rawResult === "string") {
       return {
         result: rawResult,
         validatedParams,
@@ -9189,7 +9722,7 @@ function getMockManager(options) {
 
 // src/testing/mock-stream.ts
 init_constants();
-function sleep(ms) {
+function sleep2(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
 function generateInvocationId() {
@@ -9270,7 +9803,7 @@ ${blockParams}${GADGET_END_PREFIX}`;
 }
 async function* createMockStream(response) {
   if (response.delayMs) {
-    await sleep(response.delayMs);
+    await sleep2(response.delayMs);
   }
   const streamDelay = response.streamDelayMs ?? 0;
   let fullText = response.text ?? "";
@@ -9295,7 +9828,7 @@ async function* createMockStream(response) {
       }
       yield chunk;
       if (streamDelay > 0 && !isLast) {
-        await sleep(streamDelay);
+        await sleep2(streamDelay);
       }
     }
   } else {
@@ -9572,7 +10105,9 @@ var MockBuilder = class {
    */
   whenMessageContains(text3) {
     this.matchers.push(
-      (ctx) => ctx.messages.some((msg) => extractText(msg.content).toLowerCase().includes(text3.toLowerCase()))
+      (ctx) => ctx.messages.some(
+        (msg) => extractText(msg.content).toLowerCase().includes(text3.toLowerCase())
+      )
     );
     return this;
   }
@@ -10010,32 +10545,161 @@ function createMockClient(options) {
   });
 }
 
-// src/testing/mock-gadget.ts
-init_gadget();
-var MockGadgetImpl = class extends BaseGadget {
-  name;
-  description;
-  parameterSchema;
-  timeoutMs;
-  calls = [];
-  resultValue;
-  resultFn;
-  errorToThrow;
-  delayMs;
-  shouldTrackCalls;
-  constructor(config) {
-    super();
-    this.name = config.name;
-    this.description = config.description ?? `Mock gadget: ${config.name}`;
-    this.parameterSchema = config.schema;
-    this.resultValue = config.result;
-    this.resultFn = config.resultFn;
-    this.delayMs = config.delayMs ?? 0;
-    this.shouldTrackCalls = config.trackCalls ?? true;
-    this.timeoutMs = config.timeoutMs;
-    if (config.error) {
-      this.errorToThrow = typeof config.error === "string" ? new Error(config.error) : config.error;
-    }
+// src/testing/mock-conversation.ts
+var MockConversationManager = class {
+  history;
+  baseMessages;
+  replacementHistory;
+  replaceHistoryCallCount = 0;
+  addedMessages = [];
+  constructor(history = [], baseMessages = []) {
+    this.history = [...history];
+    this.baseMessages = [...baseMessages];
+  }
+  addUserMessage(content) {
+    const msg = { role: "user", content };
+    this.history.push(msg);
+    this.addedMessages.push(msg);
+  }
+  addAssistantMessage(content) {
+    const msg = { role: "assistant", content };
+    this.history.push(msg);
+    this.addedMessages.push(msg);
+  }
+  addGadgetCall(gadgetName, parameters, result) {
+    const assistantMsg = {
+      role: "assistant",
+      content: `!!!GADGET_START:${gadgetName}
+${JSON.stringify(parameters)}
+!!!GADGET_END`
+    };
+    const resultMsg = {
+      role: "user",
+      content: `Result: ${result}`
+    };
+    this.history.push(assistantMsg);
+    this.history.push(resultMsg);
+    this.addedMessages.push(assistantMsg);
+    this.addedMessages.push(resultMsg);
+  }
+  getMessages() {
+    return [...this.baseMessages, ...this.history];
+  }
+  getHistoryMessages() {
+    return [...this.history];
+  }
+  getBaseMessages() {
+    return [...this.baseMessages];
+  }
+  replaceHistory(newHistory) {
+    this.replacementHistory = [...newHistory];
+    this.history = [...newHistory];
+    this.replaceHistoryCallCount++;
+  }
+  // ============================================
+  // Test Helper Methods
+  // ============================================
+  /**
+   * Check if replaceHistory was called.
+   */
+  wasReplaceHistoryCalled() {
+    return this.replaceHistoryCallCount > 0;
+  }
+  /**
+   * Get the number of times replaceHistory was called.
+   */
+  getReplaceHistoryCallCount() {
+    return this.replaceHistoryCallCount;
+  }
+  /**
+   * Get the most recent history passed to replaceHistory.
+   * Returns undefined if replaceHistory was never called.
+   */
+  getReplacementHistory() {
+    return this.replacementHistory;
+  }
+  /**
+   * Get all messages that were added via add* methods.
+   */
+  getAddedMessages() {
+    return [...this.addedMessages];
+  }
+  /**
+   * Reset all tracking state while preserving the conversation.
+   */
+  resetTracking() {
+    this.replacementHistory = void 0;
+    this.replaceHistoryCallCount = 0;
+    this.addedMessages = [];
+  }
+  /**
+   * Completely reset the mock to initial state.
+   * Note: baseMessages cannot be changed after construction.
+   */
+  reset(history = []) {
+    this.history = [...history];
+    this.resetTracking();
+  }
+  /**
+   * Set the history directly (for test setup).
+   */
+  setHistory(messages) {
+    this.history = [...messages];
+  }
+  /**
+   * Get the current history length.
+   */
+  getHistoryLength() {
+    return this.history.length;
+  }
+  /**
+   * Get total message count (base + history).
+   */
+  getTotalMessageCount() {
+    return this.baseMessages.length + this.history.length;
+  }
+};
+function createMockConversationManager(turnCount, baseMessages = []) {
+  const history = [];
+  for (let i = 0; i < turnCount; i++) {
+    history.push({
+      role: "user",
+      content: `User message ${i + 1}: This is turn ${i + 1} of the conversation.`
+    });
+    history.push({
+      role: "assistant",
+      content: `Assistant response ${i + 1}: I acknowledge turn ${i + 1}.`
+    });
+  }
+  return new MockConversationManager(history, baseMessages);
+}
+
+// src/testing/mock-gadget.ts
+init_gadget();
+var MockGadgetImpl = class extends BaseGadget {
+  name;
+  description;
+  parameterSchema;
+  timeoutMs;
+  calls = [];
+  resultValue;
+  resultFn;
+  errorToThrow;
+  delayMs;
+  shouldTrackCalls;
+  constructor(config) {
+    super();
+    this.name = config.name;
+    this.description = config.description ?? `Mock gadget: ${config.name}`;
+    this.parameterSchema = config.schema;
+    this.resultValue = config.result;
+    this.resultFn = config.resultFn;
+    this.delayMs = config.delayMs ?? 0;
+    this.shouldTrackCalls = config.trackCalls ?? true;
+    this.timeoutMs = config.timeoutMs;
+    if (config.error) {
+      this.errorToThrow = typeof config.error === "string" ? new Error(config.error) : config.error;
+    }
   }
   async execute(params) {
     if (this.shouldTrackCalls) {
@@ -10169,7 +10833,7 @@ function createTestStream(chunks) {
 function createTextStream(text3, options) {
   return async function* () {
     if (options?.delayMs) {
-      await sleep2(options.delayMs);
+      await sleep3(options.delayMs);
     }
     const chunkSize = options?.chunkSize ?? text3.length;
     const chunks = [];
@@ -10191,7 +10855,7 @@ function createTextStream(text3, options) {
       }
       yield chunk;
       if (options?.chunkDelayMs && !isLast) {
-        await sleep2(options.chunkDelayMs);
+        await sleep3(options.chunkDelayMs);
       }
     }
   }();
@@ -10229,365 +10893,9 @@ function createErrorStream(chunksBeforeError, error) {
     throw error;
   }();
 }
-function sleep2(ms) {
-  return new Promise((resolve) => setTimeout(resolve, ms));
-}
-
-// src/testing/conversation-fixtures.ts
-function createConversation(turnCount, options) {
-  const messages = [];
-  const userPrefix = options?.userPrefix ?? "User message";
-  const assistantPrefix = options?.assistantPrefix ?? "Assistant response";
-  const contentLength = options?.contentLength ?? 100;
-  for (let i = 0; i < turnCount; i++) {
-    const padding = " ".repeat(Math.max(0, contentLength - 30));
-    messages.push({
-      role: "user",
-      content: `${userPrefix} ${i + 1}: This is turn ${i + 1} of the conversation.${padding}`
-    });
-    messages.push({
-      role: "assistant",
-      content: `${assistantPrefix} ${i + 1}: I acknowledge turn ${i + 1}.${padding}`
-    });
-  }
-  return messages;
-}
-function createConversationWithGadgets(turnCount, gadgetCallsPerTurn = 1, options) {
-  const messages = [];
-  const gadgetNames = options?.gadgetNames ?? ["search", "calculate", "read"];
-  const contentLength = options?.contentLength ?? 50;
-  let gadgetIndex = 0;
-  for (let turn = 0; turn < turnCount; turn++) {
-    messages.push({
-      role: "user",
-      content: `User request ${turn + 1}${"x".repeat(contentLength)}`
-    });
-    for (let g = 0; g < gadgetCallsPerTurn; g++) {
-      const gadgetName = gadgetNames[gadgetIndex % gadgetNames.length];
-      gadgetIndex++;
-      messages.push({
-        role: "assistant",
-        content: `!!!GADGET_START:${gadgetName}
-!!!ARG:query
-test query ${turn}-${g}
-!!!GADGET_END`
-      });
-      messages.push({
-        role: "user",
-        content: `Result: Gadget ${gadgetName} returned result for query ${turn}-${g}`
-      });
-    }
-    messages.push({
-      role: "assistant",
-      content: `Final response for turn ${turn + 1}${"y".repeat(contentLength)}`
-    });
-  }
-  return messages;
-}
-function estimateTokens(messages) {
-  return Math.ceil(
-    messages.reduce((sum, msg) => sum + (msg.content?.length ?? 0), 0) / 4
-  );
-}
-function createUserMessage(content) {
-  return { role: "user", content };
-}
-function createAssistantMessage(content) {
-  return { role: "assistant", content };
-}
-function createSystemMessage(content) {
-  return { role: "system", content };
-}
-function createMinimalConversation() {
-  return [
-    { role: "user", content: "Hello" },
-    { role: "assistant", content: "Hi there!" }
-  ];
-}
-function createLargeConversation(targetTokens, options) {
-  const tokensPerTurn = options?.tokensPerTurn ?? 200;
-  const turnsNeeded = Math.ceil(targetTokens / tokensPerTurn);
-  const charsPerMessage = Math.floor(tokensPerTurn * 4 / 2);
-  return createConversation(turnsNeeded, {
-    contentLength: charsPerMessage
-  });
-}
-
-// src/testing/mock-conversation.ts
-var MockConversationManager = class {
-  history;
-  baseMessages;
-  replacementHistory;
-  replaceHistoryCallCount = 0;
-  addedMessages = [];
-  constructor(history = [], baseMessages = []) {
-    this.history = [...history];
-    this.baseMessages = [...baseMessages];
-  }
-  addUserMessage(content) {
-    const msg = { role: "user", content };
-    this.history.push(msg);
-    this.addedMessages.push(msg);
-  }
-  addAssistantMessage(content) {
-    const msg = { role: "assistant", content };
-    this.history.push(msg);
-    this.addedMessages.push(msg);
-  }
-  addGadgetCall(gadgetName, parameters, result) {
-    const assistantMsg = {
-      role: "assistant",
-      content: `!!!GADGET_START:${gadgetName}
-${JSON.stringify(parameters)}
-!!!GADGET_END`
-    };
-    const resultMsg = {
-      role: "user",
-      content: `Result: ${result}`
-    };
-    this.history.push(assistantMsg);
-    this.history.push(resultMsg);
-    this.addedMessages.push(assistantMsg);
-    this.addedMessages.push(resultMsg);
-  }
-  getMessages() {
-    return [...this.baseMessages, ...this.history];
-  }
-  getHistoryMessages() {
-    return [...this.history];
-  }
-  getBaseMessages() {
-    return [...this.baseMessages];
-  }
-  replaceHistory(newHistory) {
-    this.replacementHistory = [...newHistory];
-    this.history = [...newHistory];
-    this.replaceHistoryCallCount++;
-  }
-  // ============================================
-  // Test Helper Methods
-  // ============================================
-  /**
-   * Check if replaceHistory was called.
-   */
-  wasReplaceHistoryCalled() {
-    return this.replaceHistoryCallCount > 0;
-  }
-  /**
-   * Get the number of times replaceHistory was called.
-   */
-  getReplaceHistoryCallCount() {
-    return this.replaceHistoryCallCount;
-  }
-  /**
-   * Get the most recent history passed to replaceHistory.
-   * Returns undefined if replaceHistory was never called.
-   */
-  getReplacementHistory() {
-    return this.replacementHistory;
-  }
-  /**
-   * Get all messages that were added via add* methods.
-   */
-  getAddedMessages() {
-    return [...this.addedMessages];
-  }
-  /**
-   * Reset all tracking state while preserving the conversation.
-   */
-  resetTracking() {
-    this.replacementHistory = void 0;
-    this.replaceHistoryCallCount = 0;
-    this.addedMessages = [];
-  }
-  /**
-   * Completely reset the mock to initial state.
-   * Note: baseMessages cannot be changed after construction.
-   */
-  reset(history = []) {
-    this.history = [...history];
-    this.resetTracking();
-  }
-  /**
-   * Set the history directly (for test setup).
-   */
-  setHistory(messages) {
-    this.history = [...messages];
-  }
-  /**
-   * Get the current history length.
-   */
-  getHistoryLength() {
-    return this.history.length;
-  }
-  /**
-   * Get total message count (base + history).
-   */
-  getTotalMessageCount() {
-    return this.baseMessages.length + this.history.length;
-  }
-};
-function createMockConversationManager(turnCount, baseMessages = []) {
-  const history = [];
-  for (let i = 0; i < turnCount; i++) {
-    history.push({
-      role: "user",
-      content: `User message ${i + 1}: This is turn ${i + 1} of the conversation.`
-    });
-    history.push({
-      role: "assistant",
-      content: `Assistant response ${i + 1}: I acknowledge turn ${i + 1}.`
-    });
-  }
-  return new MockConversationManager(history, baseMessages);
-}
-
-// src/testing/cli-helpers.ts
-var import_node_stream = require("stream");
-function createTestEnvironment(options = {}) {
-  const stdin = createMockReadable(options.stdin);
-  const stdout = new import_node_stream.PassThrough();
-  const stderr = new import_node_stream.PassThrough();
-  let exitCode;
-  return {
-    stdin,
-    stdout,
-    stderr,
-    isTTY: options.isTTY ?? false,
-    argv: options.argv ?? ["node", "llmist"],
-    env: { ...filterDefinedEnv(process.env), ...options.env },
-    get exitCode() {
-      return exitCode;
-    },
-    setExitCode: (code) => {
-      exitCode = code;
-    }
-  };
-}
-function createMockReadable(input) {
-  if (!input) {
-    const stream3 = new import_node_stream.Readable({ read() {
-    } });
-    stream3.push(null);
-    return stream3;
-  }
-  const content = Array.isArray(input) ? `${input.join("\n")}
-` : input;
-  const stream2 = new import_node_stream.Readable({ read() {
-  } });
-  stream2.push(content);
-  stream2.push(null);
-  return stream2;
-}
-function createMockWritable() {
-  const chunks = [];
-  const stream2 = new import_node_stream.Writable({
-    write(chunk, _encoding, callback) {
-      chunks.push(Buffer.from(chunk));
-      callback();
-    }
-  });
-  stream2.getData = () => Buffer.concat(chunks).toString("utf8");
-  return stream2;
-}
-async function collectOutput(stream2, timeout = 5e3) {
-  return new Promise((resolve, reject) => {
-    const chunks = [];
-    const timeoutId = setTimeout(() => {
-      resolve(Buffer.concat(chunks).toString("utf8"));
-    }, timeout);
-    stream2.on("data", (chunk) => {
-      chunks.push(Buffer.from(chunk));
-    });
-    stream2.on("end", () => {
-      clearTimeout(timeoutId);
-      resolve(Buffer.concat(chunks).toString("utf8"));
-    });
-    stream2.on("error", (err) => {
-      clearTimeout(timeoutId);
-      reject(err);
-    });
-  });
-}
-function getBufferedOutput(stream2) {
-  const chunks = [];
-  for (; ; ) {
-    const chunk = stream2.read();
-    if (chunk === null) break;
-    chunks.push(chunk);
-  }
-  return Buffer.concat(chunks).toString("utf8");
-}
-function createMockPrompt(responses) {
-  let index = 0;
-  return async (_question) => {
-    if (index >= responses.length) {
-      throw new Error(`Mock prompt exhausted: no response for question ${index + 1}`);
-    }
-    return responses[index++];
-  };
-}
-var MockPromptRecorder = class {
-  responses;
-  index = 0;
-  questions = [];
-  constructor(responses) {
-    this.responses = responses;
-  }
-  /**
-   * The prompt function to use in tests.
-   */
-  prompt = async (question) => {
-    this.questions.push(question);
-    if (this.index >= this.responses.length) {
-      throw new Error(`Mock prompt exhausted after ${this.index} questions`);
-    }
-    return this.responses[this.index++];
-  };
-  /**
-   * Get all questions that were asked.
-   */
-  getQuestions() {
-    return [...this.questions];
-  }
-  /**
-   * Get the number of questions asked.
-   */
-  getQuestionCount() {
-    return this.questions.length;
-  }
-  /**
-   * Reset the recorder state.
-   */
-  reset(newResponses) {
-    this.index = 0;
-    this.questions = [];
-    if (newResponses) {
-      this.responses = newResponses;
-    }
-  }
-};
-async function waitFor(condition, timeout = 5e3, interval = 50) {
-  const startTime = Date.now();
-  while (!condition()) {
-    if (Date.now() - startTime > timeout) {
-      throw new Error(`waitFor timed out after ${timeout}ms`);
-    }
-    await sleep3(interval);
-  }
-}
 function sleep3(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
-function filterDefinedEnv(env) {
-  const result = {};
-  for (const [key, value] of Object.entries(env)) {
-    if (value !== void 0) {
-      result[key] = value;
-    }
-  }
-  return result;
-}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   MockBuilder,