@vtstech/pi-ollama-sync 1.0.3

package/ollama-sync.js

@@ -0,0 +1,207 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// .build-npm/ollama-sync/ollama-sync.temp.ts
+var ollama_sync_temp_exports = {};
+__export(ollama_sync_temp_exports, {
+  default: () => ollama_sync_temp_default
+});
+module.exports = __toCommonJS(ollama_sync_temp_exports);
+var import_ollama = require("@vtstech/pi-shared/ollama");
+var import_format = require("@vtstech/pi-shared/format");
+var BRANDING = [
+  `  \u26A1 Pi Ollama Sync v1.0.3`,
+  `  Written by VTSTech`,
+  `  GitHub: https://github.com/VTSTech`,
+  `  Website: www.vts-tech.org`
+].join("\n");
+function getProviderConfig(existing) {
+  const defaultUrl = (0, import_ollama.getOllamaBaseUrl)();
+  const ollama = existing.providers["ollama"];
+  return {
+    baseUrl: ollama?.baseUrl ?? defaultUrl + "/v1",
+    api: ollama?.api ?? "openai-completions",
+    apiKey: ollama?.apiKey ?? "ollama",
+    compat: ollama?.compat ?? {
+      supportsDeveloperRole: false,
+      supportsReasoningEffort: false
+    }
+  };
+}
+function buildModelEntry(m) {
+  return {
+    id: m.name,
+    reasoning: (0, import_ollama.isReasoningModel)(m.name),
+    parameterSize: m.details.parameter_size,
+    quantizationLevel: m.details.quantization_level,
+    modelFamily: m.details.family || m.details.families?.[0] || "unknown"
+  };
+}
+function mergeModels(newModels, oldModels) {
+  const oldModelMap = new Map(oldModels.map((m) => [m.id, m]));
+  return newModels.map((m) => {
+    const old = oldModelMap.get(m.id);
+    if (old) {
+      const merged = { ...m };
+      for (const [k, v] of Object.entries(old)) {
+        if (!(k in m)) merged[k] = v;
+      }
+      return merged;
+    }
+    return m;
+  });
+}
+function ollama_sync_temp_default(pi) {
+  pi.registerCommand("ollama-sync", {
+    description: "Sync models from Ollama into models.json. Use: /ollama-sync [url]",
+    getArgumentCompletions: async () => {
+      const url = (0, import_ollama.getOllamaBaseUrl)();
+      return [
+        { value: url, label: url, description: "Default Ollama URL..." }
+      ];
+    },
+    async handler(args, ctx) {
+      const arg = args.trim();
+      const overrideUrl = arg || void 0;
+      ctx.ui.setStatus("ollama-sync", "Fetching models from Ollama...");
+      try {
+        const existing = (0, import_ollama.readModelsJson)();
+        const config = getProviderConfig(existing);
+        const ollamaBaseUrl = overrideUrl ? overrideUrl.replace(/\/v1$/, "").replace(/\/+$/, "") : config.baseUrl?.replace(/\/v1$/, "") ?? (0, import_ollama.getOllamaBaseUrl)();
+        const models = await (0, import_ollama.fetchOllamaModels)(ollamaBaseUrl);
+        if (models.length === 0) {
+          ctx.ui.notify("No models found in Ollama", "info");
+          ctx.ui.setStatus("ollama-sync", void 0);
+          return;
+        }
+        const sorted = [...models].sort((a, b) => a.size - b.size);
+        const newModels = sorted.map(buildModelEntry);
+        const oldIds = new Set(
+          existing.providers["ollama"]?.models?.map((m) => m.id) ?? []
+        );
+        const added = newModels.filter((m) => !oldIds.has(m.id));
+        const removed = [...oldIds].filter((id) => !newModels.some((m) => m.id === id));
+        const mergedModels = mergeModels(
+          newModels,
+          existing.providers["ollama"]?.models ?? []
+        );
+        existing.providers["ollama"] = {
+          ...config,
+          baseUrl: ollamaBaseUrl + "/v1",
+          models: mergedModels
+        };
+        (0, import_ollama.writeModelsJson)(existing);
+        const lines = [""];
+        lines.push(`  Ollama: ${ollamaBaseUrl}`);
+        lines.push(`  Synced ${newModels.length} models from Ollama`);
+        lines.push((0, import_format.section)("Synced Models"));
+        for (const m of newModels) {
+          lines.push((0, import_format.ok)(m.id));
+          lines.push(
+            `       Params: ${m.parameterSize ?? "?"} \xB7 Quant: ${m.quantizationLevel ?? "?"} \xB7 Family: ${m.modelFamily ?? "?"}`
+          );
+        }
+        if (added.length > 0 || removed.length > 0) {
+          lines.push((0, import_format.section)("Changes"));
+          if (added.length > 0) {
+            lines.push((0, import_format.ok)(`Added ${added.length}: ${added.map((m) => m.id).join(", ")}`));
+          }
+          if (removed.length > 0) {
+            lines.push((0, import_format.warn)(`Removed ${removed.length}: ${removed.join(", ")}`));
+          }
+        } else {
+          lines.push((0, import_format.info)("No changes \u2014 already in sync"));
+        }
+        lines.push("");
+        lines.push(`  Written to ${import_ollama.MODELS_JSON_PATH}`);
+        lines.push(`  Run /reload to pick up changes`);
+        lines.push(BRANDING);
+        const report = lines.join("\n");
+        const summary = [`Synced ${newModels.length} models`];
+        if (added.length > 0) summary.push(`+${added.map((m) => m.id).join(", ")}`);
+        if (removed.length > 0) summary.push(`-${removed.join(", ")}`);
+        ctx.ui.notify(summary.join(" \xB7 "), "success");
+        pi.sendMessage({
+          customType: "ollama-sync-report",
+          content: report,
+          display: { type: "content", content: report },
+          details: { timestamp: (/* @__PURE__ */ new Date()).toISOString(), added: added.length, removed: removed.length }
+        });
+      } catch (err) {
+        ctx.ui.notify(`Failed: ${err.message}`, "error");
+      }
+      ctx.ui.setStatus("ollama-sync", void 0);
+    }
+  });
+  pi.registerTool({
+    name: "ollama_sync",
+    label: "Ollama Sync",
+    description: "Sync available models from an Ollama instance into Pi's models.json config file. Supports local or remote Ollama.\n\n" + BRANDING,
+    parameters: {
+      type: "object",
+      properties: {
+        url: {
+          type: "string",
+          description: "Ollama base URL (e.g. http://192.168.1.100:11434). If omitted, uses models.json or OLLAMA_HOST env var."
+        }
+      }
+    },
+    async execute(_toolCallId, params, _signal, _onUpdate, _ctx) {
+      const overrideUrl = params?.url;
+      const existing = (0, import_ollama.readModelsJson)();
+      const config = getProviderConfig(existing);
+      const ollamaBaseUrl = overrideUrl ? overrideUrl.replace(/\/v1$/, "").replace(/\/+$/, "") : config.baseUrl?.replace(/\/v1$/, "") ?? (0, import_ollama.getOllamaBaseUrl)();
+      try {
+        const models = await (0, import_ollama.fetchOllamaModels)(ollamaBaseUrl);
+        const sorted = [...models].sort((a, b) => a.size - b.size);
+        const newModels = sorted.map(buildModelEntry);
+        const mergedModels = mergeModels(
+          newModels,
+          existing.providers["ollama"]?.models ?? []
+        );
+        existing.providers["ollama"] = {
+          ...config,
+          baseUrl: ollamaBaseUrl + "/v1",
+          models: mergedModels
+        };
+        (0, import_ollama.writeModelsJson)(existing);
+        const modelDetails = newModels.map(
+          (m) => `  \u2022 ${m.id} (${m.parameterSize}, ${m.quantizationLevel}, family: ${m.modelFamily})`
+        ).join("\n");
+        return {
+          content: [
+            {
+              type: "text",
+              text: `${BRANDING}
+
+Synced ${newModels.length} models from ${ollamaBaseUrl} to ${import_ollama.MODELS_JSON_PATH}. Run /reload to pick up changes.
+
+${modelDetails}`
+            }
+          ],
+          details: { models: newModels }
+        };
+      } catch (err) {
+        return {
+          content: [{ type: "text", text: `Error: ${err.message}` }],
+          details: {}
+        };
+      }
+    }
+  });
+}

package/package.json

@@ -0,0 +1,24 @@
+{
+  "name": "@vtstech/pi-ollama-sync",
+  "version": "1.0.3",
+  "description": "Ollama model sync extension for Pi Coding Agent",
+  "main": "ollama-sync.js",
+  "keywords": ["pi-package", "pi-extensions"],
+  "license": "MIT",
+  "access": "public",
+  "author": "VTSTech",
+  "homepage": "https://www.vts-tech.org",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/VTSTech/pi-coding-agent"
+  },
+  "dependencies": {
+    "@vtstech/pi-shared": "1.0.3"
+  },
+  "peerDependencies": {
+    "@mariozechner/pi-coding-agent": ">=0.66"
+  },
+  "pi": {
+    "extensions": ["./ollama-sync.js"]
+  }
+}