@bacnh85/pi-subagent 0.9.1 → 0.9.2

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # Changelog
 
+## 0.9.2 (2026-07-16)
+
+### Pi SDK compatibility
+
+- Removed use of the deleted `AuthStorage.inMemory()` API so delegated planner and other subagents start on Pi 0.80.10.
+
 ## 0.9.1 (2026-07-16)
 
 ### Activity-aware timeouts

package/extensions/index.ts

@@ -14,17 +14,14 @@
  */
 
 import * as path from "node:path";
-import type { Model } from "@earendil-works/pi-ai";
 import { StringEnum } from "@earendil-works/pi-ai";
 import {
-	AuthStorage,
 	CONFIG_DIR_NAME,
 	DynamicBorder,
 	type ExtensionAPI,
 	type ExtensionContext,
 	getAgentDir,
 	getMarkdownTheme,
-	ModelRegistry,
 	type ThemeColor,
 } from "@earendil-works/pi-coding-agent";
 import { Container, Markdown, SelectList, Spacer, Text } from "@earendil-works/pi-tui";
@@ -463,20 +460,9 @@ export default function (pi: ExtensionAPI) {
 				}
 			}
 
-			// Shared auth/model setup for SDK sessions
-			// ponytail: reuse parent modelRegistry instead of a fresh copy — avoids
-			// internal API casts (storeModelHeaders) and preserves env/headers/OAuth.
-			const authStorage = AuthStorage.inMemory();
 			const modelRegistry = ctx.modelRegistry;
-
-			// Helper: inject parent's API key into child auth storage
-			async function injectApiKey(model: Model<any>): Promise<void> {
-				const auth = await ctx.modelRegistry.getApiKeyAndHeaders(model);
-				if (auth.ok) {
-					if (auth.apiKey) authStorage.setRuntimeApiKey(model.provider, auth.apiKey);
-					// ponytail: headers/env stay on the parent registry — no copy needed.
-				}
-			}
+			const modelRuntime = (modelRegistry as any).runtime;
+			const authStorage = (modelRegistry as any).authStorage;
 
 			// Helper: resolve a safe child working directory.
 			function resolveChildCwd(childCwd: string | undefined): string {
@@ -566,8 +552,6 @@ export default function (pi: ExtensionAPI) {
 				let effectiveTimeoutMs: number | undefined;
 				let safeCwd: string;
 				try {
-					// Inject parent's API key so --api-key and other runtime overrides work
-					await injectApiKey(resolved.model);
 					tools = resolveChildTools(agent.tools, agent.sandbox, isReadOnly);
 					effectiveTimeoutMs = resolveChildTimeout(timeoutMs, params.timeout);
 					safeCwd = resolveChildCwd(cwd);
@@ -599,6 +583,7 @@ export default function (pi: ExtensionAPI) {
 						task,
 						tools,
 						model: resolved.model,
+						modelRuntime,
 						authStorage,
 						modelRegistry,
 						signal: parentSignal,

package/extensions/runner.ts

@@ -17,10 +17,8 @@
 import type { Message, Model } from "@earendil-works/pi-ai";
 import type { AgentMessage } from "@earendil-works/pi-agent-core";
 import {
-	AuthStorage,
 	createAgentSession,
 	createExtensionRuntime,
-	ModelRegistry,
 	type ResourceLoader,
 	SessionManager,
 	SettingsManager,
@@ -87,8 +85,11 @@ export async function runSubAgent(options: {
 	task: string;
 	tools: string[];
 	model: Model<any>;
-	authStorage: AuthStorage;
-	modelRegistry: ModelRegistry;
+	/** Pi 0.80.10's canonical credential/model runtime. */
+	modelRuntime?: unknown;
+	/** Legacy Pi SDK session options retained for 0.80.6 tests and hosts. */
+	authStorage?: unknown;
+	modelRegistry?: unknown;
 	signal?: AbortSignal;
 	agentName?: string;
 	thinkingLevel?: "off" | "minimal" | "low" | "medium" | "high" | "xhigh" | "max";
@@ -98,7 +99,7 @@ export async function runSubAgent(options: {
 	hardTimeoutMs?: number;
 }): Promise<SubAgentResult> {
 	const {
-		cwd, systemPrompt, task, tools, model, authStorage, modelRegistry, signal,
+		cwd, systemPrompt, task, tools, model, modelRuntime, authStorage, modelRegistry, signal,
 		agentName = "subagent", thinkingLevel = "off", onMessage, onProgress,
 		timeoutMs = DEFAULT_INACTIVITY_TIMEOUT_MS, hardTimeoutMs = HARD_TIMEOUT_MS,
 	} = options;
@@ -142,7 +143,12 @@ export async function runSubAgent(options: {
 			result.status = classifyStopReason(result.stopReason, !timedOut, timedOut);
 			return result;
 		}
-		const { session } = await createAgentSession({ cwd, model, thinkingLevel, authStorage, modelRegistry, resourceLoader, tools, sessionManager: SessionManager.inMemory(cwd), settingsManager });
+		const { session } = await createAgentSession({
+			cwd, model, thinkingLevel, resourceLoader, tools, sessionManager: SessionManager.inMemory(cwd), settingsManager,
+			...(modelRuntime ? { modelRuntime } : {}),
+			// Pi 0.80.10 owns credentials in ModelRuntime; older SDKs still accept these.
+			...(authStorage ? { authStorage, modelRegistry } : {}),
+		} as any);
 		let unsubscribe: (() => void) | undefined;
 		let removeAbort: (() => void) | undefined;
 		try {

package/extensions/service.ts

@@ -1,4 +1,4 @@
-import { AuthStorage, type ExtensionContext } from "@earendil-works/pi-coding-agent";
+import { type ExtensionContext } from "@earendil-works/pi-coding-agent";
 import { type AgentConfig, getModelCandidates } from "./agents.ts";
 import { runSubAgent, type SubAgentProgress, type SubAgentResult } from "./runner.ts";
 import { resolveModel } from "./model.ts";
@@ -44,13 +44,9 @@ export async function runNamedAgent(options: {
 	const { model, attempted } = await resolveModel(getModelCandidates(options.agent), options.ctx.model, options.ctx.modelRegistry);
 	if (!model) throw new Error(`No model resolved for agent "${options.agent.name}" (tried: ${attempted.join(", ") || "none"})`);
 
-	const authStorage = AuthStorage.inMemory();
 	const modelRegistry = options.ctx.modelRegistry;
-	const auth = await options.ctx.modelRegistry.getApiKeyAndHeaders(model);
-	if (auth.ok) {
-		if (auth.apiKey) authStorage.setRuntimeApiKey(model.provider, auth.apiKey);
-		// ponytail: env and headers stay on the parent modelRegistry — reuse it directly.
-	}
+	const modelRuntime = (modelRegistry as any).runtime;
+	const authStorage = (modelRegistry as any).authStorage;
 
 	// Security: validate and normalise timeout.
 	const effectiveTimeoutMs = normalizeTimeout({ requested: options.timeout }).timeoutMs;
@@ -83,6 +79,7 @@ export async function runNamedAgent(options: {
 			task: options.task,
 			tools: toolValidation.tools,
 			model,
+			modelRuntime,
 			authStorage,
 			modelRegistry,
 			signal: options.signal,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bacnh85/pi-subagent",
-  "version": "0.9.1",
+  "version": "0.9.2",
   "description": "In-process subagents for Pi with isolated SDK sessions, parallel and chained delegation, and inspectable threads.",
   "type": "module",
   "license": "MIT",