pi-free 2.0.2 → 2.0.5

package/config.ts

@@ -24,54 +24,45 @@ const _logger = createLogger("config");
 
 interface PiFreeConfig {
 	nvidia_api_key?: string;
-	cloudflare_api_token?: string;
-	cloudflare_account_id?: string;
 	ollama_api_key?: string;
-	modal_api_key?: string;
-	opencode_api_key?: string;
+	zenmux_api_key?: string;
+	crofai_api_key?: string;
 	mistral_api_key?: string;
 	groq_api_key?: string;
 	cerebras_api_key?: string;
 	xai_api_key?: string;
 	hf_token?: string;
-	openrouter_api_key?: string;
 	kilo_free_only?: boolean;
 	hidden_models?: string[];
 	free_only?: boolean;
 	kilo_show_paid?: boolean;
-	nvidia_show_paid?: boolean;
-	cloudflare_show_paid?: boolean;
 	ollama_show_paid?: boolean;
 	cline_show_paid?: boolean;
-	qwen_show_paid?: boolean;
-	modal_show_paid?: boolean;
+	zenmux_show_paid?: boolean;
+	crofai_show_paid?: boolean;
 	openrouter_show_paid?: boolean;
 	opencode_show_paid?: boolean;
 }
 
 const CONFIG_TEMPLATE: PiFreeConfig = {
 	nvidia_api_key: "",
-	cloudflare_api_token: "",
-	cloudflare_account_id: "",
 	ollama_api_key: "",
-	modal_api_key: "",
-	opencode_api_key: "",
+	zenmux_api_key: "",
+	crofai_api_key: "",
 	mistral_api_key: "",
 	groq_api_key: "",
 	cerebras_api_key: "",
 	xai_api_key: "",
 	hf_token: "",
-	openrouter_api_key: "",
+
 	kilo_free_only: false,
 	hidden_models: [],
 	free_only: true,
 	kilo_show_paid: false,
-	nvidia_show_paid: false,
-	cloudflare_show_paid: false,
 	ollama_show_paid: false,
 	cline_show_paid: false,
-	qwen_show_paid: false,
-	modal_show_paid: false,
+	zenmux_show_paid: false,
+	crofai_show_paid: false,
 	openrouter_show_paid: false,
 	opencode_show_paid: false,
 };
@@ -144,34 +135,22 @@ export function getKiloShowPaid(): boolean {
 	return resolveBool("KILO_SHOW_PAID", loadConfigFile().kilo_show_paid);
 }
 
-export function getNvidiaShowPaid(): boolean {
-	return resolveBool("NVIDIA_SHOW_PAID", loadConfigFile().nvidia_show_paid);
-}
-
 export function getClineShowPaid(): boolean {
 	return resolveBool("CLINE_SHOW_PAID", loadConfigFile().cline_show_paid);
 }
 
