@aliou/pi-ts-aperture 0.0.1 → 0.2.0

package/.github/workflows/publish.yml

@@ -57,14 +57,10 @@ jobs:
 
           let title = 'Version Packages';
           let commit = 'Version Packages';
-          if (releases.length === 1) {
-            const { name, newVersion } = releases[0];
-            title = 'Updating ' + name + ' to version ' + newVersion;
-            commit = name + '@' + newVersion;
-          } else if (releases.length > 1) {
-            const summary = releases.map(r => r.name + '@' + r.newVersion).join(', ');
-            title = 'Updating ' + summary;
-            commit = summary;
+          if (releases.length >= 1) {
+            const version = releases[0].newVersion;
+            title = version;
+            commit = version;
           }
 
           fs.appendFileSync(process.env.GITHUB_OUTPUT, 'title=' + title + '\n');

package/CHANGELOG.md

@@ -0,0 +1,28 @@
+# @aliou/pi-ts-aperture
+
+## 0.2.0
+
+### Minor Changes
+
+- 926f0a9: Improve `/aperture:setup` provider and connectivity flow.
+
+  - Add URL health check during setup (`/v1/models`) before provider selection, with retry/cancel UX.
+  - Build provider choices from Pi's runtime model registry so extension-registered providers (for example `pi-synthetic`) appear in the setup list.
+
+### Patch Changes
+
+- 2263fc2: mark pi SDK peer deps as optional to prevent koffi OOM in Gondolin VMs
+
+## 0.1.0
+
+### Minor Changes
+
+- ebb9556: Initial release. Route Pi LLM providers through Tailscale Aperture.
+
+  - `/aperture:setup` interactive wizard (base URL + provider multi-select)
+  - `/aperture:settings` settings UI for updating configuration
+  - Auto-registers selected providers with Aperture base URL on load
+
+### Patch Changes
+
+- 7388139: Fix providers not taking effect immediately after setup/settings save. Register directly on modelRegistry and re-resolve the active model when it belongs to a reconfigured provider.

package/README.md

@@ -7,7 +7,7 @@ Aperture handles API key injection and request routing server-side. This extensi
 ## Setup
 
 ```bash
-pi install @aliou/pi-ts-aperture
+pi install npm:@aliou/pi-ts-aperture
 ```
 
 Then run the setup wizard:

package/package.json

@@ -1,8 +1,7 @@
 {
   "name": "@aliou/pi-ts-aperture",
   "description": "Route Pi LLM providers through Tailscale Aperture",
-  "version": "0.0.1",
-  "packageManager": "pnpm@10.26.1",
+  "version": "0.2.0",
   "repository": {
     "type": "git",
     "url": "https://github.com/aliou/pi-ts-aperture"
@@ -11,7 +10,8 @@
     "pi-package"
   ],
   "publishConfig": {
-    "access": "public"
+    "access": "public",
+    "provenance": true
   },
   "pi": {
     "extensions": [
@@ -20,21 +20,12 @@
     "video": "https://assets.aliou.me/pi-extensions/demos/pi-ts-aperture.mp4"
   },
   "dependencies": {
-    "@aliou/pi-utils-settings": "^0.3.0"
+    "@aliou/pi-utils-settings": "^0.4.0"
   },
   "peerDependencies": {
     "@mariozechner/pi-ai": ">=0.52.12",
     "@mariozechner/pi-coding-agent": ">=0.52.12"
   },
-  "scripts": {
-    "typecheck": "tsc --noEmit",
-    "lint": "biome check",
-    "format": "biome check --write",
-    "prepare": "husky",
-    "changeset": "changeset",
-    "version": "changeset version",
-    "release": "pnpm changeset publish"
-  },
   "devDependencies": {
     "@biomejs/biome": "^2.3.13",
     "@changesets/cli": "^2.27.11",
@@ -45,10 +36,20 @@
     "husky": "^9.1.7",
     "typescript": "^5.9.3"
   },
-  "pnpm": {
-    "overrides": {
-      "@mariozechner/pi-ai": "$@mariozechner/pi-coding-agent",
-      "@mariozechner/pi-tui": "$@mariozechner/pi-coding-agent"
+  "peerDependenciesMeta": {
+    "@mariozechner/pi-coding-agent": {
+      "optional": true
+    },
+    "@mariozechner/pi-ai": {
+      "optional": true
     }
+  },
+  "scripts": {
+    "typecheck": "tsc --noEmit",
+    "lint": "biome check",
+    "format": "biome check --write",
+    "changeset": "changeset",
+    "version": "changeset version",
+    "release": "pnpm changeset publish"
   }
-}
+}

package/src/commands/settings.ts

@@ -12,14 +12,17 @@ import {
   type SettingsSection,
   setNestedValue,
 } from "@aliou/pi-utils-settings";
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import type {
+  ExtensionAPI,
+  ExtensionContext,
+} from "@mariozechner/pi-coding-agent";
 import { getSettingsListTheme } from "@mariozechner/pi-coding-agent";
 import type { ApertureConfig, ResolvedConfig } from "../config";
 import { configLoader } from "../config";
 
 export function registerApertureSettings(
   pi: ExtensionAPI,
-  onConfigChange: () => void,
+  onConfigChange: (ctx: ExtensionContext) => void,
 ): void {
   registerSettingsCommand<ApertureConfig, ResolvedConfig>(pi, {
     commandName: "aperture:settings",
@@ -89,8 +92,8 @@ export function registerApertureSettings(
       }
       return updated;
     },
-    onSave: () => {
-      onConfigChange();
+    onSave: (ctx) => {
+      onConfigChange(ctx);
     },
   });
 }

package/src/commands/setup.ts

@@ -8,12 +8,15 @@
  */
 
 import { FuzzySelector } from "@aliou/pi-utils-settings";
-import { getProviders } from "@mariozechner/pi-ai";
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import type {
+  ExtensionAPI,
+  ExtensionContext,
+} from "@mariozechner/pi-coding-agent";
 import { getSettingsListTheme } from "@mariozechner/pi-coding-agent";
-import type { Component } from "@mariozechner/pi-tui";
+import type { Component, TUI } from "@mariozechner/pi-tui";
 import { Input, Key, matchesKey } from "@mariozechner/pi-tui";
 import { configLoader } from "../config";
+import { checkApertureHealth } from "../lib/health";
 
 function normalizeUrl(url: string): string {
   let result = url.trim();
@@ -24,6 +27,102 @@ function normalizeUrl(url: string): string {
   return result.replace(/\/v1\/?$/, "").replace(/\/+$/, "");
 }
 
+const SPINNER_FRAMES = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"];
+
+/**
+ * Shows a spinner while verifying the Aperture URL is reachable.
+ * Kicks off the health check on construction and resolves done(boolean)
+ * when the check completes.
+ */
+class HealthCheckSpinner implements Component {
+  private theme: ReturnType<typeof getSettingsListTheme>;
+  private url: string;
+  private tui: TUI;
+  // true = healthy, false = failed (retry), undefined = cancelled
+  private done: (result: boolean | undefined) => void;
+  private frame = 0;
+  private timer: ReturnType<typeof setInterval>;
+  private result: { ok: boolean; error?: string } | null = null;
+
+  constructor(
+    theme: ReturnType<typeof getSettingsListTheme>,
+    url: string,
+    tui: TUI,
+    done: (result: boolean | undefined) => void,
+  ) {
+    this.theme = theme;
+    this.url = url;
+    this.tui = tui;
+    this.done = done;
+
+    // Animate spinner at ~80ms per frame.
+    this.timer = setInterval(() => {
+      this.frame = (this.frame + 1) % SPINNER_FRAMES.length;
+      this.tui.requestRender();
+    }, 80);
+
+    // Fire the check.
+    checkApertureHealth(url).then((res) => {
+      clearInterval(this.timer);
+      this.result = res;
+      this.tui.requestRender();
+
+      // On success, auto-advance after a brief pause.
+      // On failure, wait for user input (see handleInput).
+      if (res.ok) {
+        setTimeout(() => this.done(true), 600);
+      }
+    });
+  }
+
+  render(_width: number): string[] {
+    const lines: string[] = [];
+    lines.push(this.theme.label(" Aperture Setup", true));
+    lines.push("");
+
+    if (!this.result) {
+      const spinner = SPINNER_FRAMES[this.frame];
+      lines.push(
+        this.theme.hint(`  ${spinner} Checking connection to ${this.url}...`),
+      );
+    } else if (this.result.ok) {
+      lines.push(this.theme.hint(`  Connected to ${this.url}`));
+    } else {
+      lines.push(
+        this.theme.hint(`  Could not reach ${this.url}: ${this.result.error}`),
+      );
+      lines.push("");
+      lines.push(
+        this.theme.hint(
+          "  Make sure the URL is correct and you are connected to the tailnet.",
+        ),
+      );
+      lines.push("");
+      lines.push(this.theme.hint("  Enter: try another URL · Esc: cancel"));
+    }
+
+    lines.push("");
+    return lines;
+  }
+
+  invalidate() {}
+
+  handleInput(data: string) {
+    // Only handle input after a failed check.
+    if (!this.result || this.result.ok) return;
+
+    if (matchesKey(data, Key.enter)) {
+      this.done(false); // retry
+    } else if (matchesKey(data, Key.escape)) {
+      this.done(undefined); // cancel wizard
+    }
+  }
+
+  dispose() {
+    clearInterval(this.timer);
+  }
+}
+
 /**
  * Simple input prompt component for the base URL step.
  */
@@ -152,7 +251,7 @@ class ProviderMultiSelect implements Component {
 
 export function registerSetupCommand(
   pi: ExtensionAPI,
-  onConfigChange: () => void,
+  onConfigChange: (ctx: ExtensionContext) => void,
 ): void {
   pi.registerCommand("aperture:setup", {
     description: "Configure Tailscale Aperture integration",
@@ -168,17 +267,39 @@ export function registerSetupCommand(
       const config = configLoader.getConfig();
       const settingsTheme = getSettingsListTheme();
 
-      // Step 1: base URL
-      const baseUrl = await ctx.ui.custom<string | undefined>(
-        (_tui, _theme, _kb, done) => {
-          return new UrlPrompt(settingsTheme, config.baseUrl, done);
-        },
-      );
+      // Step 1: base URL + health check loop.
+      // On failure, loop back to the URL prompt so the user can retry.
+      let baseUrl: string | undefined;
+      while (true) {
+        baseUrl = await ctx.ui.custom<string | undefined>(
+          (_tui, _theme, _kb, done) => {
+            return new UrlPrompt(
+              settingsTheme,
+              baseUrl ?? config.baseUrl,
+              done,
+            );
+          },
+        );
+
+        if (!baseUrl) return;
+        const urlToCheck = baseUrl;
+
+        const result = await ctx.ui.custom<boolean | undefined>(
+          (tui, _theme, _kb, done) => {
+            return new HealthCheckSpinner(settingsTheme, urlToCheck, tui, done);
+          },
+        );
 
-      if (!baseUrl) return;
+        if (result === true) break; // healthy, proceed
+        if (result === undefined) return; // cancelled
+      }
 
       // Step 2: select providers
-      const knownProviders = getProviders();
+      // Use model registry so custom/extension providers are included.
+      const knownProviders = Array.from(
+        new Set(ctx.modelRegistry.getAll().map((model) => model.provider)),
+      ).sort((a, b) => a.localeCompare(b));
+
       const providers = await ctx.ui.custom<string[] | undefined>(
         (_tui, _theme, _kb, done) => {
           return new ProviderMultiSelect(
@@ -194,7 +315,7 @@ export function registerSetupCommand(
 
       // Step 3: save and register
       await configLoader.save("global", { baseUrl, providers });
-      onConfigChange();
+      onConfigChange(ctx);
       ctx.ui.notify(
         `Aperture configured: ${providers.length} provider(s) via ${baseUrl}`,
         "info",

package/src/index.ts

@@ -6,34 +6,72 @@
  * overrides each provider's baseUrl and sets a dummy apiKey.
  */
 
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import type {
+  ExtensionAPI,
+  ExtensionContext,
+} from "@mariozechner/pi-coding-agent";
 import { registerApertureSettings } from "./commands/settings";
 import { registerSetupCommand } from "./commands/setup";
 import { configLoader } from "./config";
 
-function registerProviders(pi: ExtensionAPI): void {
+function getApertureBaseUrl(): string | null {
   const config = configLoader.getConfig();
-  if (!config.baseUrl || config.providers.length === 0) return;
-
-  const baseUrl = config.baseUrl.replace(/\/+$/, "");
-
-  for (const provider of config.providers) {
-    pi.registerProvider(provider, {
-      baseUrl: `${baseUrl}/v1`,
-      apiKey: "-",
-    });
-  }
+  if (!config.baseUrl || config.providers.length === 0) return null;
+  return `${config.baseUrl.replace(/\/+$/, "")}/v1`;
 }
 
 export default async function (pi: ExtensionAPI): Promise<void> {
   await configLoader.load();
 
-  const onConfigChange = () => {
-    // Config is already reloaded by configLoader.save(), just re-register.
-    registerProviders(pi);
+  const config = configLoader.getConfig();
+  let lastRegisteredProviders = [...config.providers];
+
+  // At load time, pi.registerProvider() queue is flushed by the runner.
+  const baseUrl = getApertureBaseUrl();
+  if (baseUrl) {
+    for (const provider of config.providers) {
+      pi.registerProvider(provider, { baseUrl, apiKey: "-" });
+    }
+  }
+
+  const onSetupComplete = (ctx: ExtensionContext) => {
+    const cfg = configLoader.getConfig();
+    const removedProviders = lastRegisteredProviders.filter(
+      (p) => !cfg.providers.includes(p),
+    );
+
+    const url = getApertureBaseUrl();
+    if (url) {
+      for (const provider of cfg.providers) {
+        ctx.modelRegistry.registerProvider(provider, {
+          baseUrl: url,
+          apiKey: "-",
+        });
+      }
+    }
+    lastRegisteredProviders = [...cfg.providers];
+
+    // The active model is a snapshot. If it belongs to a provider we just
+    // reconfigured, re-resolve it so the new baseUrl takes effect.
+    if (ctx.model && cfg.providers.includes(ctx.model.provider)) {
+      const updated = ctx.modelRegistry.find(ctx.model.provider, ctx.model.id);
+      if (updated) {
+        ctx.ui.notify(
+          `[aperture] re-routing ${ctx.model.id} through ${updated.baseUrl}`,
+          "info",
+        );
+        pi.setModel(updated);
+      }
+    }
+
+    if (removedProviders.length > 0) {
+      ctx.ui.notify(
+        `Removed providers (${removedProviders.join(", ")}) will revert after /reload`,
+        "warning",
+      );
+    }
   };
 
-  registerProviders(pi);
-  registerSetupCommand(pi, onConfigChange);
-  registerApertureSettings(pi, onConfigChange);
+  registerSetupCommand(pi, onSetupComplete);
+  registerApertureSettings(pi, onSetupComplete);
 }

package/src/lib/health.ts

@@ -0,0 +1,30 @@
+/**
+ * Health check for the Aperture gateway.
+ *
+ * Hits GET <baseUrl>/v1/models to verify the gateway is reachable.
+ * Uses native fetch (no extra dependencies).
+ */
+
+export interface HealthCheckResult {
+  ok: boolean;
+  error?: string;
+}
+
+export async function checkApertureHealth(
+  baseUrl: string,
+): Promise<HealthCheckResult> {
+  const url = `${baseUrl.replace(/\/+$/, "")}/v1/models`;
+  try {
+    const res = await fetch(url, {
+      method: "GET",
+      signal: AbortSignal.timeout(5000),
+    });
+    if (!res.ok) {
+      return { ok: false, error: `HTTP ${res.status} ${res.statusText}` };
+    }
+    return { ok: true };
+  } catch (e: unknown) {
+    const msg = e instanceof Error ? e.message : String(e);
+    return { ok: false, error: msg };
+  }
+}

package/.envrc

@@ -1,21 +0,0 @@
-# shellcheck disable=2148,2086
-
-# Use `source_up` if part of a monorepo.
-
-use nix
-
-# Load layouts. This basically just sets up PATH and other env vars.
-# Possible values are "node", "bun", and "uv".
-#
-# Options:
-#   --deny bin1,bin2,...  Exclude binaries from local PATH, falling back to system.
-#                         Useful when a local binary shadows a system one you need.
-#
-# Examples:
-#   layout node
-#   layout node --deny pi,prettier
-#   layout uv --deny python
-
-if has node; then
-  layout node --deny pi
-fi

package/.osgrep/lancedb/chunks.lance/_transactions/2-5f641c2a-07a9-45a1-b450-72c7d4d38754.txn

@@ -1 +0,0 @@
-$5f641c2a-07a9-45a1-b450-72c7d4d38754�(&path IN ('tsconfig.json','biome.json')