@animalabs/connectome-host 0.3.1 → 0.3.7

package/src/call-ledger.ts

@@ -0,0 +1,371 @@
+/**
+ * Recent provider-call ledger for the WebUI.
+ *
+ * The provider adapter feeds this tracker compact, content-free call records.
+ * It reconstructs the cache verdict operators actually need (hit, first
+ * write, expired rewrite, uncached auxiliary call) and retains a bounded
+ * window. Older process logs are replayed on startup so a host restart does
+ * not leave the dashboard blank.
+ */
+
+import { closeSync, openSync, readSync, readdirSync, statSync } from 'node:fs';
+import { join } from 'node:path';
+import type {
+  CallLedgerRow,
+  CallLedgerSnapshot,
+  CallLedgerVerdict,
+} from './web/protocol.js';
+import { ANTHROPIC_PRICING_VERSION, priceAnthropicCall } from './call-pricing.js';
+
+export interface ProviderCallRecord {
+  timestamp: string;
+  kind: 'complete' | 'stream';
+  durationMs: number;
+  model: string;
+  messages: number;
+  inputTokens: number;
+  outputTokens: number;
+  cacheReadTokens: number;
+  cacheWriteTokens: number;
+  /** Provider-reported cache creation buckets. These remain separate because
+   *  Anthropic charges different multipliers for 5m and 1h writes. */
+  cacheWrite5mTokens?: number;
+  cacheWrite1hTokens?: number;
+  /** True only when the split came from the response usage object. */
+  cacheWriteBucketsAuthoritative?: boolean;
+  inferenceGeo?: string;
+  serviceTier?: string;
+  /** Undefined for historical compact logs written before cache summaries. */
+  cacheBreakpoints?: number;
+  /** TTLs found on cache_control blocks. `default` means provider default. */
+  cacheTtls?: string[];
+  stopReason?: string;
+  error?: string;
+}
+
+export interface CallLedgerOptions {
+  dataDir?: string;
+  defaultTtl?: '5m' | '1h';
+  maxRows?: number;
+  hydrate?: boolean;
+}
+
+type Listener = (snapshot: CallLedgerSnapshot) => void;
+
+export class CallLedger {
+  private readonly rows: CallLedgerRow[] = [];
+  private readonly maxRows: number;
+  private readonly defaultTtl: '5m' | '1h';
+  private readonly listeners = new Set<Listener>();
+  private lastCacheActivityAt: number | null = null;
+  private sequence = 0;
+
+  constructor(opts: CallLedgerOptions = {}) {
+    this.maxRows = opts.maxRows ?? 200;
+    this.defaultTtl = opts.defaultTtl ?? '5m';
+    if (opts.hydrate !== false && opts.dataDir) this.hydrate(opts.dataDir);
+  }
+
+  record(call: ProviderCallRecord): void {
+    this.append(call, true);
+  }
+
+  onUpdate(listener: Listener): () => void {
+    this.listeners.add(listener);
+    return () => this.listeners.delete(listener);
+  }
+
+  snapshot(): CallLedgerSnapshot {
+    const rows = this.rows.map((r) => ({
+      ...r,
+      tokens: { ...r.tokens },
+      cache: { ...r.cache, ttls: [...r.cache.ttls] },
+      ...(r.cost ? { cost: { ...r.cost, rates: { ...r.cost.rates } } } : {}),
+    }));
+    const byVerdict: Partial<Record<CallLedgerVerdict, number>> = {};
+    let input = 0;
+    let output = 0;
+    let cacheRead = 0;
+    let cacheWrite = 0;
+    let totalCost = 0;
+    let pricedCalls = 0;
+    for (const row of rows) {
+      byVerdict[row.verdict] = (byVerdict[row.verdict] ?? 0) + 1;
+      input += row.tokens.input;
+      output += row.tokens.output;
+      cacheRead += row.tokens.cacheRead;
+      cacheWrite += row.tokens.cacheWrite;
+      if (row.cost) {
+        totalCost += row.cost.total;
+        pricedCalls++;
+      }
+    }
+    const cacheDenominator = input + cacheRead + cacheWrite;
+    return {
+      rows,
+      summary: {
+        calls: rows.length,
+        input,
+        output,
+        cacheRead,
+        cacheWrite,
+        cacheHitRatio: cacheDenominator > 0 ? cacheRead / cacheDenominator : 0,
+        cost: {
+          total: totalCost,
+          currency: 'USD',
+          pricedCalls,
+          unpricedCalls: rows.length - pricedCalls,
+          pricingVersion: ANTHROPIC_PRICING_VERSION,
+        },
+        byVerdict,
+      },
+    };
+  }
+
+  private append(call: ProviderCallRecord, notify: boolean): void {
+    const at = Date.parse(call.timestamp);
+    const ttl = effectiveTtl(call.cacheTtls, this.defaultTtl);
+    const ttlSeconds = ttl === '1h' ? 3600 : 300;
+    const hasKnownFlags = call.cacheBreakpoints !== undefined;
+    const hasCacheControls = (call.cacheBreakpoints ?? 0) > 0;
+    const cacheActivity = call.cacheReadTokens > 0 || call.cacheWriteTokens > 0;
+    const priorCacheAt = this.lastCacheActivityAt;
+    const gapSeconds = priorCacheAt !== null && Number.isFinite(at)
+      ? Math.max(0, Math.round((at - priorCacheAt) / 1000))
+      : undefined;
+
+    const { verdict, cause } = classifyCall({
+      call,
+      hasKnownFlags,
+      hasCacheControls,
+      ttlSeconds,
+      gapSeconds,
+    });
+
+    const write5m = call.cacheWrite5mTokens ?? 0;
+    const write1h = call.cacheWrite1hTokens ?? 0;
+    const reportedSplit = write5m + write1h;
+    const splitMismatch = call.cacheWriteTokens > 0 && (
+      !call.cacheWriteBucketsAuthoritative || reportedSplit !== call.cacheWriteTokens
+    );
+    const cost = call.error ? undefined : priceAnthropicCall(call.model, call.timestamp, {
+      inputTokens: call.inputTokens,
+      outputTokens: call.outputTokens,
+      cacheReadTokens: call.cacheReadTokens,
+      cacheWrite5mTokens: write5m,
+      cacheWrite1hTokens: write1h,
+      unclassifiedCacheWriteTokens: splitMismatch
+        ? Math.max(1, Math.abs(call.cacheWriteTokens - reportedSplit))
+        : 0,
+      inferenceGeo: call.inferenceGeo,
+      serviceTier: call.serviceTier,
+    });
+
+    const row: CallLedgerRow = {
+      id: `${call.timestamp}:${++this.sequence}`,
+      timestamp: call.timestamp,
+      kind: call.kind,
+      originEstimate: call.kind === 'stream' ? 'turn~' : 'aux~',
+      model: call.model,
+      messages: call.messages,
+      durationMs: call.durationMs,
+      tokens: {
+        input: call.inputTokens,
+        output: call.outputTokens,
+        cacheRead: call.cacheReadTokens,
+        cacheWrite: call.cacheWriteTokens,
+      },
+      ...(cost ? { cost } : {}),
+      cache: {
+        breakpoints: call.cacheBreakpoints,
+        ttls: call.cacheTtls?.length ? [...new Set(call.cacheTtls)] : [],
+        effectiveTtl: ttl,
+      },
+      verdict,
+      cause,
+      ...(call.stopReason ? { stopReason: call.stopReason } : {}),
+      ...(call.error ? { error: call.error } : {}),
+    };
+
+    this.rows.push(row);
+    if (this.rows.length > this.maxRows) this.rows.splice(0, this.rows.length - this.maxRows);
+    if (cacheActivity && Number.isFinite(at)) this.lastCacheActivityAt = at;
+
+    if (notify) {
+      const snapshot = this.snapshot();
+      for (const listener of this.listeners) {
+        try { listener(snapshot); } catch { /* observability must not break inference */ }
+      }
+    }
+  }
+
+  private hydrate(dataDir: string): void {
+    let files: string[];
+    try {
+      files = readdirSync(dataDir)
+        .filter((name) => /^llm-calls\..*\.jsonl$/.test(name))
+        .map((name) => join(dataDir, name))
+        .sort((a, b) => statSync(a).mtimeMs - statSync(b).mtimeMs)
+        .slice(-4);
+    } catch {
+      return;
+    }
+
+    const calls: ProviderCallRecord[] = [];
+    for (const file of files) {
+      for (const line of readTailLines(file, 16 * 1024 * 1024)) {
+        try {
+          const parsed = parseLoggedCall(JSON.parse(line) as Record<string, unknown>);
+          if (parsed) calls.push(parsed);
+        } catch {
+          // Partial/legacy/oversized forensic lines are skipped individually.
+        }
+      }
+    }
+    calls.sort((a, b) => Date.parse(a.timestamp) - Date.parse(b.timestamp));
+    for (const call of calls.slice(-this.maxRows)) this.append(call, false);
+  }
+}
+
+function classifyCall(opts: {
+  call: ProviderCallRecord;
+  hasKnownFlags: boolean;
+  hasCacheControls: boolean;
+  ttlSeconds: number;
+  gapSeconds?: number;
+}): { verdict: CallLedgerVerdict; cause: string } {
+  const { call, hasKnownFlags, hasCacheControls, ttlSeconds, gapSeconds } = opts;
+  if (call.error) return { verdict: 'ERROR', cause: call.error };
+  if (hasKnownFlags && !hasCacheControls) {
+    return { verdict: 'uncached', cause: 'no cache_control flags in request' };
+  }
+  if (call.cacheReadTokens > 0 && call.cacheWriteTokens > 0) {
+    const pct = cachePercent(call);
+    return { verdict: 'hit+extend', cause: `${pct}% from cache; wrote ${formatTokens(call.cacheWriteTokens)} more` };
+  }
+  if (call.cacheReadTokens > 0) {
+    return { verdict: 'HIT', cause: `${cachePercent(call)}% from cache` };
+  }
+  if (call.cacheWriteTokens > 0) {
+    if (gapSeconds === undefined) {
+      return { verdict: 'first-write', cause: `created ${formatTokens(call.cacheWriteTokens)} cache tokens` };
+    }
+    if (gapSeconds > ttlSeconds) {
+      return { verdict: 'rewrite:expired', cause: `gap ${gapSeconds}s > ttl ${ttlSeconds}s` };
+    }
+    return {
+      verdict: 'rewrite:unexplained',
+      cause: `cache miss inside ttl (${gapSeconds}s <= ${ttlSeconds}s); prefix changed or was truncated`,
+    };
+  }
+  if (!hasKnownFlags) {
+    return { verdict: 'unknown', cause: 'older log has no cache_control summary' };
+  }
+  return { verdict: 'empty', cause: 'cache flags present; provider reported no cache activity' };
+}
+
+function cachePercent(call: ProviderCallRecord): number {
+  const total = call.inputTokens + call.cacheReadTokens + call.cacheWriteTokens;
+  return total > 0 ? Math.round(call.cacheReadTokens / total * 100) : 0;
+}
+
+function effectiveTtl(ttls: string[] | undefined, fallback: '5m' | '1h'): '5m' | '1h' {
+  if (ttls?.includes('1h')) return '1h';
+  if (ttls?.includes('5m') || ttls?.includes('default')) return '5m';
+  return fallback;
+}
+
+function formatTokens(n: number): string {
+  if (n < 1000) return String(n);
+  if (n < 1_000_000) return `${(n / 1000).toFixed(1)}k`;
+  return `${(n / 1_000_000).toFixed(2)}M`;
+}
+
+function parseLoggedCall(record: Record<string, unknown>): ProviderCallRecord | null {
+  const kind = record.kind;
+  const timestamp = record.timestamp;
+  const summary = record.requestSummary as Record<string, unknown> | undefined;
+  if ((kind !== 'complete' && kind !== 'stream') || typeof timestamp !== 'string' || !summary) return null;
+  const rawResponse = record.rawResponse as Record<string, unknown> | null | undefined;
+  const usage = rawResponse?.usage as Record<string, unknown> | undefined;
+  const cacheCreation = usage?.cache_creation as Record<string, unknown> | undefined;
+  const rawRequest = record.rawRequest;
+  const summarized = rawRequest ? summarizeCacheControls(rawRequest) : undefined;
+  const error = record.error as Record<string, unknown> | string | undefined;
+  return {
+    timestamp,
+    kind,
+    durationMs: number(record.durationMs),
+    model: string(summary.model),
+    messages: number(summary.messages),
+    inputTokens: number(usage?.input_tokens),
+    outputTokens: number(usage?.output_tokens),
+    cacheReadTokens: number(usage?.cache_read_input_tokens),
+    cacheWriteTokens: number(usage?.cache_creation_input_tokens),
+    cacheWrite5mTokens: optionalNumber(cacheCreation?.ephemeral_5m_input_tokens),
+    cacheWrite1hTokens: optionalNumber(cacheCreation?.ephemeral_1h_input_tokens),
+    cacheWriteBucketsAuthoritative: cacheCreation !== undefined,
+    inferenceGeo: optionalString(usage?.inference_geo) ?? optionalString(rawResponse?.inference_geo),
+    serviceTier: optionalString(usage?.service_tier) ?? optionalString(rawResponse?.service_tier),
+    cacheBreakpoints: optionalNumber(summary.cacheBreakpoints) ?? summarized?.count,
+    cacheTtls: stringArray(summary.cacheTtls) ?? summarized?.ttls,
+    stopReason: typeof rawResponse?.stop_reason === 'string' ? rawResponse.stop_reason : undefined,
+    error: typeof error === 'string'
+      ? error
+      : typeof error?.message === 'string' ? error.message : undefined,
+  };
+}
+
+/** Content-free cache_control inventory from an actual provider payload. */
+export function summarizeCacheControls(raw: unknown): { count: number; ttls: string[] } {
+  let count = 0;
+  const ttls: string[] = [];
+  const visit = (value: unknown): void => {
+    if (!value || typeof value !== 'object') return;
+    if (Array.isArray(value)) {
+      for (const item of value) visit(item);
+      return;
+    }
+    const obj = value as Record<string, unknown>;
+    if (obj.cache_control && typeof obj.cache_control === 'object') {
+      count++;
+      const ttl = (obj.cache_control as Record<string, unknown>).ttl;
+      ttls.push(typeof ttl === 'string' ? ttl : 'default');
+    }
+    for (const [key, child] of Object.entries(obj)) {
+      if (key !== 'cache_control') visit(child);
+    }
+  };
+  visit(raw);
+  return { count, ttls };
+}
+
+function readTailLines(path: string, maxBytes: number): string[] {
+  let fd: number | undefined;
+  try {
+    const size = statSync(path).size;
+    const start = Math.max(0, size - maxBytes);
+    const length = size - start;
+    if (length <= 0) return [];
+    fd = openSync(path, 'r');
+    const buffer = Buffer.allocUnsafe(length);
+    readSync(fd, buffer, 0, length, start);
+    let text = buffer.toString('utf8');
+    if (start > 0) {
+      const firstNewline = text.indexOf('\n');
+      text = firstNewline >= 0 ? text.slice(firstNewline + 1) : '';
+    }
+    return text.split('\n').filter(Boolean);
+  } catch {
+    return [];
+  } finally {
+    if (fd !== undefined) closeSync(fd);
+  }
+}
+
+const number = (v: unknown): number => typeof v === 'number' && Number.isFinite(v) ? v : 0;
+const string = (v: unknown): string => typeof v === 'string' ? v : '';
+const optionalNumber = (v: unknown): number | undefined => typeof v === 'number' && Number.isFinite(v) ? v : undefined;
+const optionalString = (v: unknown): string | undefined => typeof v === 'string' && v.length > 0 ? v : undefined;
+const stringArray = (v: unknown): string[] | undefined =>
+  Array.isArray(v) && v.every((x) => typeof x === 'string') ? v as string[] : undefined;

