talon-agent 5.1.0 → 5.2.0

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

@@ -1,55 +0,0 @@
-/**
- * Kilo model catalog — binder over the shared remote-server model-catalog
- * module (see `backend/remote-server/model-catalog/`).
- *
- * Kilo's provider-bucket API is forked from OpenCode's, so the catalog,
- * resolution, and presentation logic is the shared implementation; this file
- * only binds the Kilo SDK client + presentation knobs and re-exports the
- * result under the historical `OpenCode*`-prefixed names (retained on
- * purpose — the names match the wire shape the upstream actually emits).
- *
- * Knobs: Kilo's model picker renders through Discord StringSelectMenus,
- * which allow 25 options and values up to 100 chars with any characters —
- * Kilo ids routinely contain "/" and ":" (e.g. "inclusionai/ling-2.6-1t:free").
- */
-
-import { ensureServer, onServerStop } from "../server.js";
-import { createRemoteModelCatalogModule } from "../../remote-server/model-catalog/index.js";
-
-export type {
-  RemoteModelCatalogEntry as OpenCodeModelCatalogEntry,
-  RemoteModelCatalog as OpenCodeModelCatalog,
-  RemoteModelResolution as OpenCodeModelResolution,
-  ModelButton,
-} from "../../remote-server/model-catalog/index.js";
-export {
-  sortCatalogModels,
-  getRemoteModelSelectionValue as getOpenCodeModelSelectionValue,
-  resolveRemoteModelInput as resolveOpenCodeModelInput,
-  guessProviderID,
-  getBucketPriority,
-  normalizeModelLookup,
-  parseRemoteModelQuery as parseOpenCodeModelQuery,
-  formatRemoteUnavailableModel as formatOpenCodeUnavailableModel,
-} from "../../remote-server/model-catalog/index.js";
-
-const kiloModels = createRemoteModelCatalogModule({
-  label: "Kilo",
-  getClient: () => ensureServer(),
-  maxCallbackIdLength: 90,
-  allowCallbackSeparators: true,
-  quickPickLimit: 24,
-});
-
-// A stopped server invalidates the catalog it served.
-onServerStop(kiloModels.clearCache);
-
-export const getOpenCodeModelCatalog = kiloModels.getCatalog;
-export const clearModelCatalogCache = kiloModels.clearCache;
-export const getOpenCodeModelInfo = kiloModels.getModelInfo;
-export const getOpenCodeQuickPickModels = kiloModels.getQuickPickModels;
-export const getOpenCodeSettingsPresentation =
-  kiloModels.getSettingsPresentation;
-export const renderOpenCodeModelSummary = kiloModels.renderModelSummary;
-export const renderOpenCodeModelList = kiloModels.renderModelList;
-export const formatOpenCodeSelectionError = kiloModels.formatSelectionError;

package/src/backend/kilo/one-shot.ts

@@ -1,42 +0,0 @@
-/**
- * Kilo one-shot agent runner — used by heartbeat & dream.
- *
- * Thin binding over the shared remote-server runner (see
- * `backend/remote-server/one-shot.ts` for the design notes): Kilo
- * supplies its server bootstrap, model-selection parser, and delivery
- * suffix; everything else is the shared lifecycle.
- */
-
-import type { OneShotAgentParams, OneShotUsage } from "../../core/types.js";
-import { runRemoteOneShotAgent } from "../remote-server/one-shot.js";
-import {
-  ensureServer,
-  ensureChatMcpServer,
-  ensurePluginMcpServers,
-  buildToolOverrides,
-  disconnectChatMcpServer,
-  resolveProviderID,
-  parseStoredKiloModelSelection,
-  KILO_SYSTEM_PROMPT_SUFFIX,
-  errMsg,
-} from "./server.js";
-
-export function runOneShotAgent(
-  params: OneShotAgentParams,
-): Promise<OneShotUsage | void> {
-  return runRemoteOneShotAgent(
-    {
-      label: "Kilo",
-      systemPromptSuffix: KILO_SYSTEM_PROMPT_SUFFIX,
-      ensureServer,
-      parseModelSelection: parseStoredKiloModelSelection,
-      resolveProviderID,
-      ensureChatMcpServer,
-      ensurePluginMcpServers,
-      buildToolOverrides,
-      disconnectChatMcpServer,
-      errMsg,
-    },
-    params,
-  );
-}

