pi-ui-extend 0.1.56 → 0.1.57

package/external/pi-tools-suite/src/codex-reasoning-fix/index.ts

@@ -3,10 +3,9 @@
  * rejects HTTP 400 `Unknown parameter: 'input[N].content'` when non-message
  * items (reasoning, function_call_output) carry a spurious `content` field.
  *
- * Root cause is in pi-ai's `convertResponsesMessages`, which pushes replayed
- * items verbatim including a stray `content` placeholder. The upstream fix is
- * uncommitted in pi-mono and never shipped, so every released pi-ai version
- * (incl. 0.79.10) is affected.
+ * Stray fields can come from replayed pi-ai items or from another extension
+ * that modifies the provider payload. The sanitizer therefore must be the
+ * LAST `before_provider_request` handler registered by pi-tools-suite.
  *
  * The fix must run at the transport layer because the provider has TWO paths:
  *
@@ -16,16 +15,13 @@
  *     strip `content` from every non-message input item of each
  *     `response.create` frame, on the exact bytes leaving the socket.
  *
- *  2. SSE/fetch fallback path: once a websocket attempt fails (common at large
- *     context — full body, ~100+ input items), the transport records an SSE
- *     fallback for the session and every subsequent request goes through a plain
- *     `fetch()` POST with the FULL body (`bodyJson`). The `before_provider_request`
- *     hook should clean this, but in practice it does not reliably catch every
- *     offending item, so we also wrap `globalThis.fetch` once and strip the same
- *     spurious `content` from the request body on the exact bytes leaving the
- *     HTTP client.
+ *  2. SSE/fetch fallback path: once a websocket attempt fails, subsequent
+ *     requests use the FULL body. pi-ai >= 0.80.6 zstd-compresses that body
+ *     before fetch, so a fetch wrapper cannot inspect it. The final
+ *     `before_provider_request` sanitizer is the primary guard for this path;
+ *     the fetch wrapper remains a best-effort fallback for uncompressed bodies.
  *
- * The `before_provider_request` hook is kept as a third secondary guard.
+ * Module registration order is part of this workaround's correctness.
  *
  * Remove this whole module once an upstream pi-ai release carries the fix.
  */