package/src/call-pricing.ts

@@ -0,0 +1,119 @@
+/**
+ * Provider-call pricing used by the operator ledger.
+ *
+ * Rates are USD per million tokens and are applied to the provider's
+ * authoritative usage buckets. Cache creation MUST remain split by TTL:
+ * Anthropic bills 5m writes at 1.25x input and 1h writes at 2x input.
+ *
+ * Source: https://platform.claude.com/docs/en/about-claude/pricing
+ * Snapshot: 2026-07-13. Keep the version string/date auditable; silently
+ * changing historical prices would make old JSONL replay disagree with bills.
+ */
+
+import type { CallCostBreakdown } from './web/protocol.js';
+
+export const ANTHROPIC_PRICING_VERSION = 'anthropic-public-2026-07-13';
+
+interface BaseRate {
+  inputPerMillion: number;
+  outputPerMillion: number;
+}
+
+export interface PriceableCallUsage {
+  inputTokens: number;
+  outputTokens: number;
+  cacheReadTokens: number;
+  cacheWrite5mTokens: number;
+  cacheWrite1hTokens: number;
+  /** Non-zero means the provider reported creation tokens we could not place
+   *  in an authoritative TTL bucket. Such a call is deliberately unpriced. */
+  unclassifiedCacheWriteTokens: number;
+  inferenceGeo?: string;
+  serviceTier?: string;
+}
+
+export function priceAnthropicCall(
+  model: string,
+  timestamp: string,
+  usage: PriceableCallUsage,
+): CallCostBreakdown | undefined {
+  const rate = anthropicBaseRate(model, timestamp);
+  if (!rate || usage.unclassifiedCacheWriteTokens > 0) return undefined;
+
+  // Public list pricing covers standard service. Priority Tier is contract
+  // priced; returning no figure is safer than presenting a plausible lie.
+  const tier = usage.serviceTier?.toLowerCase();
+  if (tier && tier !== 'standard') return undefined;
+
+  const geo = usage.inferenceGeo?.toLowerCase();
+  const geoMultiplier = !geo || geo === 'global'
+    ? 1
+    : (geo === 'us' || geo === 'us-only' || geo === 'us_only') ? 1.1 : undefined;
+  if (geoMultiplier === undefined) return undefined;
+
+  const input = usage.inputTokens * rate.inputPerMillion / 1_000_000 * geoMultiplier;
+  const cacheWrite5m = usage.cacheWrite5mTokens * rate.inputPerMillion * 1.25 / 1_000_000 * geoMultiplier;
+  const cacheWrite1h = usage.cacheWrite1hTokens * rate.inputPerMillion * 2 / 1_000_000 * geoMultiplier;
+  const cacheRead = usage.cacheReadTokens * rate.inputPerMillion * 0.1 / 1_000_000 * geoMultiplier;
+  const output = usage.outputTokens * rate.outputPerMillion / 1_000_000 * geoMultiplier;
+
+  return {
+    input,
+    cacheWrite5m,
+    cacheWrite1h,
+    cacheRead,
+    output,
+    total: input + cacheWrite5m + cacheWrite1h + cacheRead + output,
+    currency: 'USD',
+    grade: 'billing',
+    pricingVersion: ANTHROPIC_PRICING_VERSION,
+    rates: {
+      inputPerMillion: rate.inputPerMillion * geoMultiplier,
+      outputPerMillion: rate.outputPerMillion * geoMultiplier,
+      cacheWrite5mPerMillion: rate.inputPerMillion * 1.25 * geoMultiplier,
+      cacheWrite1hPerMillion: rate.inputPerMillion * 2 * geoMultiplier,
+      cacheReadPerMillion: rate.inputPerMillion * 0.1 * geoMultiplier,
+    },
+  };
+}
+
+function anthropicBaseRate(model: string, timestamp: string): BaseRate | undefined {
+  // Fable 5 and Mythos 5 share pricing.
+  if (starts(model, 'claude-fable-5', 'claude-mythos-5')) return rate(10, 50);
+
+  if (starts(model,
+    'claude-opus-4-8',
+    'claude-opus-4-7',
+    'claude-opus-4-6',
+    'claude-opus-4-5',
+  )) return rate(5, 25);
+
+  // Sonnet 5 launch pricing is promotional through 2026-08-31 inclusive.
+  if (starts(model, 'claude-sonnet-5')) {
+    const at = Date.parse(timestamp);
+    const promoEnd = Date.parse('2026-09-01T00:00:00Z');
+    return Number.isFinite(at) && at < promoEnd ? rate(2, 10) : rate(3, 15);
+  }
+
+  if (starts(model,
+    'claude-sonnet-4-6',
+    'claude-sonnet-4-5',
+    'claude-sonnet-4-',
+    'claude-3-7-sonnet',
+    'claude-3-5-sonnet',
+  )) return rate(3, 15);
+
+  if (starts(model, 'claude-haiku-4-5')) return rate(1, 5);
+  if (starts(model, 'claude-3-5-haiku')) return rate(0.8, 4);
+
+  if (starts(model, 'claude-opus-4-1', 'claude-opus-4-')) return rate(15, 75);
+  return undefined;
+}
+
+function starts(model: string, ...prefixes: string[]): boolean {
+  return prefixes.some((prefix) => model.startsWith(prefix));
+}
+
+function rate(inputPerMillion: number, outputPerMillion: number): BaseRate {
+  return { inputPerMillion, outputPerMillion };
+}

