@pi-unipi/btw 2.6.2 → 2.9.0

package/extensions/btw.ts

@@ -26,6 +26,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, type UserMessage } from "@earendil-works/pi-ai";
@@ -161,6 +162,17 @@ function stripDynamicSystemPromptFooter(systemPrompt: string): string {
     .trim();
 }
 
+/**
+ * Extract the canonical ModelRuntime from the extension-facing ModelRegistry
+ * facade. The SDK (>=0.84) exposes `modelRuntime` on AgentSession and expects
+ * it in CreateAgentSessionOptions; the extension context only hands us the
+ * ModelRegistry wrapper, whose `runtime` field is the same instance the
+ * parent session uses (agent-session wires `new ModelRegistry(this._modelRuntime)`).
+ */
+function sessionModelRuntime(ctx: ExtensionCommandContext): ModelRuntime {
+  return (ctx.modelRegistry as unknown as { runtime: ModelRuntime }).runtime;
+}
+
 function createBtwResourceLoader(
   ctx: ExtensionCommandContext,
   appendSystemPrompt: string[] = [BTW_SYSTEM_PROMPT],
@@ -175,7 +187,9 @@ function createBtwResourceLoader(
     getThemes: () => ({ themes: [], diagnostics: [] }),
     getAgentsFiles: () => ({ agentsFiles: [] }),
     getSystemPrompt: () => systemPrompt,
+    getSystemPromptSource: () => undefined,
     getAppendSystemPrompt: () => appendSystemPrompt,
+    getAppendSystemPromptSources: () => [],
     extendResources: () => {},
     reload: async () => {},
   };
@@ -1339,7 +1353,7 @@ export default function (pi: ExtensionAPI) {
     const { session } = await createAgentSession({
       sessionManager: SessionManager.inMemory(),
       model: ctx.model,
-      modelRegistry: ctx.modelRegistry as AgentSession["modelRegistry"],
+      modelRuntime: sessionModelRuntime(ctx),
       thinkingLevel: pi.getThinkingLevel() as SessionThinkingLevel,
       tools: ["read", "bash", "edit", "write"],
       resourceLoader: createBtwResourceLoader(ctx),
@@ -1816,7 +1830,7 @@ export default function (pi: ExtensionAPI) {
     const { session } = await createAgentSession({
       sessionManager: SessionManager.inMemory(),
       model,
-      modelRegistry: ctx.modelRegistry as AgentSession["modelRegistry"],
+      modelRuntime: sessionModelRuntime(ctx),
       thinkingLevel: "off",
       tools: [],
       resourceLoader: createBtwResourceLoader(ctx, [BTW_SUMMARIZE_SYSTEM_PROMPT]),

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@pi-unipi/btw",
-  "version": "2.6.2",
-  "description": "A pi extension for parallel side conversations with /unipi:btw — part of the Unipi suite",
+  "version": "2.9.0",
+  "description": "A pi extension for parallel side conversations with /unipi:btw \u2014 part of the Unipi suite",
   "type": "module",
   "main": "extensions/btw.ts",
   "license": "MIT",
@@ -32,12 +32,12 @@
     "access": "public"
   },
   "peerDependencies": {
-    "@earendil-works/pi-ai": "^0.80.0",
-    "@earendil-works/pi-coding-agent": "^0.80.0",
-    "@earendil-works/pi-tui": "^0.80.0"
+    "@earendil-works/pi-ai": "^0.84.0",
+    "@earendil-works/pi-coding-agent": "^0.84.0",
+    "@earendil-works/pi-tui": "^0.84.0"
   },
   "dependencies": {
-    "@pi-unipi/core": "2.6.1"
+    "@pi-unipi/core": "2.9.0"
   },
   "pi": {
     "extensions": [