@vtstech/pi-openrouter-sync 1.2.1 → 1.2.3

package/openrouter-sync.js

@@ -1,12 +1,106 @@
-// .build-npm/openrouter-sync/openrouter-sync.temp.ts
-import {
-  MODELS_JSON_PATH as MODELS_FILE,
-  readModelsJson,
-  readModifyWriteModelsJson,
-  BUILTIN_PROVIDERS,
-  EXTENSION_VERSION
-} from "@vtstech/pi-shared/ollama";
-import { section, ok, warn } from "@vtstech/pi-shared/format";
+// shared/ollama.ts
+import * as fs from "node:fs";
+import * as path from "node:path";
+import os from "node:os";
+
+// shared/debug.ts
+var DEBUG_ENABLED = process?.env?.PI_EXTENSIONS_DEBUG === "1";
+function debugLog(module, message, ...args) {
+  if (!DEBUG_ENABLED) return;
+  const timestamp = (/* @__PURE__ */ new Date()).toISOString();
+  console.debug(`[pi-ext:${module}] ${timestamp} ${message}`, ...args);
+}
+
+// shared/ollama.ts
+var EXTENSION_VERSION = "1.2.3";
+var MODELS_JSON_PATH = path.join(os.homedir(), ".pi", "agent", "models.json");
+var _modelsJsonCache = null;
+var _ollamaBaseUrlCache = null;
+var CACHE_TTL_MS = 2e3;
+function readModelsJson() {
+  const now = Date.now();
+  if (_modelsJsonCache && now - _modelsJsonCache.ts < CACHE_TTL_MS) return _modelsJsonCache.data;
+  try {
+    if (fs.existsSync(MODELS_JSON_PATH)) {
+      const raw = fs.readFileSync(MODELS_JSON_PATH, "utf-8");
+      const data = JSON.parse(raw);
+      _modelsJsonCache = { data, ts: now };
+      return data;
+    }
+  } catch (err) {
+    debugLog("ollama", "failed to read/parse models.json", err);
+  }
+  const empty = { providers: {} };
+  _modelsJsonCache = { data: empty, ts: now };
+  return empty;
+}
+function writeModelsJson(data) {
+  const dir = path.dirname(MODELS_JSON_PATH);
+  if (!fs.existsSync(dir)) {
+    fs.mkdirSync(dir, { recursive: true });
+  }
+  const tmpPath = MODELS_JSON_PATH + ".tmp";
+  fs.writeFileSync(tmpPath, JSON.stringify(data, null, 2) + "\n", "utf-8");
+  fs.renameSync(tmpPath, MODELS_JSON_PATH);
+  _modelsJsonCache = null;
+  _ollamaBaseUrlCache = null;
+}
+var _modelsJsonLock = null;
+async function acquireModelsJsonLock() {
+  while (_modelsJsonLock) {
+    await _modelsJsonLock;
+  }
+  let releaseLock;
+  _modelsJsonLock = new Promise((resolve) => {
+    releaseLock = resolve;
+  });
+  return {
+    release: () => {
+      releaseLock();
+      _modelsJsonLock = null;
+    }
+  };
+}
+async function readModifyWriteModelsJson(modifier) {
+  const { release } = await acquireModelsJsonLock();
+  try {
+    const data = readModelsJson();
+    const modified = modifier(data);
+    if (modified === null) return false;
+    writeModelsJson(modified);
+    return true;
+  } finally {
+    release();
+  }
+}
+var BUILTIN_PROVIDERS = {
+  openrouter: { api: "openai-completions", baseUrl: "https://openrouter.ai/api/v1", envKey: "OPENROUTER_API_KEY" },
+  anthropic: { api: "anthropic-messages", baseUrl: "https://api.anthropic.com/v1", envKey: "ANTHROPIC_API_KEY" },
+  google: { api: "gemini", baseUrl: "https://generativelanguage.googleapis.com", envKey: "GOOGLE_API_KEY" },
+  openai: { api: "openai-completions", baseUrl: "https://api.openai.com/v1", envKey: "OPENAI_API_KEY" },
+  groq: { api: "openai-completions", baseUrl: "https://api.groq.com/v1", envKey: "GROQ_API_KEY" },
+  deepseek: { api: "openai-completions", baseUrl: "https://api.deepseek.com/v1", envKey: "DEEPSEEK_API_KEY" },
+  mistral: { api: "openai-completions", baseUrl: "https://api.mistral.ai/v1", envKey: "MISTRAL_API_KEY" },
+  xai: { api: "openai-completions", baseUrl: "https://api.x.ai/v1", envKey: "XAI_API_KEY" },
+  together: { api: "openai-completions", baseUrl: "https://api.together.xyz/v1", envKey: "TOGETHER_API_KEY" },
+  fireworks: { api: "openai-completions", baseUrl: "https://api.fireworks.ai/inference/v1", envKey: "FIREWORKS_API_KEY" },
+  cohere: { api: "cohere-chat", baseUrl: "https://api.cohere.com/v1", envKey: "COHERE_API_KEY" },
+  zai: { api: "openai-completions", baseUrl: "https://open.bigmodel.cn/api/paas/v4", envKey: "ZAI_API_KEY" }
+};
+
+// shared/format.ts
+function section(title) {
+  return `
+\u2500\u2500 ${title} ${"\u2500".repeat(Math.max(1, 60 - title.length - 4))}`;
+}
+function ok(msg) {
+  return `  \u2705 ${msg}`;
+}
+function warn(msg) {
+  return `  \u26A0\uFE0F  ${msg}`;
+}
+
+// extensions/openrouter-sync.ts
 var BRANDING = [
   `  \u26A1 Pi OpenRouter Sync v${EXTENSION_VERSION}`,
   `  Written by VTSTech`,
@@ -66,7 +160,7 @@ async function performSync(modelIds) {
   });
   return result;
 }
