tinker-agent 2.3.0 → 2.5.0

package/src/tools/types.ts

@@ -1,6 +1,7 @@
 import type { ToolCall } from "../agent/types";
 import type { ImageAssetRef } from "../image/image-types";
-import type { SessionId } from "../ids/runtime-id";
+import type { MessageId, SessionId } from "../ids/runtime-id";
+import type { ContextUsageSource } from "../model/model-request-preflight";
 import type {
   RecallGetPage,
   RecallSearchFilters,
@@ -310,6 +311,85 @@ export type RecallGetRawResult =
 
 export type RecallRawResult = RecallSearchRawResult | RecallGetRawResult;
 
+export type ContextMaintenancePressure = "normal" | "high" | "critical";
+
+export type ContextSwapCandidate = {
+  candidateId: MessageId;
+  label: string;
+  ordinal: number;
+  savingsBytes: number;
+};
+
+export type ContextSwapScheduledCandidate = {
+  candidateId: MessageId;
+  savingsBytes: number;
+};
+
+export type ContextSwapRejectedCandidate = {
+  candidateId: MessageId;
+  reason: string;
+};
+
+type ContextMaintenanceFailure<TOperation extends "status" | "candidates" | "swap"> = {
+  ok: false;
+  operation: TOperation;
+  error: string;
+};
+
+export type ContextStatusRawResult =
+  | {
+      ok: true;
+      operation: "status";
+      usedInputTokens: number;
+      inputBudgetTokens: number;
+      pressure: ContextMaintenancePressure;
+      triggerTokens: number;
+      source: ContextUsageSource;
+    }
+  | ContextMaintenanceFailure<"status">;
+
+export type ContextSwapCandidatesRawResult =
+  | {
+      ok: true;
+      operation: "candidates";
+      total: number;
+      candidates: readonly ContextSwapCandidate[];
+    }
+  | ContextMaintenanceFailure<"candidates">;
+
+export type ContextSwapRawResult =
+  | {
+      ok: true;
+      operation: "swap";
+      scheduled: readonly ContextSwapScheduledCandidate[];
+      rejected: readonly ContextSwapRejectedCandidate[];
+      note: string;
+    }
+  | {
+      ok: false;
+      operation: "swap";
+      scheduled: readonly ContextSwapScheduledCandidate[];
+      rejected: readonly ContextSwapRejectedCandidate[];
+      error?: string;
+    };
+
+export type ContextMaintenanceRawResult =
+  | ContextStatusRawResult
+  | ContextSwapCandidatesRawResult
+  | ContextSwapRawResult;
+
+export type ContextMaintenanceHandle = {
+  status(call: ToolCall): Promise<ContextStatusRawResult>;
+  candidates(
+    call: ToolCall,
+    input: { readonly limit: number; readonly offset: number },
+  ): Promise<ContextSwapCandidatesRawResult>;
+  swap(
+    call: ToolCall,
+    input: { readonly candidateIds: readonly MessageId[] },
+  ): Promise<ContextSwapRawResult>;
+};
+
 export type MemorySearchRawResult =
   | {
       ok: true;
@@ -425,6 +505,7 @@ export type ToolRawResultByKind = {
   web_search: WebSearchRawResult;
   web_fetch: WebFetchRawResult;
   recall: RecallRawResult;
+  context_maintenance: ContextMaintenanceRawResult;
   memory_search: MemorySearchRawResult;
   memory_get: MemoryGetRawResult;
   wait: WaitRawResult;
@@ -472,6 +553,7 @@ export function defineToolExecutor<TKind extends ToolRawResultKind>(
 
 export type ToolExecutionContext = {
   signal: AbortSignal;
+  contextMaintenance?: ContextMaintenanceHandle;
   confirmBashCommand?: (request: {
     command: string;
     reason: string;

package/src/tui/event-store.ts

@@ -170,6 +170,15 @@ export function reduceTuiProjection(
         }),
       );
     }
+    case "context.pressure_notice.sent":
+      return updateActiveTurn(state, event, policy, (turn) =>
+        appendTurnItem(turn, {
+          id: `turn-${event.turnId}-pressure-notice-${event.eventSequence}`,
+          label: "context notice",
+          text: `input pressure ${event.data.pressure} (${event.data.usedInputTokens}/${event.data.inputBudgetTokens} tokens) — model notified to self-manage`,
+          status: "text",
+        }),
+      );
     case "model.request.started":
       return updateActiveTurn(state, event, policy, (turn) =>
         event.data.attemptNumber === 1
@@ -883,6 +892,19 @@ function toolRawResultSummary(name: string, args: unknown, raw: ToolRawResult):
       return raw.mode === "search"
         ? `${base} -> ${raw.page.hits.length} historical match${raw.page.hits.length === 1 ? "" : "es"}`
         : `${base} -> ${raw.page.returnedBytes} historical bytes`;
+    case "context_maintenance":
+      if (!raw.ok) {
+        return raw.operation === "swap"
+          ? `${base} -> 0 scheduled, ${raw.rejected.length} rejected`
+          : base;
+      }
+      if (raw.operation === "status") {
+        return `${base} -> ${raw.pressure}, ${raw.usedInputTokens}/${raw.inputBudgetTokens} tokens`;
+      }
+      if (raw.operation === "candidates") {
+        return `${base} -> ${raw.candidates.length}/${raw.total} candidates`;
+      }
+      return `${base} -> ${raw.scheduled.length} scheduled, ${raw.rejected.length} rejected`;
     case "memory_search":
       if (!raw.ok) {
         return base;
@@ -980,6 +1002,7 @@ function toolRawResultBashDetail(raw: ToolRawResult): Pick<TimelineItem, "bash">
     case "web_search":
     case "web_fetch":
     case "recall":
+    case "context_maintenance":
     case "memory_search":
     case "memory_get":
     case "wait":
@@ -1021,6 +1044,7 @@ function toolRawResultDiff(
     case "web_search":
     case "web_fetch":
     case "recall":
+    case "context_maintenance":
     case "memory_search":
     case "memory_get":
     case "wait":