pi-neuralwatt-provider 1.3.1 → 1.3.3

package/index.ts

@@ -61,13 +61,12 @@ import type { SimpleStreamOptions } from "@earendil-works/pi-ai";
 import { clampThinkingLevel } from "@earendil-works/pi-ai";
 import { streamOpenAICompletions } from "@earendil-works/pi-ai/compat";
 import type { AssistantMessageEventStream } from "@earendil-works/pi-ai";
-import type { ExtensionAPI, ModelRegistry } from "@earendil-works/pi-coding-agent";
+import { getAgentDir, type ExtensionAPI, type ModelRegistry } from "@earendil-works/pi-coding-agent";
 import modelsData from "./models.json" with { type: "json" };
 import customModelsData from "./custom-models.json" with { type: "json" };
 import patchesData from "./patch.json" with { type: "json" };
 import { transformContextForImageLimit } from "./transform";
 import fs from "fs";
-import os from "os";
 import path from "path";
 
 // ─── Display Configuration ────────────────────────────────────────────────────
@@ -94,7 +93,7 @@ interface ModelOverride {
   vision?: { maxImagesPerRequest?: number };
 }
 
-const CONFIG_PATH = path.join(os.homedir(), ".pi", "agent", "extensions", "neuralwatt.json");
+const CONFIG_PATH = path.join(getAgentDir(), "extensions", "neuralwatt.json");
 
 const VALID_DISPLAY_MODES = new Set<string>(["widget", "statusbar", "off"]);
 
@@ -319,7 +318,7 @@ function buildModels(
 const PROVIDER_ID = "neuralwatt";
 export const BASE_URL = "https://api.neuralwatt.com/v1";
 const MODELS_URL = `${BASE_URL}/models`;
-const CACHE_DIR = path.join(os.homedir(), ".pi", "agent", "cache");
+const CACHE_DIR = path.join(getAgentDir(), "cache");
 const CACHE_PATH = path.join(CACHE_DIR, `${PROVIDER_ID}-models.json`);
 const LIVE_FETCH_TIMEOUT_MS = 8000;
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-neuralwatt-provider",
-  "version": "1.3.1",
+  "version": "1.3.3",
   "description": "Neuralwatt provider extension for pi - Access Kimi, GLM, Qwen, GPT-OSS, Devstral, and MiniMax models through the Neuralwatt API",
   "type": "module",
   "main": "index.ts",

package/tests/__mocks__/pi-coding-agent.ts

@@ -1,5 +1,13 @@
 // Stub for @earendil-works/pi-coding-agent peer dependency
 
+import os from "os";
+import path from "path";
+
+/** Mirrors the real getAgentDir(): respects PI_CODING_AGENT_DIR, defaults to ~/.pi/agent */
+export function getAgentDir(): string {
+  return process.env.PI_CODING_AGENT_DIR || path.join(os.homedir(), ".pi", "agent");
+}
+
 export interface ExtensionAPI {
   registerProvider(_name: string, _provider: any): void;
   on(_event: string, _handler: any): void;

package/tests/config.test.ts

@@ -1,14 +1,13 @@
 import { describe, expect, it, vi, beforeEach, afterEach } from "vitest";
 import fs from "fs";
-import os from "os";
+import { getAgentDir } from "@earendil-works/pi-coding-agent";
 import path from "path";
 
 // We test the config parsing by importing the module's loadConfig indirectly.
 // Since loadConfig reads from the filesystem at import time and on session_start,
 // we test the parseDisplayMode logic and the config file loading path directly.
 
-const CONFIG_DIR = path.join(os.homedir(), ".pi", "agent", "extensions");
-const CONFIG_PATH = path.join(CONFIG_DIR, "neuralwatt.json");
+const CONFIG_PATH = path.join(getAgentDir(), "extensions", "neuralwatt.json");
 
 // Replicate the parsing logic inline to test it without module-level side effects.
 // The real code under test is in index.ts — these unit tests validate the