@aliou/pi-neuralwatt 0.11.2 → 0.11.3

package/README.md

@@ -8,7 +8,7 @@ A Pi extension that adds [Neuralwatt](https://portal.neuralwatt.com/auth/registe
 
 ### Get API Key
 
-Sign up at [neuralwatt.com](https://portal.neuralwatt.com/auth/register?ref=NW-ALIOU-Q7MF) to get an API key.
+Sign up [here](https://portal.neuralwatt.com/auth/register?ref=NW-ALIOU-Q7MF) to get an API key (referral link).
 
 ### Configure Credentials
 

package/extensions/provider/models/refresh.ts

@@ -5,6 +5,10 @@ import type {
   RefreshModelsContext,
 } from "@earendil-works/pi-ai";
 import type { ProviderModelConfig } from "@earendil-works/pi-coding-agent";
+import {
+  persistModels,
+  readStoredModels,
+} from "../../../src/refresh-store-compat";
 import {
   ALIAS_NEURALWATT_MODEL_IDS,
   buildAliasNeuralwattModels,
@@ -121,11 +125,11 @@ function cachedEarlyAccessModels(
     .map(toProviderModel);
 }
 
-async function persistModels(
+function persistCatalog(
   context: RefreshModelsContext,
   models: ProviderModelConfig[],
-): Promise<void> {
-  await context.store.write({
+): Promise<boolean> {
+  return persistModels(context, {
     models: models.map(toStoredModel),
     checkedAt: Date.now(),
   });
@@ -140,10 +144,10 @@ export async function refreshNeuralwattModels(
     options.includeLegacyModelIds,
     options.includeAliasedModelIds,
   );
-  const stored = await context.store.read();
+  const stored = await readStoredModels(context);
 
   if (!options.includeEarlyAccessModels) {
-    await persistModels(context, baseline);
+    await persistCatalog(context, baseline);
     return baseline;
   }
 
@@ -179,6 +183,7 @@ export async function refreshNeuralwattModels(
     options.includeAliasedModelIds,
     configuredEarlyAccessModels(earlyAccess, baseline),
   );
-  await persistModels(context, catalog);
+  context.signal?.throwIfAborted();
+  await persistCatalog(context, catalog);
   return catalog;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aliou/pi-neuralwatt",
-  "version": "0.11.2",
+  "version": "0.11.3",
   "license": "MIT",
   "type": "module",
   "private": false,
@@ -46,9 +46,9 @@
     "@aliou/biome-plugins": "^0.11.0",
     "@biomejs/biome": "^2.4.15",
     "@changesets/cli": "^2.27.11",
-    "@earendil-works/pi-ai": "0.80.8",
-    "@earendil-works/pi-coding-agent": "0.80.8",
-    "@earendil-works/pi-tui": "0.80.8",
+    "@earendil-works/pi-ai": "0.84.0",
+    "@earendil-works/pi-coding-agent": "0.84.0",
+    "@earendil-works/pi-tui": "0.84.0",
     "@types/node": "^25.0.10",
     "husky": "^9.1.7",
     "ts-json-schema-generator": "^2.4.0",

package/src/refresh-store-compat.ts

@@ -0,0 +1,69 @@
+// ---------------------------------------------------------------------------
+// Backward-compat shim for the Pi coding-agent provider refresh context.
+//
+// Pi 0.84 replaced dynamic `context.store` read/write access with the
+// read-only `context.stored` snapshot and the generation-checked
+// `context.publish({ persist })` transaction. This module detects the
+// available API shape at runtime so the extension works on both <0.84 (store)
+// and >=0.84 (publish) hosts.
+//
+// Once the minimum supported @earendil-works/pi-coding-agent version is
+// >=0.84, delete this file and:
+//   - replace `readStoredModels(context)` with `context.stored`
+//   - replace `persistModels(context, entry)` with
+//     `await context.publish({ persist: entry })` (skip when aborted)
+// ---------------------------------------------------------------------------
+
+import type {
+  ModelsStoreEntry,
+  RefreshModelsContext,
+} from "@earendil-works/pi-ai";
+
+type LegacyRefreshModelsContext = RefreshModelsContext & {
+  store?: {
+    read(): Promise<ModelsStoreEntry | undefined>;
+    write(entry: ModelsStoreEntry): Promise<unknown>;
+  };
+};
+
+/**
+ * Returns the persisted catalog entry for the current provider, reading from
+ * the 0.84+ `context.stored` snapshot when available and falling back to the
+ * legacy `context.store.read()` on older hosts.
+ */
+export async function readStoredModels(
+  context: RefreshModelsContext,
+): Promise<ModelsStoreEntry | undefined> {
+  if (context.stored !== undefined) return context.stored;
+  return readLegacyStore(context);
+}
+
+function readLegacyStore(
+  context: RefreshModelsContext,
+): Promise<ModelsStoreEntry | undefined> {
+  const legacy = context as LegacyRefreshModelsContext;
+  return legacy.store ? legacy.store.read() : Promise.resolve(undefined);
+}
+
+/**
+ * Persists the catalog entry, publishing through
+ * `context.publish({ persist: entry })` on 0.84+ hosts and writing through
+ * the legacy `context.store.write(entry)` on older hosts.
+ *
+ * Returns true when the entry was persisted. On 0.84+ hosts a return value of
+ * false means a newer refresh superseded this publication (generation check).
+ */
+export async function persistModels(
+  context: RefreshModelsContext,
+  entry: ModelsStoreEntry,
+): Promise<boolean> {
+  if (typeof context.publish === "function") {
+    return context.publish({ persist: entry });
+  }
+  const legacy = context as LegacyRefreshModelsContext;
+  if (legacy.store) {
+    await legacy.store.write(entry);
+    return true;
+  }
+  return false;
+}