@@ -182,6 +178,9 @@ export function stripContentFromWireFrame(frame: unknown): { frame: Record<strin
 }
 
 export default function codexReasoningFix(pi: ExtensionAPI): void {
+	// src/index.ts deliberately registers this module last. A later payload
+	// modifier could otherwise reintroduce invalid content after sanitization,
+	// and compressed SSE bodies are too late to repair in the fetch wrapper.
 	pi.on("before_provider_request", async (event: ProviderRequestEvent, _ctx: ProviderRequestContext) => {
 		const result = stripReasoningContentFromPayload(event.payload);
 		return result === event.payload ? undefined : result;

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

@@ -132,6 +132,17 @@ function appendDcpControlToMessages(messages: unknown, text: string): unknown {
 	for (let index = messages.length - 1; index >= 0; index--) {
 		const message = messages[index] as any
 		if (!message || typeof message !== "object") continue
+		// Responses items such as reasoning/function_call_output do not accept a
+		// `content` field. Keep a function output at the tail by appending to its
+		// valid `output` string; otherwise scan back to an actual message item.
+		if (typeof message.type === "string" && message.type !== "message" && message.role === undefined) {
+			if (message.type === "function_call_output" && typeof message.output === "string") {
+				return messages.map((candidate: any, candidateIndex) => candidateIndex === index
+					? { ...candidate, output: `${candidate.output}\n\n${block}` }
+					: candidate)
+			}
+			continue
+		}
 		if (message.role === "system" || message.role === "developer") continue
 		targetIndex = index
 		break

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

@@ -9,8 +9,7 @@ type ExtensionModule = {
 	default: ExtensionFactory;
 };
 
-const MODULES: Array<{ name: string; load: () => Promise<ExtensionModule> }> = [
-	{ name: "codex-reasoning-fix", load: () => import("./codex-reasoning-fix/index") },
+export const MODULES: Array<{ name: string; load: () => Promise<ExtensionModule> }> = [
 	{ name: "coding-discipline", load: () => import("./coding-discipline/index") },
 	{ name: "ast-grep", load: () => import("./ast-grep/index") },
 	{ name: "async-subagents", load: () => import("./async-subagents/index") },
@@ -28,6 +27,9 @@ const MODULES: Array<{ name: string; load: () => Promise<ExtensionModule> }> = [
 	{ name: "prompt-commands", load: () => import("./prompt-commands/index") },
 	{ name: "skill-installer", load: () => import("./skill-installer/index") },
 	{ name: "telegram-mirror", load: () => import("./telegram-mirror/index") },
+	// Keep this last: its before_provider_request handler is the final payload
+	// sanitizer after DCP and any other provider-payload modifiers.
+	{ name: "codex-reasoning-fix", load: () => import("./codex-reasoning-fix/index") },
 ];
 
 export default async function piToolsSuite(pi: ExtensionAPI) {

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

@@ -10,16 +10,6 @@ import { getState, replaceState } from "./state/store.js";
 import { activateTodoStateScope, DEFAULT_PROMPT_GUIDELINES, DEFAULT_PROMPT_SNIPPET, publishTodoState, registerTodosCommand, registerTodoTool } from "./todo.js";
 import type { Task, TaskMutationParams } from "./tool/types.js";
 
-/**
- * Renderer-relayed signal that the session is in an auto-retry cycle.
- * Payload: `{ active: boolean }`. The SDK does not forward retry state to
- * extensions, so the renderer emits this on the extension event bus. We use it
- * to suppress the auto-nudge on intermediate retry agent_end events: nudging
- * then races the pending retry continuation and surfaces a benign
- * "Agent is already processing" extension error in other tabs.
- */
-const RETRY_ACTIVE_EVENT = "pix:retry-active";
-
 type AgentMessageLike = { role?: unknown; stopReason?: unknown; content?: unknown };
 
 const TODO_NUDGE_LIMIT = 8;
@@ -144,10 +134,10 @@ export default function (pi: ExtensionAPI) {
 	const pendingAskUserToolCallIds = new Set<string>();
 	let suppressNextNudgeForThinkingSwitch = false;
 	let inProgressAtAgentStart = new Set<number>();
-	// True while the session is in an auto-retry cycle (relayed via the
-	// extension event bus). Suppresses the auto-nudge on intermediate retry
-	// agent_end events so it doesn't race the pending retry continuation.
-	let retryActive = false;
+	// agent_end may be followed by an automatic retry, compaction, or queued
+	// continuation. Remember only whether its terminal assistant reply was a
+	// successful visible response; schedule a nudge later on agent_settled.
+	let settledNudgeEligible = false;
 
 	function registerTodoToolWithCurrentPrompt(): void {
 		const thinkingPrompt = todoThinkingEnabled ? buildThinkingPromptParts(currentModel) : {};
@@ -336,9 +326,6 @@ export default function (pi: ExtensionAPI) {
 		const delayMs = attempt === 0 ? TODO_NUDGE_INITIAL_DELAY_MS : TODO_NUDGE_IDLE_RETRY_DELAY_MS;
 		nudgeTimer = setTimeout(() => {
 			nudgeTimer = undefined;
-			// A retry-start signal may have arrived after the agent_end that
-			// scheduled this nudge. Bail out so we don't nudge mid-retry.
-			if (retryActive) return;
 			try {
 				activateTodoStateScope(ctx);
 				if (!ctx.isIdle()) {
@@ -358,9 +345,8 @@ export default function (pi: ExtensionAPI) {
 				if (nudge.signature === lastNudgedSignature) return;
 				lastNudgedSignature = nudge.signature;
 
-				// agent_end fires before Pi is fully back in idle dispatch. Sending as a
-				// normal user message on the next idle tick reliably starts a fresh turn;
-				// queueing followUp from inside agent_end can be too late to be drained.
+				// agent_settled means retries, compaction, and queued continuations are
+				// finished. Send on the next idle tick as a fresh turn.
 				pi.sendUserMessage(nudge.message);
 			} catch (err) {
 				if (isAgentBusyRaceError(err)) {
@@ -413,12 +399,6 @@ export default function (pi: ExtensionAPI) {
 		clearNudgeTimer();
 	});
 
-	pi.events.on(RETRY_ACTIVE_EVENT, (data: unknown) => {
-		const active = data != null && typeof data === "object" && (data as { active?: unknown }).active === true;
-		retryActive = active;
-		if (active) clearNudgeTimer();
-	});
-
 	pi.on("model_select", async (event) => {
 		currentModel = event.model;
 		if (todoThinkingEnabled) registerTodoToolWithCurrentPrompt();
@@ -439,7 +419,7 @@ export default function (pi: ExtensionAPI) {
 	pi.on("agent_start", async (_event, ctx) => {
 		activateTodoStateScope(ctx);
 		pendingAskUserToolCallIds.clear();
-		retryActive = false;
+		settledNudgeEligible = false;
 		inProgressAtAgentStart = new Set(selectVisibleTasks(getState()).filter((task) => task.status === "in_progress").map((task) => task.id));
 	});
 
@@ -455,26 +435,42 @@ export default function (pi: ExtensionAPI) {
 	pi.on("agent_end", async (event, ctx) => {
 		activateTodoStateScope(ctx);
 		const completedAssistantReply = hasCompletedAssistantReply((event as { messages?: readonly unknown[] } | undefined)?.messages);
+		settledNudgeEligible = completedAssistantReply;
 
 		if (suppressNextNudgeForThinkingSwitch) {
 			suppressNextNudgeForThinkingSwitch = false;
 			if (!completedAssistantReply) {
+				settledNudgeEligible = false;
 				clearNudgeTimer();
 				return;
 			}
 		}
 
 		if (pendingAskUserToolCallIds.size > 0) {
+			settledNudgeEligible = false;
 			clearNudgeTimer();
 			return;
 		}
 
 		if (completedAssistantReply && maybeRecoverCompletedCurrentTask((event as { messages?: readonly unknown[] } | undefined)?.messages, ctx)) {
+			settledNudgeEligible = false;
 			lastNudgedSignature = undefined;
 			clearNudgeTimer();
 			return;
 		}
+	});
 
+	pi.on("agent_settled", async (_event, ctx) => {
+		activateTodoStateScope(ctx);
+		if (!settledNudgeEligible) {
+			clearNudgeTimer();
+			return;
+		}
+		settledNudgeEligible = false;
+		if (pendingAskUserToolCallIds.size > 0) {
+			clearNudgeTimer();
+			return;
+		}
 		const nudge = getUnfinishedTodoNudge();
 		if (!nudge) {
 			lastNudgedSignature = undefined;

package/package.json

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