@runtypelabs/cli 0.1.10 → 0.2.0

package/README.md

@@ -10,6 +10,12 @@ Command-line interface for the Runtype AI platform.
 npm install -g @runtypelabs/cli
 ```
 
+Or run without installing:
+
+```bash
+npx @runtypelabs/cli <command>
+```
+
 ## Quick Start
 
 ### 1. Create Account & Authenticate
@@ -54,6 +60,27 @@ runtype records create --name "My Record" --type "document"
 runtype records list --type document
 ```
 
+### 4. Run Multi-Session Agent Tasks
+
+Use `runtype marathon` (or `runtype agents task`) to run long-running, multi-session agent tasks with real-time streaming output.
+
+```bash
+# Run a task — agent output streams to your terminal in real time
+runtype marathon "Code Builder" -m "Refactor the auth module to use JWT tokens"
+
+# Set a budget and session limit, save progress with a custom name
+runtype marathon agent_abc123 -m "Write integration tests for the API" \
+  --max-sessions 10 --max-cost 2.50 --name "api-tests"
+
+# Resume an interrupted task (picks up where it left off)
+runtype marathon "Code Builder" -m "Refactor the auth module to use JWT tokens" \
+  --resume --name "auth-refactor" --debug
+
+# Sync progress to a Runtype record (visible in the dashboard)
+runtype marathon "Code Builder" -m "Build the payments integration" \
+  --max-sessions 20 --track --name "payments"
+```
+
 ## Chat Commands
 
 During a chat session, you can use these special commands:

package/dist/index.js

@@ -2783,7 +2783,7 @@ var import_chalk16 = __toESM(require("chalk"));
 var import_ora11 = __toESM(require("ora"));
 var import_sdk6 = require("@runtypelabs/sdk");
 
-// src/commands/agents-run-task.ts
+// src/commands/agents-task.ts
 init_cjs_shims();
 var import_commander11 = require("commander");
 var import_chalk15 = __toESM(require("chalk"));
