pi-free 1.0.8 → 2.0.0

package/providers/fireworks.ts

@@ -1,49 +0,0 @@
-/**
- * Fireworks AI Provider Extension
- *
- * Provides access to Fireworks AI hosted models via api.fireworks.ai.
- * Uses OpenAI-compatible API - requires FIREWORKS_API_KEY.
- * Get a key at: https://app.fireworks.ai/settings/users/api-keys
- *
- * All models are credit-based (no free tier).
- * Set FIREWORKS_SHOW_PAID=true to enable.
- */
-
-import type { ProviderModelConfig } from "@mariozechner/pi-coding-agent";
-import { PROVIDER_FIREWORKS } from "../config.ts";
-import { BASE_URL_FIREWORKS } from "../constants.ts";
-import { createProvider } from "../provider-factory.ts";
-
-// =============================================================================
-// Static model list (Fireworks doesn't have a models API)
-// =============================================================================
-
-function getFireworksModels(): ProviderModelConfig[] {
-	return [
-		{
-			id: "accounts/fireworks/routers/kimi-k2p5-turbo",
-			name: "Kimi K2.5 Turbo",
-			reasoning: true,
-			input: ["text"],
-			// Actual pricing (per 1M tokens): $0.50 input, $2.00 output
-			cost: { input: 0.0005, output: 0.002, cacheRead: 0, cacheWrite: 0 },
-			contextWindow: 262144,
-			maxTokens: 131072,
-		},
-	];
-}
-
-// =============================================================================
-// Extension Entry Point
-// =============================================================================
-
-export default function (pi: Parameters<typeof createProvider>[0]) {
-	return createProvider(pi, {
-		providerId: PROVIDER_FIREWORKS,
-		baseUrl: BASE_URL_FIREWORKS,
-		apiKeyEnvVar: "FIREWORKS_API_KEY",
-		apiKeyConfigKey: "fireworks_api_key",
-		showPaidFlag: "FIREWORKS_SHOW_PAID",
-		fetchModels: async () => getFireworksModels(),
-	});
-}

package/providers/go.ts

