pi-free 2.0.0 → 2.0.1

package/config.ts

@@ -6,7 +6,7 @@
  *   2. ~/.pi/free.json
  *
  * All exported values are getter functions so that runtime changes
- * (e.g. after /{provider}-toggle or /free) are visible immediately.
+ * (e.g. after toggle-{provider}) are visible immediately.
  */
 
 import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
@@ -112,7 +112,11 @@ function ensureConfigFile(): void {
 export function loadConfigFile(): PiFreeConfig {
 	try {
 		return JSON.parse(readFileSync(CONFIG_PATH, "utf8")) as PiFreeConfig;
-	} catch {
+	} catch (err) {
+		_logger.warn("Could not parse config file — returning empty config", {
+			path: CONFIG_PATH,
+			error: err instanceof Error ? err.message : String(err),
+		});
 		return {};
 	}
 }

package/constants.ts

@@ -23,6 +23,7 @@ export const ALL_UNIQUE_PROVIDERS = [
 	/** @deprecated Qwen free tier no longer available */
 	PROVIDER_QWEN,
 	PROVIDER_MODAL,
+	PROVIDER_OLLAMA,
 ] as const;
 
 // =============================================================================

package/index.ts

@@ -8,11 +8,13 @@
  * - Kilo: OAuth-based free models
  * - Cline: Cline bot integration
  * - NVIDIA: NVIDIA NIM hosting (free tier available)
- * - Qwen: OAuth-based Qwen access
+ * - Ollama Cloud: Ollama's cloud-hosted models with usage-based free tier
+ * - Qwen: OAuth-based Qwen access (deprecated)
  * - Modal: Modal Labs hosting
  */
 
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { setupBuiltInProviderToggles } from "./lib/built-in-toggle.ts";
 import { createLogger } from "./lib/logger.ts";
 import {
 	applyGlobalFilter,
@@ -37,49 +39,6 @@ const _logger = createLogger("pi-free");
 // =============================================================================
 
 function setupGlobalCommands(pi: ExtensionAPI) {
-	// /free - Global toggle for ALL providers
-	pi.registerCommand("free", {
-		description: "Toggle free-only mode for ALL providers (on/off/status)",
-		handler: async (args, ctx) => {
-			const arg = args.trim().toLowerCase();
-			const registry = getProviderRegistry();
-
-			if (arg === "on" || arg === "true" || arg === "yes") {
-				applyGlobalFilter(pi, true);
-				ctx.ui.notify(
-					"✓ Free-only mode enabled - paid models hidden for all providers",
-					"info",
-				);
-			} else if (arg === "off" || arg === "false" || arg === "no") {
-				applyGlobalFilter(pi, false);
-				ctx.ui.notify(
-					"✓ Paid models enabled - all models visible for all providers",
-					"info",
-				);
-			} else if (arg === "status" || arg === "" || !arg) {
-				const available = await ctx.modelRegistry.getAvailable();
-				const freeCount = available.filter(isFreeModel).length;
-				const status = getGlobalFreeOnly() ? "enabled" : "disabled";
-
-				// Count by provider
-				const lines = [
-					`Free-only mode: ${status}`,
-					`${freeCount}/${available.length} models free`,
-					"",
-				];
-				for (const [id, entry] of registry) {
-					const free = entry.stored.free.length;
-					const all = entry.stored.all.length || free;
-					lines.push(`${id}: ${free}/${all} free`);
-				}
-
-				ctx.ui.notify(lines.join("\n"), "info");
-			} else {
-				ctx.ui.notify("Usage: /free [on|off|status]", "warning");
-			}
-		},
-	});
-
 	// /free-providers - Show free model counts by provider
 	pi.registerCommand("free-providers", {
 		description: "Show free/paid model counts for all pi-free providers",
@@ -166,6 +125,9 @@ export default async function (pi: ExtensionAPI) {
 	);
 	await setupDynamicBuiltInProviders(pi);
 
+	// Setup toggles for pi's built-in providers (e.g., OpenCode)
+	setupBuiltInProviderToggles(pi);
+
 	// Apply initial global filter if free-only mode is enabled
 	if (globalFreeOnly) {
 		_logger.info("[pi-free] Applying initial free-only filter");

package/lib/built-in-toggle.ts

@@ -0,0 +1,206 @@
+/**
+ * Built-in Provider Toggle Support
+ *
+ * Captures pi's built-in providers after session start and enables
+ * free/paid toggling for them via the global registry.
+ *
+ * Currently supports:
+ * - opencode (OpenCode / Zen gateway)
+ * - openrouter (OpenRouter)
+ *
+ * Usage: /toggle-opencode
+ */
+
+import type { Api, Model } from "@mariozechner/pi-ai";
+import type {
+	ExtensionAPI,
+	ProviderModelConfig,
+} from "@mariozechner/pi-coding-agent";
+import {
+	getOpencodeShowPaid,
+	getOpenrouterShowPaid,
+	saveConfig,
+} from "../config.ts";
+import { createLogger } from "./logger.ts";
+import {
+	getGlobalFreeOnly,
+	isFreeModel,
+	registerWithGlobalToggle,
+} from "./registry.ts";
+
+const _logger = createLogger("built-in-toggle");
+
+// =============================================================================
+// Configuration
+// =============================================================================
+
+interface BuiltInToggleConfig {
+	id: string;
+	getShowPaid: () => boolean;
+}
+
+const BUILT_IN_TOGGLE_PROVIDERS: BuiltInToggleConfig[] = [
+	{ id: "opencode", getShowPaid: getOpencodeShowPaid },
+	{ id: "openrouter", getShowPaid: getOpenrouterShowPaid },
+];
+
+// =============================================================================
+// State
+// =============================================================================
+
+interface BuiltInProviderState {
+	free: ProviderModelConfig[];
+	all: ProviderModelConfig[];
+	reRegister: (models: ProviderModelConfig[]) => void;
+	showPaid: boolean;
+}
+
+const providerStates = new Map<string, BuiltInProviderState>();
+let commandsRegistered = false;
+
+// =============================================================================
+// Setup
+// =============================================================================
+
+export function setupBuiltInProviderToggles(pi: ExtensionAPI): void {
+	// Register toggle commands once (available even before models load)
+	if (!commandsRegistered) {
+		for (const config of BUILT_IN_TOGGLE_PROVIDERS) {
+			registerToggleCommand(pi, config);
+		}
+		commandsRegistered = true;
+	}
+
+	// Capture built-in models on session start and apply initial filter
+	pi.on("session_start", async (_event, ctx) => {
+		const available = ctx.modelRegistry.getAvailable();
+
+		for (const config of BUILT_IN_TOGGLE_PROVIDERS) {
+			if (providerStates.has(config.id)) {
+				// Already captured this session — skip to avoid re-registering
+				continue;
+			}
+
+			const providerModels = available.filter(
+				(m: Model<Api>) => m.provider === config.id,
+			);
+			if (providerModels.length === 0) continue;
+
+			const allModels = providerModels.map(modelToProviderConfig);
+			const freeModels = allModels.filter(isFreeModel);
+
+			const baseUrl = providerModels[0].baseUrl;
+			const api = providerModels[0].api;
+			const apiKeyEnv = getApiKeyEnvForProvider(config.id);
+
+			const reRegister = (models: ProviderModelConfig[]) => {
+				pi.registerProvider(config.id, {
+					baseUrl,
+					apiKey: apiKeyEnv,
+					api,
+					models,
+				});
+			};
+
+			providerStates.set(config.id, {
+				free: freeModels,
+				all: allModels,
+				reRegister,
+				showPaid: config.getShowPaid(),
+			});
+
+			// Register with global free-only filter
+			registerWithGlobalToggle(
+				config.id,
+				{ free: freeModels, all: allModels },
+				reRegister,
+				true,
+			);
+
+			_logger.info(
+				`[built-in-toggle] ${config.id}: captured ${allModels.length} models (${freeModels.length} free)`,
+			);
+
+			// Respect global free-only setting at capture time
+			if (!getGlobalFreeOnly() && !config.getShowPaid()) {
+				// Default: show free only (same as other pi-free providers)
+				if (freeModels.length > 0) {
+					reRegister(freeModels);
+					_logger.info(
+						`[built-in-toggle] ${config.id}: applied free-only filter`,
+					);
+				}
+			}
+		}
+	});
+}
+
+// =============================================================================
+// Per-provider toggle command
+// =============================================================================
+
+function registerToggleCommand(
+	pi: ExtensionAPI,
+	config: BuiltInToggleConfig,
+): void {
+	const commandName = `toggle-${config.id}`;
+	pi.registerCommand(commandName, {
+		description: `Toggle free/paid ${config.id} models`,
+		handler: async (_args, ctx) => {
+			const state = providerStates.get(config.id);
+			if (!state) {
+				ctx.ui.notify(
+					`${config.id}: models not loaded yet. Start a session first.`,
+					"warning",
+				);
+				return;
+			}
+
+			state.showPaid = !state.showPaid;
+
+			// Persist preference
+			saveConfig({ [`${config.id}_show_paid`]: state.showPaid });
+
+			if (state.showPaid) {
+				state.reRegister(state.all);
+				ctx.ui.notify(
+					`${config.id}: showing all ${state.all.length} models`,
+					"info",
+				);
+			} else {
+				state.reRegister(state.free);
+				ctx.ui.notify(
+					`${config.id}: showing ${state.free.length} free models`,
+					"info",
+				);
+			}
+		},
+	});
+}
+
+// =============================================================================
+// Helpers
+// =============================================================================
+
+function modelToProviderConfig(m: Model<Api>): ProviderModelConfig {
+	return {
+		id: m.id,
+		name: m.name,
+		api: m.api,
+		reasoning: m.reasoning,
+		input: m.input,
+		cost: m.cost,
+		contextWindow: m.contextWindow,
+		maxTokens: m.maxTokens,
+		headers: m.headers,
+		compat: (m as any).compat,
+	};
+}
+
+function getApiKeyEnvForProvider(providerId: string): string {
+	const envMap: Record<string, string> = {
+		opencode: "OPENCODE_API_KEY",
+		openrouter: "OPENROUTER_API_KEY",
+	};
+	return envMap[providerId] || `${providerId.toUpperCase()}_API_KEY`;
+}

package/package.json

@@ -1,67 +1,67 @@
-{
-	"name": "pi-free",
-	"version": "2.0.0",
-	"type": "module",
-	"description": "AIO free models for PI: Kilo, Cline, Nvidia, Ollama Cloud and others",
-	"keywords": [
-		"pi-package",
-		"pi-extension",
-		"free-models",
-		"model-filter",
-		"nvidia-nim",
-		"kilo",
-		"cline",
-		"qwen",
-		"qwen-oauth",
-		"modal"
-	],
-	"license": "MIT",
-	"author": "Apostolos Mantzaris",
-	"homepage": "https://github.com/apmantza/pi-free#readme",
-	"bugs": {
-		"url": "https://github.com/apmantza/pi-free/issues"
-	},
-	"repository": {
-		"type": "git",
-		"url": "git+https://github.com/apmantza/pi-free.git"
-	},
-	"engines": {
-		"node": ">=20.0.0"
-	},
-	"files": [
-		"index.ts",
-		"providers/**/*.ts",
-		"lib/**/*.ts",
-		"provider-failover/**/*.ts",
-		"config.ts",
-		"constants.ts",
-		"provider-factory.ts",
-		"provider-helper.ts",
-		"README.md",
-		"LICENSE",
-		"CHANGELOG.md",
-		"scripts/check-extensions.mjs"
-	],
-	"scripts": {
-		"check": "node scripts/check-extensions.mjs",
-		"test": "vitest",
-		"test:ui": "vitest --ui",
-		"test:run": "vitest run"
-	},
-	"peerDependencies": {
-		"@mariozechner/pi-ai": "*",
-		"@mariozechner/pi-coding-agent": "*",
-		"@mariozechner/pi-tui": "*"
-	},
-	"devDependencies": {
-		"@vitest/ui": "^1.0.0",
-		"tsx": "^4.0.0",
-		"typescript": "^6.0.2",
-		"vitest": "^1.0.0"
-	},
-	"pi": {
-		"extensions": [
-			"./index.ts"
-		]
-	}
-}
+{
+	"name": "pi-free",
+	"version": "2.0.1",
+	"type": "module",
+	"description": "AIO free models for PI: Kilo, Cline, Nvidia, Ollama Cloud and others",
+	"keywords": [
+		"pi-package",
+		"pi-extension",
+		"free-models",
+		"model-filter",
+		"nvidia-nim",
+		"kilo",
+		"cline",
+		"qwen",
+		"qwen-oauth",
+		"modal"
+	],
+	"license": "MIT",
+	"author": "Apostolos Mantzaris",
+	"homepage": "https://github.com/apmantza/pi-free#readme",
+	"bugs": {
+		"url": "https://github.com/apmantza/pi-free/issues"
+	},
+	"repository": {
+		"type": "git",
+		"url": "git+https://github.com/apmantza/pi-free.git"
+	},
+	"engines": {
+		"node": ">=20.0.0"
+	},
+	"files": [
+		"index.ts",
+		"providers/**/*.ts",
+		"lib/**/*.ts",
+		"provider-failover/**/*.ts",
+		"config.ts",
+		"constants.ts",
+		"provider-factory.ts",
+		"provider-helper.ts",
+		"README.md",
+		"LICENSE",
+		"CHANGELOG.md",
+		"scripts/check-extensions.mjs"
+	],
+	"scripts": {
+		"check": "node scripts/check-extensions.mjs",
+		"test": "vitest",
+		"test:ui": "vitest --ui",
+		"test:run": "vitest run"
+	},
+	"peerDependencies": {
+		"@mariozechner/pi-ai": "*",
+		"@mariozechner/pi-coding-agent": "*",
+		"@mariozechner/pi-tui": "*"
+	},
+	"devDependencies": {
+		"@vitest/ui": "^1.0.0",
+		"tsx": "^4.0.0",
+		"typescript": "^6.0.2",
+		"vitest": "^1.0.0"
+	},
+	"pi": {
+		"extensions": [
+			"./index.ts"
+		]
+	}
+}