@hasna/computer 0.1.1 → 0.1.3

package/dist/cli/index.js

@@ -2211,10 +2211,15 @@ import { mkdir } from "fs/promises";
 import { tmpdir } from "os";
 import { join } from "path";
 import { readFile, unlink } from "fs/promises";
-async function captureScreenshot() {
+async function captureScreenshot(displayNumber) {
   const timestamp = Date.now();
   const tmpPath = join(tmpdir(), `computer-screenshot-${timestamp}.png`);
-  const proc = Bun.spawn(["screencapture", "-x", "-C", "-t", "png", tmpPath], {
+  const args = ["screencapture", "-x", "-C", "-t", "png"];
+  if (displayNumber) {
+    args.push(`-D${displayNumber}`);
+  }
+  args.push(tmpPath);
+  const proc = Bun.spawn(args, {
     stdout: "pipe",
     stderr: "pipe"
   });
@@ -2437,19 +2442,23 @@ function getScrollHelperPath() {
 
 // src/drivers/mac/index.ts
 class MacDriver {
+  displayNumber;
+  constructor(opts) {
+    this.displayNumber = opts?.displayNumber;
+  }
   async getScreenSize() {
     return getScreenSize();
   }
   async screenshot() {
-    return captureScreenshot();
+    return captureScreenshot(this.displayNumber);
   }
   async execute(action) {
     return executeAction(action);
   }
   async dispose() {}
 }
-function createMacDriver() {
-  return new MacDriver;
+function createMacDriver(opts) {
+  return new MacDriver(opts);
 }
 
 // node_modules/@anthropic-ai/sdk/version.mjs
@@ -21386,6 +21395,7 @@ function getDb() {
       total_tokens_in INTEGER NOT NULL DEFAULT 0,
       total_tokens_out INTEGER NOT NULL DEFAULT 0,
       total_duration_ms INTEGER NOT NULL DEFAULT 0,
+      tags TEXT,
       error TEXT,
       created_at TEXT NOT NULL,
       completed_at TEXT
@@ -21441,9 +21451,9 @@ function getDb() {
 async function createSession(session) {
   const d = getDb();
   d.prepare(`
-    INSERT INTO sessions (id, task, provider, model, status, steps, total_tokens_in, total_tokens_out, total_duration_ms, error, created_at, completed_at)
-    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-  `).run(session.id, session.task, session.provider, session.model, session.status, session.steps, session.total_tokens_in, session.total_tokens_out, session.total_duration_ms, session.error ?? null, session.created_at, session.completed_at ?? null);
+    INSERT INTO sessions (id, task, provider, model, status, steps, total_tokens_in, total_tokens_out, total_duration_ms, tags, error, created_at, completed_at)
+    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+  `).run(session.id, session.task, session.provider, session.model, session.status, session.steps, session.total_tokens_in, session.total_tokens_out, session.total_duration_ms, session.tags?.length ? JSON.stringify(session.tags) : null, session.error ?? null, session.created_at, session.completed_at ?? null);
 }
 async function updateSession(session) {
   const d = getDb();
@@ -21460,14 +21470,29 @@ async function logAction(params) {
     VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
   `).run(params.session_id, params.step, params.action.type, JSON.stringify(params.action), params.reasoning, params.screenshot_path ?? null, params.success ? 1 : 0, params.error ?? null, params.duration_ms, params.tokens_in ?? null, params.tokens_out ?? null);
 }
+function getSession(id) {
+  const d = getDb();
+  const row = d.prepare("SELECT * FROM sessions WHERE id = ?").get(id);
+  if (!row)
+    return null;
+  return rowToSession(row);
+}
 function listSessions(opts) {
   const d = getDb();
   let sql = "SELECT * FROM sessions";
   const params = [];
+  const conditions = [];
   if (opts?.status) {
-    sql += " WHERE status = ?";
+    conditions.push("status = ?");
     params.push(opts.status);
   }
+  if (opts?.tag) {
+    conditions.push("tags LIKE ?");
+    params.push(`%"${opts.tag}"%`);
+  }
+  if (conditions.length > 0) {
+    sql += " WHERE " + conditions.join(" AND ");
+  }
   sql += " ORDER BY created_at DESC";
   if (opts?.limit) {
     sql += " LIMIT ?";
@@ -21544,12 +21569,104 @@ function rowToSession(row) {
     total_tokens_in: row.total_tokens_in,
     total_tokens_out: row.total_tokens_out,
     total_duration_ms: row.total_duration_ms,
+    tags: row.tags ? JSON.parse(row.tags) : undefined,
     error: row.error,
     created_at: row.created_at,
     completed_at: row.completed_at
   };
 }
 
+// src/lib/integrations.ts
+async function saveToRecordings(session, logs) {
+  try {
+    const { saveRecording } = await import("@hasna/recordings");
+    await saveRecording({
+      title: `Computer Use: ${session.task.slice(0, 100)}`,
+      type: "computer-use",
+      source: "computer",
+      duration_ms: session.total_duration_ms,
+      metadata: {
+        session_id: session.id,
+        provider: session.provider,
+        model: session.model,
+        steps: session.steps,
+        tokens_in: session.total_tokens_in,
+        tokens_out: session.total_tokens_out,
+        status: session.status,
+        tags: session.tags
+      },
+      transcript: logs.map((l) => ({
+        step: l.step,
+        action: l.action.type,
+        reasoning: l.reasoning?.slice(0, 200),
+        success: l.success,
+        timestamp: l.created_at
+      }))
+    });
+    return true;
+  } catch {
+    return false;
+  }
+}
+async function registerWithSessions(session) {
+  try {
+    const mod = await import("@hasna/sessions");
+    const registerSession = mod.registerSession ?? mod.createSession ?? mod.saveSession;
+    if (typeof registerSession !== "function")
+      return false;
+    await registerSession({
+      id: session.id,
+      type: "computer-use",
+      source: "computer",
+      status: session.status,
+      metadata: {
+        task: session.task,
+        provider: session.provider,
+        model: session.model,
+        steps: session.steps
+      },
+      started_at: session.created_at,
+      ended_at: session.completed_at
+    });
+    return true;
+  } catch {
+    return false;
+  }
+}
+async function pushToLogs(session, logs) {
+  try {
+    const mod = await import("@hasna/logs");
+    const pushBatch = mod.logPushBatch ?? mod.pushBatch;
+    if (typeof pushBatch !== "function")
+      return false;
+    await pushBatch(logs.map((l) => ({
+      level: l.success ? "info" : "error",
+      source: "computer",
+      message: `[${l.action.type}] ${l.reasoning?.slice(0, 100) ?? ""}`,
+      metadata: {
+        session_id: session.id,
+        step: l.step,
+        action_type: l.action.type,
+        success: l.success,
+        error: l.error,
+        duration_ms: l.duration_ms
+      },
+      timestamp: l.created_at
+    })));
+    return true;
+  } catch {
+    return false;
+  }
+}
+async function runPostSessionIntegrations(session, logs) {
+  const [recordings, sessions, logsPushed] = await Promise.all([
+    saveToRecordings(session, logs),
+    registerWithSessions(session),
+    pushToLogs(session, logs)
+  ]);
+  return { recordings, sessions, logs: logsPushed };
+}
+
 // src/agent/loop.ts
 var DEFAULT_MAX_STEPS = 50;
 async function runTask(options) {
@@ -21563,10 +21680,12 @@ async function runTask(options) {
     systemPrompt,
     screenshotMaxWidth,
     dryRun = false,
+    tags,
+    displayNumber,
     onStep,
     onDone
   } = options;
-  const driver = createMacDriver();
+  const driver = createMacDriver({ displayNumber });
   const provider = createProvider(providerName, { model });
   const config = loadConfig();
   const safetyConfig = config.safety;
@@ -21577,6 +21696,7 @@ async function runTask(options) {
     provider: providerName,
     model: model ?? (providerName === "anthropic" ? "claude-sonnet-4-5-20250514" : "computer-use-preview"),
     status: "running",
+    tags,
     steps: 0,
     total_tokens_in: 0,
     total_tokens_out: 0,
@@ -21629,6 +21749,8 @@ async function runTask(options) {
         });
         await updateSession(session);
         onStep?.(step, response, { success: true, duration_ms: 0 });
+        const logs = getActionLogs(sessionId);
+        await runPostSessionIntegrations(session, logs).catch(() => {});
         onDone?.(session);
         await driver.dispose();
         return session;
@@ -21694,6 +21816,8 @@ async function runTask(options) {
     session.completed_at = new Date().toISOString();
     session.error = `Reached max steps (${maxSteps})`;
     await updateSession(session);
+    const endLogs = getActionLogs(sessionId);
+    await runPostSessionIntegrations(session, endLogs).catch(() => {});
     onDone?.(session);
     await driver.dispose();
     return session;
@@ -21703,6 +21827,8 @@ async function runTask(options) {
     session.total_duration_ms = Date.now() - startTime;
     session.completed_at = new Date().toISOString();
     await updateSession(session);
+    const errLogs = getActionLogs(sessionId);
+    await runPostSessionIntegrations(session, errLogs).catch(() => {});
     onDone?.(session);
     await driver.dispose();
     return session;
@@ -21775,10 +21901,65 @@ function findPricing(model) {
   return { input: 3, output: 12 };
 }
 
+// src/lib/terminal-image.ts
+function detectProtocol() {
+  const term = process.env.TERM_PROGRAM ?? "";
+  const termInfo = process.env.TERM ?? "";
+  if (term === "iTerm.app" || term === "WezTerm") {
+    return "iterm2";
+  }
+  if (termInfo.includes("kitty") || process.env.KITTY_PID) {
+    return "kitty";
+  }
+  return "none";
+}
+function renderInlineImage(base64, opts) {
+  const protocol = detectProtocol();
+  const width = opts?.width ?? 40;
+  const height = opts?.height ?? 15;
+  switch (protocol) {
+    case "iterm2":
+      return renderIterm2(base64, width, height, opts?.preserveAspectRatio ?? true);
+    case "kitty":
+      return renderKitty(base64, width, height);
+    case "none":
+      return "";
+  }
+}
+function supportsInlineImages() {
+  return detectProtocol() !== "none";
+}
+function renderIterm2(base64, width, height, preserveAspectRatio) {
+  const params = [
+    `width=${width}`,
+    `height=${height}`,
+    `preserveAspectRatio=${preserveAspectRatio ? 1 : 0}`,
+    "inline=1"
+  ].join(";");
+  return `\x1B]1337;File=${params}:${base64}\x07
+`;
+}
+function renderKitty(base64, width, height) {
+  const chunkSize = 4096;
+  const chunks = [];
+  for (let i = 0;i < base64.length; i += chunkSize) {
+    const chunk = base64.slice(i, i + chunkSize);
+    const isLast = i + chunkSize >= base64.length;
+    const more = isLast ? 0 : 1;
+    if (i === 0) {
+      chunks.push(`\x1B_Gf=100,a=T,m=${more},c=${width},r=${height};${chunk}\x1B\\`);
+    } else {
+      chunks.push(`\x1B_Gm=${more};${chunk}\x1B\\`);
+    }
+  }
+  return chunks.join("") + `
+`;
+}
+
 // src/cli/index.ts
 var program2 = new Command;
 program2.name("computer").description("Open-source computer use for AI agents \u2014 control your Mac with AI").version("0.1.0");
-program2.command("run").description("Run a computer use task").argument("<task>", "Natural language description of the task").option("-p, --provider <provider>", "AI provider (anthropic|openai)", "anthropic").option("-m, --model <model>", "Model to use").option("-s, --max-steps <n>", "Maximum number of steps", "50").option("--save-screenshots", "Save screenshots to disk", false).option("--screenshots-dir <dir>", "Directory to save screenshots").option("--system-prompt <prompt>", "Custom system prompt").option("--max-width <px>", "Max screenshot width for AI model (default: 1280)", "1280").option("--dry-run", "Plan actions without executing them", false).action(async (task, opts) => {
+program2.command("run").description("Run a computer use task").argument("<task>", "Natural language description of the task").option("-p, --provider <provider>", "AI provider (anthropic|openai)", "anthropic").option("-m, --model <model>", "Model to use").option("-s, --max-steps <n>", "Maximum number of steps", "50").option("--save-screenshots", "Save screenshots to disk", false).option("--screenshots-dir <dir>", "Directory to save screenshots").option("--system-prompt <prompt>", "Custom system prompt").option("--max-width <px>", "Max screenshot width for AI model (default: 1280)", "1280").option("--dry-run", "Plan actions without executing them", false).option("--no-preview", "Disable inline screenshot preview in terminal").option("--tag <tags...>", "Tag this session (can specify multiple)").option("--display <n>", "Display number to capture (1=main, 2=secondary)").action(async (task, opts) => {
   const cfg = loadConfig();
   const provider = opts.provider ?? cfg.provider;
   const maxSteps = parseInt(opts.maxSteps) || cfg.maxSteps;
@@ -21797,6 +21978,8 @@ program2.command("run").description("Run a computer use task").argument("<task>"
     systemPrompt: opts.systemPrompt,
     screenshotMaxWidth: maxWidth,
     dryRun: opts.dryRun,
+    tags: opts.tag,
+    displayNumber: opts.display ? parseInt(opts.display) : undefined,
     onStep: (step, response, result) => {
       const status = result.success ? chalk.green("OK") : chalk.red("FAIL");
       const actionDesc = response.action ? `${response.action.type}${response.action.type === "click" ? ` (${response.action.point.x},${response.action.point.y})` : ""}` : "done";
@@ -21806,6 +21989,11 @@ program2.command("run").description("Run a computer use task").argument("<task>"
         const short = response.reasoning.slice(0, 120).replace(/\n/g, " ");
         console.log(chalk.dim(`      ${short}${response.reasoning.length > 120 ? "..." : ""}`));
       }
+      if (opts.preview !== false && result.screenshot && supportsInlineImages()) {
+        const img = renderInlineImage(result.screenshot.base64, { width: 40, height: 12 });
+        if (img)
+          process.stdout.write(img);
+      }
     },
     onDone: (session2) => {
       console.log();
@@ -21834,10 +22022,11 @@ program2.command("screenshot").description("Take a screenshot of the current scr
     console.log(chalk.dim(`Base64 length: ${ss.base64.length} chars`));
   }
 });
-program2.command("sessions").description("List computer use sessions").option("-n, --limit <n>", "Number of sessions to show", "20").option("--status <status>", "Filter by status").action(async (opts) => {
+program2.command("sessions").description("List computer use sessions").option("-n, --limit <n>", "Number of sessions to show", "20").option("--status <status>", "Filter by status").option("--tag <tag>", "Filter by tag").action(async (opts) => {
   const sessions = listSessions({
     limit: parseInt(opts.limit),
-    status: opts.status
+    status: opts.status,
+    tag: opts.tag
   });
   if (sessions.length === 0) {
     console.log(chalk.dim("No sessions found."));
@@ -21845,7 +22034,8 @@ program2.command("sessions").description("List computer use sessions").option("-
   }
   for (const s of sessions) {
     const statusColor = s.status === "completed" ? chalk.green : s.status === "failed" ? chalk.red : s.status === "running" ? chalk.yellow : chalk.dim;
-    console.log(`${chalk.dim(s.id.slice(0, 8))} ${statusColor(s.status.padEnd(10))} ${chalk.cyan(s.provider.padEnd(10))} ${s.steps} steps  ${chalk.dim(s.created_at)}`);
+    const tagStr = s.tags?.length ? chalk.magenta(` [${s.tags.join(", ")}]`) : "";
+    console.log(`${chalk.dim(s.id.slice(0, 8))} ${statusColor(s.status.padEnd(10))} ${chalk.cyan(s.provider.padEnd(10))} ${s.steps} steps${tagStr}  ${chalk.dim(s.created_at)}`);
     console.log(chalk.dim(`  ${s.task.slice(0, 100)}`));
   }
 });
@@ -21899,6 +22089,80 @@ program2.command("stats").description("Show usage statistics").action(async () =
   console.log(`  Steps:     ${stats.total_steps}`);
   console.log(`  Tokens:    ${stats.total_tokens.toLocaleString()}`);
 });
+program2.command("watch").description("Live-stream what the agent sees (polls running session)").argument("[id]", "Session ID to watch (default: latest running)").option("-i, --interval <ms>", "Poll interval in milliseconds", "500").option("--no-preview", "Disable inline screenshot preview").action(async (id, opts) => {
+  const interval = parseInt(opts.interval);
+  let sessionId = id;
+  if (!sessionId) {
+    const running = listSessions({ status: "running", limit: 1 });
+    if (running.length === 0) {
+      console.log(chalk.yellow("No running sessions found. Start one with `computer run <task>`."));
+      process.exit(0);
+    }
+    sessionId = running[0].id;
+  } else {
+    const all = listSessions({ limit: 100 });
+    const match = all.find((s) => s.id.startsWith(sessionId));
+    if (match)
+      sessionId = match.id;
+  }
+  console.log(chalk.bold.cyan("computer watch") + ` \u2014 session ${chalk.dim(sessionId.slice(0, 8))}`);
+  console.log(chalk.dim(`Polling every ${interval}ms. Press Ctrl+C to stop.
+`));
+  let lastStep = -1;
+  const poll = async () => {
+    const session = getSession(sessionId);
+    if (!session) {
+      console.log(chalk.red("Session not found."));
+      process.exit(1);
+    }
+    const logs = getActionLogs(sessionId);
+    const newLogs = logs.filter((l) => l.step > lastStep);
+    for (const log of newLogs) {
+      const status = log.success ? chalk.green("OK") : chalk.red("FAIL");
+      console.log(chalk.dim(`[${String(log.step + 1).padStart(3)}]`) + ` ${status} ${chalk.yellow(log.action.type)}` + chalk.dim(` (${log.duration_ms}ms)`));
+      if (log.reasoning) {
+        const short = log.reasoning.slice(0, 120).replace(/\n/g, " ");
+        console.log(chalk.dim(`      ${short}${log.reasoning.length > 120 ? "..." : ""}`));
+      }
+      if (log.error) {
+        console.log(chalk.red(`      Error: ${log.error}`));
+      }
+      if (opts.preview !== false && log.screenshot_path && supportsInlineImages()) {
+        try {
+          const { readFileSync: readFileSync3 } = await import("fs");
+          const imgData = readFileSync3(log.screenshot_path);
+          const b64 = imgData.toString("base64");
+          const img = renderInlineImage(b64, { width: 40, height: 12 });
+          if (img)
+            process.stdout.write(img);
+        } catch {}
+      }
+      lastStep = log.step;
+    }
+    if (session.status !== "running") {
+      console.log();
+      const totalCost = formatCost(calculateCost(session.model, session.total_tokens_in, session.total_tokens_out));
+      if (session.status === "completed" && !session.error) {
+        console.log(chalk.green.bold("Session completed."));
+      } else if (session.status === "completed") {
+        console.log(chalk.yellow.bold(`Session finished: ${session.error}`));
+      } else {
+        console.log(chalk.red.bold(`Session ${session.status}: ${session.error}`));
+      }
+      console.log(chalk.dim(`Steps: ${session.steps} | Tokens: ${(session.total_tokens_in + session.total_tokens_out).toLocaleString()} | Cost: ${totalCost} | Duration: ${(session.total_duration_ms / 1000).toFixed(1)}s`));
+      process.exit(0);
+    }
+  };
+  await poll();
+  const timer = setInterval(poll, interval);
+  process.on("SIGINT", () => {
+    clearInterval(timer);
+    console.log(chalk.dim(`
+Stopped watching.`));
+    process.exit(0);
+  });
+  await new Promise(() => {});
+});
 program2.command("search").description("Search sessions by task text").argument("<query>", "Search query").option("-n, --limit <n>", "Max results", "20").action(async (query, opts) => {
   const sessions = searchSessions(query, parseInt(opts.limit));
   if (sessions.length === 0) {
@@ -21907,7 +22171,8 @@ program2.command("search").description("Search sessions by task text").argument(
   }
   for (const s of sessions) {
     const statusColor = s.status === "completed" ? chalk.green : s.status === "failed" ? chalk.red : chalk.dim;
-    console.log(`${chalk.dim(s.id.slice(0, 8))} ${statusColor(s.status.padEnd(10))} ${chalk.cyan(s.provider.padEnd(10))} ${s.steps} steps  ${chalk.dim(s.created_at)}`);
+    const tagStr = s.tags?.length ? chalk.magenta(` [${s.tags.join(", ")}]`) : "";
+    console.log(`${chalk.dim(s.id.slice(0, 8))} ${statusColor(s.status.padEnd(10))} ${chalk.cyan(s.provider.padEnd(10))} ${s.steps} steps${tagStr}  ${chalk.dim(s.created_at)}`);
     console.log(chalk.dim(`  ${s.task.slice(0, 100)}`));
   }
 });
@@ -21949,5 +22214,167 @@ configCmd.command("reset").description("Reset config to defaults").action(async
   saveConfig2(DEFAULT_CONFIG2);
   console.log(chalk.green("Config reset to defaults."));
 });
+program2.command("replay").description("Replay a session \u2014 show actions and screenshots in sequence").argument("<id>", "Session ID (or prefix)").option("--speed <x>", "Replay speed multiplier (default: 2)", "2").option("--no-preview", "Disable inline screenshot preview").action(async (id, opts) => {
+  const sessions = listSessions({ limit: 1000 });
+  const session = sessions.find((s) => s.id.startsWith(id));
+  if (!session) {
+    console.log(chalk.red(`Session not found: ${id}`));
+    process.exit(1);
+  }
+  const logs = getActionLogs(session.id);
+  if (logs.length === 0) {
+    console.log(chalk.dim("No action logs for this session."));
+    process.exit(0);
+  }
+  const speed = parseFloat(opts.speed);
+  const totalCost = formatCost(calculateCost(session.model, session.total_tokens_in, session.total_tokens_out));
+  console.log(chalk.bold.cyan("computer replay") + ` \u2014 ${chalk.dim(session.id.slice(0, 8))}`);
+  console.log(chalk.dim(`Task: ${session.task}`));
+  console.log(chalk.dim(`Provider: ${session.provider} | ${logs.length} steps | Speed: ${speed}x
+`));
+  for (let i = 0;i < logs.length; i++) {
+    const log = logs[i];
+    const status = log.success ? chalk.green("OK") : chalk.red("FAIL");
+    console.log(chalk.dim(`[${String(log.step + 1).padStart(3)}]`) + ` ${status} ${chalk.yellow(log.action.type)}` + chalk.dim(` (${log.duration_ms}ms)`));
+    if (log.reasoning) {
+      const short = log.reasoning.slice(0, 120).replace(/\n/g, " ");
+      console.log(chalk.dim(`      ${short}${log.reasoning.length > 120 ? "..." : ""}`));
+    }
+    if (log.error) {
+      console.log(chalk.red(`      Error: ${log.error}`));
+    }
+    if (opts.preview !== false && log.screenshot_path && supportsInlineImages()) {
+      try {
+        const { readFileSync: readFileSync3 } = await import("fs");
+        const imgData = readFileSync3(log.screenshot_path);
+        const b64 = imgData.toString("base64");
+        const img = renderInlineImage(b64, { width: 40, height: 12 });
+        if (img)
+          process.stdout.write(img);
+      } catch {}
+    }
+    if (i < logs.length - 1) {
+      const nextLog = logs[i + 1];
+      const delay = Math.max(100, Math.round(log.duration_ms / speed));
+      await new Promise((r) => setTimeout(r, delay));
+    }
+  }
+  console.log();
+  console.log(chalk.dim(`Replay complete. ${logs.length} steps | Cost: ${totalCost} | Duration: ${(session.total_duration_ms / 1000).toFixed(1)}s`));
+});
+program2.command("completions").description("Generate shell completions").argument("<shell>", "Shell type (zsh or bash)").action(async (shell) => {
+  if (shell === "zsh") {
+    console.log(generateZshCompletions());
+  } else if (shell === "bash") {
+    console.log(generateBashCompletions());
+  } else {
+    console.error(chalk.red(`Unknown shell: ${shell}. Use "zsh" or "bash".`));
+    process.exit(1);
+  }
+});
 registerCloudCommands(program2, "computer");
 program2.parse();
+function generateZshCompletions() {
+  return `#compdef computer
+# Zsh completions for @hasna/computer
+# Install: computer completions zsh > ~/.zsh/completions/_computer
+
+_computer() {
+  local -a commands
+  commands=(
+    'run:Run a computer use task'
+    'screenshot:Take a screenshot'
+    'sessions:List sessions'
+    'session:Show session details'
+    'delete:Delete a session'
+    'stats:Show usage statistics'
+    'watch:Live-stream agent activity'
+    'search:Search sessions'
+    'config:View or modify configuration'
+    'completions:Generate shell completions'
+    'cloud:Cloud sync and feedback'
+  )
+
+  _arguments -C \\
+    '1:command:->command' \\
+    '*::arg:->args'
+
+  case "$state" in
+    command)
+      _describe 'command' commands
+      ;;
+    args)
+      case "$words[1]" in
+        run)
+          _arguments \\
+            '-p[AI provider]:provider:(anthropic openai)' \\
+            '-m[Model to use]:model:' \\
+            '-s[Max steps]:steps:' \\
+            '--save-screenshots[Save screenshots]' \\
+            '--dry-run[Plan without executing]' \\
+            '--tag[Tag session]:tag:' \\
+            '--max-width[Max screenshot width]:width:' \\
+            '--no-preview[Disable inline preview]' \\
+            '1:task:'
+          ;;
+        sessions)
+          _arguments \\
+            '-n[Limit]:limit:' \\
+            '--status[Filter by status]:status:(running completed failed cancelled)' \\
+            '--tag[Filter by tag]:tag:'
+          ;;
+        config)
+          local -a config_cmds
+          config_cmds=(show get set path edit reset)
+          _describe 'config command' config_cmds
+          ;;
+        completions)
+          _arguments '1:shell:(zsh bash)'
+          ;;
+      esac
+      ;;
+  esac
+}
+
+_computer "$@"`;
+}
+function generateBashCompletions() {
+  return `# Bash completions for @hasna/computer
+# Install: computer completions bash >> ~/.bashrc
+
+_computer_completions() {
+  local cur prev commands
+  COMPREPLY=()
+  cur="\${COMP_WORDS[COMP_CWORD]}"
+  prev="\${COMP_WORDS[COMP_CWORD-1]}"
+  commands="run screenshot sessions session delete stats watch search config completions cloud"
+
+  if [ "$COMP_CWORD" -eq 1 ]; then
+    COMPREPLY=( $(compgen -W "$commands" -- "$cur") )
+    return 0
+  fi
+
+  case "\${COMP_WORDS[1]}" in
+    run)
+      COMPREPLY=( $(compgen -W "-p --provider -m --model -s --max-steps --save-screenshots --dry-run --tag --max-width --no-preview" -- "$cur") )
+      ;;
+    sessions)
+      COMPREPLY=( $(compgen -W "-n --limit --status --tag" -- "$cur") )
+      ;;
+    config)
+      COMPREPLY=( $(compgen -W "show get set path edit reset" -- "$cur") )
+      ;;
+    completions)
+      COMPREPLY=( $(compgen -W "zsh bash" -- "$cur") )
+      ;;
+    --provider|-p)
+      COMPREPLY=( $(compgen -W "anthropic openai" -- "$cur") )
+      ;;
+    --status)
+      COMPREPLY=( $(compgen -W "running completed failed cancelled" -- "$cur") )
+      ;;
+  esac
+}
+
+complete -F _computer_completions computer`;
+}

package/dist/db/agents.d.ts

@@ -0,0 +1,26 @@
+export interface Agent {
+    id: string;
+    name: string;
+    description?: string;
+    capabilities?: string[];
+    focus?: string;
+    last_heartbeat: string;
+    created_at: string;
+}
+/** Ensure agents table exists */
+export declare function ensureAgentsTable(): void;
+/** Register or update an agent */
+export declare function registerAgent(agent: {
+    name: string;
+    description?: string;
+    capabilities?: string[];
+}): Agent;
+/** Send a heartbeat for an agent */
+export declare function heartbeat(agentId: string): boolean;
+/** Set agent focus */
+export declare function setFocus(agentId: string, focus: string): boolean;
+/** Get an agent by ID */
+export declare function getAgent(id: string): Agent | null;
+/** List all agents */
+export declare function listAgents(): Agent[];
+//# sourceMappingURL=agents.d.ts.map

package/dist/db/agents.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agents.d.ts","sourceRoot":"","sources":["../../src/db/agents.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,KAAK;IACpB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,cAAc,EAAE,MAAM,CAAC;IACvB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,iCAAiC;AACjC,wBAAgB,iBAAiB,IAAI,IAAI,CAaxC;AAED,kCAAkC;AAClC,wBAAgB,aAAa,CAAC,KAAK,EAAE;IACnC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;CACzB,GAAG,KAAK,CA0BR;AAED,oCAAoC;AACpC,wBAAgB,SAAS,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAQlD;AAED,sBAAsB;AACtB,wBAAgB,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAKhE;AAED,yBAAyB;AACzB,wBAAgB,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,KAAK,GAAG,IAAI,CAMjD;AAED,sBAAsB;AACtB,wBAAgB,UAAU,IAAI,KAAK,EAAE,CAKpC"}

package/dist/db/index.d.ts

@@ -24,6 +24,7 @@ export declare function getSession(id: string): Session | null;
 /** List sessions */
 export declare function listSessions(opts?: {
     status?: SessionStatus;
+    tag?: string;
     limit?: number;
     offset?: number;
 }): Session[];

package/dist/db/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/db/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,KAAK,SAAS,EAAE,MAAM,cAAc,CAAC;AAE7D,OAAO,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAMzF,0DAA0D;AAC1D,wBAAgB,KAAK,IAAI,SAAS,CA4EjC;AAED,2BAA2B;AAC3B,wBAAsB,aAAa,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAmBnE;AAED,uBAAuB;AACvB,wBAAsB,aAAa,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAgBnE;AAED,qCAAqC;AACrC,wBAAsB,SAAS,CAAC,MAAM,EAAE;IACtC,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,YAAY,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,GAAG,OAAO,CAAC,IAAI,CAAC,CAkBhB;AAED,0BAA0B;AAC1B,wBAAgB,UAAU,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,GAAG,IAAI,CAKrD;AAED,oBAAoB;AACpB,wBAAgB,YAAY,CAAC,IAAI,CAAC,EAAE;IAClC,MAAM,CAAC,EAAE,aAAa,CAAC;IACvB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,GAAG,OAAO,EAAE,CAsBZ;AAED,0CAA0C;AAC1C,wBAAgB,cAAc,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,GAAE,MAAW,GAAG,OAAO,EAAE,CAU3E;AAED,kDAAkD;AAClD,wBAAgB,gBAAgB,CAC9B,KAAK,EAAE,MAAM,EACb,KAAK,GAAE,MAAW,GACjB,SAAS,EAAE,CAuBb;AAED,oCAAoC;AACpC,wBAAgB,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,SAAS,EAAE,CAoB5D;AAED,oCAAoC;AACpC,wBAAgB,aAAa,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAKjD;AAED,wBAAwB;AACxB,wBAAgB,QAAQ,IAAI;IAC1B,cAAc,EAAE,MAAM,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;CACtB,CAmBA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/db/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,KAAK,SAAS,EAAE,MAAM,cAAc,CAAC;AAE7D,OAAO,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAMzF,0DAA0D;AAC1D,wBAAgB,KAAK,IAAI,SAAS,CA6EjC;AAED,2BAA2B;AAC3B,wBAAsB,aAAa,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAoBnE;AAED,uBAAuB;AACvB,wBAAsB,aAAa,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAgBnE;AAED,qCAAqC;AACrC,wBAAsB,SAAS,CAAC,MAAM,EAAE;IACtC,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,YAAY,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,GAAG,OAAO,CAAC,IAAI,CAAC,CAkBhB;AAED,0BAA0B;AAC1B,wBAAgB,UAAU,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,GAAG,IAAI,CAKrD;AAED,oBAAoB;AACpB,wBAAgB,YAAY,CAAC,IAAI,CAAC,EAAE;IAClC,MAAM,CAAC,EAAE,aAAa,CAAC;IACvB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,GAAG,OAAO,EAAE,CA8BZ;AAED,0CAA0C;AAC1C,wBAAgB,cAAc,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,GAAE,MAAW,GAAG,OAAO,EAAE,CAU3E;AAED,kDAAkD;AAClD,wBAAgB,gBAAgB,CAC9B,KAAK,EAAE,MAAM,EACb,KAAK,GAAE,MAAW,GACjB,SAAS,EAAE,CAuBb;AAED,oCAAoC;AACpC,wBAAgB,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,SAAS,EAAE,CAoB5D;AAED,oCAAoC;AACpC,wBAAgB,aAAa,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAKjD;AAED,wBAAwB;AACxB,wBAAgB,QAAQ,IAAI;IAC1B,cAAc,EAAE,MAAM,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;CACtB,CAmBA"}