pi-free 1.0.9 → 2.0.1

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/lib/json-persistence.ts

@@ -5,6 +5,9 @@
 
 import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
 import { dirname } from "node:path";
+import { createLogger } from "./logger.ts";
+
+const _logger = createLogger("json-persistence");
 
 export interface JSONStore<T> {
 	load(): T;
@@ -28,8 +31,7 @@ export function createJSONStore<T extends object>(
 				return cached;
 			}
 		} catch (err) {
-			// Silently fail and return default
-			void err;
+			_logger.warn("Failed to load JSON store, using default", { filepath, error: err });
 		}
 		cached = defaultValue;
 		return cached;
@@ -44,8 +46,7 @@ export function createJSONStore<T extends object>(
 			}
 			writeFileSync(filepath, `${JSON.stringify(data, null, 2)}\n`, "utf-8");
 		} catch (err) {
-			// Silently fail - persistence is best-effort
-			void err;
+			_logger.warn("Failed to save JSON store", { filepath, error: err });
 		}
 	}
 
@@ -71,8 +72,8 @@ export function createJSONLStore<T extends object>(
 					.filter((line) => line.trim())
 					.map((line) => JSON.parse(line) as T);
 			}
-		} catch {
-			// Return empty on error
+		} catch (err) {
+			_logger.warn("Failed to load JSONL store, using empty array", { filepath, error: err });
 		}
 		return [];
 	}
@@ -85,16 +86,16 @@ export function createJSONLStore<T extends object>(
 			}
 			const line = JSON.stringify(entry);
 			writeFileSync(filepath, `${line}\n`, { flag: "a", encoding: "utf-8" });
-		} catch {
-			// Silently fail
+		} catch (err) {
+			_logger.warn("Failed to append to JSONL store", { filepath, error: err });
 		}
 	}
 
 	function clear(): void {
 		try {
 			writeFileSync(filepath, "", "utf-8");
-		} catch {
-			// Silently fail
+		} catch (err) {
+			_logger.warn("Failed to clear JSONL store", { filepath, error: err });
 		}
 	}
 

package/lib/logger.ts

@@ -63,8 +63,8 @@ function appendToFile(line: string): void {
 			mkdirSync(dir, { recursive: true });
 		}
 		appendFileSync(LOG_PATH, `${line}\n`, "utf8");
-	} catch {
-		// Never throw from logger
+	} catch (err) {
+		console.error("Failed to write to log file:", err);
 	}
 }
 

package/lib/model-enhancer.ts

@@ -1,20 +1,20 @@
-/**
- * Model name enhancement helper
- * Adds Coding Index scores to model names for display in /model
- */
-
-import type { ProviderModelConfig } from "@mariozechner/pi-coding-agent";
-import { enhanceModelNameWithCodingIndex } from "../provider-failover/hardcoded-benchmarks.ts";
-
-/**
- * Enhance model names with Coding Index scores
- * Use this before registering providers to show CI in /model list
- */
-export function enhanceModelsWithCodingIndex(
-	models: ProviderModelConfig[],
-): ProviderModelConfig[] {
-	return models.map((m) => ({
-		...m,
-		name: enhanceModelNameWithCodingIndex(m.name, m.id),
-	}));
-}
+/**
+ * Model name enhancement helper
+ * Adds Coding Index scores to model names for display in /model
+ */
+
+import type { ProviderModelConfig } from "@mariozechner/pi-coding-agent";
+import { enhanceModelNameWithCodingIndex } from "../provider-failover/benchmark-lookup.ts";
+
+/**
+ * Enhance model names with Coding Index scores
+ * Use this before registering providers to show CI in /model list
+ */
+export function enhanceModelsWithCodingIndex(
+	models: ProviderModelConfig[],
+): ProviderModelConfig[] {
+	return models.map((m) => ({
+		...m,
+		name: enhanceModelNameWithCodingIndex(m.name, m.id),
+	}));
+}

package/lib/open-browser.ts

@@ -0,0 +1,41 @@
+/**
+ * Cross-platform browser opener
+ *
+ * Opens a URL in the user's default browser. Handles URL-unsafe characters
+ * on Windows by using PowerShell's Start-Process instead of cmd.exe.
+ */
+
+import { spawn } from "node:child_process";
+
+/**
+ * Open a URL in the user's default browser.
+ *
+ * - Windows: uses PowerShell Start-Process (cmd.exe interprets & as command separator)
+ * - macOS: uses `open`
+ * - Linux/BSD: uses `xdg-open`
+ */
+export function openBrowser(url: string): void {
+	try {
+		if (process.platform === "win32") {
+			// PowerShell's Start-Process treats the URL as a literal string,
+			// unlike cmd.exe which interprets & as a command separator.
+			spawn(
+				"powershell.exe",
+				[
+					"-NoProfile",
+					"-NonInteractive",
+					"-Command",
+					`Start-Process "${url.replace(/"/g, '\\"')}"`,
+				],
+				{ detached: true, shell: false, windowsHide: true },
+			).unref();
+		} else if (process.platform === "darwin") {
+			spawn("open", [url], { detached: true }).unref();
+		} else {
+			spawn("xdg-open", [url], { detached: true }).unref();
+		}
+	} catch (err) {
+		// Best-effort — browser opening is non-critical
+		console.debug("Failed to open browser:", err);
+	}
+}

