pi-free 1.0.8 → 2.0.0

package/providers/openrouter.ts

@@ -1,175 +0,0 @@
-/**
- * OpenRouter Provider Extension
- *
- * Provides access to 29+ free models and 300+ paid models via OpenRouter.
- * Requires OPENROUTER_API_KEY (free account at https://openrouter.ai).
- *
- * By default only free (:free) models are shown.
- * Set OPENROUTER_OPENROUTER_SHOW_PAID=true to also include paid models.
- */
-
-import type {
-	ExtensionAPI,
-	ProviderModelConfig,
-} from "@mariozechner/pi-coding-agent";
-import {
-	applyHidden,
-	OPENROUTER_API_KEY as CONFIG_API_KEY,
-	OPENROUTER_SHOW_PAID,
-	PROVIDER_OPENROUTER,
-} from "../config.ts";
-import {
-	BASE_URL_OPENROUTER,
-	DEFAULT_FETCH_TIMEOUT_MS,
-	DEFAULT_MIN_SIZE_B,
-} from "../constants.ts";
-import { fetchOpenRouterMetrics } from "../usage/metrics.ts";
-import {
-	type StoredModels,
-	setupProvider,
-	createCtxReRegister,
-} from "../provider-helper.ts";
-import { createLogger } from "../lib/logger.ts";
-import { cleanModelName, isUsableModel, logWarning } from "../lib/util.ts";
-import { fetchOpenRouterModelsWithFree } from "./model-fetcher.ts";
-
-const _logger = createLogger("openrouter");
-
-const OPENROUTER_CONFIG = {
-	providerId: PROVIDER_OPENROUTER,
-	baseUrl: BASE_URL_OPENROUTER,
-	apiKey: "OPENROUTER_API_KEY",
-	headers: {
-		"HTTP-Referer": "https://github.com/apmantza/pi-free",
-		"X-Title": "Pi",
-	},
-};
-
-// =============================================================================
-// Fetch
-// =============================================================================
-
-async function fetchOpenRouterModels(apiKey: string): Promise<{
-	free: ProviderModelConfig[];
-	all: ProviderModelConfig[];
-}> {
-	const { free, all } = await fetchOpenRouterModelsWithFree({
-		baseUrl: BASE_URL_OPENROUTER,
-		apiKey,
-		extraHeaders: {
-			"HTTP-Referer": "https://github.com/apmantza/pi-free",
-			"X-Title": "Pi",
-		},
-	});
-
-	return { free: applyHidden(free), all: applyHidden(all) };
-}
-
-// =============================================================================
-// Extension Entry Point
-// =============================================================================
-
-export default async function (pi: ExtensionAPI) {
-	const apiKey = CONFIG_API_KEY;
-
-	// Shared model storage (references held by setupProvider for commands)
-	const stored: StoredModels = { free: [], all: [] };
-
-	// Re-registration function - will be set in session_start with ctx
-	let reRegisterFn: (models: ProviderModelConfig[]) => void = () => {};
-
-	// Wire up shared boilerplate (commands, model_select, turn_end)
-	setupProvider(
-		pi,
-		{
-			providerId: PROVIDER_OPENROUTER,
-			initialShowPaid: OPENROUTER_SHOW_PAID,
-			reRegister: (models) => reRegisterFn(models),
-		},
-		stored,
-	);
-
-	// Check in session_start if user already has auth for this provider
-	// If yes: filter their models to free-only, use their key
-	// If no: use our extension's key with filtered models
-	pi.on("session_start", async (_event, ctx) => {
-		const allModels = ctx.modelRegistry.getAll();
-		const availableModels = ctx.modelRegistry.getAvailable();
-		const existingModels = allModels.filter(
-			(m) => m.provider === PROVIDER_OPENROUTER,
-		);
-		const hasExistingAuth = availableModels.some(
-			(m) => m.provider === PROVIDER_OPENROUTER,
-		);
-
-		if (hasExistingAuth && existingModels.length > 0) {
-			// User has existing auth - filter to free models, use their key
-			const freeModels = existingModels
-				.filter((m) => (m.cost?.input ?? 0) === 0)
-				.filter((m) => isUsableModel(m.id, DEFAULT_MIN_SIZE_B))
-				.map((m) => ({
-					id: m.id,
-					name: cleanModelName(m.name),
-					reasoning: m.reasoning,
-					input: m.input,
-					cost: m.cost,
-					contextWindow: m.contextWindow,
-					maxTokens: m.maxTokens,
-				}));
-
-			if (freeModels.length === 0) {
-				_logger.warn(
-					"[openrouter] No free models available from existing auth",
-				);
-				return;
-			}
-
-			// Store for command toggle
-			stored.free = freeModels;
-			stored.all = existingModels;
-
-			// Create re-register function using ctx
-			reRegisterFn = createCtxReRegister(ctx as any, OPENROUTER_CONFIG);
-			reRegisterFn(freeModels);
-			return;
-		}
-
-		// User doesn't have existing auth — use our extension's key
-		if (apiKey) {
-			process.env.OPENROUTER_API_KEY = apiKey;
-		} else {
-			_logger.warn(
-				"[openrouter] No API key found — set OPENROUTER_API_KEY or add openrouter_api_key to ~/.pi/free.json. Free key at https://openrouter.ai",
-			);
-			return;
-		}
-
-		let models: ProviderModelConfig[] = [];
-		let fetchResult: {
-			free: ProviderModelConfig[];
-			all: ProviderModelConfig[];
-		} | null = null;
-
-		try {
-			fetchResult = await fetchOpenRouterModels(apiKey);
-			models = OPENROUTER_SHOW_PAID ? fetchResult.all : fetchResult.free;
-		} catch (error) {
-			logWarning("openrouter", "Failed to fetch models", error);
-		}
-
-		if (models.length === 0) return;
-
-		// Store for command toggle
-		if (fetchResult) {
-			stored.free = fetchResult.free;
-			stored.all = fetchResult.all;
-		}
-
-		// Create re-register function using ctx and register
-		reRegisterFn = createCtxReRegister(ctx as any, OPENROUTER_CONFIG);
-		reRegisterFn(models);
-
-		// Fetch and cache metrics (used internally, not displayed)
-		await fetchOpenRouterMetrics();
-	});
-}