-/** @deprecated Qwen provider is deprecated. */
-export function getQwenShowPaid(): boolean {
-	return resolveBool("QWEN_SHOW_PAID", loadConfigFile().qwen_show_paid);
+export function getZenmuxShowPaid(): boolean {
+	return resolveBool("ZENMUX_SHOW_PAID", loadConfigFile().zenmux_show_paid);
 }
 
-export function getModalShowPaid(): boolean {
-	return resolveBool("MODAL_SHOW_PAID", loadConfigFile().modal_show_paid);
+export function getCrofaiShowPaid(): boolean {
+	return resolveBool("CROFAI_SHOW_PAID", loadConfigFile().crofai_show_paid);
 }
 
 export function getOllamaShowPaid(): boolean {
 	return resolveBool("OLLAMA_SHOW_PAID", loadConfigFile().ollama_show_paid);
 }
 
-export function getCloudflareShowPaid(): boolean {
-	return resolveBool(
-		"CLOUDFLARE_SHOW_PAID",
-		loadConfigFile().cloudflare_show_paid,
-	);
-}
-
 export function getOpenrouterShowPaid(): boolean {
 	return resolveBool(
 		"OPENROUTER_SHOW_PAID",
@@ -203,27 +182,16 @@ export function getNvidiaApiKey(): string | undefined {
 	return resolve("NVIDIA_API_KEY", loadConfigFile().nvidia_api_key);
 }
 
-export function getModalApiKey(): string | undefined {
-	return resolve("MODAL_API_KEY", loadConfigFile().modal_api_key);
-}
-
-export function getOllamaApiKey(): string | undefined {
-	return resolve("OLLAMA_API_KEY", loadConfigFile().ollama_api_key);
+export function getZenmuxApiKey(): string | undefined {
+	return resolve("ZENMUX_API_KEY", loadConfigFile().zenmux_api_key);
 }
 
-export function getCloudflareApiToken(): string | undefined {
-	return resolve("CLOUDFLARE_API_TOKEN", loadConfigFile().cloudflare_api_token);
-}
-
-export function getCloudflareAccountId(): string | undefined {
-	return resolve(
-		"CLOUDFLARE_ACCOUNT_ID",
-		loadConfigFile().cloudflare_account_id,
-	);
+export function getCrofaiApiKey(): string | undefined {
+	return resolve("CROFAI_API_KEY", loadConfigFile().crofai_api_key);
 }
 
-export function getOpencodeApiKey(): string | undefined {
-	return resolve("OPENCODE_API_KEY", loadConfigFile().opencode_api_key);
+export function getOllamaApiKey(): string | undefined {
+	return resolve("OLLAMA_API_KEY", loadConfigFile().ollama_api_key);
 }
 
 export function getMistralApiKey(): string | undefined {
@@ -246,8 +214,12 @@ export function getHfToken(): string | undefined {
 	return resolve("HF_TOKEN", loadConfigFile().hf_token);
 }
 
+/**
+ * OpenRouter key — pi's built-in provider reads from ~/.pi/agent/auth.json.
+ * pi-free only checks the env var to avoid stale keys from free.json.
+ */
 export function getOpenrouterApiKey(): string | undefined {
-	return resolve("OPENROUTER_API_KEY", loadConfigFile().openrouter_api_key);
+	return process.env.OPENROUTER_API_KEY;
 }
 
 // =============================================================================

package/constants.ts

@@ -15,6 +15,8 @@ export const PROVIDER_OLLAMA = "ollama-cloud";
 /** @deprecated Qwen provider is deprecated. The 1,000 req/day free tier is no longer available. */
 export const PROVIDER_QWEN = "qwen";
 export const PROVIDER_MODAL = "modal";
+export const PROVIDER_ZENMUX = "zenmux";
+export const PROVIDER_CROFAI = "crofai";
 
 export const ALL_UNIQUE_PROVIDERS = [
 	PROVIDER_KILO,
@@ -24,6 +26,8 @@ export const ALL_UNIQUE_PROVIDERS = [
 	PROVIDER_QWEN,
 	PROVIDER_MODAL,
 	PROVIDER_OLLAMA,
+	PROVIDER_ZENMUX,
+	PROVIDER_CROFAI,
 ] as const;
 
 // =============================================================================
@@ -38,6 +42,8 @@ export const BASE_URL_CLINE = "https://api.cline.bot/api/v1";
 export const BASE_URL_MODAL = "https://api.us-west-2.modal.direct/v1";
 export const BASE_URL_QWEN =
 	"https://dashscope.aliyuncs.com/compatible-mode/v1";
+export const BASE_URL_ZENMUX = "https://zenmux.ai/api/v1";
+export const BASE_URL_CROFAI = "https://crof.ai/v1";
 
 /** Cline fetches free models from OpenRouter */
 export const BASE_URL_OPENROUTER = "https://openrouter.ai/api/v1";

package/index.ts

@@ -1,148 +1,175 @@
-/**
- * Pi-Free Providers Index
- *
- * Provides free model filtering for ALL providers (built-in + extension)
- * plus unique free/paid providers not covered by pi's built-in providers.
- *
- * Unique providers:
- * - Kilo: OAuth-based free models
- * - Cline: Cline bot integration
- * - NVIDIA: NVIDIA NIM hosting (free tier available)
- * - 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,
-	getGlobalFreeOnly,
-	getProviderRegistry,
-	isFreeModel,
-	registerWithGlobalToggle,
-} from "./lib/registry.ts";
-// Import unique provider extensions (only providers NOT built into pi)
-import cline from "./providers/cline/cline.ts";
-import cloudflare from "./providers/cloudflare/cloudflare.ts";
-import kilo from "./providers/kilo/kilo.ts";
-import modal from "./providers/modal/modal.ts";
-import nvidia from "./providers/nvidia/nvidia.ts";
-import ollama from "./providers/ollama/ollama.ts";
-import qwen from "./providers/qwen/qwen.ts";
-
-const _logger = createLogger("pi-free");
-
-// =============================================================================
-// Global Commands
-// =============================================================================
-
-function setupGlobalCommands(pi: ExtensionAPI) {
-	// /free-providers - Show free model counts by provider
-	pi.registerCommand("free-providers", {
-		description: "Show free/paid model counts for all pi-free providers",
-		handler: async (_args, ctx) => {
-			const lines = ["📊 Pi-Free Providers:", ""];
-			const registry = getProviderRegistry();
-
-			// Providers known to not expose pricing via API (all models show as "free")
-			// OpenRouter and OpenCode expose actual pricing
-			const noPricingApi = new Set([
-				"mistral",
-				"xai",
-				"huggingface",
-				"groq",
-				"cerebras",
-			]);
-			// Freemium providers - all models share a free tier quota
-			const freemiumProviders = new Set(["nvidia"]);
-
-			for (const [id, entry] of registry) {
-				const free = entry.stored.free.length;
-				const all = entry.stored.all.length || free;
-				const indicator = entry.hasKey ? "🔑" : "🆓";
-				const paid = all - free;
-
-				if (freemiumProviders.has(id)) {
-					// Freemium: all models share a free tier (e.g., 1,000 reqs/month)
-					lines.push(`${indicator} ${id}: ${all} models (freemium)`);
-				} else if (noPricingApi.has(id)) {
-					// Provider doesn't expose pricing - can't determine free vs paid
-					lines.push(
-						`${indicator} ${id}: ${all} models (pricing not exposed by API)`,
-					);
-				} else if (paid === 0 && free > 0) {
-					// All models are actually free
-					lines.push(`${indicator} ${id}: ${free} free models`);
-				} else {
-					// Mix of free and paid
-					lines.push(
-						`${indicator} ${id}: ${free} free / ${paid} paid (${all} total)`,
-					);
-				}
-			}
-
-			if (registry.size === 0) {
-				lines.push("(No providers registered yet)");
-			}
-
-			ctx.ui.notify(lines.join("\n"), "info");
-		},
-	});
-}
-
-// =============================================================================
-// Main Entry Point
-// =============================================================================
-
-export default async function (pi: ExtensionAPI) {
-	const globalFreeOnly = getGlobalFreeOnly();
-	_logger.info(`[pi-free] Initializing (global free-only: ${globalFreeOnly})`);
-
-	// Setup global commands first
-	setupGlobalCommands(pi);
-
-	// Load all unique providers
-	// Each provider will register itself with the global toggle system
-	await Promise.allSettled([
-		cloudflare(pi),
-		modal(pi),
-		nvidia(pi),
-		kilo(pi),
-		ollama(pi),
-		// Qwen is deprecated
-		qwen(pi).catch((err) => {
-			_logger.warn("[pi-free] Qwen provider failed to load (deprecated)", err);
-		}),
-		cline(pi),
-	]);
-
-	// Setup dynamic built-in providers (Mistral, Groq, Cerebras, xAI, Hugging Face, OpenRouter)
-	// These only activate if the user has configured API keys (OpenRouter works without key too)
-	const { setupDynamicBuiltInProviders } = await import(
-		"./providers/dynamic-built-in/index.ts"
-	);
-	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");
-		await applyGlobalFilter(pi, true);
-	}
-
-	const registry = getProviderRegistry();
-	_logger.info(`[pi-free] Loaded with ${registry.size} providers`);
-}
-
-// Re-export registry helpers so consumers don't need deep imports
-export {
-	applyGlobalFilter,
-	getGlobalFreeOnly,
-	getProviderRegistry,
-	isFreeModel,
-	registerWithGlobalToggle,
-};
+/**
+ * Pi-Free Providers Index
+ *
+ * Provides free model filtering for ALL providers (built-in + extension)
+ * plus unique free/paid providers not covered by pi's built-in providers.
+ *
+ * Unique providers:
+ * - Kilo: OAuth-based free models
+ * - Cline: Cline bot integration
+ * - NVIDIA: NVIDIA NIM hosting (free tier available)
+ * - Ollama Cloud: Ollama's cloud-hosted models with usage-based free tier
+ * - ZenMux: Unified AI API gateway with 200+ models
+ */
+
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { setupBuiltInProviderToggles } from "./lib/built-in-toggle.ts";
+import { createLogger } from "./lib/logger.ts";
+import {
+	applyGlobalFilter,
+	getGlobalFreeOnly,
+	getProviderRegistry,
+	isFreeModel,
+	registerWithGlobalToggle,
+} from "./lib/registry.ts";
+// Import unique provider extensions (only providers NOT built into pi)
+import cline from "./providers/cline/cline.ts";
+import crofai from "./providers/crofai/crofai.ts";
+import kilo from "./providers/kilo/kilo.ts";
+import nvidia from "./providers/nvidia/nvidia.ts";
+import ollama from "./providers/ollama/ollama.ts";
+import zenmux from "./providers/zenmux/zenmux.ts";
+
+const _logger = createLogger("pi-free");
+
+// =============================================================================
+// Global Commands
+// =============================================================================
+
+function setupGlobalCommands(pi: ExtensionAPI) {
+	// /toggle-free - Global free-only mode toggle
+	pi.registerCommand("toggle-free", {
+		description: "Toggle global free-only mode for all providers",
+		handler: async (_args, ctx) => {
+			const current = getGlobalFreeOnly();
+			const next = !current;
+			applyGlobalFilter(pi, next);
+
+			const registry = getProviderRegistry();
+			const providerCount = registry.size;
+
+			if (next) {
+				const totalFree = [...registry.values()].reduce(
+					(sum, e) => sum + e.stored.free.length,
+					0,
+				);
+				ctx.ui.notify(
+					`Free-only mode: ON (${totalFree} free models across ${providerCount} providers)`,
+					"info",
+				);
+			} else {
+				const totalAll = [...registry.values()].reduce(
+					(sum, e) => sum + (e.stored.all.length || e.stored.free.length),
+					0,
+				);
+				ctx.ui.notify(
+					`Free-only mode: OFF (all ${totalAll} models visible across ${providerCount} providers)`,
+					"info",
+				);
+			}
+		},
+	});
+
+	// /free-providers - Show free model counts by provider
+	pi.registerCommand("free-providers", {
+		description: "Show free/paid model counts for all pi-free providers",
+		handler: async (_args, ctx) => {
+			const lines = ["📊 Pi-Free Providers:", ""];
+			const registry = getProviderRegistry();
+
+			// Providers known to not expose pricing via API (all models show as "free")
+			// OpenRouter and OpenCode expose actual pricing
+			const noPricingApi = new Set([
+				"mistral",
+				"xai",
+				"huggingface",
+				"groq",
+				"cerebras",
+			]);
+			// Freemium providers - all models share a free tier quota
+			const freemiumProviders = new Set(["nvidia"]);
+
+			for (const [id, entry] of registry) {
+				const free = entry.stored.free.length;
+				const all = entry.stored.all.length || free;
+				const indicator = entry.hasKey ? "🔑" : "🆓";
+				const paid = all - free;
+
+				if (freemiumProviders.has(id)) {
+					// Freemium: all models share a free tier (e.g., 1,000 reqs/month)
+					lines.push(`${indicator} ${id}: ${all} models (freemium)`);
+				} else if (noPricingApi.has(id)) {
+					// Provider doesn't expose pricing - can't determine free vs paid
+					lines.push(
+						`${indicator} ${id}: ${all} models (pricing not exposed by API)`,
+					);
+				} else if (paid === 0 && free > 0) {
+					// All models are actually free
+					lines.push(`${indicator} ${id}: ${free} free models`);
+				} else {
+					// Mix of free and paid
+					lines.push(
+						`${indicator} ${id}: ${free} free / ${paid} paid (${all} total)`,
+					);
+				}
+			}
+
+			if (registry.size === 0) {
+				lines.push("(No providers registered yet)");
+			}
+
+			ctx.ui.notify(lines.join("\n"), "info");
+		},
+	});
+}
+
+// =============================================================================
+// Main Entry Point
+// =============================================================================
+
+export default async function (pi: ExtensionAPI) {
+	const globalFreeOnly = getGlobalFreeOnly();
+	_logger.info(`[pi-free] Initializing (global free-only: ${globalFreeOnly})`);
+
+	// Setup global commands first
+	setupGlobalCommands(pi);
+
+	// Load all unique providers
+	// Each provider will register itself with the global toggle system
+	await Promise.allSettled([
+		nvidia(pi),
+		kilo(pi),
+		ollama(pi),
+		cline(pi),
+		zenmux(pi),
+		crofai(pi),
+	]);
+
+	// Setup dynamic built-in providers (Mistral, Groq, Cerebras, xAI, Hugging Face)
+	// These only activate if the user has configured API keys (OpenRouter works without key too)
+	const { setupDynamicBuiltInProviders } = await import(
+		"./providers/dynamic-built-in/index.ts"
+	);
+	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");
+		applyGlobalFilter(pi, true);
+	}
+
+	const registry = getProviderRegistry();
+	_logger.info(`[pi-free] Loaded with ${registry.size} providers`);
+}
+
+// Re-export registry helpers so consumers don't need deep imports
+export {
+	applyGlobalFilter,
+	getGlobalFreeOnly,
+	getProviderRegistry,
+	isFreeModel,
+	registerWithGlobalToggle,
+};

package/lib/built-in-toggle.ts

@@ -60,6 +60,7 @@ export function setupBuiltInProviderToggles(pi: ExtensionAPI): void {
 		for (const config of BUILT_IN_TOGGLE_PROVIDERS) {
 			registerToggleCommand(pi, config);
 		}
+		setupStatusBar(pi);
 		commandsRegistered = true;
 	}
 
@@ -79,7 +80,9 @@ export function setupBuiltInProviderToggles(pi: ExtensionAPI): void {
 			if (providerModels.length === 0) continue;
 
 			const allModels = providerModels.map(modelToProviderConfig);
-			const freeModels = allModels.filter(isFreeModel);
+			const freeModels = allModels.filter((m) =>
+				isFreeModel({ ...m, provider: config.id }, allModels),
+			);
 
 			const baseUrl = providerModels[0].baseUrl;
 			const api = providerModels[0].api;
@@ -178,6 +181,42 @@ function modelToProviderConfig(m: Model<Api>): ProviderModelConfig {
 	};
 }
 
+// =============================================================================
+// Status bar for provider selection
+// =============================================================================
+
+function setupStatusBar(pi: ExtensionAPI): void {
+	pi.on("model_select", (_event, ctx) => {
+		const selected = _event.model?.provider;
+
+		// Clear status for all built-in toggle providers
+		for (const config of BUILT_IN_TOGGLE_PROVIDERS) {
+			if (selected !== config.id) {
+				ctx.ui.setStatus(`${config.id}-status`, undefined);
+			}
+		}
+
+		if (!selected) return;
+
+		const state = providerStates.get(selected);
+		if (!state) return;
+
+		const free = state.stored.free.length;
+		const total = state.stored.all.length;
+		const paid = total - free;
+		const mode = state.toggleState.getCurrentMode();
+		let status: string;
+		if (paid === 0) {
+			status = `${selected}: ${free} free models`;
+		} else if (mode === "all") {
+			status = `${selected}: ${total} models (free + paid)`;
+		} else {
+			status = `${selected}: ${free} free \u00b7 ${paid} paid`;
+		}
+		ctx.ui.setStatus(`${selected}-status`, status);
+	});
+}
+
 function getApiKeyEnvForProvider(providerId: string): string {
 	const envMap: Record<string, string> = {
 		opencode: "OPENCODE_API_KEY",