@skydiveai/pi-extensions 0.1.255-beta.17 → 0.1.255-beta.19

package/dist/index.d.mts

@@ -1,6 +1,6 @@
 /// <reference types="node" />
 import { A2AOptions, Middleware, ProtocolsOptions } from "@skydiveai/pi-server";
-import { ExtensionFactory } from "@earendil-works/pi-coding-agent";
+import { AgentSession, ExtensionFactory } from "@earendil-works/pi-coding-agent";
 import { Type } from "typebox";
 import { Client } from "@modelcontextprotocol/sdk/client/index.js";
 import * as _$qs from "qs";
@@ -139,7 +139,7 @@ declare function platformExtensions({
 }): ExtensionFactory[];
 //#endregion
 //#region src/tool-update-loop.d.ts
-type SessionLike = {
+type SessionLike = Pick<AgentSession, 'model' | 'setModel'> & {
   reload: () => Promise<void>;
   bindExtensions: (vars: Record<string, unknown>) => Promise<void>;
   sendCustomMessage: (msg: {
@@ -153,7 +153,7 @@ type SessionLike = {
 };
 type SessionWithAgent = {
   agent: unknown;
-} & SessionLike;
+};
 declare function installToolUpdateAutoStop({
   session,
   log
@@ -163,10 +163,12 @@ declare function installToolUpdateAutoStop({
 }): void;
 declare function runToolUpdateLoop({
   session,
-  log
+  log,
+  signal
 }: {
   session: SessionLike;
   log: Logger;
+  signal: AbortSignal;
 }): Promise<void>;
 //#endregion
 export { type CreateHarnessOptions, all, createHarness, createHealthHandler, createPlatformEnvMiddleware, installToolUpdateAutoStop, isPlatformConfigLoaded, loadPlatformConfig, localToolsExtension, _default as mcpExtension, memoryExtension, platformExtensions, providerApiCompatExtension, runToolUpdateLoop, soulExtension, toolCallEnvExtension };

package/dist/index.mjs

@@ -2223,8 +2223,9 @@ function installToolUpdateAutoStop({ session, log }) {
 		};
 	};
 }
-async function runToolUpdateLoop({ session, log }) {
+async function runToolUpdateLoop({ session, log, signal }) {
 	for (let i = 0; i < MAX_TOOL_UPDATE_CONTINUATIONS; i++) {
+		if (signal.aborted) return;
 		const mcpUpdate = consumePendingMcpUpdate();
 		const localToolsUpdate = consumePendingLocalToolsUpdate();
 		if (!mcpUpdate && !localToolsUpdate) return;
@@ -2248,8 +2249,13 @@ async function runToolUpdateLoop({ session, log }) {
 			} : null,
 			formatted_content: content
 		}, "injecting tool-update continuation prompt");
+		const model = session.model;
 		await session.reload();
+		if (signal.aborted) return;
 		await session.bindExtensions({});
+		if (signal.aborted) return;
+		if (model) await session.setModel(model);
+		if (signal.aborted) return;
 		const customType = mcpUpdate && localToolsUpdate ? "tool-update" : mcpUpdate ? "mcp-tool-update" : "local-tools-update";
 		await session.sendCustomMessage({
 			customType,
@@ -4014,6 +4020,10 @@ function createHarness(options) {
 	const serverOptions = {
 		...options,
 		onSessionSetup: options.onSessionSetup ?? installToolUpdateAutoStop,
+		postPrompt: async (args) => {
+			await runToolUpdateLoop(args);
+			if (!args.signal.aborted) await options.postPrompt?.(args);
+		},
 		onSteer: async (args) => {
 			options.onSteer?.(args);
 			await notifyBashSteerArrival(args);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@skydiveai/pi-extensions",
-  "version": "0.1.255-beta.17",
+  "version": "0.1.255-beta.19",
   "homepage": "https://skydive.com",
   "license": "MIT",
   "author": "Create, Inc.",