@pi-unipi/unipi 2.0.2 → 2.0.3

package/README.md

@@ -1,6 +1,6 @@
 # Unipi
 
-18 packages that turn Pi into a full development workstation. Structured workflows, persistent memory, parallel agents, web research, notifications, context management, and a live status bar — all wired together through a shared event system.
+20 workspace packages that turn Pi into a full development workstation. Structured workflows, persistent memory, parallel agents, web research, notifications, context management, command autocomplete, and a live status bar — all wired together through a shared event system.
 
 One command installs everything:
 ```bash
@@ -17,6 +17,8 @@ pi install npm:@pi-unipi/unipi
 
 **[Compactor](./packages/compactor/README.md)** — Zero-LLM context engine. 6-stage pipeline hits 95%+ token reduction at zero API cost. Session continuity, sandbox execution, FTS5 search.
 
+**[CocoIndex](./packages/cocoindex/README.md)** — Project indexing and semantic code search backed by CocoIndex and LanceDB. Agent tools and slash commands for status, init, update, and search.
+
 **[Subagents](./packages/subagents/README.md)** — Parallel execution with file locking. Spawn background agents to research, fix, or build while the main agent keeps going.
 
 **[Web API](./packages/web-api/README.md)** — Web search, page reading, content summarization. Smart-fetch engine with browser-grade TLS fingerprinting — free, no API key. Paid providers as fallbacks.
@@ -43,6 +45,8 @@ pi install npm:@pi-unipi/unipi
 
 **[Input Shortcuts](./packages/input-shortcuts/README.md)** — Keyboard shortcuts via vim-style chord overlay. Stash/restore, undo/redo, clipboard, thinking toggle.
 
+**[Command Enchantment](./packages/autocomplete/README.md)** — Enhanced `/unipi:*` autocomplete with package grouping, descriptions, colors, and registry audits that catch stale command docs before release.
+
 ## Architecture
 
 Packages discover each other through events, not direct imports. Core defines the event types and constants. Every package emits `MODULE_READY` on load and subscribes to events it cares about.
@@ -78,6 +82,7 @@ Coexists triggers enhance behavior when packages are installed together. Workflo
 | Ralph | `/unipi:ralph` | start, stop, resume, status |
 | Memory | `/unipi:memory-` | process, search, consolidate, forget |
 | Compactor | `/unipi:` | lossless-compact, session-recall, compact-stats, compact-settings, compact-preset, compact-help |
+| CocoIndex | `/unipi:cocoindex-` | init, update, status, search, settings |
 | Notify | `/unipi:notify-` | settings, test, set-tg, set-ntfy |
 | MCP | `/unipi:mcp-` | add, settings, sync, status |
 | Web | `/unipi:web-` | settings, cache-clear |
@@ -121,6 +126,7 @@ unipi/
 │   ├── ralph/          # Iterative loops
 │   ├── memory/         # SQLite + vector search
 │   ├── compactor/      # Context engine
+│   ├── cocoindex/      # Project indexing and semantic search
 │   ├── subagents/      # Parallel execution
 │   ├── web-api/        # Web research
 │   ├── mcp/            # MCP server integration
@@ -134,6 +140,7 @@ unipi/
 │   ├── utility/        # Diagnostics, diff rendering
 │   ├── updater/        # Auto-update, browsers
 │   ├── input-shortcuts/ # Keyboard shortcuts
+│   ├── autocomplete/   # Enhanced command autocomplete
 │   └── unipi/          # Umbrella package
 ├── .unipi/             # Runtime data (specs, plans, worktrees)
 └── CHANGELOG.md

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-unipi/unipi",
-  "version": "2.0.2",
+  "version": "2.0.3",
   "description": "All-in-one extension suite for Pi coding agent",
   "type": "module",
   "license": "MIT",

package/packages/ask-user/README.md

@@ -73,6 +73,10 @@ When the user chooses a `new_session` option:
 
 The tool result is rendered as `queued compact → ...` or `queued direct → ...`. If automatic delivery fails, ask-user falls back to editor prefill and warns you to press Enter.
 
+### History Expansion
+
+Completed `ask_user` calls stay readable in chat history. The collapsed result shows the selected answer; press Ctrl+O on the tool result to expand the original question, context, and options that were presented.
+
 ### Keyboard Controls
 
 | Mode | Keys |

package/packages/ask-user/ask-ui.ts

@@ -621,7 +621,7 @@ export function createRenderCall() {
  * Create a renderResult function for the ask_user tool.
  */
 export function createRenderResult() {
-  return (result: AgentToolResult<unknown>, _options: unknown, theme: Theme, _context: unknown) => {
+  return (result: AgentToolResult<unknown>, options: unknown, theme: Theme, _context: unknown) => {
     const details = result.details as Record<string, unknown> | undefined;
     if (!details) {
       const content = result.content as unknown as Array<Record<string, unknown>> | undefined;
@@ -634,103 +634,110 @@ export function createRenderResult() {
       return new Text(theme.fg("warning", "No response"), 0, 0);
     }
 
-    switch (response.kind) {
-      case "cancelled":
-        return new Text(theme.fg("warning", "Cancelled"), 0, 0);
-      case "timed_out":
-        return new Text(theme.fg("warning", "Timed out"), 0, 0);
-      case "freeform":
-        return new Text(
-          theme.fg("success", "✓ ") +
+    const renderOptionSummary = (): string[] => {
+      const rawOptions = (details as { options?: unknown }).options;
+      if (!Array.isArray(rawOptions) || rawOptions.length === 0) return [];
+      return rawOptions.map((opt, index) => {
+        if (typeof opt === "string") {
+          return theme.fg("dim", `  ${index + 1}. `) + theme.fg("text", opt);
+        }
+        const record = opt as Record<string, unknown>;
+        const label = String(record.label ?? record.value ?? `Option ${index + 1}`);
+        const value = typeof record.value === "string" && record.value !== label
+          ? theme.fg("dim", ` (${record.value})`)
+          : "";
+        const action = typeof record.action === "string" && record.action !== "select"
+          ? theme.fg("dim", ` [${record.action}]`)
+          : "";
+        const description = typeof record.description === "string" && record.description.trim()
+          ? `\n${theme.fg("muted", `     ${record.description}`)}`
+          : "";
+        return theme.fg("dim", `  ${index + 1}. `) + theme.fg("text", label) + value + action + description;
+      });
+    };
+
+    const answerText = (() => {
+      switch (response.kind) {
+        case "cancelled":
+          return theme.fg("warning", "Cancelled");
+        case "timed_out":
+          return theme.fg("warning", "Timed out");
+        case "freeform":
+          return theme.fg("success", "✓ ") +
             theme.fg("muted", "(wrote) ") +
-            theme.fg("accent", response.text || ""),
-          0,
-          0,
-        );
-      case "selection": {
-        const selections = response.selections || [];
-        const display =
-          selections.length === 1
+            theme.fg("accent", response.text || "");
+        case "selection": {
+          const selections = response.selections || [];
+          const display = selections.length === 1 ? selections[0] : selections.join(", ");
+          return theme.fg("success", "✓ ") + theme.fg("accent", display);
+        }
+        case "combined": {
+          const selections = response.selections || [];
+          const selDisplay = selections.length === 1
             ? selections[0]
             : selections.join(", ");
-        return new Text(
-          theme.fg("success", "✓ ") + theme.fg("accent", display),
-          0,
-          0,
-        );
-      }
-      case "combined": {
-        const selections = response.selections || [];
-        const selDisplay = selections.length === 1
-          ? selections[0]
-          : selections.join(", ");
-        return new Text(
-          theme.fg("success", "✓ ") +
+          return theme.fg("success", "✓ ") +
             theme.fg("accent", selDisplay) +
             theme.fg("muted", " and wrote ") +
-            theme.fg("accent", response.text || ""),
-          0,
-          0,
-        );
-      }
-      case "end_turn":
-        return new Text(
-          theme.fg("success", "✓ ") + theme.fg("muted", "end turn"),
-          0,
-          0,
-        );
-      case "new_session": {
-        const prefill = response.prefill || "";
-        if (response.launchStatus === "editor_prefill") {
-          const label = response.launchedWith === "compact"
-            ? "⚠ compact editor prefill → "
-            : "⚠ direct editor prefill → ";
-          return new Text(
-            theme.fg("warning", label) + theme.fg("accent", prefill),
-            0,
-            0,
-          );
-        }
-        if (response.launchStatus === "failed") {
-          const label = response.launchedWith === "compact"
-            ? "handoff failed (compact) → "
-            : "handoff failed (direct) → ";
-          return new Text(
-            theme.fg("error", label) + theme.fg("accent", prefill),
-            0,
-            0,
-          );
+            theme.fg("accent", response.text || "");
         }
-        if (response.launchedWith === "compact") {
-          return new Text(
-            theme.fg("success", "✓ queued compact → ") +
-              theme.fg("accent", prefill),
-            0,
-            0,
-          );
-        }
-        if (response.launchedWith === "direct") {
-          return new Text(
-            theme.fg("success", "✓ queued direct → ") +
-              theme.fg("accent", prefill),
-            0,
-            0,
-          );
-        }
-        return new Text(
-          theme.fg("success", "✓ ") +
+        case "end_turn":
+          return theme.fg("success", "✓ ") + theme.fg("muted", "end turn");
+        case "new_session": {
+          const prefill = response.prefill || "";
+          if (response.launchStatus === "editor_prefill") {
+            const label = response.launchedWith === "compact"
+              ? "⚠ compact editor prefill → "
+              : "⚠ direct editor prefill → ";
+            return theme.fg("warning", label) + theme.fg("accent", prefill);
+          }
+          if (response.launchStatus === "failed") {
+            const label = response.launchedWith === "compact"
+              ? "handoff failed (compact) → "
+              : "handoff failed (direct) → ";
+            return theme.fg("error", label) + theme.fg("accent", prefill);
+          }
+          if (response.launchedWith === "compact") {
+            return theme.fg("success", "✓ queued compact → ") + theme.fg("accent", prefill);
+          }
+          if (response.launchedWith === "direct") {
+            return theme.fg("success", "✓ queued direct → ") + theme.fg("accent", prefill);
+          }
+          return theme.fg("success", "✓ ") +
             theme.fg("muted", "new session") +
-            (prefill ? theme.fg("accent", `: ${prefill}`) : ""),
-          0,
-          0,
-        );
+            (prefill ? theme.fg("accent", `: ${prefill}`) : "");
+        }
+        default:
+          return theme.fg("text", JSON.stringify(response));
       }
-      default:
-        return new Text(
-          theme.fg("text", JSON.stringify(response)),
-          0,
-          0,
-        );
+    })();
+
+    const expanded = typeof options === "object" && options !== null && "expanded" in options
+      ? Boolean((options as { expanded?: boolean }).expanded)
+      : false;
+    if (!expanded) {
+      const question = typeof (details as { question?: unknown }).question === "string"
+        ? (details as { question: string }).question
+        : "";
+      const expandHint = question
+        ? theme.fg("dim", "  · Ctrl+O question/options")
+        : "";
+      return new Text(answerText + expandHint, 0, 0);
+    }
+
+    const lines = [answerText];
+    const question = (details as { question?: unknown }).question;
+    const context = (details as { context?: unknown }).context;
+    if (typeof question === "string" && question.trim()) {
+      lines.push(theme.fg("muted", "Question: ") + theme.fg("text", question));
     }
+    if (typeof context === "string" && context.trim()) {
+      lines.push(theme.fg("muted", "Context: ") + theme.fg("text", context));
+    }
+    const optionLines = renderOptionSummary();
+    if (optionLines.length > 0) {
+      lines.push(theme.fg("muted", "Options:"), ...optionLines);
+    }
+    return new Text(lines.join("\n"), 0, 0);
   };
 }

package/packages/ask-user/tools.ts

@@ -252,6 +252,13 @@ export function registerAskUserTools(pi: ExtensionAPI): void {
         action: (opt.action as NormalizedOption["action"]) ?? "select",
         prefill: opt.prefill,
       }));
+      const detailsBase = {
+        question,
+        context,
+        options: normalizedOptions,
+        allowMultiple,
+        allowFreeform,
+      };
 
       // Emit ASK_USER_PROMPT event if notifyOnAsk is enabled
       if (settings.notifyOnAsk) {
@@ -286,8 +293,7 @@ export function registerAskUserTools(pi: ExtensionAPI): void {
             },
           ],
           details: {
-            question,
-            options: normalizedOptions.map((o) => o.label),
+            ...detailsBase,
             response: {
               kind: "cancelled",
             } as AskUserResponse,
@@ -348,8 +354,7 @@ export function registerAskUserTools(pi: ExtensionAPI): void {
           return {
             content: [{ type: "text", text: "User cancelled the session launch" }],
             details: {
-              question,
-              options: normalizedOptions.map((o) => o.label),
+              ...detailsBase,
               response: {
                 kind: "cancelled",
                 comment: "Session launcher cancelled",
@@ -374,8 +379,7 @@ export function registerAskUserTools(pi: ExtensionAPI): void {
           return {
             content: [{ type: "text", text: "Session launch cancelled: no prefill message was provided." }],
             details: {
-              question,
-              options: normalizedOptions.map((o) => o.label),
+              ...detailsBase,
               response: {
                 kind: "cancelled",
                 comment: "Session launcher had no prefill to queue",
@@ -404,8 +408,7 @@ export function registerAskUserTools(pi: ExtensionAPI): void {
         return {
           content: [{ type: "text", text: contentText }],
           details: {
-            question,
-            options: normalizedOptions.map((o) => o.label),
+            ...detailsBase,
             response: {
               ...response,
               prefill: handoff.prefill ?? prefill,
@@ -421,8 +424,7 @@ export function registerAskUserTools(pi: ExtensionAPI): void {
       return {
         content: [{ type: "text", text: contentText }],
         details: {
-          question,
-          options: normalizedOptions.map((o) => o.label),
+          ...detailsBase,
           response,
         },
       };

package/packages/autocomplete/README.md

@@ -0,0 +1,36 @@
+# @pi-unipi/command-enchantment
+
+Enhanced autocomplete for Unipi commands. It wraps Pi's base autocomplete provider and makes `/unipi:*` suggestions easier to scan with package grouping, stable sorting, short descriptions, and package colors.
+
+## Commands
+
+Command Enchantment has no user commands. It improves the editor autocomplete experience automatically when the package is installed.
+
+## What It Does
+
+- Groups `/unipi:*` commands by package so workflow, memory, web, footer, and other commands are visually distinct.
+- Sorts matches in predictable tiers: exact Unipi matches first, then other Unipi matches, then system commands.
+- Preserves dynamic argument completions from command providers, including workflow document and worktree suggestions.
+- Ships an audit test that checks registered Unipi commands are represented in the autocomplete registry and have descriptions.
+
+## Development Checks
+
+Run the registry audit before releases:
+
+```bash
+npm --workspace packages/autocomplete test -- src/__tests__/command-registry.audit.test.ts
+```
+
+Run all autocomplete tests:
+
+```bash
+npm --workspace packages/autocomplete test
+```
+
+## Configuration
+
+Autocomplete enhancement is enabled by default. The package stores its toggle in the Unipi config and can be disabled by setting `autocompleteEnhanced` to `false`.
+
+## License
+
+MIT

package/packages/workflow/README.md

@@ -97,6 +97,8 @@ For small tasks that skip the full flow:
 /unipi:worktree-merge feat/new-feature
 ```
 