@@ -2810,206 +2810,221 @@ function loadState(filePath) {
 function saveState(filePath, state) {
   const dir = path4.dirname(filePath);
   fs3.mkdirSync(dir, { recursive: true });
-  state.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
   fs3.writeFileSync(filePath, JSON.stringify(state, null, 2));
 }
-var runTaskCommand = new import_commander11.Command("run-task").description("Run a long-task agent with local state persistence").argument("<agent>", "Agent ID or name").requiredOption("-m, --message <text>", "Task message for the agent").option("--max-sessions <n>", "Maximum sessions", "50").option("--max-cost <n>", "Budget in USD").option("--name <name>", "Task name (used for state file, defaults to agent ID)").option("--state-dir <path>", "Directory for state files (default: .runtype/tasks/)").option("--resume", "Resume from existing local state").option("--track-progress", "Also sync progress to the Runtype dashboard").option("--debug", "Show debug output from each session").option("--json", "Output final result as JSON").action(
-  async (agent, options) => {
-    const apiKey = await ensureAuth();
-    if (!apiKey) return;
-    const client = new import_sdk5.RuntypeClient({
-      apiKey,
-      baseUrl: getApiUrl()
-    });
-    let agentId = agent;
-    if (!agent.startsWith("agent_")) {
-      const spinner = (0, import_ora10.default)("Looking up agent by name...").start();
-      try {
-        const list = await client.agents.list();
-        const found = list.data.find(
-          (a) => a.name.toLowerCase() === agent.toLowerCase()
-        );
-        if (found) {
-          agentId = found.id;
-          spinner.succeed(`Found agent: ${import_chalk15.default.green(agentId)}`);
-        } else {
-          spinner.fail(`No agent found with name "${agent}"`);
-          console.log(import_chalk15.default.gray(`  Create one with: runtype agents create -n "${agent}"`));
-          process.exit(1);
-        }
-      } catch (error) {
-        spinner.fail("Failed to look up agent");
-        const message = error instanceof Error ? error.message : "Unknown error";
-        console.error(import_chalk15.default.red(message));
+async function taskAction(agent, options) {
+  const apiKey = await ensureAuth();
+  if (!apiKey) return;
+  const client = new import_sdk5.RuntypeClient({
+    apiKey,
+    baseUrl: getApiUrl()
+  });
+  let agentId = agent;
+  if (!agent.startsWith("agent_")) {
+    const spinner = (0, import_ora10.default)("Looking up agent by name...").start();
+    try {
+      const list = await client.agents.list();
+      const found = list.data.find(
+        (a) => a.name.toLowerCase() === agent.toLowerCase()
+      );
+      if (found) {
+        agentId = found.id;
+        spinner.succeed(`Found agent: ${import_chalk15.default.green(agentId)}`);
+      } else {
+        spinner.fail(`No agent found with name "${agent}"`);
+        console.log(import_chalk15.default.gray(`  Create one with: runtype agents create -n "${agent}"`));
         process.exit(1);
       }
+    } catch (error) {
+      spinner.fail("Failed to look up agent");
+      const errMsg = error instanceof Error ? error.message : "Unknown error";
+      console.error(import_chalk15.default.red(errMsg));
+      process.exit(1);
     }
-    const taskName = options.name || agentId;
-    const filePath = stateFilePath(taskName, options.stateDir);
-    const maxSessions = parseInt(options.maxSessions, 10);
-    const maxCost = options.maxCost ? parseFloat(options.maxCost) : void 0;
-    let localState;
-    let priorSessionCount = 0;
-    let priorCost = 0;
-    if (options.resume) {
-      const existing = loadState(filePath);
-      if (!existing) {
-        console.error(import_chalk15.default.red(`No state file found at ${filePath}`));
-        console.log(import_chalk15.default.gray("  Run without --resume to start a new task."));
-        process.exit(1);
-      }
-      if (existing.status === "complete") {
-        console.log(import_chalk15.default.yellow("This task already completed."));
-        if (options.json) printJson(existing);
-        return;
-      }
-      localState = existing;
-      localState.status = "running";
-      priorSessionCount = localState.sessionCount;
-      priorCost = localState.totalCost;
-      console.log(
-        import_chalk15.default.cyan(
-          `Resuming from session ${priorSessionCount} ($${priorCost.toFixed(4)} spent)`
-        )
-      );
-    } else {
-      localState = {
-        agentId,
-        taskName,
-        message: options.message,
-        status: "running",
-        sessionCount: 0,
-        totalCost: 0,
-        context: {},
-        lastOutput: "",
-        sessions: [],
-        createdAt: (/* @__PURE__ */ new Date()).toISOString(),
-        updatedAt: (/* @__PURE__ */ new Date()).toISOString()
-      };
+  }
+  const taskName = options.name || agentId;
+  const filePath = stateFilePath(taskName, options.stateDir);
+  const maxSessions = parseInt(options.maxSessions, 10);
+  const maxCost = options.maxCost ? parseFloat(options.maxCost) : void 0;
+  let priorSessionCount = 0;
+  let priorCost = 0;
+  if (options.resume) {
+    const existing = loadState(filePath);
+    if (!existing) {
+      console.error(import_chalk15.default.red(`No state file found at ${filePath}`));
+      console.log(import_chalk15.default.gray("  Run without --resume to start a new task."));
+      process.exit(1);
+    }
+    if (existing.status === "complete") {
+      console.log(import_chalk15.default.yellow("This task already completed."));
+      if (options.json) printJson(existing);
+      return;
     }
-    let interrupted = false;
-    const onSigint = () => {
-      if (interrupted) process.exit(1);
-      interrupted = true;
-      console.log(import_chalk15.default.yellow("\n\nInterrupted \u2014 saving state..."));
-      localState.status = "paused";
-      saveState(filePath, localState);
+    priorSessionCount = existing.sessionCount;
+    priorCost = existing.totalCost;
+    console.log(
+      import_chalk15.default.cyan(
+        `Resuming from session ${priorSessionCount} ($${priorCost.toFixed(4)} spent)`
+      )
+    );
+  }
+  let lastKnownState = null;
+  let interrupted = false;
+  const onSigint = () => {
+    if (interrupted) process.exit(1);
+    interrupted = true;
+    console.log(import_chalk15.default.yellow("\n\nInterrupted \u2014 saving state..."));
+    if (lastKnownState) {
+      lastKnownState.status = "paused";
+      saveState(filePath, lastKnownState);
       console.log(import_chalk15.default.green(`State saved to ${filePath}`));
-      console.log(
-        import_chalk15.default.gray(
-          `  Resume with: runtype agents run-task ${agent} -m "${options.message}" --resume${options.name ? ` --name ${options.name}` : ""}`
-        )
-      );
-      process.exit(0);
-    };
-    process.on("SIGINT", onSigint);
-    const remainingSessions = maxSessions - priorSessionCount;
-    console.log(import_chalk15.default.cyan(`
-Running task "${taskName}" on ${agentId}`));
+    }
     console.log(
       import_chalk15.default.gray(
-        `  Sessions: ${priorSessionCount > 0 ? `${priorSessionCount} done, ` : ""}${remainingSessions} remaining${maxCost ? ` | Budget: $${maxCost.toFixed(2)}` : ""}`
+        `  Resume with: runtype marathon ${agent} -m "${options.message}" --resume${options.name ? ` --name ${options.name}` : ""}`
       )
     );
-    console.log(import_chalk15.default.gray(`  State: ${filePath}
+    process.exit(0);
+  };
+  process.on("SIGINT", onSigint);
+  const remainingSessions = maxSessions - priorSessionCount;
+  const remainingCost = maxCost ? maxCost - priorCost : void 0;
+  console.log(import_chalk15.default.cyan(`
+Running task "${taskName}" on ${agentId}`));
+  console.log(
+    import_chalk15.default.gray(
+      `  Sessions: ${priorSessionCount > 0 ? `${priorSessionCount} done, ` : ""}${remainingSessions} remaining${maxCost ? ` | Budget: $${maxCost.toFixed(2)}` : ""}`
+    )
+  );
+  console.log(import_chalk15.default.gray(`  State: ${filePath}
 `));
-    try {
-      const result = await client.agents.runTask(agentId, {
-        buildMessages: (state) => {
-          const merged = { ...localState.context, ...state.context };
-          const isFirstSession = state.sessionCount === 1;
-          const totalSession = priorSessionCount + state.sessionCount;
-          if (isFirstSession && !options.resume) {
-            return [{ role: "user", content: options.message }];
-          }
-          const progress = merged.progressNotes || merged.progress;
-          const progressStr = progress ? typeof progress === "string" ? progress : JSON.stringify(progress) : "No structured progress captured yet";
-          const resumeCtx = options.resume && isFirstSession ? `
-
-Resuming from session ${priorSessionCount}. Previous context: ${JSON.stringify(localState.context)}` : "";
-          return [
-            {
-              role: "user",
-              content: `${options.message}${resumeCtx}
-
-Progress so far: ${progressStr}
-Session ${totalSession} of ${maxSessions}.`
-            }
-          ];
-        },
-        maxSessions: remainingSessions,
-        maxCost,
-        trackProgress: options.trackProgress ? taskName : void 0,
-        debugMode: options.debug,
-        onSession: (session) => {
-          const sessionCost = session.sessionCount > 1 ? session.totalCost - localState.sessions.slice(-session.sessionCount + 1).reduce((s, e) => s + e.cost, 0) : session.totalCost;
-          const thisCost = sessionCost > 0 ? sessionCost : session.totalCost / session.sessionCount;
-          localState.sessionCount = priorSessionCount + session.sessionCount;
-          localState.totalCost = priorCost + session.totalCost;
-          Object.assign(localState.context, session.context);
-          localState.lastOutput = session.lastOutput;
-          localState.status = session.status;
-          localState.sessions.push({
-            index: localState.sessionCount,
-            cost: thisCost,
-            stopReason: session.lastStopReason,
-            outputPreview: session.lastOutput.slice(0, 200),
-            at: (/* @__PURE__ */ new Date()).toISOString()
-          });
-          saveState(filePath, localState);
-          const total = localState.sessionCount;
-          const costStr = import_chalk15.default.yellow(`$${localState.totalCost.toFixed(4)}`);
-          const reasonColor = session.lastStopReason === "complete" ? import_chalk15.default.green : import_chalk15.default.gray;
+  try {
+    let currentSession = priorSessionCount;
+    const streamCallbacks = {
+      onAgentStart: () => {
+        currentSession++;
+        console.log(import_chalk15.default.cyan(`
+\u2500\u2500 Session ${currentSession} \u2500\u2500
+`));
+      },
+      onTurnDelta: (event) => {
+        if (event.contentType === "text") {
+          process.stdout.write(event.delta);
+        } else if (event.contentType === "thinking" && options.debug) {
+          process.stdout.write(import_chalk15.default.dim(event.delta));
+        }
+      },
+      onToolStart: (event) => {
+        if (options.debug) {
+          console.log(import_chalk15.default.gray(`
+  [tool] ${event.toolName}...`));
+        }
+      },
+      onToolComplete: (event) => {
+        if (options.debug) {
+          const status = event.success ? import_chalk15.default.green("ok") : import_chalk15.default.red("failed");
+          console.log(import_chalk15.default.gray(`  [tool] ${event.toolName} \u2192 ${status}`));
+        }
+      },
+      onError: (event) => {
+        console.error(import_chalk15.default.red(`
+  Error: ${event.error}`));
+      }
+    };
+    const result = await client.agents.runTask(agentId, {
+      message: options.message,
+      maxSessions: remainingSessions,
+      maxCost: remainingCost,
+      debugMode: options.debug,
+      stream: true,
+      streamCallbacks,
+      trackProgress: options.track ? taskName : void 0,
+      onSession: (state) => {
+        const adjustedState = {
+          ...state,
+          sessionCount: priorSessionCount + state.sessionCount,
+          totalCost: priorCost + state.totalCost
+        };
+        lastKnownState = adjustedState;
+        saveState(filePath, adjustedState);
+        const latest = state.sessions[state.sessions.length - 1];
+        if (latest) {
+          const total = adjustedState.sessionCount;
+          const costStr = import_chalk15.default.yellow(`$${adjustedState.totalCost.toFixed(4)}`);
+          const reasonColor = latest.stopReason === "complete" ? import_chalk15.default.green : import_chalk15.default.gray;
           console.log(
-            `  ${import_chalk15.default.dim(`[${total}/${maxSessions}]`)} ${reasonColor(session.lastStopReason)} | total: ${costStr}`
+            `
+  ${import_chalk15.default.dim(`[${total}/${maxSessions}]`)} ${reasonColor(latest.stopReason)} | total: ${costStr}`
           );
-          if (session.lastOutput) {
-            const preview = session.lastOutput.slice(0, 120).replace(/\n/g, " ");
-            console.log(
-              `    ${import_chalk15.default.dim(preview)}${session.lastOutput.length > 120 ? "..." : ""}`
-            );
-          }
         }
-      });
-      localState.status = result.status;
-      localState.totalCost = priorCost + result.totalCost;
-      Object.assign(localState.context, result.context);
-      saveState(filePath, localState);
-      process.removeListener("SIGINT", onSigint);
-      console.log();
-      const statusColor = result.status === "complete" ? import_chalk15.default.green : result.status === "budget_exceeded" ? import_chalk15.default.red : import_chalk15.default.yellow;
-      console.log(`Status:     ${statusColor(localState.status)}`);
-      console.log(`Sessions:   ${localState.sessionCount}`);
-      console.log(`Total cost: ${import_chalk15.default.yellow(`$${localState.totalCost.toFixed(4)}`)}`);
-      console.log(`State:      ${import_chalk15.default.gray(filePath)}`);
-      if (localState.status === "paused" || localState.status === "max_sessions") {
-        console.log(
-          import_chalk15.default.gray(
-            `
-Resume: runtype agents run-task ${agent} -m "${options.message}" --resume${options.name ? ` --name ${options.name}` : ""}`
-          )
-        );
-      }
-      if (options.json) {
-        console.log();
-        printJson(localState);
+        if (interrupted) return false;
       }
-    } catch (error) {
-      localState.status = "paused";
-      saveState(filePath, localState);
-      process.removeListener("SIGINT", onSigint);
-      const message = error instanceof Error ? error.message : "Unknown error";
-      console.error(import_chalk15.default.red(`
-Task failed: ${message}`));
-      console.log(import_chalk15.default.gray(`State saved to ${filePath} \u2014 resume with --resume`));
-      process.exit(1);
+    });
+    const finalState = {
+      agentId,
+      agentName: result.sessions[0]?.stopReason ? agentId : agentId,
+      taskName,
+      status: result.status,
+      sessionCount: priorSessionCount + result.sessionCount,
+      totalCost: priorCost + result.totalCost,
+      lastOutput: result.lastOutput,
+      lastStopReason: result.sessions[result.sessions.length - 1]?.stopReason || "complete",
+      sessions: result.sessions,
+      startedAt: (/* @__PURE__ */ new Date()).toISOString(),
+      updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+    };
+    saveState(filePath, finalState);
+    process.removeListener("SIGINT", onSigint);
+    console.log();
+    const statusColor = result.status === "complete" ? import_chalk15.default.green : result.status === "budget_exceeded" ? import_chalk15.default.red : import_chalk15.default.yellow;
+    console.log(`Status:     ${statusColor(result.status)}`);
+    console.log(`Sessions:   ${finalState.sessionCount}`);
+    console.log(`Total cost: ${import_chalk15.default.yellow(`$${finalState.totalCost.toFixed(4)}`)}`);
+    console.log(`State:      ${import_chalk15.default.gray(filePath)}`);
+    if (result.recordId) {
+      console.log(`Record:     ${import_chalk15.default.gray(result.recordId)}`);
+    }
+    if (result.status === "paused" || result.status === "max_sessions") {
+      console.log(
+        import_chalk15.default.gray(
+          `
+Resume: runtype marathon ${agent} -m "${options.message}" --resume${options.name ? ` --name ${options.name}` : ""}`
+        )
+      );
+    }
+    if (options.json) {
+      console.log();
+      printJson(finalState);
     }
+  } catch (error) {
+    const stateAtError = lastKnownState;
+    if (stateAtError) {
+      stateAtError.status = "paused";
+      saveState(filePath, stateAtError);
+    }
+    process.removeListener("SIGINT", onSigint);
+    const errMsg = error instanceof Error ? error.message : "Unknown error";
+    console.error(import_chalk15.default.red(`
+Task failed: ${errMsg}`));
+    console.log(import_chalk15.default.gray(`State saved to ${filePath} \u2014 resume with --resume`));
+    process.exit(1);
   }
+}
+function applyTaskOptions(cmd) {
+  return cmd.argument("<agent>", "Agent ID or name").requiredOption("-m, --message <text>", "Task message for the agent").option("--max-sessions <n>", "Maximum sessions", "50").option("--max-cost <n>", "Budget in USD").option("--name <name>", "Task name (used for state file, defaults to agent name)").option("--state-dir <path>", "Directory for state files (default: .runtype/tasks/)").option("--resume", "Resume from existing local state").option("--track", "Sync progress to a Runtype record (visible in dashboard)").option("--debug", "Show debug output from each session").option("--json", "Output final result as JSON").action(taskAction);
+}
+var taskCommand = applyTaskOptions(
+  new import_commander11.Command("task").description("Run a multi-session agent task")
 );
+function createMarathonCommand() {
+  return applyTaskOptions(
+    new import_commander11.Command("marathon").description("Run a multi-session agent task (alias for agents task)")
+  );
+}
 
 // src/commands/agents.ts
 var agentsCommand = new import_commander12.Command("agents").description("Manage agents");
-agentsCommand.addCommand(runTaskCommand);
+agentsCommand.addCommand(taskCommand);
 agentsCommand.command("list").description("List all agents").option("--json", "Output as JSON").option("--limit <n>", "Limit results", "20").action(async (options) => {
   const apiKey = await ensureAuth();
   if (!apiKey) return;
@@ -4077,6 +4092,7 @@ program.addCommand(apiKeysCommand);
 program.addCommand(analyticsCommand);
 program.addCommand(billingCommand);
 program.addCommand(flowVersionsCommand);
+program.addCommand(createMarathonCommand());
 program.exitOverride();
 try {
   if (!process.argv.slice(2).length) {