llmist 1.7.0 → 2.1.0

package/dist/cli.js

@@ -1,5 +1,5 @@
 #!/usr/bin/env node
-import "./chunk-JGORHSHC.js";
+import "./chunk-LSCCBXS7.js";
 import {
   AgentBuilder,
   BaseGadget,
@@ -27,7 +27,7 @@ import {
   resolveModel,
   schemaToJSONSchema,
   validateGadgetSchema
-} from "./chunk-E52IO2NO.js";
+} from "./chunk-PDYVT3FI.js";
 
 // src/cli/constants.ts
 var CLI_NAME = "llmist";
@@ -51,7 +51,6 @@ var OPTION_FLAGS = {
   logFile: "--log-file <path>",
   logReset: "--log-reset",
   logLlmRequests: "--log-llm-requests [dir]",
-  logLlmResponses: "--log-llm-responses [dir]",
   noBuiltins: "--no-builtins",
   noBuiltinInteraction: "--no-builtin-interaction",
   quiet: "-q, --quiet",
@@ -70,8 +69,7 @@ var OPTION_DESCRIPTIONS = {
   logLevel: "Log level: silly, trace, debug, info, warn, error, fatal.",
   logFile: "Path to log file. When set, logs are written to file instead of stderr.",
   logReset: "Reset (truncate) the log file at session start instead of appending.",
-  logLlmRequests: "Save raw LLM requests as plain text. Optional dir, defaults to ~/.llmist/logs/requests/",
-  logLlmResponses: "Save raw LLM responses as plain text. Optional dir, defaults to ~/.llmist/logs/responses/",
+  logLlmRequests: "Save LLM requests/responses to session directories. Optional dir, defaults to ~/.llmist/logs/requests/",
   noBuiltins: "Disable built-in gadgets (AskUser, TellUser).",
   noBuiltinInteraction: "Disable interactive gadgets (AskUser) while keeping TellUser.",
   quiet: "Suppress all output except content (text and TellUser messages).",
@@ -88,7 +86,7 @@ import { Command, InvalidArgumentError as InvalidArgumentError2 } from "commande
 // package.json
 var package_default = {
   name: "llmist",
-  version: "1.6.2",
+  version: "2.0.0",
   description: "Universal TypeScript LLM client with streaming-first agent framework. Works with any model - no structured outputs or native tool calling required. Implements its own flexible grammar for function calling.",
   type: "module",
   main: "dist/index.cjs",
@@ -261,6 +259,14 @@ ${addedLines}`;
 }
 
 // src/cli/approval/context-providers.ts
+function formatGadgetSummary(gadgetName, params) {
+  const paramEntries = Object.entries(params);
+  if (paramEntries.length === 0) {
+    return `${gadgetName}()`;
+  }
+  const paramStr = paramEntries.map(([k, v]) => `${k}=${JSON.stringify(v)}`).join(", ");
+  return `${gadgetName}(${paramStr})`;
+}
 var WriteFileContextProvider = class {
   gadgetName = "WriteFile";
   async getContext(params) {
@@ -269,14 +275,14 @@ var WriteFileContextProvider = class {
     const resolvedPath = resolve(process.cwd(), filePath);
     if (!existsSync(resolvedPath)) {
       return {
-        summary: `Create new file: ${filePath}`,
+        summary: formatGadgetSummary(this.gadgetName, params),
         details: formatNewFileDiff(filePath, newContent)
       };
     }
     const oldContent = readFileSync(resolvedPath, "utf-8");
     const diff = createPatch(filePath, oldContent, newContent, "original", "modified");
     return {
-      summary: `Modify: ${filePath}`,
+      summary: formatGadgetSummary(this.gadgetName, params),
       details: diff
     };
   }
@@ -290,37 +296,27 @@ var EditFileContextProvider = class {
       const newContent = String(params.content);
       if (!existsSync(resolvedPath)) {
         return {
-          summary: `Create new file: ${filePath}`,
+          summary: formatGadgetSummary(this.gadgetName, params),
           details: formatNewFileDiff(filePath, newContent)
         };
       }
       const oldContent = readFileSync(resolvedPath, "utf-8");
       const diff = createPatch(filePath, oldContent, newContent, "original", "modified");
       return {
-        summary: `Modify: ${filePath}`,
+        summary: formatGadgetSummary(this.gadgetName, params),
         details: diff
       };
     }
     if ("commands" in params) {
       const commands = String(params.commands);
       return {
-        summary: `Edit: ${filePath}`,
+        summary: formatGadgetSummary(this.gadgetName, params),
         details: `Commands:
 ${commands}`
       };
     }
     return {
-      summary: `Edit: ${filePath}`
-    };
-  }
-};
-var RunCommandContextProvider = class {
-  gadgetName = "RunCommand";
-  async getContext(params) {
-    const command = String(params.command ?? "");
-    const cwd = params.cwd ? ` (in ${params.cwd})` : "";
-    return {
-      summary: `Execute: ${command}${cwd}`
+      summary: formatGadgetSummary(this.gadgetName, params)
     };
   }
 };
@@ -329,27 +325,15 @@ var DefaultContextProvider = class {
     this.gadgetName = gadgetName;
   }
   async getContext(params) {
-    const paramEntries = Object.entries(params);
-    if (paramEntries.length === 0) {
-      return {
-        summary: `${this.gadgetName}()`
-      };
-    }
-    const formatValue = (value) => {
-      const MAX_LEN = 50;
-      const str = JSON.stringify(value);
-      return str.length > MAX_LEN ? `${str.slice(0, MAX_LEN - 3)}...` : str;
-    };
-    const paramStr = paramEntries.map(([k, v]) => `${k}=${formatValue(v)}`).join(", ");
     return {
-      summary: `${this.gadgetName}(${paramStr})`
+      summary: formatGadgetSummary(this.gadgetName, params)
     };
   }
 };
 var builtinContextProviders = [
   new WriteFileContextProvider(),
-  new EditFileContextProvider(),
-  new RunCommandContextProvider()
+  new EditFileContextProvider()
+  // Note: RunCommand uses DefaultContextProvider - no custom details needed
 ];
 
 // src/cli/approval/manager.ts
@@ -360,11 +344,13 @@ var ApprovalManager = class {
    * @param config - Approval configuration with per-gadget modes
    * @param env - CLI environment for I/O operations
    * @param progress - Optional progress indicator to pause during prompts
+   * @param keyboard - Optional keyboard coordinator to disable ESC listener during prompts
    */
-  constructor(config, env, progress) {
+  constructor(config, env, progress, keyboard) {
     this.config = config;
     this.env = env;
     this.progress = progress;
+    this.keyboard = keyboard;
     for (const provider of builtinContextProviders) {
       this.registerProvider(provider);
     }
@@ -433,26 +419,34 @@ var ApprovalManager = class {
     const provider = this.providers.get(gadgetName.toLowerCase()) ?? new DefaultContextProvider(gadgetName);
     const context = await provider.getContext(params);
     this.progress?.pause();
-    this.env.stderr.write(`
+    if (this.keyboard?.cleanupEsc) {
+      this.keyboard.cleanupEsc();
+      this.keyboard.cleanupEsc = null;
+    }
+    try {
+      this.env.stderr.write(`
 ${chalk2.yellow("\u{1F512} Approval required:")} ${context.summary}
 `);
-    if (context.details) {
-      this.env.stderr.write(`
+      if (context.details) {
+        this.env.stderr.write(`
 ${renderColoredDiff(context.details)}
 `);
-    }
-    const response = await this.prompt("   \u23CE approve, or type to reject: ");
-    const isApproved = response === "" || response.toLowerCase() === "y";
-    if (isApproved) {
-      this.env.stderr.write(`   ${chalk2.green("\u2713 Approved")}
+      }
+      const response = await this.prompt("   \u23CE approve, or type to reject: ");
+      const isApproved = response === "" || response.toLowerCase() === "y";
+      if (isApproved) {
+        this.env.stderr.write(`   ${chalk2.green("\u2713 Approved")}
 
 `);
-      return { approved: true };
-    }
-    this.env.stderr.write(`   ${chalk2.red("\u2717 Denied")}
+        return { approved: true };
+      }
+      this.env.stderr.write(`   ${chalk2.red("\u2717 Denied")}
 
 `);
-    return { approved: false, reason: response || "Rejected by user" };
+      return { approved: false, reason: response || "Rejected by user" };
+    } finally {
+      this.keyboard?.restore();
+    }
   }
   /**
    * Prompts for user input.
@@ -891,49 +885,77 @@ error: ${message}`;
 import { z as z6 } from "zod";
 var runCommand = createGadget({
   name: "RunCommand",
-  description: "Execute a shell command and return its output. Returns both stdout and stderr combined with the exit status.",
+  description: "Execute a command with arguments and return its output. Uses argv array to bypass shell - arguments are passed directly without interpretation. Returns stdout/stderr combined with exit status.",
   schema: z6.object({
-    command: z6.string().describe("The shell command to execute"),
+    argv: z6.array(z6.string()).describe("Command and arguments as array (e.g., ['git', 'commit', '-m', 'message'])"),
     cwd: z6.string().optional().describe("Working directory for the command (default: current directory)"),
     timeout: z6.number().default(3e4).describe("Timeout in milliseconds (default: 30000)")
   }),
   examples: [
     {
-      params: { command: "ls -la", timeout: 3e4 },
+      params: { argv: ["ls", "-la"], timeout: 3e4 },
       output: "status=0\n\ntotal 24\ndrwxr-xr-x  5 user  staff   160 Nov 27 10:00 .\ndrwxr-xr-x  3 user  staff    96 Nov 27 09:00 ..\n-rw-r--r--  1 user  staff  1024 Nov 27 10:00 package.json",
       comment: "List directory contents with details"
     },
     {
-      params: { command: "echo 'Hello World'", timeout: 3e4 },
+      params: { argv: ["echo", "Hello World"], timeout: 3e4 },
       output: "status=0\n\nHello World",
-      comment: "Simple echo command"
+      comment: "Echo without shell - argument passed directly"
     },
     {
-      params: { command: "cat nonexistent.txt", timeout: 3e4 },
+      params: { argv: ["cat", "nonexistent.txt"], timeout: 3e4 },
       output: "status=1\n\ncat: nonexistent.txt: No such file or directory",
       comment: "Command that fails returns non-zero status"
     },
     {
-      params: { command: "pwd", cwd: "/tmp", timeout: 3e4 },
+      params: { argv: ["pwd"], cwd: "/tmp", timeout: 3e4 },
       output: "status=0\n\n/tmp",
       comment: "Execute command in a specific directory"
+    },
+    {
+      params: { argv: ["gh", "pr", "review", "123", "--comment", "--body", "Review with `backticks` and 'quotes'"], timeout: 3e4 },
+      output: "status=0\n\n(no output)",
+      comment: "Complex arguments with special characters - no escaping needed"
+    },
+    {
+      params: {
+        argv: [
+          "gh",
+          "pr",
+          "review",
+          "123",
+          "--approve",
+          "--body",
+          "## Review Summary\n\n**Looks good!**\n\n- Clean code\n- Tests pass"
+        ],
+        timeout: 3e4
+      },
+      output: "status=0\n\nApproving pull request #123",
+      comment: "Multiline body: --body flag and content must be SEPARATE array elements"
     }
   ],
-  execute: async ({ command, cwd, timeout }) => {
+  execute: async ({ argv, cwd, timeout }) => {
     const workingDir = cwd ?? process.cwd();
+    if (argv.length === 0) {
+      return "status=1\n\nerror: argv array cannot be empty";
+    }
+    let timeoutId;
     try {
-      const proc = Bun.spawn(["sh", "-c", command], {
+      const proc = Bun.spawn(argv, {
         cwd: workingDir,
         stdout: "pipe",
         stderr: "pipe"
       });
       const timeoutPromise = new Promise((_, reject) => {
-        setTimeout(() => {
+        timeoutId = setTimeout(() => {
           proc.kill();
           reject(new Error(`Command timed out after ${timeout}ms`));
         }, timeout);
       });
       const exitCode = await Promise.race([proc.exited, timeoutPromise]);
+      if (timeoutId) {
+        clearTimeout(timeoutId);
+      }
       const stdout = await new Response(proc.stdout).text();
       const stderr = await new Response(proc.stderr).text();
       const output = [stdout, stderr].filter(Boolean).join("\n").trim();
@@ -941,6 +963,9 @@ var runCommand = createGadget({
 
 ${output || "(no output)"}`;
     } catch (error) {
+      if (timeoutId) {
+        clearTimeout(timeoutId);
+      }
       const message = error instanceof Error ? error.message : String(error);
       return `status=1
 
@@ -1113,6 +1138,30 @@ async function writeLogFile(dir, filename, content) {
   await mkdir(dir, { recursive: true });
   await writeFile2(join(dir, filename), content, "utf-8");
 }
+function formatSessionTimestamp(date = /* @__PURE__ */ new Date()) {
+  const pad = (n) => n.toString().padStart(2, "0");
+  const year = date.getFullYear();
+  const month = pad(date.getMonth() + 1);
+  const day = pad(date.getDate());
+  const hours = pad(date.getHours());
+  const minutes = pad(date.getMinutes());
+  const seconds = pad(date.getSeconds());
+  return `${year}-${month}-${day}_${hours}-${minutes}-${seconds}`;
+}
+async function createSessionDir(baseDir) {
+  const timestamp = formatSessionTimestamp();
+  const sessionDir = join(baseDir, timestamp);
+  try {
+    await mkdir(sessionDir, { recursive: true });
+    return sessionDir;
+  } catch (error) {
+    console.warn(`[llmist] Failed to create log session directory: ${sessionDir}`, error);
+    return void 0;
+  }
+}
+function formatCallNumber(n) {
+  return n.toString().padStart(4, "0");
+}
 
 // src/cli/utils.ts
 init_constants();
@@ -1276,7 +1325,7 @@ function formatBytes(bytes) {
   }
   return `${(bytes / (1024 * 1024)).toFixed(1)} MB`;
 }
-function formatGadgetSummary(result) {
+function formatGadgetSummary2(result) {
   const gadgetLabel = chalk3.magenta.bold(result.gadgetName);
   const timeLabel = chalk3.dim(`${Math.round(result.executionTimeMs)}ms`);
   const paramsStr = formatParametersInline(result.parameters);
@@ -1361,12 +1410,21 @@ function isInteractive(stream) {
 }
 var ESC_KEY = 27;
 var ESC_TIMEOUT_MS = 50;
-function createEscKeyListener(stdin, onEsc) {
+var CTRL_C = 3;
+function createEscKeyListener(stdin, onEsc, onCtrlC) {
   if (!stdin.isTTY || typeof stdin.setRawMode !== "function") {
     return null;
   }
   let escTimeout = null;
   const handleData = (data) => {
+    if (data[0] === CTRL_C && onCtrlC) {
+      if (escTimeout) {
+        clearTimeout(escTimeout);
+        escTimeout = null;
+      }
+      onCtrlC();
+      return;
+    }
     if (data[0] === ESC_KEY) {
       if (data.length === 1) {
         escTimeout = setTimeout(() => {
@@ -1814,7 +1872,7 @@ function addCompleteOptions(cmd, defaults) {
     OPTION_DESCRIPTIONS.maxTokens,
     createNumericParser({ label: "Max tokens", integer: true, min: 1 }),
     defaults?.["max-tokens"]
-  ).option(OPTION_FLAGS.quiet, OPTION_DESCRIPTIONS.quiet, defaults?.quiet).option(OPTION_FLAGS.logLlmRequests, OPTION_DESCRIPTIONS.logLlmRequests, defaults?.["log-llm-requests"]).option(OPTION_FLAGS.logLlmResponses, OPTION_DESCRIPTIONS.logLlmResponses, defaults?.["log-llm-responses"]);
+  ).option(OPTION_FLAGS.quiet, OPTION_DESCRIPTIONS.quiet, defaults?.quiet).option(OPTION_FLAGS.logLlmRequests, OPTION_DESCRIPTIONS.logLlmRequests, defaults?.["log-llm-requests"]);
 }
 function addAgentOptions(cmd, defaults) {
   const gadgetAccumulator = (value, previous = []) => [
@@ -1838,7 +1896,7 @@ function addAgentOptions(cmd, defaults) {
     OPTION_FLAGS.noBuiltinInteraction,
     OPTION_DESCRIPTIONS.noBuiltinInteraction,
     defaults?.["builtin-interaction"] !== false
-  ).option(OPTION_FLAGS.quiet, OPTION_DESCRIPTIONS.quiet, defaults?.quiet).option(OPTION_FLAGS.logLlmRequests, OPTION_DESCRIPTIONS.logLlmRequests, defaults?.["log-llm-requests"]).option(OPTION_FLAGS.logLlmResponses, OPTION_DESCRIPTIONS.logLlmResponses, defaults?.["log-llm-responses"]).option(OPTION_FLAGS.docker, OPTION_DESCRIPTIONS.docker).option(OPTION_FLAGS.dockerRo, OPTION_DESCRIPTIONS.dockerRo).option(OPTION_FLAGS.noDocker, OPTION_DESCRIPTIONS.noDocker).option(OPTION_FLAGS.dockerDev, OPTION_DESCRIPTIONS.dockerDev);
+  ).option(OPTION_FLAGS.quiet, OPTION_DESCRIPTIONS.quiet, defaults?.quiet).option(OPTION_FLAGS.logLlmRequests, OPTION_DESCRIPTIONS.logLlmRequests, defaults?.["log-llm-requests"]).option(OPTION_FLAGS.docker, OPTION_DESCRIPTIONS.docker).option(OPTION_FLAGS.dockerRo, OPTION_DESCRIPTIONS.dockerRo).option(OPTION_FLAGS.noDocker, OPTION_DESCRIPTIONS.noDocker).option(OPTION_FLAGS.dockerDev, OPTION_DESCRIPTIONS.dockerDev);
 }
 function configToCompleteOptions(config) {
   const result = {};
@@ -1848,7 +1906,6 @@ function configToCompleteOptions(config) {
   if (config["max-tokens"] !== void 0) result.maxTokens = config["max-tokens"];
   if (config.quiet !== void 0) result.quiet = config.quiet;
   if (config["log-llm-requests"] !== void 0) result.logLlmRequests = config["log-llm-requests"];
-  if (config["log-llm-responses"] !== void 0) result.logLlmResponses = config["log-llm-responses"];
   return result;
 }
 function configToAgentOptions(config) {
@@ -1872,7 +1929,6 @@ function configToAgentOptions(config) {
     result.gadgetApproval = config["gadget-approval"];
   if (config.quiet !== void 0) result.quiet = config.quiet;
   if (config["log-llm-requests"] !== void 0) result.logLlmRequests = config["log-llm-requests"];
-  if (config["log-llm-responses"] !== void 0) result.logLlmResponses = config["log-llm-responses"];
   if (config.docker !== void 0) result.docker = config.docker;
   if (config["docker-cwd-permission"] !== void 0)
     result.dockerCwdPermission = config["docker-cwd-permission"];
@@ -1890,7 +1946,8 @@ var DOCKER_CONFIG_KEYS = /* @__PURE__ */ new Set([
   "env-vars",
   "image-name",
   "dev-mode",
-  "dev-source"
+  "dev-source",
+  "docker-args"
 ]);
 var DEFAULT_IMAGE_NAME = "llmist-sandbox";
 var DEFAULT_CWD_PERMISSION = "rw";
@@ -2005,7 +2062,6 @@ var COMPLETE_CONFIG_KEYS = /* @__PURE__ */ new Set([
   "log-file",
   "log-reset",
   "log-llm-requests",
-  "log-llm-responses",
   "type",
   // Allowed for inheritance compatibility, ignored for built-in commands
   "docker",
@@ -2038,7 +2094,6 @@ var AGENT_CONFIG_KEYS = /* @__PURE__ */ new Set([
   "log-file",
   "log-reset",
   "log-llm-requests",
-  "log-llm-responses",
   "type",
   // Allowed for inheritance compatibility, ignored for built-in commands
   "docker",
@@ -2226,13 +2281,6 @@ function validateCompleteConfig(raw, section) {
       section
     );
   }
-  if ("log-llm-responses" in rawObj) {
-    result["log-llm-responses"] = validateStringOrBoolean(
-      rawObj["log-llm-responses"],
-      "log-llm-responses",
-      section
-    );
-  }
   return result;
 }
 function validateAgentConfig(raw, section) {
@@ -2311,13 +2359,6 @@ function validateAgentConfig(raw, section) {
       section
     );
   }
-  if ("log-llm-responses" in rawObj) {
-    result["log-llm-responses"] = validateStringOrBoolean(
-      rawObj["log-llm-responses"],
-      "log-llm-responses",
-      section
-    );
-  }
   return result;
 }
 function validateStringOrBoolean(value, field, section) {
@@ -2759,6 +2800,9 @@ function validateDockerConfig(raw, section) {
   if ("dev-source" in rawObj) {
     result["dev-source"] = validateString2(rawObj["dev-source"], "dev-source", section);
   }
+  if ("docker-args" in rawObj) {
+    result["docker-args"] = validateStringArray2(rawObj["docker-args"], "docker-args", section);
+  }
   return result;
 }
 
@@ -2770,6 +2814,8 @@ FROM oven/bun:1-debian
 
 # Install essential tools
 RUN apt-get update && apt-get install -y --no-install-recommends \\
+    # ed for EditFile gadget (line-oriented editor)
+    ed \\
     # ripgrep for fast file searching
     ripgrep \\
     # git for version control operations
@@ -2802,6 +2848,7 @@ FROM oven/bun:1-debian
 
 # Install essential tools (same as production)
 RUN apt-get update && apt-get install -y --no-install-recommends \\
+    ed \\
     ripgrep \\
     git \\
     curl \\
@@ -3036,6 +3083,9 @@ function buildDockerRunArgs(ctx, imageName, devMode) {
       }
     }
   }
+  if (ctx.config["docker-args"]) {
+    args.push(...ctx.config["docker-args"]);
+  }
   args.push(imageName);
   args.push(...ctx.forwardArgs);
   return args;
@@ -3202,6 +3252,8 @@ async function executeAgent(promptArg, options, env) {
       env.stderr.write(chalk5.yellow(`
 [Cancelled] ${progress.formatStats()}
 `));
+    } else {
+      handleQuit();
     }
   };
   const keyboard = {
@@ -3209,7 +3261,7 @@ async function executeAgent(promptArg, options, env) {
     cleanupSigint: null,
     restore: () => {
       if (stdinIsInteractive && stdinStream.isTTY && !wasCancelled) {
-        keyboard.cleanupEsc = createEscKeyListener(stdinStream, handleCancel);
+        keyboard.cleanupEsc = createEscKeyListener(stdinStream, handleCancel, handleCancel);
       }
     }
   };
@@ -3234,7 +3286,7 @@ async function executeAgent(promptArg, options, env) {
     process.exit(130);
   };
   if (stdinIsInteractive && stdinStream.isTTY) {
-    keyboard.cleanupEsc = createEscKeyListener(stdinStream, handleCancel);
+    keyboard.cleanupEsc = createEscKeyListener(stdinStream, handleCancel, handleCancel);
   }
   keyboard.cleanupSigint = createSigintListener(
     handleCancel,
@@ -3260,11 +3312,11 @@ async function executeAgent(promptArg, options, env) {
     gadgetApprovals,
     defaultMode: "allowed"
   };
-  const approvalManager = new ApprovalManager(approvalConfig, env, progress);
+  const approvalManager = new ApprovalManager(approvalConfig, env, progress, keyboard);
   let usage;
   let iterations = 0;
-  const llmRequestsDir = resolveLogDir(options.logLlmRequests, "requests");
-  const llmResponsesDir = resolveLogDir(options.logLlmResponses, "responses");
+  const llmLogsBaseDir = resolveLogDir(options.logLlmRequests, "requests");
+  let llmSessionDir;
   let llmCallCounter = 0;
   const countMessagesTokens = async (model, messages) => {
     try {
@@ -3296,10 +3348,19 @@ async function executeAgent(promptArg, options, env) {
         );
         progress.startCall(context.options.model, inputTokens);
         progress.setInputTokens(inputTokens, false);
-        if (llmRequestsDir) {
-          const filename = `${Date.now()}_call_${llmCallCounter}.request.txt`;
-          const content = formatLlmRequest(context.options.messages);
-          await writeLogFile(llmRequestsDir, filename, content);
+      },
+      // onLLMCallReady: Log the exact request being sent to the LLM
+      // This fires AFTER controller modifications (e.g., trailing messages)
+      onLLMCallReady: async (context) => {
+        if (llmLogsBaseDir) {
+          if (!llmSessionDir) {
+            llmSessionDir = await createSessionDir(llmLogsBaseDir);
+          }
+          if (llmSessionDir) {
+            const filename = `${formatCallNumber(llmCallCounter)}.request`;
+            const content = formatLlmRequest(context.options.messages);
+            await writeLogFile(llmSessionDir, filename, content);
+          }
         }
       },
       // onStreamChunk: Real-time updates as LLM generates tokens
@@ -3364,9 +3425,9 @@ async function executeAgent(promptArg, options, env) {
 `);
           }
         }
-        if (llmResponsesDir) {
-          const filename = `${Date.now()}_call_${llmCallCounter}.response.txt`;
-          await writeLogFile(llmResponsesDir, filename, context.rawResponse);
+        if (llmSessionDir) {
+          const filename = `${formatCallNumber(llmCallCounter)}.response`;
+          await writeLogFile(llmSessionDir, filename, context.rawResponse);
         }
       }
     },
@@ -3463,6 +3524,13 @@ Denied: ${result.reason ?? "by user"}`
     parameterMapping: (text) => ({ message: text, done: false, type: "info" }),
     resultMapping: (text) => `\u2139\uFE0F  ${text}`
   });
+  builder.withTrailingMessage(
+    (ctx) => [
+      `[Iteration ${ctx.iteration + 1}/${ctx.maxIterations}]`,
+      "Think carefully: what gadget invocations can you make in parallel right now?",
+      "Maximize efficiency by batching independent operations in a single response."
+    ].join(" ")
+  );
   const agent = builder.ask(prompt);
   let textBuffer = "";
   const flushTextBuffer = () => {
@@ -3488,7 +3556,7 @@ Denied: ${result.reason ?? "by user"}`
           }
         } else {
           const tokenCount = await countGadgetOutputTokens(event.result.result);
-          env.stderr.write(`${formatGadgetSummary({ ...event.result, tokenCount })}
+          env.stderr.write(`${formatGadgetSummary2({ ...event.result, tokenCount })}
 `);
         }
       }
@@ -3500,7 +3568,10 @@ Denied: ${result.reason ?? "by user"}`
   } finally {
     isStreaming = false;
     keyboard.cleanupEsc?.();
-    keyboard.cleanupSigint?.();
+    if (keyboard.cleanupSigint) {
+      keyboard.cleanupSigint();
+      process.once("SIGINT", () => process.exit(130));
+    }
   }
   flushTextBuffer();
   progress.complete();
@@ -3548,13 +3619,15 @@ async function executeComplete(promptArg, options, env) {
   }
   builder.addUser(prompt);
   const messages = builder.build();
-  const llmRequestsDir = resolveLogDir(options.logLlmRequests, "requests");
-  const llmResponsesDir = resolveLogDir(options.logLlmResponses, "responses");
-  const timestamp = Date.now();
-  if (llmRequestsDir) {
-    const filename = `${timestamp}_complete.request.txt`;
-    const content = formatLlmRequest(messages);
-    await writeLogFile(llmRequestsDir, filename, content);
+  const llmLogsBaseDir = resolveLogDir(options.logLlmRequests, "requests");
+  let llmSessionDir;
+  if (llmLogsBaseDir) {
+    llmSessionDir = await createSessionDir(llmLogsBaseDir);
+    if (llmSessionDir) {
+      const filename = "0001.request";
+      const content = formatLlmRequest(messages);
+      await writeLogFile(llmSessionDir, filename, content);
+    }
   }
   const stream = client.stream({
     model,
@@ -3593,9 +3666,9 @@ async function executeComplete(promptArg, options, env) {
   progress.endCall(usage);
   progress.complete();
   printer.ensureNewline();
-  if (llmResponsesDir) {
-    const filename = `${timestamp}_complete.response.txt`;
-    await writeLogFile(llmResponsesDir, filename, accumulatedResponse);
+  if (llmSessionDir) {
+    const filename = "0001.response";
+    await writeLogFile(llmSessionDir, filename, accumulatedResponse);
   }
   if (stderrTTY && !options.quiet) {
     const summary = renderSummary({ finishReason, usage, cost: progress.getTotalCost() });
@@ -3830,9 +3903,11 @@ ${issues}`);
   env.stderr.write(chalk7.dim("\nExecuting...\n"));
   const startTime = Date.now();
   let result;
+  let cost;
   try {
+    let rawResult;
     if (gadget.timeoutMs && gadget.timeoutMs > 0) {
-      result = await Promise.race([
+      rawResult = await Promise.race([
         Promise.resolve(gadget.execute(params)),
         new Promise(
           (_, reject) => setTimeout(
@@ -3842,15 +3917,18 @@ ${issues}`);
         )
       ]);
     } else {
-      result = await Promise.resolve(gadget.execute(params));
+      rawResult = await Promise.resolve(gadget.execute(params));
     }
+    result = typeof rawResult === "string" ? rawResult : rawResult.result;
+    cost = typeof rawResult === "object" ? rawResult.cost : void 0;
   } catch (error) {
     const message = error instanceof Error ? error.message : String(error);
     throw new Error(`Execution failed: ${message}`);
   }
   const elapsed = Date.now() - startTime;
+  const costInfo = cost !== void 0 && cost > 0 ? ` (Cost: $${cost.toFixed(6)})` : "";
   env.stderr.write(chalk7.green(`
-\u2713 Completed in ${elapsed}ms
+\u2713 Completed in ${elapsed}ms${costInfo}
 
 `));
   formatOutput(result, options, env.stdout);