@hasna/computer 0.1.1 → 0.1.3

package/dist/index.js

@@ -1,4 +1,6 @@
 // @bun
+var __require = import.meta.require;
+
 // src/agent/loop.ts
 import { randomUUID } from "crypto";
 import { mkdir } from "fs/promises";
@@ -7,10 +9,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"
   });
@@ -233,19 +240,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
@@ -9586,7 +9597,7 @@ var __export = (target, all) => {
       set: __exportSetter.bind(all, name)
     });
 };
-var __require = /* @__PURE__ */ createRequire(import.meta.url);
+var __require2 = /* @__PURE__ */ createRequire(import.meta.url);
 var require_postgres_array = __commonJS((exports) => {
   exports.parse = function(source, transform) {
     return new ArrayParser(source, transform).parse();
@@ -10526,7 +10537,7 @@ var require_defaults = __commonJS((exports, module) => {
 });
 var require_utils = __commonJS((exports, module) => {
   var defaults2 = require_defaults();
-  var util = __require("util");
+  var util = __require2("util");
   var { isDate } = util.types || util;
   function escapeElement(elementRepresentation) {
     const escaped = elementRepresentation.replace(/\\/g, "\\\\").replace(/"/g, "\\\"");
@@ -10682,7 +10693,7 @@ var require_utils = __commonJS((exports, module) => {
   };
 });
 var require_utils_legacy = __commonJS((exports, module) => {
-  var nodeCrypto = __require("crypto");
+  var nodeCrypto = __require2("crypto");
   function md5(string) {
     return nodeCrypto.createHash("md5").update(string, "utf-8").digest("hex");
   }
@@ -10715,7 +10726,7 @@ var require_utils_legacy = __commonJS((exports, module) => {
   };
 });
 var require_utils_webcrypto = __commonJS((exports, module) => {
-  var nodeCrypto = __require("crypto");
+  var nodeCrypto = __require2("crypto");
   module.exports = {
     postgresMd5PasswordHash,
     randomBytes,
@@ -11125,7 +11136,7 @@ var require_pg_connection_string = __commonJS((exports, module) => {
     if (config.sslcert || config.sslkey || config.sslrootcert || config.sslmode) {
       config.ssl = {};
     }
-    const fs = config.sslcert || config.sslkey || config.sslrootcert ? __require("fs") : null;
+    const fs = config.sslcert || config.sslkey || config.sslrootcert ? __require2("fs") : null;
     if (config.sslcert) {
       config.ssl.cert = fs.readFileSync(config.sslcert).toString();
     }
@@ -11248,7 +11259,7 @@ See https://www.postgresql.org/docs/current/libpq-ssl.html for libpq SSL mode de
   parse.parseIntoClientConfig = parseIntoClientConfig;
 });
 var require_connection_parameters = __commonJS((exports, module) => {
-  var dns = __require("dns");
+  var dns = __require2("dns");
   var defaults2 = require_defaults();
   var parse = require_pg_connection_string().parse;
   var val = function(key, config, envVar) {
@@ -11471,7 +11482,7 @@ var require_result = __commonJS((exports, module) => {
   module.exports = Result;
 });
 var require_query = __commonJS((exports, module) => {
-  var { EventEmitter } = __require("events");
+  var { EventEmitter } = __require2("events");
   var Result = require_result();
   var utils = require_utils();
 
@@ -12454,11 +12465,11 @@ var require_stream = __commonJS((exports, module) => {
   };
   function getNodejsStreamFuncs() {
     function getStream2(ssl) {
-      const net = __require("net");
+      const net = __require2("net");
       return new net.Socket;
     }
     function getSecureStream2(options) {
-      const tls = __require("tls");
+      const tls = __require2("tls");
       return tls.connect(options);
     }
     return {
@@ -12500,7 +12511,7 @@ var require_stream = __commonJS((exports, module) => {
   }
 });
 var require_connection = __commonJS((exports, module) => {
-  var EventEmitter = __require("events").EventEmitter;
+  var EventEmitter = __require2("events").EventEmitter;
   var { parse, serialize } = require_dist();
   var { getStream, getSecureStream } = require_stream();
   var flushBuffer = serialize.flush();
@@ -12573,7 +12584,7 @@ var require_connection = __commonJS((exports, module) => {
             options.key = self.ssl.key;
           }
         }
-        const net = __require("net");
+        const net = __require2("net");
         if (net.isIP && net.isIP(host) === 0) {
           options.servername = host;
         }
@@ -12676,8 +12687,8 @@ var require_connection = __commonJS((exports, module) => {
   module.exports = Connection;
 });
 var require_split2 = __commonJS((exports, module) => {
-  var { Transform } = __require("stream");
-  var { StringDecoder } = __require("string_decoder");
+  var { Transform } = __require2("stream");
+  var { StringDecoder } = __require2("string_decoder");
   var kLast = Symbol("last");
   var kDecoder = Symbol("decoder");
   function transform(chunk, enc, cb) {
@@ -12773,10 +12784,10 @@ var require_split2 = __commonJS((exports, module) => {
   module.exports = split;
 });
 var require_helper = __commonJS((exports, module) => {
-  var path = __require("path");
-  var Stream3 = __require("stream").Stream;
+  var path = __require2("path");
+  var Stream3 = __require2("stream").Stream;
   var split = require_split2();
-  var util = __require("util");
+  var util = __require2("util");
   var defaultPort = 5432;
   var isWin = process.platform === "win32";
   var warnStream = process.stderr;
@@ -12935,8 +12946,8 @@ var require_helper = __commonJS((exports, module) => {
   };
 });
 var require_lib = __commonJS((exports, module) => {
-  var path = __require("path");
-  var fs = __require("fs");
+  var path = __require2("path");
+  var fs = __require2("fs");
   var helper = require_helper();
   module.exports = function(connInfo, cb) {
     var file = helper.getFileName();
@@ -12951,9 +12962,9 @@ var require_lib = __commonJS((exports, module) => {
   module.exports.warnTo = helper.warnTo;
 });
 var require_client = __commonJS((exports, module) => {
-  var EventEmitter = __require("events").EventEmitter;
+  var EventEmitter = __require2("events").EventEmitter;
   var utils = require_utils();
-  var nodeUtils = __require("util");
+  var nodeUtils = __require2("util");
   var sasl = require_sasl();
   var TypeOverrides = require_type_overrides();
   var ConnectionParameters = require_connection_parameters();
@@ -13530,7 +13541,7 @@ var require_client = __commonJS((exports, module) => {
   module.exports = Client;
 });
 var require_pg_pool = __commonJS((exports, module) => {
-  var EventEmitter = __require("events").EventEmitter;
+  var EventEmitter = __require2("events").EventEmitter;
   var NOOP = function() {};
   var removeWhere = (list, predicate) => {
     const i = list.findIndex(predicate);
@@ -13940,8 +13951,8 @@ var require_pg_pool = __commonJS((exports, module) => {
   module.exports = Pool;
 });
 var require_query2 = __commonJS((exports, module) => {
-  var EventEmitter = __require("events").EventEmitter;
-  var util = __require("util");
+  var EventEmitter = __require2("events").EventEmitter;
+  var util = __require2("util");
   var utils = require_utils();
   var NativeQuery = module.exports = function(config, values, callback) {
     EventEmitter.call(this);
@@ -14074,7 +14085,7 @@ var require_query2 = __commonJS((exports, module) => {
   };
 });
 var require_client2 = __commonJS((exports, module) => {
-  var nodeUtils = __require("util");
+  var nodeUtils = __require2("util");
   var Native;
   try {
     Native = (() => {
@@ -14084,8 +14095,8 @@ var require_client2 = __commonJS((exports, module) => {
     throw e;
   }
   var TypeOverrides = require_type_overrides();
-  var EventEmitter = __require("events").EventEmitter;
-  var util = __require("util");
+  var EventEmitter = __require2("events").EventEmitter;
+  var util = __require2("util");
   var ConnectionParameters = require_connection_parameters();
   var NativeQuery = require_query2();
   var queryQueueLengthDeprecationNotice = nodeUtils.deprecate(() => {}, "Calling client.query() when the client is already executing a query is deprecated and will be removed in pg@9.0. Use async/await or an external async flow control mechanism instead.");
@@ -18579,6 +18590,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
@@ -18634,9 +18646,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();
@@ -18664,10 +18676,18 @@ 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 ?";
@@ -18768,12 +18788,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) {
@@ -18787,10 +18899,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;
@@ -18801,6 +18915,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,
@@ -18853,6 +18968,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;
@@ -18918,6 +19035,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;
@@ -18927,6 +19046,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;
@@ -18953,6 +19074,152 @@ function remapCoordinates(action, from, to) {
       break;
   }
 }
+// src/drivers/mac/accessibility.ts
+import { join as join6, dirname as dirname2 } from "path";
+import { existsSync as existsSync4 } from "fs";
+import { fileURLToPath as fileURLToPath2 } from "url";
+async function queryAccessibilityTree(opts) {
+  const helperPath = getAccessibilityHelperPath();
+  const args = [helperPath];
+  if (opts?.app) {
+    args.push("--app", opts.app);
+  }
+  if (opts?.focusedOnly) {
+    args.push("--focused");
+  }
+  if (opts?.depth !== undefined) {
+    args.push("--depth", String(opts.depth));
+  }
+  const proc = Bun.spawn(args, {
+    stdout: "pipe",
+    stderr: "pipe"
+  });
+  await proc.exited;
+  if (proc.exitCode !== 0) {
+    const stderr = await new Response(proc.stderr).text();
+    throw new Error(`accessibility query failed: ${stderr}`);
+  }
+  const stdout = await new Response(proc.stdout).text();
+  try {
+    return JSON.parse(stdout);
+  } catch {
+    return [];
+  }
+}
+function summarizeAccessibilityTree(elements) {
+  if (elements.length === 0)
+    return "No accessibility information available.";
+  const lines = ["UI Elements:"];
+  for (const el of elements) {
+    const parts = [];
+    if (el.role)
+      parts.push(el.role);
+    if (el.title)
+      parts.push(`"${el.title}"`);
+    if (el.label)
+      parts.push(`(${el.label})`);
+    if (el.value)
+      parts.push(`value="${el.value}"`);
+    const pos = `at (${el.x + Math.round(el.width / 2)}, ${el.y + Math.round(el.height / 2)})`;
+    const size = `${el.width}x${el.height}`;
+    const flags = [];
+    if (!el.enabled)
+      flags.push("disabled");
+    if (el.focused)
+      flags.push("focused");
+    lines.push(`  - ${parts.join(" ")} ${pos} [${size}]${flags.length ? ` (${flags.join(", ")})` : ""}`);
+  }
+  return lines.join(`
+`);
+}
+var _helperPath = null;
+function getAccessibilityHelperPath() {
+  if (_helperPath)
+    return _helperPath;
+  const candidates = [
+    join6(dirname2(fileURLToPath2(import.meta.url)), "..", "..", "..", "helpers", "accessibility"),
+    join6(dirname2(fileURLToPath2(import.meta.url)), "..", "helpers", "accessibility"),
+    join6(process.env.HOME ?? "~", ".hasna", "computer", "helpers", "accessibility")
+  ];
+  for (const candidate of candidates) {
+    if (existsSync4(candidate)) {
+      _helperPath = candidate;
+      return candidate;
+    }
+  }
+  throw new Error("Accessibility helper not found. Run `swiftc helpers/accessibility.swift -o helpers/accessibility -framework AppKit` from the project root.");
+}
+// src/db/agents.ts
+function ensureAgentsTable() {
+  const d = getDb();
+  d.exec(`
+    CREATE TABLE IF NOT EXISTS agents (
+      id TEXT PRIMARY KEY,
+      name TEXT NOT NULL UNIQUE,
+      description TEXT,
+      capabilities TEXT,
+      focus TEXT,
+      last_heartbeat TEXT NOT NULL,
+      created_at TEXT NOT NULL DEFAULT (datetime('now'))
+    );
+  `);
+}
+function registerAgent(agent) {
+  ensureAgentsTable();
+  const d = getDb();
+  const now = new Date().toISOString();
+  const id = agent.name;
+  const existing = d.prepare("SELECT * FROM agents WHERE name = ?").get(agent.name);
+  if (existing) {
+    d.prepare(`
+      UPDATE agents SET description = ?, capabilities = ?, last_heartbeat = ?
+      WHERE name = ?
+    `).run(agent.description ?? existing.description, agent.capabilities ? JSON.stringify(agent.capabilities) : existing.capabilities, now, agent.name);
+  } else {
+    d.prepare(`
+      INSERT INTO agents (id, name, description, capabilities, last_heartbeat, created_at)
+      VALUES (?, ?, ?, ?, ?, ?)
+    `).run(id, agent.name, agent.description ?? null, agent.capabilities ? JSON.stringify(agent.capabilities) : null, now, now);
+  }
+  return getAgent(id);
+}
+function heartbeat(agentId) {
+  ensureAgentsTable();
+  const d = getDb();
+  const result = d.prepare("UPDATE agents SET last_heartbeat = ? WHERE id = ?").run(new Date().toISOString(), agentId);
+  return result.changes > 0;
+}
+function setFocus(agentId, focus) {
+  ensureAgentsTable();
+  const d = getDb();
+  const result = d.prepare("UPDATE agents SET focus = ? WHERE id = ?").run(focus, agentId);
+  return result.changes > 0;
+}
+function getAgent(id) {
+  ensureAgentsTable();
+  const d = getDb();
+  const row = d.prepare("SELECT * FROM agents WHERE id = ?").get(id);
+  if (!row)
+    return null;
+  return rowToAgent(row);
+}
+function listAgents() {
+  ensureAgentsTable();
+  const d = getDb();
+  const rows = d.prepare("SELECT * FROM agents ORDER BY last_heartbeat DESC").all();
+  return rows.map(rowToAgent);
+}
+function rowToAgent(row) {
+  return {
+    id: row.id,
+    name: row.name,
+    description: row.description,
+    capabilities: row.capabilities ? JSON.parse(row.capabilities) : undefined,
+    focus: row.focus,
+    last_heartbeat: row.last_heartbeat,
+    created_at: row.created_at
+  };
+}
 // src/lib/pricing.ts
 var MODEL_PRICING = {
   "claude-sonnet-4-5-20250514": { input: 3, output: 15 },
@@ -18997,22 +19264,88 @@ function findPricing(model) {
 function listPricing() {
   return { ...MODEL_PRICING };
 }
+// 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("") + `
+`;
+}
 export {
   updateSession,
+  supportsInlineImages,
+  summarizeAccessibilityTree,
   stepCost,
+  setFocus,
   setConfigValue,
   searchSessions,
   searchActionLogs,
   screenshotsMatch,
   scaleScreenshot,
+  saveToRecordings,
   saveScreenshotToFile,
   saveConfig,
   runTask,
+  runPostSessionIntegrations,
   resetRateLimiter,
+  renderInlineImage,
+  registerWithSessions,
+  registerAgent,
+  queryAccessibilityTree,
+  pushToLogs,
   logAction,
   loadConfig,
   listSessions,
   listPricing,
+  listAgents,
+  heartbeat,
   getStats,
   getSession,
   getScreenSize,
@@ -19020,9 +19353,11 @@ export {
   getDb,
   getConfigValue,
   getConfigPath,
+  getAgent,
   getActionLogs,
   formatCost,
   executeAction,
+  detectProtocol,
   deleteSession,
   createSession,
   createProvider,