package/providers/qwen.ts

@@ -1,127 +0,0 @@
-/**
- * Qwen OAuth Provider Extension
- *
- * Provides free access to Qwen 3.6 Plus via OAuth device flow.
- * 1,000 free API calls/day — run /login qwen to authenticate.
- *
- * Usage:
- *   pi install git:github.com/apmantza/pi-free
- *   # Then /login qwen, select qwen model
- */
-
-import type { OAuthCredentials, Model, Api } from "@mariozechner/pi-ai";
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
-import { PROVIDER_QWEN, URL_QWEN_TOS } from "../constants.ts";
-import {
-	enhanceWithCI,
-	type StoredModels,
-	setupProvider,
-	createReRegister,
-} from "../provider-helper.ts";
-import { incrementRequestCount } from "../usage/metrics.ts";
-import { logWarning } from "../lib/util.ts";
-import { createLogger } from "../lib/logger.ts";
-import { loginQwen, refreshQwenToken, getQwenBaseUrl } from "./qwen-auth.ts";
-import { fetchQwenModels } from "./qwen-models.ts";
-
-const _logger = createLogger("qwen");
-
-// =============================================================================
-// Constants
-// =============================================================================
-
-// Mirrors qwen-code's DEFAULT_QWEN_BASE_URL (used when resource_url is absent).
-const DEFAULT_BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1";
-
-// Headers required by DashScope's OpenAI-compatible API for OAuth tokens.
-// Replicates DashScopeOpenAICompatibleProvider.buildHeaders() from qwen-code.
-const DASHSCOPE_HEADERS = {
-	"X-DashScope-AuthType": "qwen-oauth",
-	"X-DashScope-CacheControl": "enable",
-	"X-DashScope-UserAgent": "QwenCode/0.0.5 (pi-free)",
-	"Client-Code": "QwenCode",
-};
-
-// =============================================================================
-// Extension entry point
-// =============================================================================
-
-export default async function (pi: ExtensionAPI) {
-	// Fetch static free-tier models
-	let models = await fetchQwenModels().catch((err) => {
-		logWarning("qwen", "Failed to load models at startup", err);
-		return [];
-	});
-
-	if (models.length === 0) {
-		logWarning("qwen", "No models available, skipping provider");
-		return;
-	}
-
-	const stored: StoredModels = { free: models, all: models };
-
-	// OAuth config for Qwen
-	const oauthConfig = {
-		name: "Qwen",
-		login: loginQwen,
-		refreshToken: refreshQwenToken,
-		getApiKey: (cred: OAuthCredentials) => cred.access,
-		modifyModels: (models: Model<Api>[], cred: OAuthCredentials) => {
-			// Mirror qwen-code: resolve baseUrl from resource_url per-credential.
-			// Chinese accounts → dashscope.aliyuncs.com/v1
-			// International accounts → portal.qwen.ai/v1 (or custom endpoint)
-			const baseUrl = getQwenBaseUrl(cred);
-			_logger.info("Qwen OAuth modifyModels called", {
-				baseUrl,
-				resource_url: cred.resource_url,
-				modelCount: models.length,
-			});
-			if (baseUrl === DEFAULT_BASE_URL) return models;
-			return (models as Model<Api>[]).map((m) => ({ ...m, baseUrl }));
-		},
-	};
-
-	// Register provider with OpenAI-compatible API
-	function registerProvider(m = models) {
-		pi.registerProvider(PROVIDER_QWEN, {
-			baseUrl: DEFAULT_BASE_URL,
-			apiKey: "QWEN_API_KEY",
-			api: "openai-completions" as const,
-			headers: {
-				"User-Agent": "pi-free",
-				...DASHSCOPE_HEADERS,
-			},
-			models: enhanceWithCI(m),
-			oauth: oauthConfig,
-		});
-	}
-
-	registerProvider();
-
-	// Wire up shared boilerplate (commands, model_select, turn_end, ToS)
-	const reRegister = createReRegister(pi, {
-		providerId: PROVIDER_QWEN,
-		baseUrl: DEFAULT_BASE_URL,
-		apiKey: "QWEN_API_KEY",
-		oauth: oauthConfig as any,
-	});
-
-	setupProvider(
-		pi,
-		{
-			providerId: PROVIDER_QWEN,
-			tosUrl: URL_QWEN_TOS,
-			initialShowPaid: false,
-			reRegister: (m) => {
-				reRegister(m);
-			},
-		},
-		stored,
-	);
-
-	// Keep lightweight request counting for now (internal only).
-	pi.on("turn_end", async (_event, ctx) => {
-		if (ctx.model?.provider !== PROVIDER_QWEN) return;
-		incrementRequestCount(PROVIDER_QWEN);
-	});
-}