package/src/backend/kilo/server.ts

@@ -1,98 +0,0 @@
-/**
- * Kilo server lifecycle — the Kilo profile bound through
- * `backend/remote-server/server-bindings.ts`.
- *
- * Kilo is a fork of OpenCode with the same HTTP API, so the server spawn,
- * MCP registration, session creation, and provider resolution are the
- * shared family implementation. This module holds only what is Kilo's:
- * the SDK constructors, port 4097, the text-or-tools delivery contract,
- * and the `kilo/`-prefix model parser. The exports keep their historical
- * names — the models module, one-shot runner, tests, and
- * `vi.mock("../backend/kilo/server.js")` all address them.
- */
-
-import {
-  createKiloClient,
-  createKiloServer,
-  type KiloClient,
-} from "@kilocode/sdk/v2";
-import {
-  bindRemoteServer,
-  type RemoteModelSelection,
-} from "../remote-server/server-bindings.js";
-
-/**
- * Parse the stored model-selection string into a `{providerID?, modelID}`
- * pair.
- *
- * Kilo model ids frequently contain `/` and `:` inside the model.id itself
- * (e.g. `inclusionai/ling-2.6-1t:free`, `deepseek/deepseek-v4-flash:free`).
- * A naive `provider/model` splitter mis-treats those vendor prefixes as
- * the provider, so we generally return the whole string as the model id
- * and let `resolveProviderID` look up the real provider from the live
- * catalog.
- *
- * Exception: if the value starts with the literal `kilo/` prefix
- * (Talon's old hint that "this is a kilo-routed model"), strip it AND
- * pin providerID to `"kilo"`. Otherwise the upstream Kilo router gets
- * `kilo/deepseek/deepseek-v4-flash:free` as the model id and concats
- * its own provider in front, producing
- * `Model not found: opencode/kilo/deepseek/deepseek-v4-flash:free`.
- */
-export function parseStoredKiloModelSelection(
-  value: string,
-): RemoteModelSelection {
-  const trimmed = value.trim();
-  if (trimmed.startsWith("kilo/")) {
-    return {
-      providerID: "kilo",
-      modelID: trimmed.slice("kilo/".length),
-    };
-  }
-  return {
-    providerID: undefined,
-    modelID: trimmed,
-  };
-}
-
-/**
- * Kilo delivery model: the reply reaches the user either as a `text` part
- * (what most Kilo-routed models emit by default — DeepSeek, GLM,
- * openrouter routes) or through a delivery tool (`end_turn` / `send` /
- * `react`) when reply-to targeting, buttons, photos, or polls are needed.
- * Both routes work; the shared text-or-tools contract documents the
- * choice.
- */
-const kilo = bindRemoteServer<KiloClient>({
-  label: "Kilo",
-  defaultPort: 4097,
-  portEnv: "KILO_PORT",
-  deliveryContract: "text-or-tools",
-  createClient: (baseUrl) => createKiloClient({ baseUrl, throwOnError: true }),
-  createServer: ({ hostname, port, timeout }) =>
-    createKiloServer({ hostname, port, timeout }),
-  parseModelSelection: parseStoredKiloModelSelection,
-});
-
-export const KILO_BASE_URL = kilo.baseUrl;
-export const kiloSystemPromptSuffix = kilo.systemPromptSuffix;
-export const KILO_SYSTEM_PROMPT_SUFFIX = kilo.defaultSystemPromptSuffix;
-export const initKiloAgent = kilo.init;
-export const stopKiloServer = kilo.stop;
-export const {
-  onServerStop,
-  ensureServer,
-  trackActiveTurn,
-  ensureChatMcpServer,
-  ensurePluginMcpServers,
-  buildToolOverrides,
-  disconnectChatMcpServer,
-  refreshPluginMcpServers,
-  updateSystemPrompt,
-  ensureSession,
-  warmSession,
-  resolveProviderID,
-  getConfig,
-  getRegisteredMcpServerNames,
-  errMsg,
-} = kilo;

package/src/backend/kilo/sessions.ts

