pi-free 1.0.8 → 2.0.0

package/config.ts

@@ -5,67 +5,75 @@
  *   1. Environment variable
  *   2. ~/.pi/free.json
  *
- * Per-provider paid model flags:
- *   KILO_SHOW_PAID=true or kilo_show_paid: true
- *   OPENROUTER_SHOW_PAID=true or openrouter_show_paid: true
- *   NVIDIA_SHOW_PAID=true or nvidia_show_paid: true
- *   FIREWORKS_SHOW_PAID=true or fireworks_show_paid: true
- *   CLINE_SHOW_PAID=true or cline_show_paid: true
- *   GO_SHOW_PAID=true or go_show_paid: true
- *   OLLAMA_SHOW_PAID=true or ollama_show_paid: true
- *
- * PI_FREE_KILO_FREE_ONLY=true — restrict Kilo to free models even after login.
+ * All exported values are getter functions so that runtime changes
+ * (e.g. after /{provider}-toggle or /free) are visible immediately.
  */
 
 import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
 import { join } from "node:path";
+import {
+	PROVIDER_CLINE,
+	PROVIDER_KILO,
+	PROVIDER_MODAL,
+	PROVIDER_NVIDIA,
+	PROVIDER_QWEN,
+} from "./constants.ts";
 import { createLogger } from "./lib/logger.ts";
 
 const _logger = createLogger("config");
 
 interface PiFreeConfig {
-	openrouter_api_key?: string;
 	nvidia_api_key?: string;
-	opencode_api_key?: string;
-	opencode_go_api_key?: string;
-	fireworks_api_key?: string;
-	mistral_api_key?: string;
+	cloudflare_api_token?: string;
+	cloudflare_account_id?: string;
 	ollama_api_key?: string;
 	modal_api_key?: string;
+	opencode_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[];
-	// Per-provider paid model flags
+	free_only?: boolean;
 	kilo_show_paid?: boolean;
-	openrouter_show_paid?: boolean;
 	nvidia_show_paid?: boolean;
-	fireworks_show_paid?: boolean;
-	cline_show_paid?: boolean;
-	zen_show_paid?: boolean;
-	go_show_paid?: boolean;
-	mistral_show_paid?: boolean;
+	cloudflare_show_paid?: boolean;
 	ollama_show_paid?: boolean;
+	cline_show_paid?: boolean;
+	qwen_show_paid?: boolean;
+	modal_show_paid?: boolean;
+	openrouter_show_paid?: boolean;
+	opencode_show_paid?: boolean;
 }
 
 const CONFIG_TEMPLATE: PiFreeConfig = {
-	openrouter_api_key: "",
 	nvidia_api_key: "",
-	opencode_api_key: "",
-	opencode_go_api_key: "",
-	fireworks_api_key: "",
-	mistral_api_key: "",
+	cloudflare_api_token: "",
+	cloudflare_account_id: "",
 	ollama_api_key: "",
 	modal_api_key: "",
+	opencode_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,
-	openrouter_show_paid: false,
 	nvidia_show_paid: false,
-	fireworks_show_paid: false,
-	cline_show_paid: false,
-	zen_show_paid: false,
-	go_show_paid: false,
-	mistral_show_paid: false,
+	cloudflare_show_paid: false,
 	ollama_show_paid: false,
+	cline_show_paid: false,
+	qwen_show_paid: false,
+	modal_show_paid: false,
+	openrouter_show_paid: false,
+	opencode_show_paid: false,
 };
 
 const PI_DIR = join(process.env.HOME || process.env.USERPROFILE || "", ".pi");
