talon-agent 3.32.1 → 3.33.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "talon-agent",
-  "version": "3.32.1",
+  "version": "3.33.1",
   "description": "Multi-frontend AI agent with full tool access, streaming, cron jobs, and plugin system",
   "author": "Dylan Neve",
   "license": "MIT",

package/src/backend/claude-sdk/constants.ts

@@ -6,14 +6,9 @@
  * additions live next to the rest of the chat-handler config).
  */
 
-import {
-  ALLOWED_TOOLS_CORE,
-  ALLOWED_TOOLS_BACKGROUND,
-} from "../../core/constants.js";
+import { ALLOWED_TOOLS_CORE } from "../../core/constants.js";
 
 // Re-export so existing backend imports keep working
-export { ALLOWED_TOOLS_CORE, ALLOWED_TOOLS_BACKGROUND };
-
 /**
  * Whitelist of built-in tools available to the main chat handler.
  *

package/src/backend/claude-sdk/index.ts

@@ -8,14 +8,5 @@
 export { initAgent, updateSystemPrompt } from "./state.js";
 export { warmSession } from "./warm.js";
 export { getActiveQuery } from "./handler.js";
-export {
-  buildMcpServers,
-  buildPluginMcpServers,
-  getActiveFrontends,
-} from "./options.js";
-export { getBridgePort } from "./state.js";
-export {
-  runOneShotAgent,
-  evictOrphanSubprocesses,
-  initClaudeOneShot,
-} from "./one-shot.js";
+export { buildMcpServers, buildPluginMcpServers } from "./options.js";
+export { runOneShotAgent, evictOrphanSubprocesses } from "./one-shot.js";

package/src/backend/claude-sdk/models/parsing.ts

@@ -167,7 +167,7 @@ export function describeSdkModel(model: SdkModelInfo): ParsedModelIdentity {
   };
 }
 
-export function buildFamilyKey(identity: ParsedModelIdentity): string | null {
+function buildFamilyKey(identity: ParsedModelIdentity): string | null {
   return identity.family
     ? `${identity.family}:${identity.version ?? "*"}`
     : null;
@@ -178,7 +178,7 @@ export function buildFamilyKey(identity: ParsedModelIdentity): string | null {
  * context size. Base and 1M variants of one family+version land in separate
  * buckets so both surface in the picker, while redundant longhand ids collapse.
  */