@@ -1,37 +0,0 @@
-/**
- * Kilo session helpers — the shared `remote-server/session-helpers.ts`
- * bound to the Kilo client, re-exported under Kilo-named symbols for the
- * handler / index / test imports.
- */
-
-import type { KiloClient } from "@kilocode/sdk/v2";
-import {
-  extractPartsSummary as extractPartsSummaryShared,
-  extractAssistantUsage as extractAssistantUsageShared,
-  summarizeAssistantMessages,
-  getSessionSnapshot,
-  type RemoteSessionSnapshot,
-  type RemoteSessionClient,
-} from "../remote-server/session-helpers.js";
-import { ensureServer } from "./server.js";
-
-/** Snapshot of a Kilo session's lifetime + last-turn assistant info. */
-export type KiloSessionSnapshot = RemoteSessionSnapshot;
-
-export const extractPartsSummary = extractPartsSummaryShared;
-export const extractAssistantUsage = extractAssistantUsageShared;
-
-/** Summarise a batch of session messages into per-turn usage totals. */
-export const summarizeKiloAssistantMessages = summarizeAssistantMessages;
-
-/**
- * Build a {@link KiloSessionSnapshot} for the given session id.
- * Returns `undefined` if no session id was provided.
- */
-export async function getKiloSessionSnapshot(
-  sessionId: string | undefined,
-): Promise<KiloSessionSnapshot | undefined> {
-  if (!sessionId) return undefined;
-  const oc: KiloClient = await ensureServer();
-  return getSessionSnapshot(oc as unknown as RemoteSessionClient, sessionId);
-}

package/src/backend/opencode/factory.ts

@@ -1,53 +0,0 @@
-/**
- * OpenCode backend factory — wires OpenCode into the registry.
- *
- * Side-effect import (`import "./factory.js"` from `bootstrap.ts`) calls
- * `registerBackend(...)` at module load, making OpenCode available under
- * `config.backend === "opencode"`. The capability wiring is the shared
- * remote-server composition; only the bound functions are OpenCode's.
- */
-
-import { registerBackend } from "../../core/agent-runtime/backend-registry.js";
-import { createRemoteBackendFactory } from "../remote-server/factory.js";
-import {
-  initOpenCodeAgent,
-  stopOpenCodeServer,
-  handleMessage,
-  runOneShotAgent,
-  getOpenCodeSessionSnapshot,
-  resolveModel,
-  getModelInfo,
-  getSettingsPresentation,
-  getProviders,
-  getProviderModels,
-  formatModelError,
-  listModels,
-  refreshPluginMcpServers,
-  updateSystemPrompt,
-  warmSession,
-} from "./index.js";
-
-registerBackend(
-  createRemoteBackendFactory({
-    id: "opencode",
-    label: "OpenCode",
-    sdkPackage: "@opencode-ai/sdk",
-    init: initOpenCodeAgent,
-    stop: stopOpenCodeServer,
-    handleMessage,
-    runOneShotAgent,
-    getSessionSnapshot: getOpenCodeSessionSnapshot,
-    models: {
-      resolveModel,
-      getModelInfo,
-      getSettingsPresentation,
-      getProviders,
-      getProviderModels,
-      formatModelError,
-      listModels,
-    },
-    refreshPluginMcpServers,
-    warmSession,
-    updateSystemPrompt,
-  }),
-);

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

@@ -1,2 +0,0 @@
-/** OpenCode message handler barrel. */
-export { handleMessage } from "./message.js";

package/src/backend/opencode/handler/message.ts

@@ -1,44 +0,0 @@
-/**
- * OpenCode main message handler — the shared remote-server chat turn
- * bound to OpenCode's server bindings.
- */
-
-import type {
-  QueryParams,
-  QueryResult,
-} from "../../runtime/turn/handler-types.js";
-import { runRemoteChatTurn } from "../../remote-server/chat-turn.js";
-import {
-  ensureServer,
-  trackActiveTurn,
-  ensureSession,
-  ensureChatMcpServer,
-  ensurePluginMcpServers,
-  buildToolOverrides,
-  resolveProviderID,
-  parseStoredOpenCodeModelSelection,
-  getConfig,
-  opencodeSystemPromptSuffix,
-} from "../server.js";
-
-export function handleMessage(params: QueryParams): Promise<QueryResult> {
-  // Assembled per call, not at module load: tests mock `../server.js`
-  // with a partial surface and only touch the members they need.
-  return runRemoteChatTurn(
-    {
-      id: "opencode",
-      label: "OpenCode",
-      getConfig,
-      ensureServer,
-      trackActiveTurn,
-      parseModelSelection: parseStoredOpenCodeModelSelection,
-      resolveProviderID,
-      ensureSession,
-      ensureChatMcpServer,
-      ensurePluginMcpServers,
-      buildToolOverrides,
-      systemPromptSuffix: opencodeSystemPromptSuffix,
-    },
-    params,
-  );
-}