package/lib/provider-cache.ts

@@ -0,0 +1,106 @@
+/**
+ * Provider Model Cache
+ *
+ * Caches provider model lists to disk for faster startup and offline use.
+ *
+ * Flow:
+ * 1. On session_start: fetch fresh models from API, save to cache
+ * 2. On extension load: register cached models immediately (shows in --list-models)
+ * 3. If API fails: use cached models as fallback
+ */
+
+import { homedir } from "node:os";
+import { join } from "node:path";
+import { createJSONStore } from "./json-persistence.ts";
+import { createLogger } from "./logger.ts";
+import type { ProviderModelConfig } from "./types.ts";
+
+const _logger = createLogger("provider-cache");
+
+// =============================================================================
+// Types
+// =============================================================================
+
+export interface CachedProviderModels {
+	/** Provider ID */
+	provider: string;
+	/** Cached model list */
+	models: ProviderModelConfig[];
+	/** When these models were fetched */
+	fetchedAt: string; // ISO timestamp
+}
+
+interface CacheData {
+	providers: Record<string, CachedProviderModels>;
+}
+
+// =============================================================================
+// Cache Store
+// =============================================================================
+
+const CACHE_FILE = join(homedir(), ".pi", "provider-cache.json");
+
+const _cache = createJSONStore<CacheData>(CACHE_FILE, { providers: {} });
+
+/**
+ * Load cached models for a provider.
+ * Returns undefined if no cache exists.
+ */
+export function loadProviderCache(
+	providerId: string,
+): ProviderModelConfig[] | undefined {
+	const data = _cache.load();
+	const cached = data.providers[providerId];
+
+	if (!cached) {
+		return undefined;
+	}
+
+	_logger.debug(`Loaded cached models for ${providerId}`, {
+		count: cached.models.length,
+		fetchedAt: cached.fetchedAt,
+	});
+
+	return cached.models;
+}
+
+/**
+ * Save models to cache for a provider.
+ */
+export function saveProviderCache(
+	providerId: string,
+	models: ProviderModelConfig[],
+): void {
+	const data = _cache.load();
+
+	data.providers[providerId] = {
+		provider: providerId,
+		models,
+		fetchedAt: new Date().toISOString(),
+	};
+
+	_cache.save(data);
+
+	_logger.debug(`Saved ${models.length} models to cache for ${providerId}`);
+}
+
+/**
+ * Clear cached models for a provider.
+ */
+export function clearProviderCache(providerId: string): void {
+	const data = _cache.load();
+
+	if (data.providers[providerId]) {
+		delete data.providers[providerId];
+		_cache.save(data);
+		_logger.debug(`Cleared cache for ${providerId}`);
+	}
+}
+
+/**
+ * Clear all provider caches.
+ */
+export function clearAllProviderCaches(): void {
+	_cache.save({ providers: {} });
+	_logger.debug("Cleared all provider caches");
+}

package/lib/registry.ts

