pi-ui-extend 0.1.57 → 0.1.59

package/external/pi-tools-suite/src/dcp/pruner-types.ts

@@ -17,6 +17,36 @@ export interface MessageCompressionCandidate {
   reason: string;
 }
 
+export interface EmergencyCurrentTurnStats {
+  totalPairs: number;
+  totalPairTokens: number;
+  eligiblePairs: number;
+  eligibleTokens: number;
+  eligibleRecoverableTokens: number;
+  preservedPairs: number;
+  preservedTokens: number;
+  preservedRecentPairs: number;
+  preservedRecentTokens: number;
+  preservedUnseenPairs: number;
+  preservedUnseenTokens: number;
+  preservedProtectedPairs: number;
+  preservedProtectedTokens: number;
+}
+
+export interface EmergencyCurrentTurnOutput {
+  toolCallId: string;
+  messageId?: string;
+  toolName: string;
+  tokenEstimate: number;
+  recoverableTokens: number;
+  resultIndex: number;
+}
+
+export interface EmergencyCurrentTurnSelection {
+  eligible: EmergencyCurrentTurnOutput[];
+  stats: EmergencyCurrentTurnStats;
+}
+
 export type DcpNudgeType = "context-strong" | "context-soft" | "turn" | "iteration";
 
 export interface NudgeThresholds {

package/external/pi-tools-suite/src/dcp/pruner.ts

@@ -12,10 +12,18 @@ import {
 
 export type {
   CompressionCandidate,
+  EmergencyCurrentTurnSelection,
+  EmergencyCurrentTurnStats,
   MessageCompressionCandidate,
   MessagePriority,
   NudgeThresholds,
 } from "./pruner-types.js";
+export {
+  analyzeEmergencyCurrentTurn,
+  emergencyPressureState,
+  emergencyCurrentTurnMessageCandidates,
+  pruneEmergencyCurrentTurn,
+} from "./pruner-emergency.js";
 export {
   estimateTokens,
   getActiveSummaryTokenEstimate,

package/external/pi-tools-suite/src/dcp/state.ts

@@ -145,6 +145,8 @@ export interface DcpState {
   prunedToolIds: Set<string>
   /** toolCallId → reason used for human-readable pruning placeholders/stats. */
   prunedToolReasons: Map<string, string>
+  /** Tool results included in at least one provider request. */
+  providerSeenToolIds: Set<string>
 
   // ── Compression ────────────────────────────────────────────────────────────
   /** All compression blocks (both active and soft-deleted) */
@@ -228,12 +230,10 @@ export interface DcpState {
    */
   lastContextWindow?: number
   /**
-   * How many consecutive `context-strong` nudges have been emitted without a
-   * subsequent successful `compress` (model- or DCP-initiated). When this
-   * reaches `compress.autoCompress.patience` while context is above the
-   * emergency threshold, the auto-compress fallback creates a block without
-   * waiting for the model. Reset to 0 on any successful compression, when
-   * pressure drops below the emergency threshold, or on a window change.
+   * How many consecutive emergency context reminders have been delivered
+   * without a successful compression or emergency prune. This advances even
+   * when no normal compression candidate exists so current-turn fallback
+   * patience cannot be reset by the exact condition it is designed to handle.
    */
   consecutiveIgnoredStrongNudges: number
 }
@@ -248,6 +248,7 @@ export function createState(): DcpState {
     toolCalls: new Map(),
     prunedToolIds: new Set(),
     prunedToolReasons: new Map(),
+    providerSeenToolIds: new Set(),
     compressionBlocks: [],
     nextBlockId: 1,
     messageIdSnapshot: new Map(),
@@ -279,6 +280,7 @@ export function resetState(state: DcpState): void {
   state.toolCalls.clear()
   state.prunedToolIds.clear()
   state.prunedToolReasons.clear()
+  state.providerSeenToolIds.clear()
   state.compressionBlocks = []
   state.nextBlockId = 1
   state.messageIdSnapshot.clear()
@@ -401,6 +403,8 @@ export interface SerializedDcpState {
   nextBlockId: number
   prunedToolIds: string[]
   prunedToolReasons: Array<[string, string]>
+  /** Tool result IDs known to have appeared in a provider request. */
+  providerSeenToolIds?: string[]
   /** Full tool records — present in legacy snapshots. */
   toolCalls?: ToolRecord[]
   /** Compact tool records — present in new compact snapshots. */
@@ -597,11 +601,16 @@ export function serializeState(state: DcpState): SerializedDcpState {
     seen.add(record.toolCallId)
   }
 
+  const persistedToolCallIds = new Set(compactToolCalls.map((record) => record.toolCallId))
+  const providerSeenToolIds = Array.from(state.providerSeenToolIds)
+    .filter((toolCallId) => persistedToolCallIds.has(toolCallId))
+
   return {
     compressionBlocks: state.compressionBlocks,
     nextBlockId: state.nextBlockId,
     prunedToolIds: Array.from(state.prunedToolIds),
     prunedToolReasons: Array.from(state.prunedToolReasons.entries()),
+    providerSeenToolIds,
     compactToolCalls,
     totalToolCallCount: allRecords.length,
     tokensSaved: state.tokensSaved,
@@ -661,6 +670,12 @@ export function restoreState(state: DcpState, data: unknown): void {
     )
   }
 
+  if (Array.isArray(saved.providerSeenToolIds)) {
+    state.providerSeenToolIds = new Set(
+      saved.providerSeenToolIds.filter((id): id is string => typeof id === "string"),
+    )
+  }
+
   if (Array.isArray(saved.compactToolCalls)) {
     // New compact format: restore CompactToolRecords as ToolRecords with
     // synthetic inputArgs derived from inputStringValues.

package/external/pi-tools-suite/src/default-pi-tools-suite-config.ts

@@ -38,6 +38,20 @@ export const DEFAULT_PI_TOOLS_SUITE_CONFIG_JSONC = String.raw`{
     "debug": false,
     "debugLog": { "maxBytes": 5242880, "maxBackups": 3 },
     "manualMode": { "enabled": false, "automaticStrategies": true },
+    "strategies": {
+      "emergencyCurrentTurnPruning": {
+        // Disabling this turns off same-turn candidates and lossy pruning;
+        // non-destructive emergency reminders remain active.
+        "enabled": true,
+        "hardContextPercent": 0.82,
+        "targetContextPercent": 0.70,
+        "patience": 2,
+        "keepRecentToolPairs": 8,
+        "minOutputTokens": 500,
+        "maxSuggestions": 8,
+        "protectedTools": []
+      }
+    },
     "modelOverrides": {
       "openai-codex/gpt-5*": {
         "compress": {

package/external/pi-tools-suite/src/telegram-mirror/events.ts

@@ -18,7 +18,11 @@ export interface RendererSink {
 }
 
 export function registerPixEventHandlers(pi: ExtensionAPI, hooks: PixMirrorHooks): void {
+	let turnActive = false;
+
 	pi.on("agent_start", (_event, ctx) => {
+		if (turnActive) return;
+		turnActive = true;
 		hooks.getRenderer()?.push({ kind: "turn_start", instance: hooks.describeInstance(ctx as ExtensionContext | undefined) });
 	});
 
@@ -33,16 +37,18 @@ export function registerPixEventHandlers(pi: ExtensionAPI, hooks: PixMirrorHooks
 		// user-visible assistant answer, not internal reasoning/tools.
 	});
 
-	pi.on("agent_end", () => {
+	pi.on("agent_settled", () => {
+		if (!turnActive) return;
+		turnActive = false;
 		hooks.getRenderer()?.push({ kind: "turn_end", reason: "end" });
-		hooks.notifyAgentEnd();
+		hooks.notifyAgentSettled();
 	});
 }
 
 export interface PixMirrorHooks {
 	getRenderer(): RendererSink | undefined;
 	describeInstance(ctx: ExtensionContext | undefined): RendererInstance | undefined;
-	notifyAgentEnd(): void;
+	notifyAgentSettled(): void;
 }
 
 export function captureAbortableContext(ctx: ExtensionContext | undefined, hooks: ContextCapture): void {
@@ -59,4 +65,3 @@ export interface ContextCapture {
 	capturePending(fn: () => boolean): void;
 	captureCompact(fn: () => void): void;
 }
-

package/external/pi-tools-suite/src/telegram-mirror/index.ts

@@ -166,7 +166,7 @@ export default function telegramMirror(pi: ExtensionAPI): void {
 	const hooks: PixMirrorHooks = {
 		getRenderer: () => eventSink,
 		describeInstance: (ctx) => describeInstance(ctx),
-		notifyAgentEnd: () => undefined,
+		notifyAgentSettled: () => undefined,
 	};
 
 	registerPixEventHandlers(pi, hooks);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-ui-extend",
-  "version": "0.1.57",
+  "version": "0.1.59",
   "private": false,
   "type": "module",
   "bin": {