@billjr99/pi-openai-compat 1.0.0 → 1.0.2

package/index.ts

@@ -56,36 +56,127 @@ const TEMPLATES: Record<string, {
   displayName: string;
   baseUrl: string;
   keyless: boolean;
-  keyHint?: string;
+  /** If set, only models whose id appears in this list are kept after fetching. */
+  modelFilter?: string[];
+  /** If true, prompt the user to confirm/edit the base URL (like ollama/custom). */
+  promptUrl?: boolean;
 }> = {
   openrouter: {
     displayName: "OpenRouter",
     baseUrl: "https://openrouter.ai/api/v1",
     keyless: false,
-    keyHint: "sk-or-... from openrouter.ai/keys",
   },
   nvidia_nim: {
     displayName: "NVIDIA NIM",
     baseUrl: "https://integrate.api.nvidia.com/v1",
     keyless: false,
-    keyHint: "nvapi-... from build.nvidia.com",
   },
   nous: {
     displayName: "Nous Research Portal",
     baseUrl: "https://inference-api.nousresearch.com/v1",
     keyless: false,
-    keyHint: "Your Nous Portal API key",
+  },
+  google: {
+    displayName: "Google Gemini",
+    baseUrl: "https://generativelanguage.googleapis.com/v1beta/openai",
+    keyless: false,
+    modelFilter: [
+      "gemini-2.5-flash",
+      "gemini-2.5-flash-lite",
+      "gemini-3-flash-preview",
+      "gemini-3.1-flash-lite-preview",
+      "gemini-3.1-pro-preview",
+    ],
+  },
+  cerebras: {
+    displayName: "Cerebras",
+    baseUrl: "https://api.cerebras.ai/v1",
+    keyless: false,
+    modelFilter: ["qwen3-235b"],
+  },
+  github_models: {
+    displayName: "GitHub Models",
+    baseUrl: "https://models.inference.ai.azure.com",
+    keyless: false,
+    modelFilter: ["gpt-4o", "openai/gpt-4.1"],
+  },
+  sambanova: {
+    displayName: "SambaNova",
+    baseUrl: "https://api.sambanova.ai/v1",
+    keyless: false,
+    modelFilter: [
+      "Meta-Llama-3.3-70B-Instruct",
+      "DeepSeek-V3.1",
+      "DeepSeek-V3.2",
+      "gpt-oss-120b",
+      "Llama-4-Maverick-17B-128E-Instruct",
+      "gemma-3-12b-it",
+    ],
+  },
+  mistral: {
+    displayName: "Mistral",
+    baseUrl: "https://api.mistral.ai/v1",
+    keyless: false,
+    modelFilter: [
+      "mistral-large-latest",
+      "mistral-medium-latest",
+      "magistral-medium-latest",
+      "codestral-latest",
+      "devstral-latest",
+    ],
+  },
+  groq: {
+    displayName: "Groq",
+    baseUrl: "https://api.groq.com/openai/v1",
+    keyless: false,
+    modelFilter: [
+      "llama-3.3-70b-versatile",
+      "meta-llama/llama-4-scout-17b-16e-instruct",
+      "openai/gpt-oss-120b",
+      "openai/gpt-oss-20b",
+      "qwen/qwen3-32b",
+      "llama-3.1-8b-instant",
+    ],
+  },
+  cloudflare: {
+    displayName: "Cloudflare Workers AI",
+    baseUrl: "https://api.cloudflare.com/client/v4/accounts/YOUR_ACCOUNT_ID/ai/v1",
+    keyless: false,
+    promptUrl: true,
+    modelFilter: [
+      "@cf/meta/llama-3.3-70b-instruct-fp8-fast",
+      "@cf/meta/llama-4-scout-17b-16e-instruct",
+      "@cf/openai/gpt-oss-120b",
+      "@cf/zai-org/glm-4.7-flash",
+      "@cf/moonshotai/kimi-k2.5",
+      "@cf/moonshotai/kimi-k2.6",
+      "@cf/qwen/qwen3-30b-a3b-fp8",
+      "@cf/deepseek-ai/deepseek-r1-distill-qwen-32b",
+      "@cf/ibm-granite/granite-4.0-h-micro",
+    ],
+  },
+  zhipu: {
+    displayName: "Zhipu (Z.ai / BigModel)",
+    baseUrl: "https://open.bigmodel.cn/api/paas/v4",
+    keyless: false,
+    modelFilter: ["glm-4.5-flash", "glm-4.7-flash"],
+  },
+  cohere: {
+    displayName: "Cohere",
+    baseUrl: "https://api.cohere.com/compatibility/v1",
+    keyless: false,
   },
   ollama: {
     displayName: "Ollama (local, keyless)",
     baseUrl: "http://localhost:11434/v1",
     keyless: true,
+    promptUrl: true,
   },
   custom: {
     displayName: "Custom Endpoint",
     baseUrl: "",
     keyless: false,
-    keyHint: "Bearer token or API key (leave blank if keyless)",
+    promptUrl: true,
   },
 };
 
