@maxiaochao/pi-toolkit 0.5.0 → 0.5.2

package/CHANGELOG.md

@@ -1,3 +1,11 @@
+## 2026-09-11 - v0.5.2
+
+- fix btw: side-session and summarize passed a nonexistent `modelRegistry` option to `createAgentSession`, so a fresh ModelRuntime was created that didn't know extension-registered providers (e.g. volcengine-plan) and prompting failed with "No API key found"; now the main session's ModelRuntime is reused via `modelRuntime`
+
+## 2026-09-11 - v0.5.1
+
+- fix pi-worktree: remove now cleans up the branch automatically — merged branches are deleted without prompting, unmerged ones confirm once (one prompt covers both worktree removal and branch deletion), so recreating the same name no longer silently reuses a stale branch
+
 ## 2026-09-11 - v0.5.0
 
 - fix pi-worktree: new worktrees now branch from the current branch instead of a stale origin/main (worktree content no longer misses current work); add start --base <ref> to branch from an explicit ref

package/bin/pi-worktree

@@ -144,12 +144,31 @@ cmd_remove() {
   [[ -d "$dir" ]] || die "worktree not found: $dir"
   # refuse to remove if the branch is not merged into main
   local branch; branch=$(git -C "$dir" branch --show-current)
+  local rm_anyway=0
   if [[ -n "$branch" ]] && ! git merge-base --is-ancestor "$branch" main 2>/dev/null; then
     warn "branch '$branch' does not appear merged into 'main' (or no main branch exists)."
     read -rp "Remove anyway? [y/N] " ans
     [[ "$ans" =~ ^[Yy]$ ]] || die "aborted"
+    rm_anyway=1
   fi
   git worktree remove "$dir" && ok "removed: $dir"
+  # The branch survives `git worktree remove`; recreating the same name would
+  # silently reuse this stale branch. Delete merged branches automatically.
+  if [[ -n "$branch" ]] && git show-ref --verify --quiet "refs/heads/$branch"; then
+    if git branch -d "$branch" 2>/dev/null; then
+      ok "deleted merged branch: $branch"
+    elif [[ "$rm_anyway" == 1 ]]; then
+      git branch -D "$branch" && ok "deleted unmerged branch: $branch (confirmed above)"
+    else
+      warn "branch '$branch' has unmerged commits."
+      read -rp "Delete anyway? [y/N] " ans
+      if [[ "$ans" =~ ^[Yy]$ ]]; then
+        git branch -D "$branch" && ok "deleted branch: $branch"
+      else
+        warn "branch '$branch' kept; 'pi-worktree start $branch' will reuse it as-is"
+      fi
+    fi
+  fi
 }
 
 cd_wt() {

package/extensions/btw.ts

@@ -9,6 +9,7 @@ import {
 	type ExtensionAPI,
 	type ExtensionCommandContext,
 	type ExtensionContext,
+	type ModelRuntime,
 	type ResourceLoader,
 } from "@earendil-works/pi-coding-agent";
 import { type AssistantMessage, type Message, type ThinkingLevel as AiThinkingLevel } from "@earendil-works/pi-ai";
@@ -633,7 +634,7 @@ export default function (pi: ExtensionAPI) {
 		const { session } = await createAgentSession({
 			sessionManager: SessionManager.inMemory(),
 			model: ctx.model,
-			modelRegistry: ctx.modelRegistry as AgentSession["modelRegistry"],
+			modelRuntime: getParentModelRuntime(ctx),
 			thinkingLevel: pi.getThinkingLevel() as SessionThinkingLevel,
 			tools: ["read", "bash", "edit", "write"],
 			resourceLoader: createBtwResourceLoader(ctx),
@@ -809,6 +810,14 @@ export default function (pi: ExtensionAPI) {
 			});
 	}
 
+	function getParentModelRuntime(ctx: ExtensionCommandContext | ExtensionContext): ModelRuntime {
+		// The ModelRegistry facade keeps its runtime in a JS property that is
+		// private at the type level. Reusing it is required: a fresh ModelRuntime
+		// would not know about extension-registered providers (e.g. volcengine-plan),
+		// causing "No API key found" even when auth.json has the credential.
+		return (ctx.modelRegistry as unknown as { runtime: ModelRuntime }).runtime;
+	}
+
 	async function summarizeThread(ctx: ExtensionContext, items: BtwDetails[]): Promise<string> {
 		const model = ctx.model;
 		if (!model) {
@@ -823,7 +832,7 @@ export default function (pi: ExtensionAPI) {
 		const { session } = await createAgentSession({
 			sessionManager: SessionManager.inMemory(),
 			model,
-			modelRegistry: ctx.modelRegistry as AgentSession["modelRegistry"],
+			modelRuntime: getParentModelRuntime(ctx),
 			thinkingLevel: "off",
 			tools: [],
 			resourceLoader: createBtwResourceLoader(ctx, [BTW_SUMMARY_PROMPT]),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@maxiaochao/pi-toolkit",
-  "version": "0.5.0",
+  "version": "0.5.2",
   "description": "Personal pi coding agent toolkit: /todos, /btw, /cache_export extensions, browser automation, and bundled theme/skills.",
   "type": "module",
   "keywords": [