@monotykamary/pi-supervisor 0.5.22 → 0.5.23

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # Changelog
 
+## 0.5.23
+
+- Pin development SDKs to Pi 1.0.0 while retaining host-owned wildcard peers.
+- Exercise actual SDK and bundled host lifecycle, identity, native nested validation/loadouts, reload and shutdown offline.
+- Fix real child sessions losing parent-native providers: forward through the public model registry, preserve cwd, scope reuse to the parent registry, and honor pre-aborted/active cancellation.
+
 ## 0.5.22
 
 - Validate against Pi 0.99.0, including an offline real-host package-loading probe.

package/README.md

@@ -11,6 +11,10 @@ _Observe every turn, steer when the agent drifts, signal when the goal is reache
 
 </div>
 
+## Pi 1.0 compatibility (0.5.23)
+
+Development SDKs are pinned to **1.0.0**; host-provided dependencies remain wildcard peers. Run `bun run test:pi` for offline real-host registrations, prompt/tool loadouts, nested calls, reload and shutdown. Set `PI1_HOST_PACKAGE` to an installed Pi package directory and `PI1_HOST_ENTRY=bundle` to check its bundled runtime.
+
 ---
 
 > A supervisor as the intelligent overseer keeping the agent on track.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@monotykamary/pi-supervisor",
-  "version": "0.5.22",
+  "version": "0.5.23",
   "description": "A pi extension that supervises the chat and steers it towards a defined outcome",
   "author": "tintinweb (forked by monotykamary)",
   "license": "MIT",
@@ -41,7 +41,7 @@
   "devDependencies": {
     "@commitlint/cli": "21.0.1",
     "@commitlint/config-conventional": "21.0.1",
-    "@earendil-works/pi-tui": "0.99.0",
+    "@earendil-works/pi-tui": "1.0.0",
     "@types/node": "25.9.1",
     "knip": "6.14.1",
     "lint-staged": "17.0.5",
@@ -50,8 +50,8 @@
     "standard-version": "9.5.0",
     "typescript": "6.0.3",
     "vitest": "4.1.7",
-    "@earendil-works/pi-coding-agent": "0.99.0",
-    "@earendil-works/pi-ai": "0.99.0",
+    "@earendil-works/pi-coding-agent": "1.0.0",
+    "@earendil-works/pi-ai": "1.0.0",
     "typebox": "1.3.27"
   },
   "scripts": {
@@ -88,8 +88,8 @@
     "fast-xml-builder": "1.2.0",
     "protobufjs": "8.4.0",
     "ws": "8.20.1",
-    "@earendil-works/pi-ai": "0.99.0",
-    "@earendil-works/pi-coding-agent": "0.99.0",
-    "@earendil-works/pi-tui": "0.99.0"
+    "@earendil-works/pi-ai": "1.0.0",
+    "@earendil-works/pi-coding-agent": "1.0.0",
+    "@earendil-works/pi-tui": "1.0.0"
   }
 }

package/src/session/supervisor-session.ts