package/src/backend/opencode/index.ts

@@ -1,42 +0,0 @@
-/**
- * OpenCode backend — uses the OpenCode SDK as an alternative to Claude Agent SDK.
- *
- * Re-exports from focused sub-modules:
- *   models.ts   — model catalog, search, resolution, presentation
- *   sessions.ts — message parsing, usage summaries, snapshots
- *   server.ts   — server lifecycle, MCP, session management
- *   handler.ts  — main message handler
- */
-
-export {
-  getOpenCodeModelSelectionValue,
-  resolveOpenCodeModelInput,
-  formatOpenCodeSelectionError,
-} from "./models/index.js";
-
-export {
-  summarizeOpenCodeAssistantMessages,
-  getOpenCodeSessionSnapshot,
-} from "./sessions.js";
-
-export {
-  initOpenCodeAgent,
-  stopOpenCodeServer,
-  refreshPluginMcpServers,
-  updateSystemPrompt,
-  warmSession,
-} from "./server.js";
-
-export { handleMessage } from "./handler/index.js";
-
-export {
-  resolveModel,
-  getModelInfo,
-  getSettingsPresentation,
-  getProviders,
-  getProviderModels,
-  formatModelError,
-  listModels,
-} from "./model-provider.js";
-
-export { runOneShotAgent } from "./one-shot.js";

package/src/backend/opencode/model-provider.ts

@@ -1,36 +0,0 @@
-/**
- * OpenCode model provider — adapts the internal catalog to the Backend model
- * interface via the shared remote-server factory. Each export matches a
- * method on `Backend.models`.
- */
-
-import { createRemoteModelProvider } from "../remote-server/model-catalog/index.js";
-import {
-  getOpenCodeModelCatalog,
-  getOpenCodeModelInfo,
-  getOpenCodeModelSelectionValue,
-  resolveOpenCodeModelInput,
-  getOpenCodeSettingsPresentation,
-  formatOpenCodeUnavailableModel,
-} from "./models/index.js";
-
-const provider = createRemoteModelProvider({
-  label: "OpenCode",
-  getCatalog: (forceRefresh) => getOpenCodeModelCatalog(forceRefresh),
-  getModelInfo: (id) => getOpenCodeModelInfo(id),
-  resolveModelInput: (query, catalog) =>
-    resolveOpenCodeModelInput(query, catalog),
-  getSelectionValue: (model, catalog) =>
-    getOpenCodeModelSelectionValue(model, catalog),
-  formatUnavailableModel: (model) => formatOpenCodeUnavailableModel(model),
-  getSettingsPresentation: (activeModel, callbackPrefix) =>
-    getOpenCodeSettingsPresentation(activeModel, callbackPrefix),
-});
-
-export const resolveModel = provider.resolveModel;
-export const getModelInfo = provider.getModelInfo;
-export const getSettingsPresentation = provider.getSettingsPresentation;
-export const getProviders = provider.getProviders;
-export const getProviderModels = provider.getProviderModels;
-export const listModels = provider.listModels;
-export const formatModelError = provider.formatModelError;

package/src/backend/opencode/models/index.ts