@@ -0,0 +1,144 @@
+/**
+ * Global Provider Registry for pi-free.
+ *
+ * Decoupled from index.ts so providers can import toggle logic
+ * without creating a circular dependency.
+ */
+
+import type {
+	ExtensionAPI,
+	ProviderModelConfig,
+} from "@mariozechner/pi-coding-agent";
+import { getFreeOnly, saveConfig } from "../config.ts";
+import { createLogger } from "./logger.ts";
+
+const _logger = createLogger("pi-free");
+
+// =============================================================================
+// Types
+// =============================================================================
+
+interface ProviderEntry {
+	id: string;
+	stored: { free: ProviderModelConfig[]; all: ProviderModelConfig[] };
+	reRegister: (models: ProviderModelConfig[]) => void;
+	hasKey: boolean;
+}
+
+// =============================================================================
+// State
+// =============================================================================
+
+const providerRegistry = new Map<string, ProviderEntry>();
+let globalFreeOnly = getFreeOnly();
+
+// Providers that expose actual per-model pricing via API
+const PRICING_EXPOSED_PROVIDERS = new Set([
+	"openrouter",
+	"opencode",
+	"kilo",
+	"cline",
+]);
+
+// =============================================================================
+// Free-model detection
+// =============================================================================
+
+/**
+ * Check if a model is free.
+ *
+ * For providers with pricing APIs: uses cost (input === 0 && output === 0)
+ * For providers without pricing: ONLY uses name-based check (name includes "free")
+ */
+export function isFreeModel(
+	model: ProviderModelConfig & { provider?: string },
+): boolean {
+	const provider = model.provider;
+	const hasPricing = provider && PRICING_EXPOSED_PROVIDERS.has(provider);
+
+	// For providers WITH pricing API: cost-based check
+	if (hasPricing) {
+		if ((model.cost?.input ?? 0) === 0 && (model.cost?.output ?? 0) === 0) {
+			return true;
+		}
+	}
+
+	// For providers WITHOUT pricing API: ONLY name-based check
+	if (model.name.toLowerCase().includes("free")) {
+		return true;
+	}
+
+	return false;
+}
+
+// =============================================================================
+// Registration
+// =============================================================================
+
+/** Register a provider with the global free/paid toggle system */
+export function registerWithGlobalToggle(
+	providerId: string,
+	stored: { free: ProviderModelConfig[]; all: ProviderModelConfig[] },
+	reRegister: (models: ProviderModelConfig[]) => void,
+	hasKey: boolean = false,
+): void {
+	providerRegistry.set(providerId, {
+		id: providerId,
+		stored,
+		reRegister,
+		hasKey,
+	});
+	_logger.info(
+		`[pi-free] Registered ${providerId} with global toggle (${stored.free.length} free, ${stored.all.length} total)`,
+	);
+}
+
+/** Get current global free-only state */
+export function getGlobalFreeOnly(): boolean {
+	return globalFreeOnly;
+}
+
+/** Access the raw registry (used by /free-providers command) */
+export function getProviderRegistry(): ReadonlyMap<string, ProviderEntry> {
+	return providerRegistry;
+}
+
+// =============================================================================
+// Global filter application
+// =============================================================================
+
+export function applyGlobalFilter(_pi: ExtensionAPI, freeOnly: boolean): void {
+	globalFreeOnly = freeOnly;
+	saveConfig({ free_only: freeOnly });
+
+	for (const [providerId, entry] of providerRegistry) {
+		try {
+			if (freeOnly) {
+				// Show only free models
+				if (entry.stored.free.length > 0) {
+					entry.reRegister(entry.stored.free);
+					_logger.info(
+						`[pi-free] ${providerId}: filtered to ${entry.stored.free.length} free models`,
+					);
+				} else {
+					_logger.warn(`[pi-free] ${providerId}: no free models available`);
+				}
+			} else {
+				// Show all models (paid + free)
+				const allModels =
+					entry.stored.all.length > 0 ? entry.stored.all : entry.stored.free;
+				if (allModels.length > 0) {
+					entry.reRegister(allModels);
+					_logger.info(
+						`[pi-free] ${providerId}: showing all ${allModels.length} models`,
+					);
+				}
+			}
+		} catch (err) {
+			_logger.error(
+				`[pi-free] Failed to apply filter to ${providerId}`,
+				err instanceof Error ? { error: err.message } : { error: String(err) },
+			);
+		}
+	}
+}

package/package.json

@@ -1,22 +1,19 @@
 {
 	"name": "pi-free",
-	"version": "1.0.9",
+	"version": "2.0.1",
 	"type": "module",
-	"description": "AIO Free AI models for Pi - Access free models from Kilo, Zen, OpenRouter, NVIDIA, Cline, Mistral, Ollama, and Qwen",
+	"description": "AIO free models for PI: Kilo, Cline, Nvidia, Ollama Cloud and others",
 	"keywords": [
 		"pi-package",
 		"pi-extension",
 		"free-models",
-		"ai-providers",
-		"openrouter",
+		"model-filter",
 		"nvidia-nim",
 		"kilo",
 		"cline",
-		"ollama",
-		"mistral",
-		"fireworks",
 		"qwen",
-		"qwen-oauth"
+		"qwen-oauth",
+		"modal"
 	],
 	"license": "MIT",
 	"author": "Apostolos Mantzaris",
@@ -32,11 +29,10 @@
 		"node": ">=20.0.0"
 	},
 	"files": [
+		"index.ts",
 		"providers/**/*.ts",
 		"lib/**/*.ts",
-		"usage/**/*.ts",
 		"provider-failover/**/*.ts",
-		"widget/**/*.ts",
 		"config.ts",
 		"constants.ts",
 		"provider-factory.ts",
@@ -55,8 +51,7 @@
 	"peerDependencies": {
 		"@mariozechner/pi-ai": "*",
 		"@mariozechner/pi-coding-agent": "*",
-		"@mariozechner/pi-tui": "*",
-		"@sinclair/typebox": "*"
+		"@mariozechner/pi-tui": "*"
 	},
 	"devDependencies": {
 		"@vitest/ui": "^1.0.0",
@@ -66,17 +61,7 @@
 	},
 	"pi": {
 		"extensions": [
-			"./providers/kilo.ts",
-			"./providers/zen.ts",
-			"./providers/go.ts",
-			"./providers/openrouter.ts",
-			"./providers/nvidia.ts",
-			"./providers/cline.ts",
-			"./providers/fireworks.ts",
-			"./providers/mistral.ts",
-			"./providers/ollama.ts",
-			"./providers/qwen.ts",
-			"./providers/modal.ts"
+			"./index.ts"
 		]
 	}
 }