@vtstech/pi-api 1.2.2 → 1.2.3

package/api.js

@@ -1,7 +1,178 @@
-// .build-npm/api/api.temp.ts
-import { section, ok, info, warn } from "@vtstech/pi-shared/format";
-import { readModelsJson, readModifyWriteModelsJson, getOllamaBaseUrl, BUILTIN_PROVIDERS, EXTENSION_VERSION, isLocalProvider } from "@vtstech/pi-shared/ollama";
-import { SETTINGS_PATH, readSettings, writeSettings } from "@vtstech/pi-shared/config-io";
+// 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}`;
+}
+function info(msg) {
+  return `  \u2139\uFE0F  ${msg}`;
+}
+
+// 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 getOllamaBaseUrl() {
+  const now = Date.now();
+  if (_ollamaBaseUrlCache && now - _ollamaBaseUrlCache.ts < CACHE_TTL_MS) return _ollamaBaseUrlCache.data;
+  try {
+    if (fs.existsSync(MODELS_JSON_PATH)) {
+      const raw = fs.readFileSync(MODELS_JSON_PATH, "utf-8");
+      const config = JSON.parse(raw);
+      const baseUrl = config?.providers?.["ollama"]?.baseUrl;
+      if (baseUrl) {
+        const result = baseUrl.replace(/\/v1\/?$/, "");
+        _ollamaBaseUrlCache = { data: result, ts: now };
+        return result;
+      }
+    }
+  } catch (err) {
+    debugLog("ollama", "failed to parse models.json for base URL", err);
+  }
+  if (process.env.OLLAMA_HOST) {
+    const result = `http://${process.env.OLLAMA_HOST.replace(/^https?:\/\//, "")}`;
+    _ollamaBaseUrlCache = { data: result, ts: now };
+    return result;
+  }
+  const fallback = "http://localhost:11434";
+  _ollamaBaseUrlCache = { data: fallback, ts: now };
+  return fallback;
+}
+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" }
+};
+function isLocalProvider(baseUrl, providerName) {
+  if (providerName === "ollama") return true;
+  const url = baseUrl || "";
+  return url.includes("localhost") || url.includes("127.0.0.1") || url.includes("0.0.0.0");
+}
+
+// shared/config-io.ts
+import * as fs2 from "fs";
+import * as path2 from "path";
+import os2 from "os";
+var PI_AGENT_DIR = path2.join(os2.homedir(), ".pi", "agent");
+function readJsonConfig(filePath, defaultValue = {}) {
+  try {
+    if (fs2.existsSync(filePath)) {
+      return JSON.parse(fs2.readFileSync(filePath, "utf-8"));
+    }
+  } catch (err) {
+    debugLog("config-io", `failed to read config: ${filePath}`, err);
+  }
+  return defaultValue;
+}
+function writeJsonConfig(filePath, data) {
+  const dir = path2.dirname(filePath);
+  if (!fs2.existsSync(dir)) fs2.mkdirSync(dir, { recursive: true });
+  const content = JSON.stringify(data, null, 2) + "\n";
+  const tmpPath = filePath + ".tmp";
+  try {
+    fs2.writeFileSync(tmpPath, content, "utf-8");
+    fs2.renameSync(tmpPath, filePath);
+  } catch {
+    fs2.writeFileSync(filePath, content, "utf-8");
+  }
+}
+var SETTINGS_PATH = path2.join(PI_AGENT_DIR, "settings.json");
+var SECURITY_PATH = path2.join(PI_AGENT_DIR, "security.json");
+var REACT_MODE_PATH = path2.join(PI_AGENT_DIR, "react-mode.json");
+var MODEL_TEST_CONFIG_PATH = path2.join(PI_AGENT_DIR, "model-test-config.json");
+function readSettings() {
+  return readJsonConfig(SETTINGS_PATH);
+}
+function writeSettings(data) {
+  writeJsonConfig(SETTINGS_PATH, data);
+}
+
+// extensions/api.ts
 var API_MODES = {
   "anthropic-messages": "Anthropic Claude API and compatibles",
   "openai-completions": "OpenAI Chat Completions API and compatibles",
@@ -64,7 +235,7 @@ function resolveProvider(config, explicit, currentProvider) {
   if (!provider) return null;
   return { name: local.name, config: provider };
 }
-function api_temp_default(pi) {
+function api_default(pi) {
   const branding = [
     `  \u26A1 Pi API Mode Switcher v${EXTENSION_VERSION}`,
     `  Written by VTSTech`,
@@ -609,5 +780,5 @@ function api_temp_default(pi) {
   });
 }
 export {
-  api_temp_default as default
+  api_default as default
 };

package/package.json

@@ -1,9 +1,14 @@
 {
   "name": "@vtstech/pi-api",
-  "version": "1.2.2",
+  "version": "1.2.3",
   "description": "API Mode Switcher extension for Pi Coding Agent",
   "main": "api.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.2"
-  },
   "peerDependencies": {
     "@mariozechner/pi-coding-agent": ">=0.66"
   },
   "pi": {
-    "extensions": ["./api.js"]
+    "extensions": [
+      "./api.js"
+    ]
   }
 }