@hk_net/pi-thinking-command 0.1.3 → 0.1.5

package/README.md

@@ -1,6 +1,6 @@
 # `/thinking` command extension for pi
 
-Adds a `/thinking` slash command to pi for changing the active thinking/reasoning level from inside a session.
+Adds a `/thinking` slash command to pi `>=0.80.10` for changing the active thinking/reasoning level from inside a session.
 
 Normally in pi, changing the thinking/reasoning level means opening the settings menu, navigating to **Thinking**, and selecting the requested level. This extension is a convenience shortcut for that workflow, so you can switch levels directly with commands such as `/thinking low` or `/thinking xhigh`.
 
@@ -36,7 +36,7 @@ After installing, restart pi or run:
 ## Usage
 
 ```text
-/thinking [off|minimal|low|medium|high|xhigh]
+/thinking [off|minimal|low|medium|high|xhigh|max]
 ```
 
 Examples:
@@ -47,6 +47,7 @@ Examples:
 /thinking medium
 /thinking high
 /thinking xhigh
+/thinking max
 ```
 
 If no argument is provided, `/thinking` sets the level to `medium`.
@@ -58,7 +59,8 @@ If no argument is provided, `/thinking` sets the level to `medium`.
 - `low` — light reasoning
 - `medium` — balanced default reasoning
 - `high` — more reasoning for harder tasks
-- `xhigh` — maximum reasoning budget
+- `xhigh` — very high reasoning budget
+- `max` — maximum reasoning budget
 
 > **Note:** If the current model does not support the requested level, pi clamps to the
 > nearest supported level (searching both up and down the level hierarchy).
@@ -67,6 +69,7 @@ If no argument is provided, `/thinking` sets the level to `medium`.
 
 - Registers the `/thinking` command.
 - Provides argument completions for all supported levels.
+- Autocompletes `/th...` to `/thinking` without Pi's trailing-space insertion, then presents the level picker.
 - Shows a `thinking` status item with the current level.
 - Updates the status item when the thinking level changes.
 - Validates input and displays an error for unknown levels.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hk_net/pi-thinking-command",
-  "version": "0.1.3",
+  "version": "0.1.5",
   "description": "Pi extension that adds a /thinking slash command for changing reasoning level",
   "type": "module",
   "private": false,
@@ -26,8 +26,8 @@
     "README.md"
   ],
   "peerDependencies": {
-    "@earendil-works/pi-coding-agent": ">=0.80.2",
-    "@earendil-works/pi-tui": ">=0.80.2"
+    "@earendil-works/pi-coding-agent": ">=0.80.10",
+    "@earendil-works/pi-tui": ">=0.80.10"
   },
   "pi": {
     "extensions": [

package/thinking-shortcut.ts

@@ -1,9 +1,9 @@
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import type { AutocompleteItem } from "@earendil-works/pi-tui";
 
-type ThinkingLevel = "off" | "minimal" | "low" | "medium" | "high" | "xhigh";
+type ThinkingLevel = Parameters<ExtensionAPI["setThinkingLevel"]>[0];
 
-const THINKING_LEVELS = ["off", "minimal", "low", "medium", "high", "xhigh"] as const satisfies readonly ThinkingLevel[];
+const THINKING_LEVELS = ["off", "minimal", "low", "medium", "high", "xhigh", "max"] as const satisfies readonly ThinkingLevel[];
 
 const THINKING_LEVEL_DESCRIPTIONS: Record<ThinkingLevel, string> = {
   off: "Disable extended thinking",
@@ -11,15 +11,16 @@ const THINKING_LEVEL_DESCRIPTIONS: Record<ThinkingLevel, string> = {
   low: "Light reasoning",
   medium: "Balanced default reasoning",
   high: "More reasoning for harder tasks",
-  xhigh: "Maximum reasoning budget",
+  xhigh: "Very high reasoning budget",
+  max: "Maximum reasoning budget",
 };
 
-function normalizeThinkingLevel(input: string): ThinkingLevel | undefined {
+export function normalizeThinkingLevel(input: string): ThinkingLevel | undefined {
   const normalized = input.toLowerCase().trim();
   return THINKING_LEVELS.find((level) => level === normalized);
 }
 
-function getThinkingLevelCompletions(prefix: string): AutocompleteItem[] | null {
+export function getThinkingLevelCompletions(prefix: string): AutocompleteItem[] | null {
   const normalizedPrefix = prefix.toLowerCase().trimStart();
   const matches = THINKING_LEVELS.filter((level) => level.startsWith(normalizedPrefix));
 
@@ -32,8 +33,71 @@ function getThinkingLevelCompletions(prefix: string): AutocompleteItem[] | null
   }));
 }
 
+export function isThinkingCommandPrefix(value: string): boolean {
+  return value.startsWith("/th") && "/thinking".startsWith(value);
+}
+
 export default function thinkingShortcutExtension(pi: ExtensionAPI) {
   pi.on("session_start", (_event, ctx) => {
+    ctx.ui.addAutocompleteProvider((current) => ({
+      async getSuggestions(lines, cursorLine, cursorCol, options) {
+        const line = lines[cursorLine] ?? "";
+        const beforeCursor = line.slice(0, cursorCol);
+        if (cursorCol === line.length && isThinkingCommandPrefix(beforeCursor)) {
+          if (beforeCursor === "/thinking") {
+            return { prefix: "", items: getThinkingLevelCompletions("") ?? [] };
+          }
+          return {
+            prefix: beforeCursor,
+            items: [{ value: "thinking", label: "thinking", description: "Set the thinking level" }],
+          };
+        }
+        if (beforeCursor === "/thinking ") {
+          return { prefix: "", items: getThinkingLevelCompletions("") ?? [] };
+        }
+        return current.getSuggestions(lines, cursorLine, cursorCol, options);
+      },
+
+      applyCompletion(lines, cursorLine, cursorCol, item, prefix) {
+        const line = lines[cursorLine] ?? "";
+        const beforeCursor = line.slice(0, cursorCol);
+        if (cursorCol === line.length && isThinkingCommandPrefix(beforeCursor)) {
+          if (beforeCursor !== "/thinking") {
+            return {
+              lines: [...lines.slice(0, cursorLine), "/thinking", ...lines.slice(cursorLine + 1)],
+              cursorLine,
+              cursorCol: "/thinking".length,
+            };
+          }
+          const nextLine = `/thinking ${item.value}`;
+          return {
+            lines: [...lines.slice(0, cursorLine), nextLine, ...lines.slice(cursorLine + 1)],
+            cursorLine,
+            cursorCol: nextLine.length,
+          };
+        }
+        if (beforeCursor === "/thinking " && cursorCol === line.length) {
+          const nextLine = `/thinking ${item.value}`;
+          return {
+            lines: [...lines.slice(0, cursorLine), nextLine, ...lines.slice(cursorLine + 1)],
+            cursorLine,
+            cursorCol: nextLine.length,
+          };
+        }
+        return current.applyCompletion(lines, cursorLine, cursorCol, item, prefix);
+      },
+
+      shouldTriggerFileCompletion(lines, cursorLine, cursorCol) {
+        const line = lines[cursorLine] ?? "";
+        const beforeCursor = line.slice(0, cursorCol);
+        if (
+          (isThinkingCommandPrefix(beforeCursor) || beforeCursor === "/thinking ") &&
+          cursorCol === line.length
+        ) return false;
+        return current.shouldTriggerFileCompletion?.(lines, cursorLine, cursorCol) ?? true;
+      },
+    }));
+
     ctx.ui.setStatus("thinking", `level: ${pi.getThinkingLevel()}`);
 
     ctx.ui.addAutocompleteProvider((current) => ({
@@ -69,7 +133,7 @@ export default function thinkingShortcutExtension(pi: ExtensionAPI) {
   });
 
   pi.registerCommand("thinking", {
-    description: "Set the thinking level. Usage: /thinking [off|minimal|low|medium|high|xhigh]",
+    description: "Set the thinking level. Usage: /thinking [off|minimal|low|medium|high|xhigh|max]",
     getArgumentCompletions: getThinkingLevelCompletions,
     handler: async (args, ctx) => {
       const input = args?.trim() || "medium";