package/src/commands.ts

@@ -197,7 +197,13 @@ export function handleCommand(command: string, app: AppContext): CommandResult {
       return handleCheckout(framework, args[0]);
 
     case 'history':
-      return handleHistory(framework);
+      return handleHistory(framework, args[0]);
+
+    case 'branchto':
+      return handleBranchTo(app, args[0]);
+
+    case 'find':
+      return handleFind(framework, args.join(' '));
 
     case 'mcp':
       return handleMcp(args);
@@ -793,7 +799,55 @@ function handleCheckout(framework: AgentFramework, name?: string): CommandResult
   };
 }
 
-function handleHistory(framework: AgentFramework): CommandResult {
+function handleFind(framework: AgentFramework, needle: string): CommandResult {
+  const cm = getAgentCM(framework);
+  if (!cm) return { lines: [{ text: 'No agent context manager.', style: 'system' }] };
+  if (!needle) return { lines: [{ text: 'Usage: /find <text>', style: 'system' }] };
+  const { messages } = cm.queryMessages({});
+  const lines: Line[] = [{ text: `--- Find "${needle}" in ${messages.length} msgs ---`, style: 'system' }];
+  const needleLc = needle.toLowerCase();
+  for (const msg of messages) {
+    // Search text blocks plus stringified tool i/o (case-insensitive), so ids
+    // buried in tool_use inputs / tool_result payloads are findable too.
+    const full = msg.content
+      .map((b) => b.type === 'text' ? b.text : JSON.stringify(b))
+      .join(' ');
+    const at = full.toLowerCase().indexOf(needleLc);
+    if (at >= 0) {
+      const snip = full.slice(Math.max(0, at - 30), at + needle.length + 45).replace(/\s+/g, ' ');
+      lines.push({ text: `  [${msg.id}] ${msg.participant}: ...${snip}...`, style: 'system' });
+    }
+  }
+  if (lines.length === 1) lines.push({ text: '  (no matches)', style: 'system' });
+  return { lines };
+}
+
+function handleBranchTo(app: AppContext, messageId: string | undefined): CommandResult {
+  const cm = getAgentCM(app.framework);
+  if (!cm) return { lines: [{ text: 'No agent context manager.', style: 'system' }] };
+  if (!messageId) return { lines: [{ text: 'Usage: /branchto <messageId>', style: 'system' }] };
+  const { messages } = cm.queryMessages({});
+  const target = messages.find(m => String(m.id) === String(messageId));
+  if (!target) return { lines: [{ text: `No message with id ${messageId} on current branch.`, style: 'system' }] };
+  const newBranchName = `rollback-at-${messageId}-${Date.now()}`;
+  let createdBranchName: string;
+  try {
+    createdBranchName = cm.branchAt(String(messageId), newBranchName);
+  } catch (err) {
+    return { lines: [{ text: `branchto failed (branchAt): ${err}`, style: 'system' }] };
+  }
+  const asyncWork = Promise.resolve(cm.switchBranch(createdBranchName))
+    .then(() => ({
+      lines: [{ text: `Branched at [${messageId}] -> "${createdBranchName}" and switched. Everything after [${messageId}] is dropped on this branch; old branch preserved.`, style: 'system' as const }],
+      branchChanged: true,
+    }))
+    .catch((err: unknown) => ({
+      lines: [{ text: `branchto failed (switchBranch): ${err}`, style: 'system' as const }],
+    }));
+  return { lines: [{ text: `Branching at [${messageId}] -> "${createdBranchName}"...`, style: 'system' }], asyncWork };
+}
+
+function handleHistory(framework: AgentFramework, countArg?: string): CommandResult {
   const cm = getAgentCM(framework);
   if (!cm) return { lines: [{ text: 'No agent context manager.', style: 'system' }] };
 
@@ -801,7 +855,7 @@ function handleHistory(framework: AgentFramework): CommandResult {
   const lines: Line[] = [{ text: `--- History (${messages.length} messages) ---`, style: 'system' }];
 
   // Show the last 20 messages in summary form
-  const recent = messages.slice(-20);
+  const recent = messages.slice(-(countArg && Number(countArg) > 0 ? Number(countArg) : 20));
   for (const msg of recent) {
     const text = msg.content
       .filter((b): b is { type: 'text'; text: string } => b.type === 'text')