diff --git a/packages/pi-turn-limit/src/turn-limit.ts b/packages/pi-turn-limit/src/turn-limit.ts index 97bfb72..f36a12e 100644 --- a/packages/pi-turn-limit/src/turn-limit.ts +++ b/packages/pi-turn-limit/src/turn-limit.ts @@ -36,12 +36,19 @@ export default function (pi: ExtensionAPI) { let turnCount = 0; let maxTurns = getMaxTurns(); + pi.on("session_start", async (event, ctx) => { + // On reload, show the widget immediately + if (event.reason === "reload" && ctx.hasUI) { + ctx.ui.setWidget("turn-limit", [`Turns: ${turnCount}/${maxTurns}`]); + } + }); + pi.on("agent_start", async (_event, ctx) => { // Reset counter for each new user prompt turnCount = 0; - // Clear widget on new session + // Show initial widget state on fresh session if (ctx.hasUI) { - ctx.ui.setWidget("turn-limit", undefined); + ctx.ui.setWidget("turn-limit", [`Turns: ${turnCount}/${maxTurns}`]); } });