@@ -266,18 +357,26 @@ export default async function (pi: ExtensionAPI) {
       const key = keys[labels.indexOf(selectedLabel)];
       const tpl = TEMPLATES[key];
 
-      // Step 2 — base URL (prompted for ollama / custom)
+      // Step 2 — base URL
       let baseUrl = tpl.baseUrl;
-      if (key === "ollama" || key === "custom") {
-        const defaultUrl = tpl.baseUrl;
+      if (key === "cloudflare") {
+        // Ask for just the account ID and splice it into the template URL.
         const entered = await ctx.ui.input(
-          "Base URL",
-          key === "ollama"
-            ? `Ollama base URL — press Enter for default (${defaultUrl}):`
-            : "Base URL of your endpoint (e.g. https://api.example.com/v1):",
-          defaultUrl
+          "Account ID",
+          "Your Cloudflare Account ID (find it on the Cloudflare dashboard overview page):",
+          ""
         );
-        if (entered === null) { ctx.ui.notify("Login cancelled.", "info"); return; }
+        if (entered == null) { ctx.ui.notify("Login cancelled.", "info"); return; }
+        const accountId = entered.trim();
+        if (!accountId) { ctx.ui.notify("Account ID cannot be empty.", "error"); return; }
+        baseUrl = tpl.baseUrl.replace("YOUR_ACCOUNT_ID", accountId);
+      } else if (tpl.promptUrl) {
+        const defaultUrl = tpl.baseUrl;
+        const prompt = key === "ollama"
+          ? `Ollama base URL — press Enter for default (${defaultUrl}):`
+          : "Base URL of your endpoint (e.g. https://api.example.com/v1):";
+        const entered = await ctx.ui.input("Base URL", prompt, defaultUrl);
+        if (entered == null) { ctx.ui.notify("Login cancelled.", "info"); return; }
         baseUrl = (entered.trim() || defaultUrl).replace(/\/+$/, "");
         if (!baseUrl) { ctx.ui.notify("Base URL cannot be empty.", "error"); return; }
       }
@@ -287,10 +386,10 @@ export default async function (pi: ExtensionAPI) {
       if (!tpl.keyless) {
         const entered = await ctx.ui.input(
           "API Key",
-          `${tpl.keyHint ?? "Your API key"} — leave blank if keyless:`,
+          "Your API key — leave blank if keyless:",
           ""
         );
-        if (entered === null) { ctx.ui.notify("Login cancelled.", "info"); return; }
+        if (entered == null) { ctx.ui.notify("Login cancelled.", "info"); return; }
         apiKey = entered.trim() || null;
       }
 
@@ -308,6 +407,20 @@ export default async function (pi: ExtensionAPI) {
         return;
       }
 
+      // Apply model filter if the template defines one (keeps only free/known models).
+      if (tpl.modelFilter && tpl.modelFilter.length > 0) {
+        const filterSet = new Set(tpl.modelFilter);
+        const filtered = models.filter((m) => filterSet.has(m.id));
+        if (filtered.length > 0) {
+          models = filtered;
+        } else {
+          ctx.ui.notify(
+            `None of the expected free models were found — registering all ${models.length} model(s) returned by the provider. The provider may have renamed their models.`,
+            "warning"
+          );
+        }
+      }
+
       // Step 5 — save to config and register with pi
       config.providers[key] = {
         displayName: tpl.displayName,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@billjr99/pi-openai-compat",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "pi-coding-agent extension: OpenAI-compatible endpoint support (OpenRouter, NVIDIA NIM, Nous Portal, Ollama, custom)",
   "author": "Bill Mongan <https://github.com/BillJr99>",
   "license": "MIT",