-export function buildVariantKey(identity: ParsedModelIdentity): string | null {
+function buildVariantKey(identity: ParsedModelIdentity): string | null {
   const familyKey = buildFamilyKey(identity);
   if (!familyKey) return null;
   return `${familyKey}:${identity.isOneMillion ? "1m" : "base"}`;

package/src/backend/claude-sdk/plan-usage.ts

@@ -163,8 +163,3 @@ export async function getPlanUsage(): Promise<PlanUsage | undefined> {
 export function invalidatePlanUsage(): void {
   if (cache) cache.fetchedAt = 0;
 }
-
-export function resetPlanUsageForTest(): void {
-  cache = undefined;
-  inFlight = undefined;
-}

package/src/backend/claude-sdk/stream.ts

@@ -214,7 +214,7 @@ export function processStreamDelta(
 }
 
 /** A tool call extracted from an assistant message. */
-export type ToolCall = {
+type ToolCall = {
   /** The SDK's tool_use block id — matches the later tool_result. */
   id: string;
   name: string;

package/src/backend/codex/auth.ts

@@ -45,7 +45,7 @@ import { existsSync, readFileSync } from "node:fs";
 import { join } from "node:path";
 
 /** Detected auth mode. */
-export type CodexAuthMode = "api-key" | "chatgpt" | "none";
+type CodexAuthMode = "api-key" | "chatgpt" | "none";
 
 export type CodexApiKeySource =
   | "env:CODEX_API_KEY"
@@ -80,7 +80,7 @@ export interface DetectCodexAuthInput {
   env?: NodeJS.ProcessEnv;
 }
 
-export function normalizeCodexBaseUrl(
+function normalizeCodexBaseUrl(
   baseUrl: string | undefined,
 ): string | undefined {
   const trimmed = baseUrl?.trim();

package/src/backend/codex/constants.ts

@@ -46,27 +46,6 @@ export const CODEX_DEFAULT_MODEL = "gpt-5-codex";
  */
 export const CODEX_CHATGPT_DEFAULT_MODEL = "gpt-5.5";
 
-/**
- * Set of Codex models that require an API key (i.e. won't work under
- * ChatGPT OAuth). Used by the handler's recovery ladder to detect a
- * model-not-supported error and automatically fall back to a
- * chatgpt-compatible model on retry.
- *
- * Keep this list in sync with `CODEX_MODELS` in `models.ts` — every
- * entry there with `apiKeyOnly: true` should appear here.
- */
-export const CODEX_API_KEY_ONLY_MODELS: ReadonlySet<string> = new Set([
-  "gpt-5-codex",
-]);
-
-/**
- * Default working directory for thread runs. Codex enforces a git-repo
- * check by default; we set `skipGitRepoCheck: true` and use a known
- * directory under the user's Talon workspace.
- */
-export const CODEX_DEFAULT_WORKING_DIRECTORY =
-  process.env.HOME ?? process.cwd();
-
 /**
  * ThreadOptions permission settings shared by both the chat handler and
  * the heartbeat/dream one-shot path.

package/src/backend/codex/handler/index.ts

@@ -12,4 +12,3 @@
 
 export { handleMessage } from "./message.js";
 export { getActiveAbort } from "./state.js";
-export { CodexUsageExhaustedError } from "./usage.js";

package/src/backend/codex/mcp-config.ts

@@ -49,7 +49,7 @@ import { frontendsForChat } from "../shared/frontends.js";
  *                 short-circuits to `true` for this server. Equivalent
  *                 to saying "Talon trusts this server, don't ask."
  */
-export type CodexToolApprovalMode = "auto" | "prompt" | "approve";
+type CodexToolApprovalMode = "auto" | "prompt" | "approve";
 
 /** TOML-compatible record shape Codex's CLI accepts (HTTP transport). */
 export interface CodexMcpServer {
@@ -100,7 +100,7 @@ export function asCodexConfig(
  * reasoning. Exposed as a module-level constant rather than a
  * literal-everywhere so behaviour changes happen in exactly one place.
  */
-export const TALON_MCP_DEFAULT_APPROVAL: CodexToolApprovalMode = "approve";
+const TALON_MCP_DEFAULT_APPROVAL: CodexToolApprovalMode = "approve";
 
 /**
  * Build the Codex `mcp_servers` config map for a given chat.

package/src/backend/codex/plan-usage.ts

@@ -161,8 +161,3 @@ export async function getPlanUsage(): Promise<PlanUsage | undefined> {
   if (loaded) cache = { value: loaded, fetchedAt: loaded.fetchedAt };
   return loaded ?? cache?.value;
 }
-
-export function resetCodexPlanUsageForTest(): void {
-  cache = undefined;
-  inFlight = undefined;
-}

package/src/backend/codex/state.ts

@@ -20,7 +20,7 @@ import type { FrontendName } from "../../core/agent-runtime/backend-registry.js"
  * window, description); the api-key path (`/v1/models`) is sparse and
  * leaves these fields undefined.
  */
-export interface DiscoveredModelMetadata {
+interface DiscoveredModelMetadata {
   displayName?: string;
   contextWindow?: number;
   description?: string;

package/src/backend/codex/token-usage.ts

@@ -47,7 +47,7 @@ export interface CodexLastTokenUsage {
  * want per-turn usage must snapshot before the turn and diff after
  * (see the terminator-abort fallback in `handler.ts`).
  */
-export interface CodexTokenTotals {
+interface CodexTokenTotals {
   inputTokens: number;
   cachedInputTokens: number;
   outputTokens: number;

package/src/backend/kilo/factory.ts

@@ -3,131 +3,51 @@
  *
  * Side-effect import (`import "./factory.js"` from `bootstrap.ts`) calls
  * `registerBackend(...)` at module load, making Kilo available under
- * `config.backend === "kilo"`.
- *
- * Returns a composed `Backend` with capability slots for chat,
- * background, models, and usage.
+ * `config.backend === "kilo"`. The capability wiring is the shared
+ * remote-server composition; only the bound functions are Kilo's.
  */
 
 import { registerBackend } from "../../core/agent-runtime/backend-registry.js";
-import type { BackendFactory } from "../../core/agent-runtime/backend-registry.js";
-import { log } from "../../util/log.js";
-import { handlerToEvents } from "../shared/handler-to-events.js";
-import { interruptChatTurn } from "../shared/turn-interrupt.js";
-import {
-  composeBackend,
-  type ChatBackend,
-  type BackgroundRunner,
-  type ModelCatalog,
-  type SessionBackend,
-  type SystemControl,
-  type ToolRuntime,
-  type UsageTelemetry,
-} from "../../core/agent-runtime/capabilities.js";
-
+import { createRemoteBackendFactory } from "../remote-server/factory.js";
 import {
   initKiloAgent,
   stopKiloServer,
-  handleMessage as kiloHandleMessage,
-  runOneShotAgent as kiloRunOneShotAgent,
+  handleMessage,
+  runOneShotAgent,
   getKiloSessionSnapshot,
-  resolveModel as kiloResolveModel,
-  getModelInfo as kiloGetModelInfo,
-  getSettingsPresentation as kiloGetSettingsPresentation,
-  getProviders as kiloGetProviders,
-  getProviderModels as kiloGetProviderModels,
-  formatModelError as kiloFormatModelError,
-  listModels as kiloListModels,
-  refreshPluginMcpServers as kiloRefreshPluginMcpServers,
-  updateSystemPrompt as kiloUpdateSystemPrompt,
-  warmSession as kiloWarmSession,
+  resolveModel,
+  getModelInfo,
+  getSettingsPresentation,
+  getProviders,
+  getProviderModels,
+  formatModelError,
+  listModels,
+  refreshPluginMcpServers,
+  updateSystemPrompt,
+  warmSession,
 } from "./index.js";
 
-const kiloFactory: BackendFactory = {
-  id: "kilo",
-  label: "Kilo",
-
-  async init(config, ctx) {
-    initKiloAgent(config, ctx.getBridgePort, ctx.frontendName);
-    log("bot", "Backend: Kilo (@kilocode/sdk)");
-
-    const chat: ChatBackend = {
-      runChatTurn: (params) =>
-        handlerToEvents((p) => kiloHandleMessage(p), params),
-      interruptChatTurn: (chatId) => interruptChatTurn(chatId),
-    };
-
-    const background: BackgroundRunner = {
-      runOneShotAgent: (p) => kiloRunOneShotAgent(p),
-      // Kilo runs a long-lived HTTP server — no per-query subprocesses.
-      // `evictOrphanSubprocesses` is intentionally not implemented.
-    };
-
-    const models: ModelCatalog = {
-      resolveModelInfo: (q) => kiloResolveModel(q),
-      // Catalog-driven backend with no canonical default — fall
-      // through to `config.backendDefaults.kilo`.
-      getDefaultModelId: () => undefined,
-      getRawModelInfo: (id) => kiloGetModelInfo(id),
-      getSettingsPresentation: (m, options) =>
-        kiloGetSettingsPresentation(m, options),
-      getProviders: () => kiloGetProviders(),
-      getProviderModels: (p, pg, ps) => kiloGetProviderModels(p, pg, ps),
-      formatModelError: (q, r) => kiloFormatModelError(q, r),
-      listModels: (f) => kiloListModels(f),
-    };
-
-    const usage: UsageTelemetry = {
-      getSessionSnapshot: async (sessionId) => {
-        const snap = await getKiloSessionSnapshot(sessionId);
-        if (!snap) return undefined;
-        return {
-          inputTokens: snap.usage?.totalInputTokens,
-          outputTokens: snap.usage?.totalOutputTokens,
-          cacheRead: snap.usage?.totalCacheRead,
-          cacheWrite: snap.usage?.totalCacheWrite,
-          contextModelId: snap.assistant?.modelID,
-        };
-      },
-    };
-
-    const tools: ToolRuntime = {
-      refreshTools: (chatId) => kiloRefreshPluginMcpServers(chatId),
-    };
-
-    // Session state lives on the Kilo server, and `/reset` already clears
-    // Talon's stored id centrally (storage/sessions.ts), so the next turn
-    // creates a fresh one — no `resetChat` needed. `warmSession` front-loads
-    // that creation plus the plugin-MCP sweep, matching what the Claude
-    // backend does after a reset.
-    const sessions: SessionBackend = {
-      warmSession: (chatId) => kiloWarmSession(chatId),
-    };
-
-    const control: SystemControl = {
-      updateSystemPrompt: (prompt) => kiloUpdateSystemPrompt(prompt),
-    };
-
-    const backend = composeBackend({
-      id: "kilo",
-      label: "Kilo",
-      cacheMetrics: "readwrite",
-      chat,
-      background,
-      models,
-      sessions,
-      tools,
-      usage,
-      control,
-    });
-
-    return {
-      backend,
-      cleanup: () => {
-        stopKiloServer();
-      },
-    };
-  },
-};
-
-registerBackend(kiloFactory);
+registerBackend(
+  createRemoteBackendFactory({
+    id: "kilo",
+    label: "Kilo",
+    sdkPackage: "@kilocode/sdk",
+    init: initKiloAgent,
+    stop: stopKiloServer,
+    handleMessage,
+    runOneShotAgent,
+    getSessionSnapshot: getKiloSessionSnapshot,
+    models: {
+      resolveModel,
+      getModelInfo,
+      getSettingsPresentation,
+      getProviders,
+      getProviderModels,
+      formatModelError,
+      listModels,
+    },
+    refreshPluginMcpServers,
+    warmSession,
+    updateSystemPrompt,
+  }),
+);

package/src/backend/kilo/handler/index.ts

@@ -1,12 +1,2 @@
-/**
- * Kilo message handler.
- *
- * Split by responsibility:
- *   - `state`   — in-flight session registry (activeSessions)
- *   - `turn`    — runKiloTurn + SSE subscription + last-assistant lookup
- *   - `message` — the handleMessage entry point + post-loop accounting
- *
- * Re-exports the same public surface the old single-file module exposed.
- */
-
+/** Kilo message handler barrel. */
 export { handleMessage } from "./message.js";