-function openrouter_sync_temp_default(pi) {
+function openrouter_sync_default(pi) {
   pi.registerCommand("openrouter-sync", {
     description: "Add OpenRouter model(s) to models.json. Use: /or-sync <url-or-id> [url-or-id ...]",
     async handler(args, ctx) {
@@ -94,7 +188,7 @@ function openrouter_sync_temp_default(pi) {
           for (const id of skipped) lines.push(warn(id));
         }
         lines.push("");
-        lines.push(`  Written to ${MODELS_FILE}`);
+        lines.push(`  Written to ${MODELS_JSON_PATH}`);
         lines.push(`  Run /reload to pick up changes`);
         lines.push(BRANDING);
         const report = lines.join("\n");
@@ -152,7 +246,7 @@ function openrouter_sync_temp_default(pi) {
           ...skipped.length > 0 ? ["\nSkipped (already present):"] : [],
           ...skipped.map((id) => `  = ${id}`),
           "",
-          `Written to ${MODELS_FILE}. Run /reload to pick up changes.`,
+          `Written to ${MODELS_JSON_PATH}. Run /reload to pick up changes.`,
           "",
           "Current openrouter models:",
           modelList
@@ -171,5 +265,5 @@ function openrouter_sync_temp_default(pi) {
   });
 }
 export {
-  openrouter_sync_temp_default as default
+  openrouter_sync_default as default
 };

package/package.json

@@ -1,9 +1,14 @@
 {
   "name": "@vtstech/pi-openrouter-sync",
-  "version": "1.2.1",
+  "version": "1.2.3",
   "description": "OpenRouter model sync extension for Pi Coding Agent — add models from OpenRouter URLs or IDs",
   "main": "openrouter-sync.js",
-  "keywords": ["pi-package", "pi", "pi-coding-agent", "pi-extensions"],
+  "keywords": [
+    "pi-package",
+    "pi",
+    "pi-coding-agent",
+    "pi-extensions"
+  ],
   "license": "MIT",
   "access": "public",
   "type": "module",
@@ -13,13 +18,12 @@
     "type": "git",
     "url": "https://github.com/VTSTech/pi-coding-agent"
   },
-  "dependencies": {
-    "@vtstech/pi-shared": "1.2.1"
-  },
   "peerDependencies": {
     "@mariozechner/pi-coding-agent": ">=0.66"
   },
   "pi": {
-    "extensions": ["./openrouter-sync.js"]
+    "extensions": [
+      "./openrouter-sync.js"
+    ]
   }
 }