pi-ui-extend 0.1.55 → 0.1.57

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;
@@ -27,7 +17,7 @@ const TODO_NUDGE_INITIAL_DELAY_MS = 0;
 const TODO_NUDGE_IDLE_RETRY_DELAY_MS = 100;
 const TODO_NUDGE_MAX_IDLE_ATTEMPTS = 40;
 const ASK_USER_TOOL_NAMES = new Set(["ask_user", "ask_user_question", "question"]);
-const TODO_THINKING_LEVELS = ["off", "minimal", "low", "medium", "high", "xhigh"] as const;
+const TODO_THINKING_LEVELS = ["off", "minimal", "low", "medium", "high", "xhigh", "max"] as const;
 const TODO_THINKING_RESTORE_METADATA_KEY = "__piTodoRestoreThinking";
 
 function isStaleExtensionContextError(error: unknown): boolean {
@@ -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/external/pi-tools-suite/src/todo/state/state-reducer.ts

@@ -30,7 +30,7 @@ function uniqueNumbers(values: number[] | undefined): number[] {
 }
 
 function isTodoThinkingLevel(value: unknown): value is TodoThinkingLevel {
-	return value === "off" || value === "minimal" || value === "low" || value === "medium" || value === "high" || value === "xhigh";
+	return value === "off" || value === "minimal" || value === "low" || value === "medium" || value === "high" || value === "xhigh" || value === "max";
 }
 
 function findTask(state: TaskState, id: number): Task | undefined {

package/external/pi-tools-suite/src/todo/tool/types.ts

@@ -24,7 +24,7 @@ export const MSG_NO_TODOS = "No todos yet. Ask the agent to add some!";
 // ---------------------------------------------------------------------------
 
 export type TaskStatus = "pending" | "in_progress" | "deferred" | "completed" | "deleted";
-export type TodoThinkingLevel = "off" | "minimal" | "low" | "medium" | "high" | "xhigh";
+export type TodoThinkingLevel = "off" | "minimal" | "low" | "medium" | "high" | "xhigh" | "max";
 
 export type TaskAction = "create" | "update" | "batch_create" | "batch_update" | "list" | "get" | "delete" | "clear" | "export" | "import";
 
@@ -104,7 +104,7 @@ export const TodoParamsSchema = Type.Object({
 		}),
 	),
 	thinking: Type.Optional(
-		StringEnum(["off", "minimal", "low", "medium", "high", "xhigh"] as const, {
+		StringEnum(["off", "minimal", "low", "medium", "high", "xhigh", "max"] as const, {
 			description: "Per-task thinking level used when todoThinking is enabled and this task is in_progress",
 		}),
 	),

package/package.json

@@ -1,9 +1,10 @@
 {
   "name": "pi-ui-extend",
-  "version": "0.1.55",
+  "version": "0.1.57",
   "private": false,
   "type": "module",
   "bin": {
+    "pi-ui-extend": "./bin/pix.mjs",
     "pix": "./bin/pix.mjs"
   },
   "exports": {
@@ -64,9 +65,9 @@
     "prepublishOnly": "npm run check && npm run build:pix && npm run generate-schemas"
   },
   "dependencies": {
-    "@earendil-works/pi-ai": "0.80.3",
-    "@earendil-works/pi-coding-agent": "0.80.3",
-    "@earendil-works/pi-tui": "0.80.3",
+    "@earendil-works/pi-ai": "0.80.6",
+    "@earendil-works/pi-coding-agent": "0.80.6",
+    "@earendil-works/pi-tui": "0.80.6",
     "@mariozechner/clipboard": "^0.3.9",
     "jsonc-parser": "3.3.1",
     "typebox": "1.1.38",

package/schemas/pix.json

@@ -46,6 +46,10 @@
             {
               "type": "string",
               "const": "xhigh"
+            },
+            {
+              "type": "string",
+              "const": "max"
             }
           ],
           "description": "Default model thinking budget level."