+Worktree command arguments autocomplete from `.unipi/worktrees`. Suggestions are cached after the first scan in a Pi session so large worktree directories do not slow down repeated `/unipi:worktree-merge` completions.
+
 ## Special Triggers
 
 Workflow skills detect installed packages and enhance their behavior automatically. This is the coexists system — each package adds capabilities without requiring configuration.

package/packages/workflow/commands.ts

@@ -10,6 +10,8 @@ import { readFileSync, readdirSync, existsSync, statSync } from "fs";
 import { join, basename } from "path";
 import { UNIPI_PREFIX, WORKFLOW_COMMANDS, getToolsForCommand, getSandboxLevel, type SandboxLevel } from "@pi-unipi/core";
 
+type CompletionItem = { value: string; label: string; description: string };
+
 /** Options for command registration */
 export interface WorkflowCommandOptions {
   /** Check if ralph module is detected */
@@ -36,7 +38,7 @@ interface WorkflowCommand {
 /**
  * Suggest spec files from .unipi/docs/specs/ for plan command.
  */
-function suggestSpecFiles(prefix: string): { value: string; label: string; description: string }[] {
+function suggestSpecFiles(prefix: string): CompletionItem[] {
   const specsDir = join(process.cwd(), ".unipi", "docs", "specs");
   if (!existsSync(specsDir)) return [];
 
@@ -61,7 +63,7 @@ function suggestSpecFiles(prefix: string): { value: string; label: string; descr
 /**
  * Suggest plan files from .unipi/docs/plans/ for work and review-work commands.
  */
-function suggestPlanFiles(prefix: string): { value: string; label: string; description: string }[] {
+function suggestPlanFiles(prefix: string): CompletionItem[] {
   const plansDir = join(process.cwd(), ".unipi", "docs", "plans");
   if (!existsSync(plansDir)) return [];
 
@@ -86,7 +88,7 @@ function suggestPlanFiles(prefix: string): { value: string; label: string; descr
 /**
  * Suggest debug files from .unipi/docs/debug/ for fix command.
  */
-function suggestDebugFiles(prefix: string): { value: string; label: string; description: string }[] {
+function suggestDebugFiles(prefix: string): CompletionItem[] {
   const debugDir = join(process.cwd(), ".unipi", "docs", "debug");
   if (!existsSync(debugDir)) return [];
 
@@ -111,7 +113,7 @@ function suggestDebugFiles(prefix: string): { value: string; label: string; desc
 /**
  * Suggest chore files from .unipi/docs/chore/ for chore-execute command.
  */
-function suggestChoreFiles(prefix: string): { value: string; label: string; description: string }[] {
+function suggestChoreFiles(prefix: string): CompletionItem[] {
   const choreDir = join(process.cwd(), ".unipi", "docs", "chore");
   if (!existsSync(choreDir)) return [];
 
@@ -133,16 +135,29 @@ function suggestChoreFiles(prefix: string): { value: string; label: string; desc
   }
 }
 
+/** Cached per-cwd worktree suggestions. Worktree autocomplete can be invoked on every
+ * keystroke, so the recursive scan is intentionally paid only once per session/cwd.
+ */
+let worktreeSuggestionsCache: { cwd: string; items: CompletionItem[] } | null = null;
+
 /**
  * Suggest existing worktree names for merge/list commands.
  * Recursively scans for actual git worktrees (directories containing .git files).
  */
-function suggestWorktrees(): { value: string; label: string; description: string }[] {
-  const worktreesDir = join(process.cwd(), ".unipi", "worktrees");
-  if (!existsSync(worktreesDir)) return [];
+function suggestWorktrees(): CompletionItem[] {
+  const cwd = process.cwd();
+  if (worktreeSuggestionsCache?.cwd === cwd) {
+    return worktreeSuggestionsCache.items;
+  }
+
+  const worktreesDir = join(cwd, ".unipi", "worktrees");
+  if (!existsSync(worktreesDir)) {
+    worktreeSuggestionsCache = { cwd, items: [] };
+    return worktreeSuggestionsCache.items;
+  }
 
   try {
-    const results: { value: string; label: string; description: string }[] = [];
+    const results: CompletionItem[] = [];
 
     /**
      * Recursively find worktree directories (those containing a .git file).
@@ -176,9 +191,11 @@ function suggestWorktrees(): { value: string; label: string; description: string
     }
 
     findWorktrees(worktreesDir, "");
-    return results;
+    worktreeSuggestionsCache = { cwd, items: results };
+    return worktreeSuggestionsCache.items;
   } catch {
-    return [];
+    worktreeSuggestionsCache = { cwd, items: [] };
+    return worktreeSuggestionsCache.items;
   }
 }
 
@@ -334,7 +351,7 @@ export function registerWorkflowCommands(
     pi.registerCommand(fullCommand, {
       description: cmd.description,
       getArgumentCompletions: (prefix: string) => {
-        let items: { value: string; label: string; description: string }[] | null = null;
+        let items: CompletionItem[] | null = null;
 
         // Plan command: suggest spec files
         if (cmd.name === WORKFLOW_COMMANDS.PLAN) {