pi-ui-extend 0.1.46 → 0.1.47

package/dist/app/session/tabs-controller.d.ts

@@ -107,6 +107,7 @@ export declare class AppTabsController {
     private observeRuntimeForTab;
     private shouldSyncTabFromRuntimeEvent;
     private shouldScheduleDelayedSyncForRuntimeEvent;
+    private shouldInvalidateCachedViewForRuntimeEvent;
     private scheduleDelayedRuntimeSync;
     private clearRuntimeRefreshTimers;
     private clearHistoryReloadTimers;

package/dist/app/session/tabs-controller.js

@@ -641,8 +641,7 @@ export class AppTabsController {
         void this.saveTabs();
         this.scheduleTabPrewarm();
         const cachedView = this.sessionViewsByTabId.get(target.id);
-        const cachedViewNeedsHistoryReload = this.tabIdsNeedingHistoryReload.has(target.id)
-            && this.sessionActivity(targetRuntime.session) !== "running";
+        const cachedViewNeedsHistoryReload = this.tabIdsNeedingHistoryReload.has(target.id);
         if (cachedView && this.host.restoreSessionView && !cachedViewNeedsHistoryReload) {
             this.host.restoreSessionView(cachedView);
             this.restoreDeferredUserMessages(target.id);
@@ -883,9 +882,11 @@ export class AppTabsController {
             return;
         existing?.unsubscribe();
         const unsubscribe = runtime.session.subscribe((event) => {
+            if (this.shouldInvalidateCachedViewForRuntimeEvent(event)) {
+                this.tabIdsNeedingHistoryReload.add(tabId);
+            }
             if (this.shouldScheduleDelayedSyncForRuntimeEvent(event)) {
                 this.scheduleDelayedRuntimeSync(tabId, runtime);
-                this.tabIdsNeedingHistoryReload.add(tabId);
             }
             if (!this.shouldSyncTabFromRuntimeEvent(event))
                 return;
@@ -905,6 +906,17 @@ export class AppTabsController {
             || event.type === "turn_end"
             || event.type === "compaction_end";
     }
+    shouldInvalidateCachedViewForRuntimeEvent(event) {
+        return event.type === "message_start"
+            || event.type === "message_update"
+            || event.type === "message_end"
+            || event.type === "tool_execution_start"
+            || event.type === "tool_execution_update"
+            || event.type === "tool_execution_end"
+            || event.type === "agent_end"
+            || event.type === "turn_end"
+            || event.type === "compaction_end";
+    }
     scheduleDelayedRuntimeSync(tabId, runtime) {
         this.clearRuntimeRefreshTimers(tabId);
         for (const delayMs of [0, 100, 500, 1500, 3000]) {

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

@@ -123,12 +123,28 @@ function appendTextToContent(content: unknown, text: string): unknown {
 
 function appendDcpControlToMessages(messages: unknown, text: string): unknown {
 	if (!Array.isArray(messages)) return messages
-	const existingIndex = messages.findIndex((message: any) =>
-		message?.role === "system" || message?.role === "developer"
-	)
 	const block = `${DCP_PROVIDER_CONTROL_HEADER}\n${text}`
-	if (existingIndex >= 0) {
-		return messages.map((message: any, index) => index === existingIndex
+
+	// Keep DCP's volatile ID map at the tail of the provider context instead of
+	// mutating the stable system/developer prefix. Prefix cache reuse is much
+	// better when only the newest transcript item changes on each request.
+	let targetIndex = -1
+	for (let index = messages.length - 1; index >= 0; index--) {
+		const message = messages[index] as any
+		if (!message || typeof message !== "object") continue
+		if (message.role === "system" || message.role === "developer") continue
+		targetIndex = index
+		break
+	}
+
+	if (targetIndex < 0) {
+		targetIndex = messages.findIndex((message: any) =>
+			message?.role === "system" || message?.role === "developer"
+		)
+	}
+
+	if (targetIndex >= 0) {
+		return messages.map((message: any, index) => index === targetIndex
 			? { ...message, content: appendTextToContent(message.content, block) }
 			: message)
 	}
@@ -155,10 +171,6 @@ function appendDcpControlToProviderPayload(payload: unknown, text: string): unkn
 	if (!payload || typeof payload !== "object") return payload
 	const record = payload as Record<string, unknown>
 
-	if ("system" in record) {
-		return { ...record, system: appendDcpControlToAnthropicSystem(record.system, text) }
-	}
-
 	if (Array.isArray(record.input)) {
 		return { ...record, input: appendDcpControlToMessages(record.input, text) }
 	}
@@ -167,6 +179,10 @@ function appendDcpControlToProviderPayload(payload: unknown, text: string): unkn
 		return { ...record, messages: appendDcpControlToMessages(record.messages, text) }
 	}
 
+	if ("system" in record) {
+		return { ...record, system: appendDcpControlToAnthropicSystem(record.system, text) }
+	}
+
 	if (record.config && typeof record.config === "object") {
 		const config = record.config as Record<string, unknown>
 		return {

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

@@ -46,27 +46,87 @@ function priorityForMessage(tokenEstimate: number, config: DcpConfig | undefined
   return "low";
 }
 
-function controlLineForMessageId(id: string, state: DcpState): string {
-  const meta = state.messageMetaSnapshot.get(id);
-  if (!meta) return `- ${id}`;
-
-  const details: string[] = [meta.role];
-  if (meta.blockId !== undefined) details.push(`block=b${meta.blockId}`);
-  if (meta.toolName) details.push(`tool=${meta.toolName}`);
-  if (meta.priority) details.push(`priority=${meta.priority}`);
-  return `- ${id}: ${details.join(", ")}`;
+function messageIdIndex(id: string): number | undefined {
+  const match = /^m(\d+)$/.exec(id);
+  if (!match) return undefined;
+  const index = Number.parseInt(match[1]!, 10);
+  return Number.isFinite(index) ? index : undefined;
+}
+
+function summarizeMessageIds(ids: string[]): string {
+  if (ids.length === 0) return "none";
+
+  const ranges: string[] = [];
+  let start = ids[0]!;
+  let previous = ids[0]!;
+  let previousIndex = messageIdIndex(previous);
+
+  const flush = () => {
+    const startIndex = messageIdIndex(start);
+    const endIndex = messageIdIndex(previous);
+    if (
+      startIndex !== undefined &&
+      endIndex !== undefined &&
+      endIndex - startIndex >= 2
+    ) {
+      ranges.push(`${start}..${previous}`);
+    } else if (start === previous) {
+      ranges.push(start);
+    } else {
+      ranges.push(start, previous);
+    }
+  };
+
+  for (let i = 1; i < ids.length; i++) {
+    const id = ids[i]!;
+    const index = messageIdIndex(id);
+    if (previousIndex !== undefined && index === previousIndex + 1) {
+      previous = id;
+      previousIndex = index;
+      continue;
+    }
+
+    flush();
+    start = id;
+    previous = id;
+    previousIndex = index;
+  }
+
+  flush();
+  return ranges.join(", ");
+}
+
+function compactPriorityHint(ids: string[], state: DcpState): string | undefined {
+  const high: string[] = [];
+  const medium: string[] = [];
+  const blocks: string[] = [];
+
+  for (const id of ids) {
+    const meta = state.messageMetaSnapshot.get(id);
+    if (!meta) continue;
+    if (meta.priority === "high") high.push(id);
+    else if (meta.priority === "medium") medium.push(id);
+    if (meta.blockId !== undefined) blocks.push(`${id}=b${meta.blockId}`);
+  }
+
+  const parts: string[] = [];
+  if (high.length > 0) parts.push(`high=${summarizeMessageIds(high)}`);
+  if (medium.length > 0) parts.push(`medium=${summarizeMessageIds(medium)}`);
+  if (blocks.length > 0) parts.push(`blocks=${blocks.join(",")}`);
+  return parts.length > 0 ? `Hints: ${parts.join("; ")}` : undefined;
 }
 
 export function buildMessageIdControlText(state: DcpState): string | undefined {
   const ids = [...state.messageIdSnapshot.keys()];
   if (ids.length === 0) return undefined;
 
+  const hint = compactPriorityHint(ids, state);
+
   return [
     "<dcp-message-ids>",
-    "DCP metadata for the preceding conversation messages. These IDs are model-visible but UI-hidden control data.",
-    "Use only these current IDs with the compress tool; do not quote or output this metadata.",
-    `Current raw message IDs: ${ids.join(", ")}`,
-    ...ids.map((id) => controlLineForMessageId(id, state)),
+    "DCP metadata for the preceding conversation messages. IDs follow current message order; use only these IDs with compress; do not quote/output.",
+    `Current raw message IDs: ${summarizeMessageIds(ids)}${ids.length > 2 ? ` (${ids.length} messages)` : ""}`,
+    ...(hint ? [hint] : []),
     "</dcp-message-ids>",
   ].join("\n");
 }

package/package.json

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