@@ -1,216 +0,0 @@
-/**
- * OpenCode Go Provider Extension
- *
- * Provides access to curated open coding models via the OpenCode Go gateway.
- * Requires an OpenCode Go subscription ($5 first month, then $10/month).
- * Set OPENCODE_GO_API_KEY (or opencode_go_api_key in ~/.pi/free.json) for access.
- *
- * Models available:
- *   - GLM-5
- *   - Kimi K2.5
- *   - MiMo-V2-Pro
- *   - MiMo-V2-Omni
- *   - MiniMax M2.7
- *   - MiniMax M2.5
- *
- * See: https://opencode.ai/docs/go
- */
-
-import type {
-	ExtensionAPI,
-	ProviderModelConfig,
-} from "@mariozechner/pi-coding-agent";
-import {
-	applyHidden,
-	GO_SHOW_PAID,
-	OPENCODE_API_KEY,
-	OPENCODE_GO_API_KEY as CONFIG_API_KEY,
-	PROVIDER_GO,
-} from "../config.ts";
-import {
-	BASE_URL_GO,
-	URL_GO_TOS,
-} from "../constants.ts";
-import {
-	type StoredModels,
-	setupProvider,
-	createCtxReRegister,
-} from "../provider-helper.ts";
-import { createOpenCodeSessionTracker } from "./opencode-session.ts";
-
-const GO_CONFIG = {
-	providerId: PROVIDER_GO,
-	baseUrl: BASE_URL_GO,
-	apiKey: "PI_FREE_GO_API_KEY",
-	headers: {
-		"X-Title": "Pi",
-		"HTTP-Referer": "https://opencode.ai/",
-	},
-};
-
-const session = createOpenCodeSessionTracker();
-
-// =============================================================================
-// Static fallback models (from OpenCode Go docs)
-// Used when /models API is unavailable
-// =============================================================================
-
-const STATIC_GO_MODELS: ProviderModelConfig[] = [
-	{
-		id: "glm-5",
-		name: "GLM-5",
-		reasoning: true,
-		input: ["text"],
-		cost: { input: 0.5, output: 2, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 128000,
-		maxTokens: 16384,
-	},
-	{
-		id: "kimi-k2.5",
-		name: "Kimi K2.5",
-		reasoning: true,
-		input: ["text"],
-		cost: { input: 0.3, output: 1.5, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 128000,
-		maxTokens: 16384,
-	},
-	{
-		id: "mimo-v2-pro",
-		name: "MiMo V2 Pro",
-		reasoning: false,
-		input: ["text"],
-		cost: { input: 0.4, output: 1.2, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 128000,
-		maxTokens: 16384,
-	},
-	{
-		id: "mimo-v2-omni",
-		name: "MiMo V2 Omni",
-		reasoning: false,
-		input: ["text", "image"],
-		cost: { input: 0.25, output: 0.8, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 128000,
-		maxTokens: 16384,
-	},
-	{
-		id: "minimax-m2.7",
-		name: "MiniMax M2.7",
-		reasoning: true,
-		input: ["text"],
-		cost: { input: 0.1, output: 0.4, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 200000,
-		maxTokens: 16384,
-	},
-	{
-		id: "minimax-m2.5",
-		name: "MiniMax M2.5",
-		reasoning: true,
-		input: ["text"],
-		cost: { input: 0.05, output: 0.2, cacheRead: 0, cacheWrite: 0 },
-		contextWindow: 200000,
-		maxTokens: 16384,
-	},
-];
-
-// =============================================================================
-// Fetch helpers
-// =============================================================================
-
-// OpenCode Go does not have a /models endpoint.
-// Models are only accessible via the chat completions API.
-// Static models are used directly (see below).
-
-// =============================================================================
-// Main fetch
-// =============================================================================
-
-/**
- * OpenCode Go does not have a /models endpoint.
- * Models are only accessible via chat completions.
- * So we use the static fallback models directly.
- */
-async function fetchGoModels(_token: string): Promise<{
-	all: ProviderModelConfig[];
-	useStaticFallback: boolean;
-}> {
-	// Go has no /models API - use static models
-	return { all: applyHidden(STATIC_GO_MODELS), useStaticFallback: true };
-}
-
-// =============================================================================
-// Extension Entry Point
-// =============================================================================
-
-export default async function (pi: ExtensionAPI) {
-	const token = CONFIG_API_KEY ?? OPENCODE_API_KEY ?? "";
-	const hasKey = !!token;
-
-	// Go requires an API key (no free tier)
-	if (!hasKey) {
-		pi.on("session_start", async () => {
-			// No key - don't register provider
-		});
-		return;
-	}
-
-	// Use a private env var so we don't conflict with other providers
-	const GO_KEY_VAR = "PI_FREE_GO_API_KEY";
-
-	// Shared model storage
-	const stored: StoredModels = { free: [], all: [] };
-
-	// Re-registration function
-	let reRegisterFn: (models: ProviderModelConfig[]) => void = () => {};
-
-	// Wire up shared boilerplate
-	setupProvider(
-		pi,
-		{
-			providerId: PROVIDER_GO,
-			tosUrl: URL_GO_TOS,
-			hasKey,
-			initialShowPaid: GO_SHOW_PAID,
-			reRegister: (models) => reRegisterFn(models),
-		},
-		stored,
-	);
-
-	// Register provider on session start
-	pi.on("session_start", async (_event, ctx) => {
-		// Set up the env var
-		process.env[GO_KEY_VAR] = token;
-
-		// OpenCode Go has no /models endpoint - use static fallback
-		const models = applyHidden(STATIC_GO_MODELS);
-
-		stored.all = models;
-		stored.free = models; // Go has no free tier
-
-		if (models.length === 0) {
-			return;
-		}
-
-		// Generate session ID for this session (used in headers)
-		const sessionId = session.getSessionId();
-
-		// Create re-register function with session headers (same as zen)
-		const sessionConfig = {
-			...GO_CONFIG,
-			headers: {
-				...GO_CONFIG.headers,
-				"x-opencode-session": sessionId,
-				"x-session-affinity": sessionId,
-			},
-		};
-		reRegisterFn = createCtxReRegister(ctx as any, sessionConfig);
-
-		// Register our provider
-		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_GO) return;
-		session.nextRequestId();
-	});
-}

package/providers/kilo.ts

@@ -1,146 +0,0 @@
-/**
- * Kilo Provider Extension
- *
- * Provides access to 300+ AI models via the Kilo Gateway (OpenRouter-compatible).
- * Free models available immediately; /login kilo for full access.
- *
- * Usage:
- *   pi install git:github.com/apmantza/pi-free
- *   # Then /login kilo, or set KILO_API_KEY=...
- */
-
-import type { Api, Model, OAuthCredentials } from "@mariozechner/pi-ai";
-import type {
-	ExtensionAPI,
-	ProviderModelConfig,
-} from "@mariozechner/pi-coding-agent";
-import { KILO_FREE_ONLY, KILO_SHOW_PAID, PROVIDER_KILO } from "../config.ts";
-import { URL_KILO_TOS } from "../constants.ts";
-import {
-	enhanceWithCI,
-	type StoredModels,
-	setupProvider,
-	createReRegister,
-	createCtxReRegister,
-} from "../provider-helper.ts";
-import { cleanModelName, logWarning } from "../lib/util.ts";
-import { loginKilo, refreshKiloToken } from "./kilo-auth.ts";
-import { fetchKiloModels, KILO_GATEWAY_BASE } from "./kilo-models.ts";
-
-const KILO_PROVIDER_CONFIG = {
-	providerId: PROVIDER_KILO,
-	baseUrl: KILO_GATEWAY_BASE,
-	apiKey: "KILO_API_KEY",
-	headers: {
-		"X-KILOCODE-EDITORNAME": "Pi",
-	},
-};
-
-export default async function (pi: ExtensionAPI) {
-	let freeModels: ProviderModelConfig[] = [];
-	try {
-		freeModels = await fetchKiloModels({ freeOnly: true });
-	} catch (error) {
-		logWarning("kilo", "Failed to fetch free models at startup", error);
-	}
-
-	let cachedAllModels: ProviderModelConfig[] = [];
-	let showPaidModels = KILO_SHOW_PAID;
-
-	// Shared model storage for setupProvider commands
-	const stored: StoredModels = { free: freeModels, all: [] };
-
-	// OAuth config for Kilo (shared across registrations)
-	const oauthConfig = {
-		name: "Kilo",
-		login: async (callbacks: any) => {
-			const cred = await loginKilo(callbacks);
-			try {
-				cachedAllModels = await fetchKiloModels({ token: cred.access });
-				stored.all = cachedAllModels;
-			} catch (error) {
-				logWarning("kilo", "Failed to fetch models after login", error);
-			}
-			return cred;
-		},
-		refreshToken: refreshKiloToken,
-		getApiKey: (cred: OAuthCredentials) => cred.access,
-		modifyModels: (models: Model<Api>[], _cred: OAuthCredentials) => {
-			if (!showPaidModels || KILO_FREE_ONLY || cachedAllModels.length === 0) {
-				return models;
-			}
-			const template = models.find((m) => m.provider === PROVIDER_KILO);
-			if (!template) return models;
-			const nonKilo = models.filter((m) => m.provider !== PROVIDER_KILO);
-			const fullModels = cachedAllModels.map((m) => ({
-				...template,
-				id: m.id,
-				name: cleanModelName(m.name),
-				reasoning: m.reasoning,
-				input: m.input,
-				cost: m.cost,
-				contextWindow: m.contextWindow,
-				maxTokens: m.maxTokens,
-			}));
-			return [...nonKilo, ...fullModels];
-		},
-	};
-
-	// Register initial provider
-	pi.registerProvider(PROVIDER_KILO, {
-		baseUrl: KILO_GATEWAY_BASE,
-		apiKey: "KILO_API_KEY",
-		api: "openai-completions" as const,
-		headers: {
-			"X-KILOCODE-EDITORNAME": "Pi",
-			"User-Agent": "pi-free-providers",
-		},
-		models: enhanceWithCI(freeModels),
-		oauth: oauthConfig,
-	});
-
-	// Wire up shared boilerplate (commands, model_select, turn_end, ToS)
-	const reRegister = createReRegister(pi, {
-		...KILO_PROVIDER_CONFIG,
-		oauth: oauthConfig as any,
-	});
-	setupProvider(
-		pi,
-		{
-			providerId: PROVIDER_KILO,
-			tosUrl: URL_KILO_TOS,
-			initialShowPaid: KILO_SHOW_PAID,
-			reRegister: (models) => {
-				showPaidModels = models === stored.all;
-				reRegister(models);
-			},
-		},
-		stored,
-	);
-
-	// Usage widget temporarily deprecated.
-
-	// ── Kilo-specific: events ────────────────────────────────────────────
-
-	pi.on("session_start", async (_event, ctx) => {
-		const cred = ctx.modelRegistry.authStorage.get(PROVIDER_KILO);
-
-		if (cred?.type === "oauth") {
-			try {
-				cachedAllModels = await fetchKiloModels({ token: cred.access });
-				stored.all = cachedAllModels;
-				if (cachedAllModels.length > 0) {
-					const ctxReRegister = createCtxReRegister(ctx as any, {
-						...KILO_PROVIDER_CONFIG,
-						oauth: oauthConfig as any,
-					});
-					const modelsToShow =
-						showPaidModels && !KILO_FREE_ONLY ? cachedAllModels : freeModels;
-					ctxReRegister(modelsToShow);
-				}
-			} catch (error) {
-				logWarning("kilo", "Failed to fetch models at session start", error);
-			}
-		}
-	});
-}

package/providers/mistral.ts

@@ -1,144 +0,0 @@
-/**
- * Mistral AI Provider Extension
- *
- * Provides access to Mistral's models via api.mistral.ai.
- * Free models available without account; paid models require MISTRAL_API_KEY.
- * Get a key at: https://console.mistral.ai/api-keys
- *
- * By default only free models are shown.
- * Set MISTRAL_SHOW_PAID=true to also include paid models.
- *
- * Note: Mistral has stricter field requirements than OpenAI —
- * only whitelisted fields are sent to avoid 400 errors.
- */
-
-import type { ProviderModelConfig } from "@mariozechner/pi-coding-agent";
-import { applyHidden, MISTRAL_SHOW_PAID, PROVIDER_MISTRAL } from "../config.ts";
-import { BASE_URL_MISTRAL } from "../constants.ts";
-import { createProvider } from "../provider-factory.ts";
-
-// =============================================================================
-// Static model list
-// =============================================================================
-
-function getMistralModels(): ProviderModelConfig[] {
-	return applyHidden([
-		// Free models
-		{
-			id: "mistral-small-latest",
-			name: "Mistral Small",
-			reasoning: false,
-			input: ["text"],
-			cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-			contextWindow: 32768,
-			maxTokens: 32768,
-		},
-		{
-			id: "open-mistral-nemo",
-			name: "Mistral Nemo (Open)",
-			reasoning: false,
-			input: ["text"],
-			cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-			contextWindow: 128000,
-			maxTokens: 128000,
-		},
-		// Paid models
-		{
-			id: "mistral-large-latest",
-			name: "Mistral Large",
-			reasoning: true,
-			input: ["text"],
-			cost: { input: 0.002, output: 0.006, cacheRead: 0, cacheWrite: 0 },
-			contextWindow: 128000,
-			maxTokens: 128000,
-		},
-		{
-			id: "ministral-3b-latest",
-			name: "Ministral 3B",
-			reasoning: false,
-			input: ["text"],
-			cost: { input: 0.00004, output: 0.00004, cacheRead: 0, cacheWrite: 0 },
-			contextWindow: 128000,
-			maxTokens: 128000,
-		},
-		{
-			id: "ministral-8b-latest",
-			name: "Ministral 8B",
-			reasoning: false,
-			input: ["text"],
-			cost: { input: 0.0001, output: 0.0001, cacheRead: 0, cacheWrite: 0 },
-			contextWindow: 128000,
-			maxTokens: 128000,
-		},
-		{
-			id: "codestral-latest",
-			name: "Codestral",
-			reasoning: false,
-			input: ["text"],
-			cost: { input: 0.0003, output: 0.0009, cacheRead: 0, cacheWrite: 0 },
-			contextWindow: 256000,
-			maxTokens: 256000,
-		},
-	]);
-}
-
-// =============================================================================
-// Mistral-specific request filtering
-// =============================================================================
-
-const MISTRAL_ALLOWED_FIELDS = new Set([
-	"model",
-	"messages",
-	"temperature",
-	"max_tokens",
-	"stream",
-	"tools",
-	"tool_choice",
-	"stop",
-	"top_p",
-	"presence_penalty",
-	"frequency_penalty",
-]);
-
-function isMistralPayload(payload: Record<string, unknown>): boolean {
-	const modelId = payload.model as string | undefined;
-	return !!modelId && (modelId.includes("mistral") || modelId.includes("nemo"));
-}
-
-function filterMistralPayload(
-	payload: Record<string, unknown>,
-): Record<string, unknown> {
-	const filtered: Record<string, unknown> = {};
-	for (const [key, value] of Object.entries(payload)) {
-		if (MISTRAL_ALLOWED_FIELDS.has(key)) {
-			filtered[key] = value;
-		}
-	}
-	return filtered;
-}
-
-// =============================================================================
-// Extension Entry Point
-// =============================================================================
-
-export default function (pi: Parameters<typeof createProvider>[0]) {
-	const allModels = getMistralModels();
-	const models = MISTRAL_SHOW_PAID
-		? allModels
-		: allModels.filter((m) => (m.cost?.input ?? 0) === 0);
-
-	// Override fetchModels to use our already-fetched models
-	return createProvider(pi, {
-		providerId: PROVIDER_MISTRAL,
-		baseUrl: BASE_URL_MISTRAL,
-		apiKeyEnvVar: "MISTRAL_API_KEY",
-		apiKeyConfigKey: "mistral_api_key",
-		fetchModels: async () => models,
-		beforeProviderRequest: (payload) => {
-			if (isMistralPayload(payload)) {
-				return filterMistralPayload(payload);
-			}
-			return undefined;
-		},
-	});
-}

package/providers/ollama.ts

@@ -1,113 +0,0 @@
-/**
- * Ollama Cloud Provider Extension
- *
- * Provides access to Ollama's cloud-hosted models via ollama.com/v1.
- * Free tier available with usage limits (resets every 5 hours + 7 days).
- * Requires OLLAMA_API_KEY from https://ollama.com/settings/keys
- *
- * Set OLLAMA_SHOW_PAID=true to enable (required since Ollama has usage limits).
- */
-
-import type { ProviderModelConfig } from "@mariozechner/pi-coding-agent";
-import { OLLAMA_API_KEY, PROVIDER_OLLAMA } from "../config.ts";
-import { BASE_URL_OLLAMA, DEFAULT_FETCH_TIMEOUT_MS } from "../constants.ts";
-import { fetchWithRetry } from "../lib/util.ts";
-import { createProvider } from "../provider-factory.ts";
-
-// =============================================================================
-// Fetch + map
-// =============================================================================
-
-async function fetchOllamaModels(): Promise<ProviderModelConfig[]> {
-	const apiKey = OLLAMA_API_KEY;
-	if (!apiKey) return [];
-
-	const response = await fetchWithRetry(
-		`${BASE_URL_OLLAMA}/models`,
-		{
-			headers: {
-				Authorization: `Bearer ${apiKey}`,
-				"User-Agent": "pi-free-providers",
-			},
-		},
-		3,
-		1000,
-		DEFAULT_FETCH_TIMEOUT_MS,
-	);
-
-	if (!response.ok) {
-		throw new Error(
-			`Failed to fetch Ollama models: ${response.status} ${response.statusText}`,
-		);
-	}
-
-	const json = (await response.json()) as {
-		data?: Array<{
-			id: string;
-			object?: string;
-			created?: number;
-			owned_by?: string;
-		}>;
-	};
-	const models = json.data ?? [];
-
-	// Filter out small models (< 30B) to keep list focused
-	return models
-		.filter((m) => {
-			const sizeMatch = m.id.match(/:(\d+)([bmt])/i);
-			if (sizeMatch) {
-				const size = parseInt(sizeMatch[1], 10);
-				const unit = sizeMatch[2].toLowerCase();
-				if (unit === "b" && size < 30) return false;
-			}
-			return true;
-		})
-		.map(mapOllamaModel);
-}
-
-function mapOllamaModel(m: { id: string }): ProviderModelConfig {
-	// Extract context window from parameter size
-	let contextWindow = 131072; // Default 128k
-	const sizeMatch = m.id.match(/:(\d+)([bmt])/i);
-	if (sizeMatch) {
-		const size = parseInt(sizeMatch[1], 10);
-		const unit = sizeMatch[2].toLowerCase();
-		if (unit === "b" && size >= 100) {
-			contextWindow = 200000;
-		}
-	}
-
-	// Clean up name for display
-	const displayName = m.id
-		.replace(/:/g, " ")
-		.replace(/-/g, " ")
-		.split(" ")
-		.filter((w) => w.length > 0)
-		.map((w) => w.charAt(0).toUpperCase() + w.slice(1))
-		.join(" ");
-
-	return {
-		id: m.id,
-		name: displayName,
-		reasoning: m.id.includes("deepseek") || m.id.includes("r1"),
-		input: ["text"],
-		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-		contextWindow,
-		maxTokens: Math.min(contextWindow / 2, 131072),
-	};
-}
-
-// =============================================================================
-// Extension Entry Point
-// =============================================================================
-
-export default function (pi: Parameters<typeof createProvider>[0]) {
-	return createProvider(pi, {
-		providerId: PROVIDER_OLLAMA,
-		baseUrl: BASE_URL_OLLAMA,
-		apiKeyEnvVar: "OLLAMA_API_KEY",
-		apiKeyConfigKey: "ollama_api_key",
-		showPaidFlag: "OLLAMA_SHOW_PAID",
-		fetchModels: fetchOllamaModels,
-	});
-}