@@ -1,54 +0,0 @@
-/**
- * OpenCode model catalog — binder over the shared remote-server model-catalog
- * module (see `backend/remote-server/model-catalog/`).
- *
- * The catalog, resolution, and presentation logic is shared with Kilo (a fork
- * of OpenCode with the same wire format); this file only binds the OpenCode
- * SDK client + presentation knobs and re-exports the result under the
- * historical `OpenCode*`-prefixed names.
- *
- * Knobs: OpenCode's model picker renders through Telegram inline keyboards —
- * callback_data caps at 64 bytes and the keyboard is tight, so quick picks
- * stay at 4 and only short separator-free ids are embedded raw.
- */
-
-import { ensureServer, onServerStop } from "../server.js";
-import { createRemoteModelCatalogModule } from "../../remote-server/model-catalog/index.js";
-
-export type {
-  RemoteModelCatalogEntry as OpenCodeModelCatalogEntry,
-  RemoteModelCatalog as OpenCodeModelCatalog,
-  RemoteModelResolution as OpenCodeModelResolution,
-  ModelButton,
-} from "../../remote-server/model-catalog/index.js";
-export {
-  sortCatalogModels,
-  getRemoteModelSelectionValue as getOpenCodeModelSelectionValue,
-  resolveRemoteModelInput as resolveOpenCodeModelInput,
-  guessProviderID,
-  getBucketPriority,
-  normalizeModelLookup,
-  parseRemoteModelQuery as parseOpenCodeModelQuery,
-  formatRemoteUnavailableModel as formatOpenCodeUnavailableModel,
-} from "../../remote-server/model-catalog/index.js";
-
-const opencodeModels = createRemoteModelCatalogModule({
-  label: "OpenCode",
-  getClient: () => ensureServer(),
-  maxCallbackIdLength: 48,
-  allowCallbackSeparators: false,
-  quickPickLimit: 4,
-});
-
-// A stopped server invalidates the catalog it served.
-onServerStop(opencodeModels.clearCache);
-
-export const getOpenCodeModelCatalog = opencodeModels.getCatalog;
-export const clearModelCatalogCache = opencodeModels.clearCache;
-export const getOpenCodeModelInfo = opencodeModels.getModelInfo;
-export const getOpenCodeQuickPickModels = opencodeModels.getQuickPickModels;
-export const getOpenCodeSettingsPresentation =
-  opencodeModels.getSettingsPresentation;
-export const renderOpenCodeModelSummary = opencodeModels.renderModelSummary;
-export const renderOpenCodeModelList = opencodeModels.renderModelList;
-export const formatOpenCodeSelectionError = opencodeModels.formatSelectionError;

package/src/backend/opencode/one-shot.ts

@@ -1,42 +0,0 @@
-/**
- * OpenCode one-shot agent runner — used by heartbeat & dream.
- *
- * Thin binding over the shared remote-server runner (see
- * `backend/remote-server/one-shot.ts` for the design notes): OpenCode
- * supplies its server bootstrap, model-selection parser, and delivery
- * suffix; everything else is the shared lifecycle.
- */
-
-import type { OneShotAgentParams, OneShotUsage } from "../../core/types.js";
-import { runRemoteOneShotAgent } from "../remote-server/one-shot.js";
-import {
-  ensureServer,
-  ensureChatMcpServer,
-  ensurePluginMcpServers,
-  buildToolOverrides,
-  disconnectChatMcpServer,
-  resolveProviderID,
-  parseStoredOpenCodeModelSelection,
-  OPENCODE_SYSTEM_PROMPT_SUFFIX,
-  errMsg,
-} from "./server.js";
-
-export function runOneShotAgent(
-  params: OneShotAgentParams,
-): Promise<OneShotUsage | void> {
-  return runRemoteOneShotAgent(
-    {
-      label: "OpenCode",
-      systemPromptSuffix: OPENCODE_SYSTEM_PROMPT_SUFFIX,
-      ensureServer,
-      parseModelSelection: parseStoredOpenCodeModelSelection,
-      resolveProviderID,
-      ensureChatMcpServer,
-      ensurePluginMcpServers,
-      buildToolOverrides,
-      disconnectChatMcpServer,
-      errMsg,
-    },
-    params,
-  );
-}

package/src/backend/opencode/server.ts

