pi-prompt-template-model 0.2.0 → 0.2.1

package/CHANGELOG.md

@@ -2,6 +2,12 @@
 
 ## [Unreleased]
 
+## [0.2.1] - 2026-01-31
+
+### Fixed
+
+- Thinking level now correctly restored after commands that switch model without a `thinking` field. Previously, running a prompt template that only specified `model` would reset thinking to "off" instead of restoring the original level (e.g., "high").
+
 ## [0.2.0] - 2025-01-31
 
 ### Added

package/index.ts

@@ -549,8 +549,9 @@ export default function promptModelExtension(pi: ExtensionAPI) {
 					return;
 				}
 
-				// Capture current model before any switching (needed for restore)
+				// Capture current state before any switching (needed for restore)
 				const savedModel = ctx.model;
+				const savedThinking = pi.getThinkingLevel();
 
 				// Resolve and switch to the first available model from the list
 				const result = await resolveAndSwitch(currentPrompt.models, ctx);
@@ -558,17 +559,15 @@ export default function promptModelExtension(pi: ExtensionAPI) {
 
 				if (!result.alreadyActive && currentPrompt.restore) {
 					previousModel = savedModel;
+					previousThinking = savedThinking;
 				}
 
 				// Set thinking level if specified
 				if (currentPrompt.thinking) {
-					const currentThinking = pi.getThinkingLevel();
-					if (currentThinking !== currentPrompt.thinking) {
-						if (currentPrompt.restore) {
-							previousThinking = currentThinking;
-						}
-						pi.setThinkingLevel(currentPrompt.thinking);
+					if (currentPrompt.restore && previousThinking === undefined && currentPrompt.thinking !== savedThinking) {
+						previousThinking = savedThinking;
 					}
+					pi.setThinkingLevel(currentPrompt.thinking);
 				}
 
 				// Set pending skill for before_agent_start handler

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-prompt-template-model",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Prompt template model selector extension for pi coding agent",
   "author": "Nico Bailon",
   "license": "MIT",