@@ -75,7 +83,6 @@ function ensureConfigFile(): void {
 	try {
 		mkdirSync(PI_DIR, { recursive: true });
 		if (existsSync(CONFIG_PATH)) {
-			// Merge: add any new template keys without touching existing values
 			const existing = JSON.parse(
 				readFileSync(CONFIG_PATH, "utf8"),
 			) as PiFreeConfig;
@@ -102,7 +109,7 @@ function ensureConfigFile(): void {
 	}
 }
 
-function loadConfigFile(): PiFreeConfig {
+export function loadConfigFile(): PiFreeConfig {
 	try {
 		return JSON.parse(readFileSync(CONFIG_PATH, "utf8")) as PiFreeConfig;
 	} catch {
@@ -111,16 +118,13 @@ function loadConfigFile(): PiFreeConfig {
 }
 
 ensureConfigFile();
-const file = loadConfigFile();
 
 // Resolve each value: env var takes priority over config file.
-// Treat empty strings in the config file as unset.
 function resolve(envKey: string, fileVal?: string): string | undefined {
 	return process.env[envKey] || (fileVal?.trim() ? fileVal : undefined);
 }
 
 // Resolve boolean flag: env var takes priority, then config file.
-// If neither is set, defaults to false (free-only mode).
 function resolveBool(envKey: string, fileVal?: boolean): boolean {
 	const envValue = process.env[envKey];
 	if (envValue === "true") return true;
@@ -128,104 +132,134 @@ function resolveBool(envKey: string, fileVal?: boolean): boolean {
 	return fileVal === true;
 }
 
-// Global fallback (deprecated, use per-provider flags)
-// Returns true only if explicitly enabled via env var
-export const SHOW_PAID = process.env.PI_FREE_SHOW_PAID === "true";
+// =============================================================================
+// Per-provider paid-model flags (getters so toggles reflect immediately)
+// =============================================================================
 
-// Per-provider paid model flags - default to false (free-only) if not set
-export const KILO_SHOW_PAID = resolveBool(
-	"KILO_SHOW_PAID",
-	file.kilo_show_paid,
-);
+export function getKiloShowPaid(): boolean {
+	return resolveBool("KILO_SHOW_PAID", loadConfigFile().kilo_show_paid);
+}
 
-export const OPENROUTER_SHOW_PAID = resolveBool(
-	"OPENROUTER_SHOW_PAID",
-	file.openrouter_show_paid,
-);
+export function getNvidiaShowPaid(): boolean {
+	return resolveBool("NVIDIA_SHOW_PAID", loadConfigFile().nvidia_show_paid);
+}
 
-export const NVIDIA_SHOW_PAID = resolveBool(
-	"NVIDIA_SHOW_PAID",
-	file.nvidia_show_paid,
-);
+export function getClineShowPaid(): boolean {
+	return resolveBool("CLINE_SHOW_PAID", loadConfigFile().cline_show_paid);
+}
 
-export const FIREWORKS_SHOW_PAID = resolveBool(
-	"FIREWORKS_SHOW_PAID",
-	file.fireworks_show_paid,
-);
+/** @deprecated Qwen provider is deprecated. */
+export function getQwenShowPaid(): boolean {
+	return resolveBool("QWEN_SHOW_PAID", loadConfigFile().qwen_show_paid);
+}
 
-export const CLINE_SHOW_PAID = resolveBool(
-	"CLINE_SHOW_PAID",
-	file.cline_show_paid,
-);
+export function getModalShowPaid(): boolean {
+	return resolveBool("MODAL_SHOW_PAID", loadConfigFile().modal_show_paid);
+}
 
-export const ZEN_SHOW_PAID = resolveBool("ZEN_SHOW_PAID", file.zen_show_paid);
+export function getOllamaShowPaid(): boolean {
+	return resolveBool("OLLAMA_SHOW_PAID", loadConfigFile().ollama_show_paid);
+}
 
-export const GO_SHOW_PAID = resolveBool("GO_SHOW_PAID", file.go_show_paid);
+export function getCloudflareShowPaid(): boolean {
+	return resolveBool(
+		"CLOUDFLARE_SHOW_PAID",
+		loadConfigFile().cloudflare_show_paid,
+	);
+}
 
-export const MISTRAL_SHOW_PAID = resolveBool(
-	"MISTRAL_SHOW_PAID",
-	file.mistral_show_paid,
-);
+export function getOpenrouterShowPaid(): boolean {
+	return resolveBool(
+		"OPENROUTER_SHOW_PAID",
+		loadConfigFile().openrouter_show_paid,
+	);
+}
 
-export const OLLAMA_SHOW_PAID = resolveBool(
-	"OLLAMA_SHOW_PAID",
-	file.ollama_show_paid,
-);
+export function getOpencodeShowPaid(): boolean {
+	return resolveBool("OPENCODE_SHOW_PAID", loadConfigFile().opencode_show_paid);
+}
 
-export const KILO_FREE_ONLY = resolveBool(
-	"PI_FREE_KILO_FREE_ONLY",
-	file.kilo_free_only,
-);
+// =============================================================================
+// Global free-only mode
+// =============================================================================
 
-const HIDDEN: Set<string> = new Set(file.hidden_models ?? []);
+export function getFreeOnly(): boolean {
+	return resolveBool("PI_FREE_ONLY", loadConfigFile().free_only);
+}
 
-/** Removes any models whose id appears in hidden_models. */
-export function applyHidden<T extends { id: string }>(models: T[]): T[] {
-	if (HIDDEN.size === 0) return models;
-	return models.filter((m) => !HIDDEN.has(m.id));
-}
-
-export const OPENROUTER_API_KEY = resolve(
-	"OPENROUTER_API_KEY",
-	file.openrouter_api_key,
-);
-export const NVIDIA_API_KEY = resolve("NVIDIA_API_KEY", file.nvidia_api_key);
-export const OPENCODE_API_KEY = resolve(
-	"OPENCODE_API_KEY",
-	file.opencode_api_key,
-);
-export const OPENCODE_GO_API_KEY = resolve(
-	"OPENCODE_GO_API_KEY",
-	file.opencode_go_api_key,
-);
-export const FIREWORKS_API_KEY = resolve(
-	"FIREWORKS_API_KEY",
-	file.fireworks_api_key,
-);
-export const MISTRAL_API_KEY = resolve("MISTRAL_API_KEY", file.mistral_api_key);
-export const OLLAMA_API_KEY = resolve("OLLAMA_API_KEY", file.ollama_api_key);
-export const MODAL_API_KEY = resolve("MODAL_API_KEY", file.modal_api_key);
+export function getKiloFreeOnly(): boolean {
+	return resolveBool("PI_FREE_KILO_FREE_ONLY", loadConfigFile().kilo_free_only);
+}
 
-// Re-export provider names for consistency
-export {
-	PROVIDER_CLINE,
-	PROVIDER_FIREWORKS,
-	PROVIDER_GO,
-	PROVIDER_KILO,
-	PROVIDER_MISTRAL,
-	PROVIDER_NVIDIA,
-	PROVIDER_OLLAMA,
-	PROVIDER_OPENROUTER,
-	PROVIDER_QWEN,
-	PROVIDER_ZEN,
-	PROVIDER_MODAL,
-} from "./constants.ts";
+// =============================================================================
+// API Keys (getters so runtime config changes are visible)
+// =============================================================================
+
+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 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 getOpencodeApiKey(): string | undefined {
+	return resolve("OPENCODE_API_KEY", loadConfigFile().opencode_api_key);
+}
+
+export function getMistralApiKey(): string | undefined {
+	return resolve("MISTRAL_API_KEY", loadConfigFile().mistral_api_key);
+}
+
+export function getGroqApiKey(): string | undefined {
+	return resolve("GROQ_API_KEY", loadConfigFile().groq_api_key);
+}
 
+export function getCerebrasApiKey(): string | undefined {
+	return resolve("CEREBRAS_API_KEY", loadConfigFile().cerebras_api_key);
+}
+
+export function getXaiApiKey(): string | undefined {
+	return resolve("XAI_API_KEY", loadConfigFile().xai_api_key);
+}
+
+export function getHfToken(): string | undefined {
+	return resolve("HF_TOKEN", loadConfigFile().hf_token);
+}
+
+export function getOpenrouterApiKey(): string | undefined {
+	return resolve("OPENROUTER_API_KEY", loadConfigFile().openrouter_api_key);
+}
+
+// =============================================================================
+// Hidden models (re-reads config on every call)
 // =============================================================================
-// Config Persistence
+
+export function applyHidden<T extends { id: string }>(models: T[]): T[] {
+	const hidden = new Set(loadConfigFile().hidden_models ?? []);
+	if (hidden.size === 0) return models;
+	return models.filter((m) => !hidden.has(m.id));
+}
+
+// =============================================================================
+// Persistence
 // =============================================================================
 
-/** Save updated config values to ~/.pi/free.json */
 export function saveConfig(updates: Partial<PiFreeConfig>): void {
 	try {
 		const existing = loadConfigFile();
@@ -243,7 +277,18 @@ export function saveConfig(updates: Partial<PiFreeConfig>): void {
 	}
 }
 
-/** Get current config values (for checking state) */
 export function getConfig(): PiFreeConfig {
 	return loadConfigFile();
 }
+
+// =============================================================================
+// Re-export provider names for consistency
+// =============================================================================
+
+export {
+	PROVIDER_CLINE,
+	PROVIDER_KILO,
+	PROVIDER_MODAL,
+	PROVIDER_NVIDIA,
+	PROVIDER_QWEN,
+};

package/constants.ts

@@ -4,31 +4,23 @@
  */
 
 // =============================================================================
-// Provider names (must match registerProvider calls)
+// Provider names (unique providers NOT built into pi)
 // =============================================================================
 
 export const PROVIDER_KILO = "kilo";
-export const PROVIDER_ZEN = "zen";
-export const PROVIDER_GO = "go";
-export const PROVIDER_OPENROUTER = "openrouter";
-export const PROVIDER_NVIDIA = "nvidia";
 export const PROVIDER_CLINE = "cline";
-export const PROVIDER_FIREWORKS = "fireworks";
-export const PROVIDER_OLLAMA = "ollama";
-export const PROVIDER_MISTRAL = "mistral";
+export const PROVIDER_NVIDIA = "nvidia";
+export const PROVIDER_CLOUDFLARE = "cloudflare";
+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 ALL_PROVIDERS = [
+export const ALL_UNIQUE_PROVIDERS = [
 	PROVIDER_KILO,
-	PROVIDER_ZEN,
-	PROVIDER_GO,
-	PROVIDER_OPENROUTER,
-	PROVIDER_NVIDIA,
 	PROVIDER_CLINE,
-	PROVIDER_FIREWORKS,
-	PROVIDER_MISTRAL,
-	PROVIDER_OLLAMA,
+	PROVIDER_NVIDIA,
+	/** @deprecated Qwen free tier no longer available */
 	PROVIDER_QWEN,
 	PROVIDER_MODAL,
 ] as const;
@@ -38,14 +30,16 @@ export const ALL_PROVIDERS = [
 // =============================================================================
 
 export const BASE_URL_KILO = "https://api.kilo.ai/api/gateway";
-export const BASE_URL_ZEN = "https://opencode.ai/zen/v1";
-export const BASE_URL_GO = "https://opencode.ai/zen/go/v1";
-export const BASE_URL_OPENROUTER = "https://openrouter.ai/api/v1";
 export const BASE_URL_NVIDIA = "https://integrate.api.nvidia.com/v1";
+export const BASE_URL_CLOUDFLARE = "https://api.cloudflare.com/client/v4";
+export const BASE_URL_OLLAMA = "https://ollama.com/v1"; // OpenAI-compatible API endpoint
 export const BASE_URL_CLINE = "https://api.cline.bot/api/v1";
-export const BASE_URL_FIREWORKS = "https://api.fireworks.ai/inference/v1";
-export const BASE_URL_OLLAMA = "https://ollama.com/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";
+
+/** Cline fetches free models from OpenRouter */
+export const BASE_URL_OPENROUTER = "https://openrouter.ai/api/v1";
 
 // =============================================================================
 // External URLs
@@ -53,12 +47,9 @@ export const BASE_URL_MODAL = "https://api.us-west-2.modal.direct/v1";
 
 export const URL_MODELS_DEV = "https://models.dev/api.json";
 export const URL_KILO_TOS = "https://kilo.ai/terms";
-export const URL_ZEN_TOS = "https://opencode.ai/terms";
-export const URL_GO_TOS = "https://opencode.ai/terms";
 export const URL_CLINE_TOS = "https://cline.bot/tos";
 export const URL_QWEN_TOS = "https://terms.alicloud.com/";
 export const URL_MODAL_TOS = "https://modal.com/terms";
-export const BASE_URL_QWEN = "https://dashscope.aliyuncs.com/compatible-mode/v1";
 
 // =============================================================================
 // Cline auth
@@ -77,47 +68,17 @@ export const DEFAULT_MIN_SIZE_B = 30; // Default minimum model size for filterin
 // Timeouts (milliseconds)
 // =============================================================================
 
-// Timeout for fetch operations
+/** Timeout for fetch operations */
 export const DEFAULT_FETCH_TIMEOUT_MS: number = 10_000;
 
-export interface TestConfig {
-	timeout: number;
-	retries: number;
-	label: string;
-}
-
-// LSP test - fixed - added missing property
-export const testConfig: TestConfig = {
-	timeout: 5000,
-	retries: 3,
-	label: "test",
-};
-
-// LSP test - fixed return type
-export function calculateTimeout(base: number): number {
-	return base * 2;
-}
-
-// LSP test - unused variable (should show hint/warning if configured)
-export function unusedParamTest(required: string, _unused: number): string {
-	return required.toUpperCase();
-}
 export const KILO_POLL_INTERVAL_MS = 3_000;
 export const KILO_TOKEN_EXPIRATION_MS = 365 * 24 * 60 * 60 * 1000; // 1 year
 
 // =============================================================================
-// Additional OpenAI-compatible providers
+// Removed providers (now built into pi):
+// - openrouter: use pi's built-in with OPENROUTER_API_KEY
+// - zen/opencode: use pi's built-in with OPENCODE_API_KEY
+// - go/opencode-go: use pi's built-in with OPENCODE_API_KEY
+// - mistral: use pi's built-in with MISTRAL_API_KEY
+// - ollama: add to ~/.pi/agent/models.json as custom provider
 // =============================================================================
-
-export const PROVIDER_GROQ = "groq";
-export const PROVIDER_TOGETHER = "together";
-export const PROVIDER_DEEPINFRA = "deepinfra";
-export const PROVIDER_PERPLEXITY = "perplexity";
-export const PROVIDER_XAI = "xai";
-
-export const BASE_URL_GROQ = "https://api.groq.com/openai/v1";
-export const BASE_URL_TOGETHER = "https://api.together.xyz/v1";
-export const BASE_URL_DEEPINFRA = "https://api.deepinfra.com/v1/openai";
-export const BASE_URL_MISTRAL = "https://api.mistral.ai/v1";
-export const BASE_URL_PERPLEXITY = "https://api.perplexity.ai";
-export const BASE_URL_XAI = "https://api.x.ai/v1";