@@ -1,80 +0,0 @@
-/**
- * OpenCode server lifecycle — the OpenCode profile bound through
- * `backend/remote-server/server-bindings.ts`.
- *
- * The server spawn, MCP registration, session creation, and provider
- * resolution are the shared remote-server family implementation (Kilo is
- * a fork with the same HTTP API). This module holds only what is
- * OpenCode's: the SDK constructors, port 4096, the text-preferred
- * delivery contract, and the fuzzy `provider/model` parser. The exports
- * keep their historical names — the models module, one-shot runner,
- * tests, and `vi.mock("../backend/opencode/server.js")` all address them.
- */
-
-import {
-  createOpencodeClient,
-  createOpencodeServer,
-  type OpencodeClient,
-} from "@opencode-ai/sdk/v2";
-import {
-  normalizeModelLookup,
-  parseRemoteModelQuery,
-} from "../remote-server/model-catalog/index.js";
-import {
-  bindRemoteServer,
-  type RemoteModelSelection,
-} from "../remote-server/server-bindings.js";
-
-/**
- * Parse the stored model-selection string into a `{providerID?, modelID}`
- * pair. The parser is fuzzy — it tries to extract a provider hint from the
- * prefix while preserving the full model id when ambiguous. See
- * `remote-server/model-catalog/` for the underlying `parseRemoteModelQuery`.
- */
-export function parseStoredOpenCodeModelSelection(
-  value: string,
-): RemoteModelSelection {
-  const { providerQuery, modelQuery } = parseRemoteModelQuery(value);
-  return {
-    providerID: providerQuery ? normalizeModelLookup(providerQuery) : undefined,
-    modelID: modelQuery,
-  };
-}
-
-// Text-preferred delivery: plain assistant text is the reply; tools only
-// for genuine side effects. Single-sourced from the shared contract
-// templates (prompts/system/contract-text-preferred.md).
-const opencode = bindRemoteServer<OpencodeClient>({
-  label: "OpenCode",
-  defaultPort: 4096,
-  portEnv: "OPENCODE_PORT",
-  deliveryContract: "text-preferred",
-  createClient: (baseUrl) =>
-    createOpencodeClient({ baseUrl, throwOnError: true }),
-  createServer: ({ hostname, port, timeout }) =>
-    createOpencodeServer({ hostname, port, timeout }),
-  parseModelSelection: parseStoredOpenCodeModelSelection,
-});
-
-export const OPENCODE_BASE_URL = opencode.baseUrl;
-export const opencodeSystemPromptSuffix = opencode.systemPromptSuffix;
-export const OPENCODE_SYSTEM_PROMPT_SUFFIX = opencode.defaultSystemPromptSuffix;
-export const initOpenCodeAgent = opencode.init;
-export const stopOpenCodeServer = opencode.stop;
-export const {
-  onServerStop,
-  ensureServer,
-  trackActiveTurn,
-  ensureChatMcpServer,
-  ensurePluginMcpServers,
-  buildToolOverrides,
-  disconnectChatMcpServer,
-  refreshPluginMcpServers,
-  updateSystemPrompt,
-  ensureSession,
-  warmSession,
-  resolveProviderID,
-  getConfig,
-  getRegisteredMcpServerNames,
-  errMsg,
-} = opencode;

package/src/backend/opencode/sessions.ts

@@ -1,35 +0,0 @@
-/**
- * OpenCode session helpers — the shared `remote-server/session-helpers.ts`
- * bound to the OpenCode client, re-exported under OpenCode-named symbols
- * for the handler / index / test imports.
- */
-
-import type { OpencodeClient } from "@opencode-ai/sdk/v2";
-import {
-  extractPartsSummary as extractPartsSummaryShared,
-  summarizeAssistantMessages,
-  getSessionSnapshot,
-  type RemoteSessionSnapshot,
-  type RemoteSessionClient,
-} from "../remote-server/session-helpers.js";
-import { ensureServer } from "./server.js";
-
-/** Snapshot of an OpenCode session's lifetime + last-turn assistant info. */
-export type OpenCodeSessionSnapshot = RemoteSessionSnapshot;
-
-export const extractPartsSummary = extractPartsSummaryShared;
-
-/** Summarise a batch of session messages into per-turn usage totals. */
-export const summarizeOpenCodeAssistantMessages = summarizeAssistantMessages;
-
-/**
- * Build an {@link OpenCodeSessionSnapshot} for the given session id.
- * Returns `undefined` if no session id was provided.
- */
-export async function getOpenCodeSessionSnapshot(
-  sessionId: string | undefined,
-): Promise<OpenCodeSessionSnapshot | undefined> {
-  if (!sessionId) return undefined;
-  const oc: OpencodeClient = await ensureServer();
-  return getSessionSnapshot(oc as unknown as RemoteSessionClient, sessionId);
-}