package/providers/zen.ts

@@ -1,371 +0,0 @@
-/**
- * OpenCode Zen Provider Extension
- *
- * Provides access to curated AI models via the OpenCode Zen gateway.
- * Free models are available immediately with no account needed.
- * Set OPENCODE_API_KEY (or opencode_api_key in ~/.pi/free.json) for paid access.
- *
- * Model list fetched directly from the Zen gateway — only returns models that
- * are actually deployed. Metadata (pricing, context) enriched from models.dev.
- */
-
-import type {
-	ExtensionAPI,
-	ProviderModelConfig,
-} from "@mariozechner/pi-coding-agent";
-import {
-	applyHidden,
-	OPENCODE_API_KEY as CONFIG_API_KEY,
-	PROVIDER_ZEN,
-	ZEN_SHOW_PAID,
-} from "../config.ts";
-import {
-	BASE_URL_ZEN,
-	DEFAULT_FETCH_TIMEOUT_MS,
-	URL_ZEN_TOS,
-} from "../constants.ts";
-import {
-	type StoredModels,
-	setupProvider,
-	createCtxReRegister,
-} from "../provider-helper.ts";
-import type { ZenGatewayModel } from "../lib/types.ts";
-import { fetchModelsDevMeta } from "./model-fetcher.ts";
-import { createOpenCodeSessionTracker } from "./opencode-session.ts";
-import { fetchWithRetry, logWarning } from "../lib/util.ts";
-
-const ZEN_CONFIG = {
-	providerId: PROVIDER_ZEN,
-	baseUrl: BASE_URL_ZEN,
-	apiKey: "PI_FREE_ZEN_API_KEY",
-	headers: {
-		"X-Title": "Pi",
-		"HTTP-Referer": "https://opencode.ai/",
-	},
-};
-
-const session = createOpenCodeSessionTracker();
-
-// =============================================================================
-// Static fallback models (from Pi's built-in + OpenCode docs)
-// Used when /models API is unavailable
-// =============================================================================
-
-const _STATIC_ZEN_MODELS: ProviderModelConfig[] = [
-	// Free models (from OpenCode Zen docs)
-	{
-		id: "big-pickle",
-		name: "Big Pickle",
-		reasoning: true,
-		input: ["text"],
-		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 200000,
-		maxTokens: 128000,
-	},
-	{
-		id: "trinity-large-preview-free",
-		name: "Trinity Large Preview Free",
-		reasoning: false,
-		input: ["text"],
-		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 128000,
-		maxTokens: 16384,
-	},
-	{
-		id: "minimax-m2.5-free",
-		name: "MiniMax M2.5 Free",
-		reasoning: true,
-		input: ["text"],
-		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 200000,
-		maxTokens: 16384,
-	},
-	{
-		id: "mimo-v2-pro-free",
-		name: "MiMo V2 Pro Free",
-		reasoning: false,
-		input: ["text"],
-		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 128000,
-		maxTokens: 16384,
-	},
-	{
-		id: "mimo-v2-omni-free",
-		name: "MiMo V2 Omni Free",
-		reasoning: false,
-		input: ["text", "image"],
-		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 128000,
-		maxTokens: 16384,
-	},
-	{
-		id: "mimo-v2-flash-free",
-		name: "MiMo V2 Flash Free",
-		reasoning: false,
-		input: ["text"],
-		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 128000,
-		maxTokens: 16384,
-	},
-	{
-		id: "nemotron-3-super-free",
-		name: "Nemotron 3 Super Free",
-		reasoning: false,
-		input: ["text"],
-		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 128000,
-		maxTokens: 16384,
-	},
-	// Paid models (available when show_paid: true and API key set)
-	{
-		id: "claude-3-5-haiku",
-		name: "Claude Haiku 3.5",
-		reasoning: false,
-		input: ["text", "image"],
-		cost: { input: 0.8, output: 4, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 200000,
-		maxTokens: 8192,
-	},
-	{
-		id: "claude-haiku-4-5",
-		name: "Claude Haiku 4.5",
-		reasoning: true,
-		input: ["text", "image"],
-		cost: { input: 1, output: 5, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 200000,
-		maxTokens: 64000,
-	},
-	{
-		id: "claude-opus-4-5",
-		name: "Claude Opus 4.5",
-		reasoning: true,
-		input: ["text", "image"],
-		cost: { input: 5, output: 25, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 200000,
-		maxTokens: 64000,
-	},
-	{
-		id: "claude-sonnet-4-5",
-		name: "Claude Sonnet 4.5",
-		reasoning: true,
-		input: ["text", "image"],
-		cost: { input: 3, output: 15, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 200000,
-		maxTokens: 64000,
-	},
-	{
-		id: "gemini-3-flash",
-		name: "Gemini 3 Flash",
-		reasoning: false,
-		input: ["text", "image"],
-		cost: { input: 0.5, output: 3, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 128000,
-		maxTokens: 16384,
-	},
-	{
-		id: "gemini-3.1-pro",
-		name: "Gemini 3.1 Pro",
-		reasoning: false,
-		input: ["text", "image"],
-		cost: { input: 2, output: 12, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 200000,
-		maxTokens: 16384,
-	},
-	{
-		id: "minimax-m2.5",
-		name: "MiniMax M2.5",
-		reasoning: true,
-		input: ["text"],
-		cost: { input: 0.3, output: 1.2, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 200000,
-		maxTokens: 16384,
-	},
-];
-
-// =============================================================================
-// Fetch helpers
-// =============================================================================
-
-// Models confirmed broken (always return empty content regardless of token budget).
-const ZEN_BROKEN_MODELS = new Set([
-	"gpt-5-nano", // always returns empty content/choices
-	"gpt-5.4-nano", // same family, same issue
-]);
-
-/** Fetch the model list from the Zen gateway — authoritative for what's deployed. */
-async function fetchGatewayModels(token: string): Promise<string[]> {
-	const response = await fetchWithRetry(
-		`${BASE_URL_ZEN}/models`,
-		{
-			headers: {
-				Authorization: `Bearer ${token}`,
-				"User-Agent": "pi-free-providers",
-			},
-		},
-		3,
-		1000,
-		DEFAULT_FETCH_TIMEOUT_MS,
-	);
-
-	if (!response.ok) {
-		throw new Error(
-			`Zen /models returned ${response.status} ${response.statusText}`,
-		);
-	}
-
-	const json = (await response.json()) as { data?: ZenGatewayModel[] };
-	return (json.data ?? [])
-		.map((m) => m.id)
-		.filter((id) => !ZEN_BROKEN_MODELS.has(id));
-}
-
-// =============================================================================
-// Main fetch
-// =============================================================================
-
-async function fetchZenModels(token: string): Promise<{
-	all: ProviderModelConfig[];
-	free: ProviderModelConfig[];
-	useStaticFallback: boolean;
-}> {
-	try {
-		const [gatewayIds, meta] = await Promise.all([
-			fetchGatewayModels(token),
-			fetchModelsDevMeta("opencode"), // Fetch only opencode provider's models
-		]);
-
-		const all: ProviderModelConfig[] = [];
-		const free: ProviderModelConfig[] = [];
-
-		for (const id of gatewayIds) {
-			const m = meta[id];
-
-			// Skip image-output models
-			if (m?.modalities?.output?.includes("image")) continue;
-
-			const config: ProviderModelConfig = {
-				id,
-				name: m?.name ?? id,
-				reasoning: m?.reasoning ?? false,
-				input: m?.modalities?.input?.includes("image")
-					? ["text", "image"]
-					: ["text"],
-				cost: {
-					input: m?.cost?.input ?? 0,
-					output: m?.cost?.output ?? 0,
-					cacheRead: m?.cost?.cache_read ?? 0,
-					cacheWrite: m?.cost?.cache_write ?? 0,
-				},
-				contextWindow: m?.limit?.context ?? 128_000,
-				maxTokens: m?.limit?.output ?? 16_384,
-			};
-
-			all.push(config);
-			if ((m?.cost?.input ?? 0) === 0) free.push(config);
-		}
-
-		return {
-			all: applyHidden(all),
-			free: applyHidden(free),
-			useStaticFallback: false,
-		};
-	} catch (error) {
-		// API failed - return special flag to skip registration
-		// Pi's built-in OpenCode provider will be used instead
-		logWarning(
-			"zen",
-			"API unavailable, letting Pi use built-in provider",
-			error,
-		);
-		return { all: [], free: [], useStaticFallback: true };
-	}
-}
-
-// =============================================================================
-// Extension Entry Point
-// =============================================================================
-
-export default async function (pi: ExtensionAPI) {
-	const hasKey = !!CONFIG_API_KEY;
-	const token = CONFIG_API_KEY ?? "public";
-
-	// Use a private env var so we don't accidentally activate Pi's built-in
-	// opencode provider, which also watches OPENCODE_API_KEY.
-	const ZEN_KEY_VAR = "PI_FREE_ZEN_API_KEY";
-
-	// Shared model storage (references held by setupProvider for commands)
-	const stored: StoredModels = { free: [], all: [] };
-
-	// Re-registration function - will be set in session_start with ctx
-	let reRegisterFn: (models: ProviderModelConfig[]) => void = () => {};
-
-	// Wire up shared boilerplate (commands, model_select, turn_end, ToS)
-	setupProvider(
-		pi,
-		{
-			providerId: PROVIDER_ZEN,
-			tosUrl: URL_ZEN_TOS,
-			hasKey,
-			initialShowPaid: ZEN_SHOW_PAID,
-			reRegister: (models) => reRegisterFn(models),
-		},
-		stored,
-	);
-
-	// Check in session_start if user already has auth for this provider
-	// If they do, we still register with session headers for better reliability
-	pi.on("session_start", async (_event, ctx) => {
-		const availableModels = ctx.modelRegistry.getAvailable();
-		const _hasExistingAuth = availableModels.some(
-			(m) => m.provider === PROVIDER_ZEN,
-		);
-
-		// Set up the env var regardless - either for our use or to supplement existing auth
-		process.env[ZEN_KEY_VAR] = token;
-
-		let models: ProviderModelConfig[] = [];
-		let _freeCount = 0;
-		let useStaticFallback = false;
-
-		try {
-			const result = await fetchZenModels(token);
-			models = hasKey && ZEN_SHOW_PAID ? result.all : result.free;
-			_freeCount = result.free.length;
-			useStaticFallback = result.useStaticFallback;
-
-			// Store for command toggle
-			stored.free = result.free;
-			stored.all = result.all;
-		} catch (error) {
-			logWarning("zen", "Failed to fetch models", error);
-		}
-
-		// If API failed, don't register our provider - let Pi use its built-in
-		if (useStaticFallback || models.length === 0) {
-			return;
-		}
-
-		// Generate session ID for this session (used in headers)
-		const sessionId = session.getSessionId();
-
-		// Create re-register function with session headers
-		const sessionConfig = {
-			...ZEN_CONFIG,
-			headers: {
-				...ZEN_CONFIG.headers,
-				"x-opencode-session": sessionId,
-				"x-session-affinity": sessionId,
-			},
-		};
-		reRegisterFn = createCtxReRegister(ctx as any, sessionConfig);
-
-		// Register our filtered provider (CI enhancement handled by provider-helper)
-		reRegisterFn(models);
-	});
-
-	// Update request count before each agent turn (for request ID generation)
-	pi.on("before_agent_start", async (_event, ctx) => {
-		if (ctx.model?.provider !== PROVIDER_ZEN) return;
-		session.nextRequestId();
-	});
-}

package/usage/commands.ts

@@ -1,17 +0,0 @@
-/**
- * Free tier usage commands
- *
- * Provides:
- * - /free-sessionusage: Current session breakdown
- * - /free-totalusage: Cumulative usage from disk
- *
- * NOTE: Commands temporarily disabled due to duplicate registration issues.
- * Use /kilo-sessionusage or /zen-sessionusage instead.
- */
-
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
-
-export function registerUsageCommands(_pi: ExtensionAPI): void {
-	// Commands disabled - Pi shows duplicate registrations across providers
-	// TODO: Find reliable way to register global commands once
-}