@aliou/pi-neuralwatt 0.5.3 → 0.6.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aliou/pi-neuralwatt",
-  "version": "0.5.3",
+  "version": "0.6.0",
   "license": "MIT",
   "type": "module",
   "private": false,
@@ -32,7 +32,7 @@
     "!src/**/*.test.ts"
   ],
   "dependencies": {
-    "@aliou/pi-utils-settings": "^0.15.0",
+    "@aliou/pi-utils-settings": "^0.17.0",
     "@aliou/pi-utils-ui": "^0.4.0"
   },
   "peerDependencies": {

package/schema.json

@@ -5,6 +5,10 @@
     "NeuralwattConfig": {
       "additionalProperties": false,
       "properties": {
+        "includeLegacyModelIds": {
+          "description": "Include legacy Neuralwatt model IDs in the model picker.",
+          "type": "boolean"
+        },
         "quotaCommand": {
           "description": "Show the quota command (/neuralwatt:quota).",
           "type": "boolean"

package/src/config.ts

@@ -28,24 +28,38 @@ export interface NeuralwattConfig {
   quotaWarnings?: boolean;
   /** Show usage in the sub-bar / status bar. */
   subBarIntegration?: boolean;
+  /** Include legacy Neuralwatt model IDs in the model picker. */
+  includeLegacyModelIds?: boolean;
 }
 
 export interface ResolvedNeuralwattConfig {
   quotaCommand: boolean;
   quotaWarnings: boolean;
   subBarIntegration: boolean;
+  includeLegacyModelIds: boolean;
 }
 
 const DEFAULTS: ResolvedNeuralwattConfig = {
   quotaCommand: true,
   quotaWarnings: true,
   subBarIntegration: true,
+  includeLegacyModelIds: false,
 };
 
 export const configLoader = new ConfigLoader<
   NeuralwattConfig,
   ResolvedNeuralwattConfig
->("neuralwatt", DEFAULTS);
+>("neuralwatt", DEFAULTS, {
+  migrations: [
+    {
+      name: "disable-legacy-model-ids-by-default",
+      shouldRun: (config) => config.includeLegacyModelIds === undefined,
+      message:
+        "[neuralwatt] legacy model IDs (ids including the provider and the quantization) are disabled by default. You can enable them with /neuralwatt:settings.",
+      run: (config) => ({ ...config, includeLegacyModelIds: false }),
+    },
+  ],
+});
 
 export const NEURALWATT_CONFIG_UPDATED_EVENT =
   "neuralwatt:config:updated" as const;
@@ -128,6 +142,23 @@ export function registerNeuralwattSettings(
             ),
           ],
         },
+        {
+          label: "Other settings",
+          items: [
+            {
+              id: "includeLegacyModelIds",
+              label: "Legacy model IDs",
+              description:
+                "Include deprecated Neuralwatt model IDs as aliases in the model picker",
+              currentValue:
+                (tabConfig?.includeLegacyModelIds ??
+                resolved.includeLegacyModelIds)
+                  ? "include"
+                  : "ignore",
+              values: ["include", "ignore"],
+            },
+          ],
+        },
       ];
     },
     onSettingChange: (id, newValue, config) => {
@@ -135,6 +166,10 @@ export function registerNeuralwattSettings(
         return null;
       }
 
+      if (id === "includeLegacyModelIds") {
+        return { ...config, includeLegacyModelIds: newValue === "include" };
+      }
+
       const enabled = newValue === "enabled";
       switch (id) {
         case "quotaCommand":

package/src/extensions/provider/index.ts

@@ -2,6 +2,7 @@ import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import {
   configLoader,
   emitConfigUpdated,
+  NEURALWATT_CONFIG_UPDATED_EVENT,
   NEURALWATT_EXTENSIONS_REGISTER_EVENT,
   NEURALWATT_EXTENSIONS_REQUEST_EVENT,
   type NeuralwattFeatureId,
@@ -16,12 +17,14 @@ import {
 } from "../../types/quota-events";
 import { fetchQuotas } from "../../utils/quotas";
 import { normalizeNeuralwattContextOverflowError } from "./context-overflow";
-import { NEURALWATT_MODELS } from "./models";
+import { getNeuralwattModels } from "./models";
 import { buildQuotasFromHeaders, fetchRequestedQuotas } from "./quota-store";
 
 const HEADER_EMIT_THROTTLE_MS = 5_000;
 
 function registerNeuralwattProvider(pi: ExtensionAPI): void {
+  const { includeLegacyModelIds } = configLoader.getConfig();
+
   pi.registerProvider("neuralwatt", {
     baseUrl: "https://api.neuralwatt.com/v1",
     apiKey: "$NEURALWATT_API_KEY",
@@ -31,7 +34,9 @@ function registerNeuralwattProvider(pi: ExtensionAPI): void {
       Referer: "https://pi.dev",
       "X-Title": "npm:@aliou/pi-neuralwatt",
     },
-    models: NEURALWATT_MODELS,
+    models: getNeuralwattModels({
+      includeLegacyModelIds,
+    }),
   });
 }
 
@@ -47,6 +52,10 @@ export default async function (pi: ExtensionAPI) {
     getLoadedFeatures: () => loadedFeatures,
   });
 
+  pi.events.on(NEURALWATT_CONFIG_UPDATED_EVENT, () => {
+    registerNeuralwattProvider(pi);
+  });
+
   let lastHeaderEmitAt = 0;
   let quotaRequestInFlight = false;
 
@@ -98,6 +107,10 @@ export default async function (pi: ExtensionAPI) {
   });
 
   pi.on("session_start", async (_event, ctx) => {
+    for (const message of configLoader.drainMessages()) {
+      ctx.ui.notify(message, "warning");
+    }
+
     loadedFeatures.clear();
     pi.events.emit(NEURALWATT_EXTENSIONS_REQUEST_EVENT, undefined);
     emitConfigUpdated(pi);

package/src/extensions/provider/models.ts

@@ -4,7 +4,7 @@
 
 import type { ProviderModelConfig } from "@earendil-works/pi-coding-agent";
 
-const CANONICAL_NEURALWATT_MODELS: ProviderModelConfig[] = [
+export const NEURALWATT_MODELS: ProviderModelConfig[] = [
   // GLM-5 Fast - ZhipuAI
   {
     id: "glm-5-fast",
@@ -235,6 +235,33 @@ const CANONICAL_NEURALWATT_MODELS: ProviderModelConfig[] = [
       requiresReasoningContentOnAssistantMessages: true,
     },
   },
+  // Kimi K2.7 Code - MoonshotAI
+  {
+    id: "moonshotai/Kimi-K2.7-Code",
+    name: "Kimi K2.7 Code",
+    reasoning: true,
+    input: ["text", "image"],
+    cost: {
+      input: 0.95,
+      output: 4.0,
+      cacheRead: 0,
+      cacheWrite: 0,
+    },
+    contextWindow: 262128,
+    maxTokens: 65536,
+    thinkingLevelMap: {
+      minimal: null,
+      low: null,
+      medium: "medium",
+      high: null,
+      xhigh: null,
+    },
+    compat: {
+      supportsDeveloperRole: false,
+      maxTokensField: "max_tokens",
+      requiresReasoningContentOnAssistantMessages: true,
+    },
+  },
   // Qwen3.6 35B Fast - Qwen
   {
     id: "qwen3.6-35b-fast",
@@ -270,9 +297,7 @@ export const LEGACY_NEURALWATT_MODEL_IDS = new Set<string>(
 const LEGACY_NEURALWATT_MODELS: ProviderModelConfig[] = Object.entries(
   LEGACY_MODEL_ALIAS_MAP,
 ).map(([legacyId, canonicalId]) => {
-  const canonical = CANONICAL_NEURALWATT_MODELS.find(
-    (model) => model.id === canonicalId,
-  );
+  const canonical = NEURALWATT_MODELS.find((model) => model.id === canonicalId);
 
   if (!canonical) {
     throw new Error(`Missing canonical model for legacy alias ${legacyId}`);
@@ -285,7 +310,11 @@ const LEGACY_NEURALWATT_MODELS: ProviderModelConfig[] = Object.entries(
   };
 });
 
-export const NEURALWATT_MODELS: ProviderModelConfig[] = [
-  ...CANONICAL_NEURALWATT_MODELS,
-  ...LEGACY_NEURALWATT_MODELS,
-];
+export function getNeuralwattModels(options?: {
+  includeLegacyModelIds?: boolean;
+}): ProviderModelConfig[] {
+  if (options?.includeLegacyModelIds)
+    return [...NEURALWATT_MODELS, ...LEGACY_NEURALWATT_MODELS];
+
+  return NEURALWATT_MODELS;
+}