@vtstech/pi-openrouter-sync 1.1.7 → 1.1.8

package/openrouter-sync.js

@@ -2,7 +2,7 @@
 import {
   MODELS_JSON_PATH as MODELS_FILE,
   readModelsJson,
-  writeModelsJson,
+  readModifyWriteModelsJson,
   BUILTIN_PROVIDERS,
   EXTENSION_VERSION
 } from "@vtstech/pi-shared/ollama";
@@ -37,6 +37,35 @@ function ensureProviderOrder(providers) {
   }
   return ordered;
 }
+async function performSync(modelIds) {
+  let result;
+  await readModifyWriteModelsJson((data) => {
+    if (!data.providers["openrouter"]) {
+      data.providers["openrouter"] = {
+        baseUrl: OR_CONFIG.baseUrl,
+        api: OR_CONFIG.api,
+        models: []
+      };
+    }
+    const orProvider = data.providers["openrouter"];
+    if (!orProvider.models) orProvider.models = [];
+    const existingIds = new Set(orProvider.models.map((m) => m.id));
+    const added = [];
+    const skipped = [];
+    for (const modelId of modelIds) {
+      if (existingIds.has(modelId)) {
+        skipped.push(modelId);
+        continue;
+      }
+      orProvider.models.push({ id: modelId });
+      added.push(modelId);
+    }
+    data.providers = ensureProviderOrder(data.providers);
+    result = { added, skipped, totalModels: orProvider.models.length };
+    return data;
+  });
+  return result;
+}
 function openrouter_sync_temp_default(pi) {
   pi.registerCommand("openrouter-sync", {
     description: "Add OpenRouter model(s) to models.json. Use: /or-sync <url-or-id> [url-or-id ...]",
@@ -51,33 +80,11 @@ function openrouter_sync_temp_default(pi) {
       }
       ctx.ui.setStatus("openrouter-sync", `Adding ${modelIds.length} model(s)...`);
       try {
-        const existing = readModelsJson();
-        if (!existing.providers["openrouter"]) {
-          existing.providers["openrouter"] = {
-            baseUrl: OR_CONFIG.baseUrl,
-            api: OR_CONFIG.api,
-            models: []
-          };
-        }
-        const orProvider = existing.providers["openrouter"];
-        if (!orProvider.models) orProvider.models = [];
-        const existingIds = new Set(orProvider.models.map((m) => m.id));
-        const added = [];
-        const skipped = [];
-        for (const modelId of modelIds) {
-          if (existingIds.has(modelId)) {
-            skipped.push(modelId);
-            continue;
-          }
-          orProvider.models.push({ id: modelId });
-          added.push(modelId);
-        }
-        existing.providers = ensureProviderOrder(existing.providers);
-        writeModelsJson(existing);
+        const { added, skipped, totalModels } = await performSync(modelIds);
         const lines = [""];
         lines.push(`  Provider: openrouter (built-in)`);
         lines.push(`  Base URL: ${OR_CONFIG.baseUrl}`);
-        lines.push(`  Total models: ${orProvider.models.length}`);
+        lines.push(`  Total models: ${totalModels}`);
         if (added.length > 0) {
           lines.push(section("Added"));
           for (const id of added) lines.push(ok(id));
@@ -131,48 +138,35 @@ function openrouter_sync_temp_default(pi) {
         };
       }
       const modelIds = parseModelIds(rawModels.join(" "));
-      const existing = readModelsJson();
-      if (!existing.providers["openrouter"]) {
-        existing.providers["openrouter"] = {
-          baseUrl: OR_CONFIG.baseUrl,
-          api: OR_CONFIG.api,
-          models: []
+      try {
+        const { added, skipped, totalModels } = await performSync(modelIds);
+        const config = readModelsJson();
+        const orProvider = config.providers["openrouter"];
+        const modelList = (orProvider?.models || []).map((m) => `  - ${m.id}`).join("\n");
+        const report = [
+          BRANDING,
+          "",
+          `Added ${added.length} model(s) to openrouter provider (${totalModels} total).`,
+          ...added.length > 0 ? ["\nAdded:"] : [],
+          ...added.map((id) => `  + ${id}`),
+          ...skipped.length > 0 ? ["\nSkipped (already present):"] : [],
+          ...skipped.map((id) => `  = ${id}`),
+          "",
+          `Written to ${MODELS_FILE}. Run /reload to pick up changes.`,
+          "",
+          "Current openrouter models:",
+          modelList
+        ].join("\n");
+        return {
+          content: [{ type: "text", text: report }],
+          details: { added: added.length, skipped: skipped.length, total: totalModels }
+        };
+      } catch (err) {
+        return {
+          content: [{ type: "text", text: `Error: ${err.message}` }],
+          details: {}
         };
       }
-      const orProvider = existing.providers["openrouter"];
-      if (!orProvider.models) orProvider.models = [];
-      const existingIds = new Set(orProvider.models.map((m) => m.id));
-      const added = [];
-      const skipped = [];
-      for (const modelId of modelIds) {
-        if (existingIds.has(modelId)) {
-          skipped.push(modelId);
-          continue;
-        }
-        orProvider.models.push({ id: modelId });
-        added.push(modelId);
-      }
-      existing.providers = ensureProviderOrder(existing.providers);
-      writeModelsJson(existing);
-      const modelList = orProvider.models.map((m) => `  - ${m.id}`).join("\n");
-      const report = [
-        BRANDING,
-        "",
-        `Added ${added.length} model(s) to openrouter provider (${orProvider.models.length} total).`,
-        ...added.length > 0 ? ["\nAdded:"] : [],
-        ...added.map((id) => `  + ${id}`),
-        ...skipped.length > 0 ? ["\nSkipped (already present):"] : [],
-        ...skipped.map((id) => `  = ${id}`),
-        "",
-        `Written to ${MODELS_FILE}. Run /reload to pick up changes.`,
-        "",
-        "Current openrouter models:",
-        modelList
-      ].join("\n");
-      return {
-        content: [{ type: "text", text: report }],
-        details: { added: added.length, skipped: skipped.length, total: orProvider.models.length }
-      };
     }
   });
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vtstech/pi-openrouter-sync",
-  "version": "1.1.7",
+  "version": "1.1.8",
   "description": "OpenRouter model sync extension for Pi Coding Agent — add models from OpenRouter URLs or IDs",
   "main": "openrouter-sync.js",
   "keywords": ["pi-extensions"],
@@ -14,7 +14,7 @@
     "url": "https://github.com/VTSTech/pi-coding-agent"
   },
   "dependencies": {
-    "@vtstech/pi-shared": "1.1.7"
+    "@vtstech/pi-shared": "1.1.8"
   },
   "peerDependencies": {
     "@mariozechner/pi-coding-agent": ">=0.66"