talon-agent 1.12.0 → 1.34.0

package/src/frontend/terminal/commands.ts

@@ -7,11 +7,28 @@
 
 import pc from "picocolors";
 import type { TalonConfig } from "../../util/config.js";
-import type { QueryBackend } from "../../core/types.js";
+import type { Backend } from "../../core/agent-runtime/capabilities.js";
 import type { Renderer } from "./renderer.js";
 import { formatTimeAgo } from "./renderer.js";
 import { isTerminalChatId } from "../../util/chat-id.js";
-import { resolveModel as coreResolveModel } from "../../core/models.js";
+import {
+  resolveModel as coreResolveModel,
+  getModels,
+} from "../../core/models/catalog.js";
+import { buildCacheDisplay } from "../shared/status-context.js";
+import {
+  getChatSettings,
+  setChatModel,
+  setChatEffort,
+  resolveModelName,
+} from "../../storage/chat-settings.js";
+import {
+  getAllSessions,
+  getSession,
+  getSessionInfo,
+  setSessionName,
+} from "../../storage/sessions.js";
+import { getLoadedPlugins } from "../../core/plugin/index.js";
 
 // ── Types ────────────────────────────────────────────────────────────────────
 
@@ -26,7 +43,7 @@ export type CommandContext = {
   /** Close the terminal (for /quit). */
   close: () => void;
   /** AI backend (available after bootstrap). */
-  backend?: QueryBackend | null;
+  backend?: Backend | null;
 };
 
 export type CommandHandler = (
@@ -96,8 +113,6 @@ export function registerBuiltinCommands(): void {
     argHint: "[name]",
     description: "Switch model",
     async handler(args, ctx) {
-      const { getChatSettings, setChatModel, resolveModelName } =
-        await import("../../storage/chat-settings.js");
       const currentModel =
         getChatSettings(ctx.chatId()).model ?? ctx.config.model;
       const be = ctx.backend;
@@ -106,7 +121,7 @@ export function registerBuiltinCommands(): void {
       const lowerArgs = trimmedArgs.toLowerCase();
 
       if (!trimmedArgs) {
-        const modelInfo = await be?.getModelInfo?.(currentModel);
+        const modelInfo = await be?.models?.getRawModelInfo?.(currentModel);
         const displayName = modelInfo?.displayName ?? currentModel;
         const details = modelInfo
           ? [
@@ -125,8 +140,8 @@ export function registerBuiltinCommands(): void {
             `  Context window: ${modelInfo.contextWindow.toLocaleString()}`,
           );
         }
-        if (be?.getProviders) {
-          const providers = await be.getProviders();
+        if (be?.models?.getProviders) {
+          const providers = await be.models?.getProviders();
           const connected = providers.filter((p) => p.connected);
           if (connected.length > 0) {
             ctx.renderer.writeln(
@@ -149,9 +164,9 @@ export function registerBuiltinCommands(): void {
       }
 
       if (lowerArgs === "free" || lowerArgs === "list" || lowerArgs === "all") {
-        if (be?.listModels) {
+        if (be?.models?.listModels) {
           const filter = lowerArgs === "free" ? "free" : "all";
-          const { models, total } = await be.listModels(filter);
+          const { models, total } = await be.models.listModels(filter);
           const list = models.slice(0, 20);
           ctx.renderer.writeSystem(
             `${filter === "free" ? "Free" : "Available"} models (${total})`,
@@ -165,7 +180,6 @@ export function registerBuiltinCommands(): void {
             ctx.renderer.writeln(`  …and ${total - list.length} more`);
           }
         } else {
-          const { getModels } = await import("../../core/models.js");
           const names = getModels()
             .map((m) => m.aliases[0] ?? m.id)
             .join(", ");
@@ -176,8 +190,8 @@ export function registerBuiltinCommands(): void {
       }
 
       if (lowerArgs === "providers") {
-        if (be?.getProviders) {
-          const providers = await be.getProviders();
+        if (be?.models?.getProviders) {
+          const providers = await be.models?.getProviders();
           ctx.renderer.writeSystem(`Providers (${providers.length})`);
           for (const p of providers.slice(0, 20)) {
             ctx.renderer.writeln(
@@ -192,11 +206,11 @@ export function registerBuiltinCommands(): void {
       }
 
       // Resolve model query via backend
-      if (be?.resolveModel) {
-        const resolution = await be.resolveModel(trimmedArgs);
+      if (be?.models?.resolveModelInfo) {
+        const resolution = await be.models?.resolveModelInfo(trimmedArgs);
         if (resolution.kind === "missing") {
           const msg =
-            be.formatModelError?.(trimmedArgs, resolution) ??
+            be.models?.formatModelError?.(trimmedArgs, resolution) ??
             `No model matched "${trimmedArgs}".`;
           ctx.renderer.writeError(msg);
           ctx.reprompt();
@@ -237,8 +251,6 @@ export function registerBuiltinCommands(): void {
     argHint: "[lvl]",
     description: "Thinking effort (off/low/medium/high/max)",
     async handler(args, ctx) {
-      const { getChatSettings, setChatEffort } =
-        await import("../../storage/chat-settings.js");
       if (!args) {
         ctx.renderer.writeSystem(
           `Effort: ${getChatSettings(ctx.chatId()).effort ?? "adaptive"}`,
@@ -260,10 +272,6 @@ export function registerBuiltinCommands(): void {
     name: "status",
     description: "Session stats",
     async handler(_args, ctx) {
-      const { getSessionInfo } = await import("../../storage/sessions.js");
-      const { getChatSettings } =
-        await import("../../storage/chat-settings.js");
-      const { getLoadedPlugins } = await import("../../core/plugin.js");
       const info = getSessionInfo(ctx.chatId());
       const u = info.usage;
       const be = ctx.backend;
@@ -279,9 +287,9 @@ export function registerBuiltinCommands(): void {
       const nameStr = info.sessionName ? `"${info.sessionName}"  ·  ` : "";
 
       // Enrich from backend when available
-      if (be?.getSessionSnapshot && info.sessionId) {
-        const snap = await be
-          .getSessionSnapshot(info.sessionId)
+      if (be?.usage?.getSessionSnapshot && info.sessionId) {
+        const snap = await be.usage
+          ?.getSessionSnapshot(info.sessionId)
           .catch(() => undefined);
         if (snap) {
           displayInputTokens = snap.inputTokens ?? displayInputTokens;
@@ -291,28 +299,29 @@ export function registerBuiltinCommands(): void {
         }
       }
 
-      if (be?.getModelInfo) {
-        const modelInfo = await be
-          .getModelInfo(activeModel)
+      const cache = buildCacheDisplay({
+        cacheMetrics: be?.cacheMetrics,
+        inputTokens: displayInputTokens,
+        cacheRead: displayCacheRead,
+        cacheWrite: displayCacheWrite,
+      });
+      if (be?.models?.getRawModelInfo) {
+        const modelInfo = await be.models
+          ?.getRawModelInfo(activeModel)
           .catch(() => undefined);
-        const label = be.backendLabel ?? "Backend";
+        const label = be.label ?? "Backend";
         if (modelInfo) {
           backendModelLine = `  ${pc.bold(label)}  ${modelInfo.displayName}  ·  ${modelInfo.providerName}${modelInfo.free ? " · free" : ""}`;
           if (modelInfo.contextWindow) {
-            backendContextLine = `  ${pc.dim(`context window ${modelInfo.contextWindow.toLocaleString()}  ·  cache ${displayCacheRead.toLocaleString()} / ${displayCacheWrite.toLocaleString()}`)}`;
+            backendContextLine = cache
+              ? `  ${pc.dim(`context window ${modelInfo.contextWindow.toLocaleString()}  ·  cache ${cache.read.toLocaleString()} / ${cache.write.toLocaleString()}`)}`
+              : `  ${pc.dim(`context window ${modelInfo.contextWindow.toLocaleString()}`)}`;
           }
         }
       }
 
-      const cacheHit =
-        displayInputTokens + displayCacheRead > 0
-          ? Math.round(
-              (displayCacheRead / (displayInputTokens + displayCacheRead)) *
-                100,
-            )
-          : 0;
       ctx.renderer.writeln(
-        `  ${pc.bold("Session")}  ${nameStr}turns ${info.turns}  ·  ${cacheHit}% cache`,
+        `  ${pc.bold("Session")}  ${nameStr}turns ${info.turns}${cache ? `  ·  ${cache.hitPct}% cache` : ""}`,
       );
       ctx.renderer.writeln(
         `  ${pc.dim(`in ${displayInputTokens.toLocaleString()}  ·  out ${displayOutputTokens.toLocaleString()} tokens`)}`,
@@ -360,7 +369,6 @@ export function registerBuiltinCommands(): void {
     name: "resume",
     description: "List & resume a past session",
     async handler(_args, ctx) {
-      const { getAllSessions } = await import("../../storage/sessions.js");
       const sessions = getAllSessions()
         .filter(
           (s) =>
@@ -420,8 +428,6 @@ export function registerBuiltinCommands(): void {
     argHint: "[name]",
     description: "Name the current session",
     async handler(args, ctx) {
-      const { getSession, setSessionName } =
-        await import("../../storage/sessions.js");
       // Ensure session exists in store (auto-creates if needed)
       getSession(ctx.chatId());
       if (!args) {

package/src/frontend/terminal/index.ts

@@ -11,13 +11,14 @@
 import pc from "picocolors";
 import type { TalonConfig } from "../../util/config.js";
 import type { ContextManager, ActionResult } from "../../core/types.js";
-import type { Gateway } from "../../core/gateway.js";
+import type { Gateway } from "../../core/engine/gateway.js";
 import { log } from "../../util/log.js";
 import {
   deriveNumericChatId,
   generateTerminalChatId,
 } from "../../util/chat-id.js";
-import { resolveModel } from "../../core/models.js";
+import { resolveModel } from "../../core/models/catalog.js";
+import { toolInputToRecord } from "../../core/agent-runtime/events.js";
 import { createRenderer } from "./renderer.js";
 import { createInput } from "./input.js";
 import {
@@ -26,6 +27,7 @@ import {
   clearCommands,
   type CommandContext,
 } from "./commands.js";
+import { buildCacheDisplay } from "../shared/status-context.js";
 
 // ── State ────────────────────────────────────────────────────────────────────
 
@@ -118,7 +120,8 @@ export function createTerminalFrontend(
   let toolCallCount = 0;
 
   const context: ContextManager = {
-    acquire: () => gateway.setContext(terminalNumericId, terminalChatId),
+    acquire: () =>
+      gateway.setContext(terminalNumericId, terminalChatId, "terminal"),
     release: () => gateway.clearContext(terminalNumericId),
     getMessageCount: (chatId: number) => gateway.getMessageCount(chatId),
   };
@@ -138,7 +141,10 @@ export function createTerminalFrontend(
     getBridgePort: () => gateway.getPort(),
 
     async init() {
-      gateway.setFrontendHandler(createActionHandler(gateway, renderer));
+      gateway.registerFrontendHandler(
+        "terminal",
+        createActionHandler(gateway, renderer),
+      );
       const port = await gateway.start(19877);
       log("bot", `Terminal gateway on port ${port}`);
     },
@@ -155,7 +161,7 @@ export function createTerminalFrontend(
       );
       renderer.writeln(`  ${pc.dim("─".repeat(renderer.cols - 2))}`);
 
-      const { execute } = await import("../../core/dispatcher.js");
+      const { execute } = await import("../../core/engine/dispatcher.js");
       const { getSessionInfo } = await import("../../storage/sessions.js");
       const input = createInput(`  ${pc.green("❯")} `);
 
@@ -216,26 +222,36 @@ export function createTerminalFrontend(
             senderName: "User",
             isGroup: false,
             source: "message",
-            onStreamDelta: (_accumulated, phase) => {
-              if (phase === "thinking" && currentPhase !== "thinking") {
-                currentPhase = "thinking";
-                renderer.updateSpinnerLabel("thinking");
-              } else if (phase === "text" && currentPhase !== "text") {
-                currentPhase = "text";
-                renderer.updateSpinnerLabel("responding");
+            onEvent: async (event) => {
+              switch (event.type) {
+                case "text_delta":
+                  if (currentPhase !== "text") {
+                    currentPhase = "text";
+                    renderer.updateSpinnerLabel("responding");
+                  }
+                  break;
+                case "reasoning":
+                  if (currentPhase !== "thinking") {
+                    currentPhase = "thinking";
+                    renderer.updateSpinnerLabel("thinking");
+                  }
+                  break;
+                case "tool_call":
+                  renderer.stopSpinner();
+                  currentPhase = "tool";
+                  toolCallCount++;
+                  renderer.renderToolCall(
+                    event.name,
+                    toolInputToRecord(event.name, event.input),
+                  );
+                  renderer.startSpinner("running tools");
+                  break;
+                case "assistant_message":
+                  renderer.stopSpinner();
+                  renderer.renderAssistantMessage(event.text);
+                  break;
               }
             },
-            onToolUse: (toolName, toolInput) => {
-              renderer.stopSpinner();
-              currentPhase = "tool";
-              toolCallCount++;
-              renderer.renderToolCall(toolName, toolInput);
-              renderer.startSpinner("running tools");
-            },
-            onTextBlock: async (blockText) => {
-              renderer.stopSpinner();
-              renderer.renderAssistantMessage(blockText);
-            },
           });
 
           renderer.stopSpinner();
@@ -247,20 +263,19 @@ export function createTerminalFrontend(
 
           const info = getSessionInfo(terminalChatId);
           const u = info.usage;
-          const cacheHit =
-            u.totalInputTokens + u.totalCacheRead > 0
-              ? Math.round(
-                  (u.totalCacheRead / (u.totalInputTokens + u.totalCacheRead)) *
-                    100,
-                )
-              : 0;
+          const cache = buildCacheDisplay({
+            cacheMetrics: gateway.backend?.cacheMetrics,
+            inputTokens: u.totalInputTokens,
+            cacheRead: u.totalCacheRead,
+            cacheWrite: u.totalCacheWrite,
+          });
           renderer.renderStatusLine(result.durationMs, toolCallCount, {
             model: modelDisplay,
             sessionName: info.sessionName,
             turns: info.turns,
             inputTokens: u.totalInputTokens,
             outputTokens: u.totalOutputTokens,
-            cacheHitPct: cacheHit,
+            cacheHitPct: cache?.hitPct,
             costUsd: u.estimatedCostUsd,
           });
           reprompt(); // readline back on, show prompt

package/src/frontend/terminal/input.ts

@@ -289,6 +289,11 @@ export function createInput(promptStr: string): InputHandler {
       paused = false;
     },
     close() {
+      if (pendingResolve) {
+        const resolve = pendingResolve;
+        pendingResolve = null;
+        resolve("");
+      }
       process.stdout.write("\x1b[?2004l");
       if (process.stdin.isTTY) process.stdin.setRawMode(false);
       process.stdin.pause();

package/src/frontend/terminal/renderer.ts

@@ -16,7 +16,7 @@ export type StatusBarInfo = {
   turns: number;
   inputTokens: number;
   outputTokens: number;
-  cacheHitPct: number;
+  cacheHitPct?: number;
   costUsd: number;
 };
 
@@ -189,8 +189,10 @@ export function createRenderer(cols?: number, displayName = "Talon"): Renderer {
     p.push(
       `${info.turns} turn${info.turns !== 1 ? "s" : ""}`,
       `${fmtTok(info.inputTokens + info.outputTokens)} tok`,
-      `${info.cacheHitPct}% cache`,
     );
+    if (typeof info.cacheHitPct === "number") {
+      p.push(`${info.cacheHitPct}% cache`);
+    }
     if (tools > 0) p.push(`${tools} tool${tools > 1 ? "s" : ""}`);
     writeln();
     writeln(`  ${pc.dim(p.join("  ·  "))}`);

package/src/index.ts

@@ -1,178 +1,20 @@
 /**
- * Talon — agentic AI harness.
- * Composition root: loads config, creates frontend + backend, wires dispatcher.
+ * Talon entry shim.
  *
- * Frontends (Telegram, Terminal) and backends (Claude, OpenCode)
- * are loaded dynamically — only the selected platform's dependencies are required.
+ * Dispatches the hidden `_mcp-launch` (MCP supervisor) and `_lua-run`
+ * (WASM Lua trigger runner) subcommands BEFORE the app graph loads —
+ * both are Talon re-invoking itself (see util/mcp-launcher.ts). The
+ * dynamic import keeps these helper processes light: they evaluate this
+ * shim and their own module, never the backends/frontends/plugins.
  */
 
-import { getFrontends } from "./util/config.js";
-import { startUploadCleanup, stopUploadCleanup } from "./util/workspace.js";
-import { flushSessions } from "./storage/sessions.js";
-import { flushChatSettings } from "./storage/chat-settings.js";
-import { flushCronJobs } from "./storage/cron-store.js";
-import { flushTriggers } from "./storage/trigger-store.js";
-import { flushHistory } from "./storage/history.js";
-import { flushMediaIndex } from "./storage/media-index.js";
-import { getActiveCount } from "./core/dispatcher.js";
-import { startPulseTimer, stopPulseTimer } from "./core/pulse.js";
-import {
-  startHeartbeatTimer,
-  stopHeartbeatTimer,
-  awaitCurrentRun as awaitHeartbeat,
-} from "./core/heartbeat.js";
-import { startCronTimer, stopCronTimer } from "./core/cron.js";
-import { shutdownTriggers } from "./core/triggers.js";
-import { startWatchdog, stopWatchdog } from "./util/watchdog.js";
-import { log, logError, logWarn } from "./util/log.js";
-import { bootstrap, initBackendAndDispatcher } from "./bootstrap.js";
-import { Gateway } from "./core/gateway.js";
-import type { Frontend } from "./bootstrap.js";
+import { MCP_LAUNCH_SUBCOMMAND, runSupervisor } from "./util/mcp-launcher.js";
+import { LUA_RUN_SUBCOMMAND, runLuaMain } from "./core/scripting/lua-runner.js";
 
-// ── Bootstrap ────────────────────────────────────────────────────────────────
-
-import { writeFileSync, unlinkSync } from "node:fs";
-import { files as pathFiles } from "./util/paths.js";
-
-const { config } = await bootstrap();
-
-// Write PID file for daemon management
-try {
-  writeFileSync(pathFiles.pid, String(process.pid));
-} catch {
-  /* ok */
-}
-
-// ── Create gateway + frontend ─────────────────────────────────────────────────
-
-const gateway = new Gateway();
-
-const selectedFrontend = getFrontends(config)[0]; // use first configured frontend
-let frontend: Frontend;
-
-if (selectedFrontend === "terminal") {
-  const { createTerminalFrontend } =
-    await import("./frontend/terminal/index.js");
-  frontend = createTerminalFrontend(config, gateway);
-  log("bot", "Frontend: Terminal");
-} else if (selectedFrontend === "teams") {
-  const { createTeamsFrontend } = await import("./frontend/teams/index.js");
-  frontend = createTeamsFrontend(config, gateway);
-  log("bot", "Frontend: Teams");
-} else if (selectedFrontend === "discord") {
-  const { createDiscordFrontend } = await import("./frontend/discord/index.js");
-  frontend = createDiscordFrontend(config, gateway);
-  log("bot", "Frontend: Discord");
+if (process.argv[2] === MCP_LAUNCH_SUBCOMMAND) {
+  await runSupervisor(process.argv.slice(3));
+} else if (process.argv[2] === LUA_RUN_SUBCOMMAND) {
+  await runLuaMain(process.argv.slice(3));
 } else {
-  const { createTelegramFrontend } =
-    await import("./frontend/telegram/index.js");
-  frontend = createTelegramFrontend(config, gateway);
-  log("bot", "Frontend: Telegram");
+  await import("./app.js");
 }
-
-// ── Create backend + wire dispatcher ─────────────────────────────────────────
-
-const { backend } = await initBackendAndDispatcher(config, frontend);
-gateway.backend = backend;
-
-// ── Graceful shutdown ────────────────────────────────────────────────────────
-
-let shuttingDown = false;
-
-const SHUTDOWN_TIMEOUT_MS = 15_000;
-
-async function gracefulShutdown(signal: string): Promise<void> {
-  if (shuttingDown) return;
-  shuttingDown = true;
-  log("shutdown", `${signal} received, shutting down gracefully...`);
-
-  const forceTimer = setTimeout(() => {
-    logError("shutdown", "Timeout exceeded, forcing exit");
-    process.exit(1);
-  }, SHUTDOWN_TIMEOUT_MS);
-  forceTimer.unref();
-
-  const pending = getActiveCount();
-  if (pending > 0) {
-    log("shutdown", `Waiting for ${pending} in-flight queries to drain...`);
-    await new Promise((r) => setTimeout(r, 5000));
-  }
-
-  await frontend.stop();
-  if (config.backend === "opencode") {
-    const { stopOpenCodeServer } = await import("./backend/opencode/index.js");
-    stopOpenCodeServer();
-  }
-  // Destroy plugins (cleanup resources)
-  if (config.plugins.length > 0) {
-    const { destroyPlugins } = await import("./core/plugin.js");
-    await destroyPlugins();
-  }
-  stopPulseTimer();
-  stopHeartbeatTimer();
-  await awaitHeartbeat();
-  stopCronTimer();
-  await shutdownTriggers();
-  stopWatchdog();
-  stopUploadCleanup();
-  flushSessions();
-  flushChatSettings();
-  flushCronJobs();
-  flushTriggers();
-  flushHistory();
-  flushMediaIndex();
-  try {
-    unlinkSync(pathFiles.pid);
-  } catch {
-    /* ok */
-  }
-  log("shutdown", "State saved");
-  process.exit(0);
-}
-
-process.on("SIGTERM", () => gracefulShutdown("SIGTERM"));
-process.on("SIGINT", () => gracefulShutdown("SIGINT"));
-
-process.on("uncaughtException", (err) => {
-  // EPIPE errors from network sockets (e.g. Telegram MTProto) are transient —
-  // gramjs will reconnect; crashing the process here is wrong.
-  if ((err as NodeJS.ErrnoException).code === "EPIPE") {
-    logWarn("bot", `Suppressed transient EPIPE error: ${err.message}`);
-    return;
-  }
-  logError("bot", "Uncaught exception", err);
-  flushSessions();
-  flushChatSettings();
-  flushCronJobs();
-  flushTriggers();
-  flushHistory();
-  flushMediaIndex();
-  process.exit(1);
-});
-
-process.on("unhandledRejection", (reason) => {
-  logWarn(
-    "bot",
-    `Unhandled rejection: ${reason instanceof Error ? reason.message : reason}`,
-  );
-});
-
-// ── Start ────────────────────────────────────────────────────────────────────
-
-async function main(): Promise<void> {
-  await frontend.init();
-  log("bot", "Starting Talon...");
-
-  if (config.pulse) startPulseTimer(config.pulseIntervalMs);
-  if (config.heartbeat) startHeartbeatTimer(config.heartbeatIntervalMinutes);
-  startCronTimer();
-  startWatchdog(config.workspace);
-  startUploadCleanup(config.workspace);
-
-  await frontend.start();
-}
-
-main().catch((err) => {
-  logError("bot", "Fatal startup error", err);
-  process.exit(1);
-});

package/src/login.ts

@@ -10,9 +10,10 @@
 
 import { TelegramClient } from "telegram";
 import { StringSession } from "telegram/sessions/index.js";
-import { existsSync, readFileSync, writeFileSync, mkdirSync } from "node:fs";
+import { existsSync, readFileSync, mkdirSync } from "node:fs";
 import { resolve, dirname } from "node:path";
 import { createInterface } from "node:readline";
+import writeFileAtomic from "write-file-atomic";
 import { files } from "./util/paths.js";
 
 // Load .env
@@ -76,7 +77,7 @@ async function main() {
   const newSession = client.session.save() as unknown as string;
   const dir = dirname(SESSION_FILE);
   if (!existsSync(dir)) mkdirSync(dir, { recursive: true });
-  writeFileSync(SESSION_FILE, newSession);
+  writeFileAtomic.sync(SESSION_FILE, newSession);
   console.log(`Session saved to ${SESSION_FILE}`);
 
   await client.disconnect();