@@ -15,6 +15,8 @@ export class SupervisorSession {
   private session: Awaited<ReturnType<typeof createAgentSession>>['session'] | null = null;
   private model: any = null;
   private systemPrompt: string = '';
+  private parentRegistry: ExtensionContext['modelRegistry'] | null = null;
+  private cwd: string | null = null;
 
   async ensureStarted(
     ctx: ExtensionContext,
@@ -26,7 +28,13 @@ export class SupervisorSession {
     const newModel = ctx.modelRegistry.find(provider, modelId);
     if (!newModel) return false;
 
-    if (this.session && this.model === newModel && this.systemPrompt === systemPrompt) {
+    if (
+      this.session &&
+      this.model === newModel &&
+      this.systemPrompt === systemPrompt &&
+      this.parentRegistry === ctx.modelRegistry &&
+      this.cwd === ctx.cwd
+    ) {
       // Session reusable
       return true;
     }
@@ -42,12 +50,45 @@ export class SupervisorSession {
       noPromptTemplates: true,
       noThemes: true,
       systemPromptOverride: () => systemPrompt,
+      noContextFiles: true,
+      appendSystemPromptOverride: () => [],
+      // A fresh SDK session otherwise loses extension/native providers owned
+      // by the parent runtime. Delegate through the public model registry so
+      // parent auth, environment and request hooks remain authoritative.
+      extensionFactories: [
+        (pi) => {
+          pi.registerProvider(newModel.provider, {
+            api: newModel.api,
+            baseUrl: newModel.baseUrl,
+            apiKey: 'supervisor-parent-runtime',
+            models: [
+              {
+                id: newModel.id,
+                name: newModel.name,
+                api: newModel.api,
+                reasoning: newModel.reasoning,
+                input: newModel.input,
+                cost: newModel.cost,
+                contextWindow: newModel.contextWindow,
+                maxTokens: newModel.maxTokens,
+              },
+            ],
+            streamSimple: (_model, context, options) => {
+              // The child credential is a local admission token, never provider
+              // authentication. Let the parent's runtime resolve real auth.
+              const { apiKey: _childKey, ...forwarded } = options ?? {};
+              return ctx.modelRegistry.streamSimple(newModel, context, forwarded);
+            },
+          });
+        },
+      ],
     });
     await loader.reload();
 
     try {
       const result = await createAgentSession({
-        sessionManager: SessionManager.inMemory(),
+        cwd: ctx.cwd,
+        sessionManager: SessionManager.inMemory(ctx.cwd),
         agentDir: getAgentDir(),
         model: newModel,
         tools: [],
@@ -56,6 +97,8 @@ export class SupervisorSession {
       this.session = result.session;
       this.model = newModel;
       this.systemPrompt = systemPrompt;
+      this.parentRegistry = ctx.modelRegistry;
+      this.cwd = ctx.cwd;
       return true;
     } catch {
       return false;
@@ -67,7 +110,7 @@ export class SupervisorSession {
     signal?: AbortSignal,
     onDelta?: (accumulated: string) => void
   ): Promise<string | null> {
-    if (!this.session) return null;
+    if (!this.session || signal?.aborted) return null;
 
     const onAbort = () => this.session?.abort();
     signal?.addEventListener('abort', onAbort, { once: true });
@@ -98,5 +141,7 @@ export class SupervisorSession {
       this.session = null;
     }
     this.model = null;
+    this.parentRegistry = null;
+    this.cwd = null;
   }
 }

package/tests/native-session.test.ts

@@ -0,0 +1,117 @@
+import { expect, it } from 'vitest';
+import { mkdtempSync, rmSync } from 'node:fs';
+import { tmpdir } from 'node:os';
+import { join } from 'node:path';
+import {
+  ModelRegistry,
+  ModelRuntime,
+  type ExtensionContext,
+} from '@earendil-works/pi-coding-agent';
+import {
+  fauxAssistantMessage,
+  fauxProvider,
+  getCurrentSystemPrompt,
+  getCurrentTools,
+} from '@earendil-works/pi-ai/compat';
+import { SupervisorSession } from '../src/session/supervisor-session.js';
+
+it('inherits the parent native provider runtime for real SDK supervisor sessions', async () => {
+  const root = mkdtempSync(join(tmpdir(), 'pi1-supervisor-'));
+  const previous = process.env.PI_CODING_AGENT_DIR;
+  const previousFetch = globalThis.fetch;
+  globalThis.fetch = async () => new Response('', { status: 503 });
+  process.env.PI_CODING_AGENT_DIR = root;
+  const supervisor = new SupervisorSession();
+  try {
+    const runtime = await ModelRuntime.create({
+      authPath: join(root, 'auth.json'),
+      modelsPath: null,
+      refreshOnCreate: false,
+    });
+    const faux = fauxProvider({
+      provider: 'supervisor-offline',
+      api: 'supervisor-offline-api',
+      models: [{ id: 'test', name: 'Offline', reasoning: false }],
+      tokenSize: { min: 100, max: 100 },
+    });
+    runtime.registerNativeProvider(faux.provider);
+    await runtime.refresh({ allowNetwork: false });
+    const ctx = { cwd: root, modelRegistry: new ModelRegistry(runtime) } as ExtensionContext;
+    faux.setResponses([
+      (context, options) => {
+        expect(options?.apiKey).not.toBe('supervisor-parent-runtime');
+        expect(getCurrentSystemPrompt(context.messages)).toContain(
+          'Analyze only the offline goal.'
+        );
+        expect(getCurrentSystemPrompt(context.messages)).toContain(root);
+        expect(getCurrentTools(context.messages)).toEqual([]);
+        return fauxAssistantMessage('local supervisor decision');
+      },
+      fauxAssistantMessage('second decision'),
+    ]);
+    expect(
+      await supervisor.ensureStarted(
+        ctx,
+        'supervisor-offline',
+        'test',
+        'Analyze only the offline goal.'
+      )
+    ).toBe(true);
+    expect(await supervisor.prompt('first analysis')).toBe('local supervisor decision');
+    expect(
+      await supervisor.ensureStarted(
+        ctx,
+        'supervisor-offline',
+        'test',
+        'Analyze only the offline goal.'
+      )
+    ).toBe(true);
+    expect(await supervisor.prompt('second analysis')).toBe('second decision');
+    expect(faux.state.callCount).toBe(2);
+    faux.setResponses([
+      (context) => {
+        expect(JSON.stringify(context.messages)).not.toContain('first analysis');
+        return fauxAssistantMessage('fresh parent decision');
+      },
+    ]);
+    const replacement = { ...ctx, modelRegistry: new ModelRegistry(runtime) };
+    expect(
+      await supervisor.ensureStarted(
+        replacement,
+        'supervisor-offline',
+        'test',
+        'Analyze only the offline goal.'
+      )
+    ).toBe(true);
+    expect(await supervisor.prompt('fresh analysis')).toBe('fresh parent decision');
+    expect(faux.state.callCount).toBe(3);
+    const alreadyAborted = new AbortController();
+    alreadyAborted.abort();
+    expect(await supervisor.prompt('must not start', alreadyAborted.signal)).toBeNull();
+    expect(faux.state.callCount).toBe(3);
+    let enter!: () => void;
+    const entered = new Promise<void>((resolve) => {
+      enter = resolve;
+    });
+    faux.setResponses([
+      async (_context, options) => {
+        enter();
+        await new Promise<void>((resolve) => {
+          options!.signal!.addEventListener('abort', () => resolve(), { once: true });
+        });
+        return fauxAssistantMessage('must not finish normally');
+      },
+    ]);
+    const cancellation = new AbortController();
+    const cancelled = supervisor.prompt('cancel local analysis', cancellation.signal);
+    await entered;
+    cancellation.abort();
+    expect(await cancelled).not.toBe('must not finish normally');
+  } finally {
+    supervisor.dispose();
+    globalThis.fetch = previousFetch;
+    if (previous === undefined) delete process.env.PI_CODING_AGENT_DIR;
+    else process.env.PI_CODING_AGENT_DIR = previous;
+    rmSync(root, { recursive: true, force: true });
+  }
+});