pi-free 1.0.8 → 2.0.0

package/providers/qwen/qwen.ts

@@ -0,0 +1,202 @@
+/**
+ * Qwen OAuth Provider Extension
+ *
+ * @deprecated This provider is deprecated. Qwen no longer offers the 1,000 free API calls/day tier.
+ * The provider remains functional for existing authenticated users but new free tier registrations
+ * are not supported. Consider using other free providers like Kilo, Cline, or NVIDIA instead.
+ *
+ * Original description (now outdated):
+ * ~~Provides free access to Qwen 3.6 Plus via OAuth device flow.
+ * 1,000 free API calls/day — run /login qwen to authenticate.~~
+ */
+
+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 { 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";
+
+// =============================================================================
+// 401 detection patterns
+// =============================================================================
+
+/** Patterns that indicate an auth failure requiring token refresh. */
+const AUTH_ERROR_PATTERNS = [
+	"invalid access token",
+	"token expired",
+	"401",
+	"unauthorized",
+	"authentication",
+] as const;
+
+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) {
+	// DEPRECATION WARNING
+	_logger.warn("Qwen provider is deprecated. The 1,000 req/day free tier is no longer available.");
+
+	// 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;
+			// modifyModels receives ALL models across providers — only patch Qwen ones.
+			const nonQwen = models.filter((m) => m.provider !== PROVIDER_QWEN);
+			const qwen = models
+				.filter((m) => m.provider === PROVIDER_QWEN)
+				.map((m) => ({ ...m, baseUrl }));
+			return [...nonQwen, ...qwen];
+		},
+	};
+
+	// 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,
+	);
+
+	// Request counting + 401 auth-error detection with forced token refresh.
+	//
+	// When Qwen returns a 401 / "invalid access token" error, the stored token
+	// may have been revoked server-side even though its expiry hasn't been reached.
+	// We force-expire the credential in auth storage so that pi-core's next
+	// getApiKey() call will trigger a token refresh via refreshToken().
+	// This mirrors qwen-code's executeWithCredentialManagement() retry logic.
+	//
+	pi.on("turn_end", async (event, ctx) => {
+		if (ctx.model?.provider !== PROVIDER_QWEN) return;
+		// NOTE: Request counting removed - usage tracking was deleted in refactor
+
+		const msg = (
+			event as { message?: { role?: string; errorMessage?: string } }
+		).message;
+
+		if (msg?.role === "assistant" && msg.errorMessage) {
+			const errLower = msg.errorMessage.toLowerCase();
+			const isAuthError = AUTH_ERROR_PATTERNS.some((p) =>
+				errLower.includes(p),
+			);
+
+			if (isAuthError) {
+				_logger.warn(
+					"Qwen auth error detected, force-expiring token for refresh",
+					{ error: msg.errorMessage.slice(0, 100) },
+				);
+
+				// Force-expire the stored credential so the next getApiKey() call
+				// triggers refreshQwenToken(). The credential object in auth.json
+				// is updated with expires = 0 (already past).
+				try {
+					const authStorage =
+						(ctx as any).modelRegistry?.authStorage;
+					if (authStorage) {
+						const cred = authStorage.get(PROVIDER_QWEN);
+						if (cred?.type === "oauth" && cred.expires > Date.now()) {
+							// Set expiry to 0 to force refresh on next request
+							authStorage.set(PROVIDER_QWEN, {
+								...cred,
+								expires: 0,
+							});
+							_logger.info(
+								"Qwen token force-expired; will refresh on next request",
+							);
+						}
+					}
+					ctx.ui.notify(
+						"Qwen: auth error detected, refreshing token…",
+						"warning",
+					);
+				} catch (e) {
+					_logger.warn("Failed to force-expire Qwen token", {
+						error: e instanceof Error ? e.message : String(e),
+					});
+				}
+			}
+		}
+	});
+}

package/provider-failover/auto-switch.ts

@@ -1,350 +0,0 @@
-/**
- * Auto-switch failover for pi-free-providers.
- * 
- * When a provider hits a 429 or capacity error, this module finds
- * an equivalent or similar model from another provider and switches to it.
- * 
- * Strategy:
- * 1. Extract the base model name/family from the failed model
- * 2. Search all available models for the same model (different provider)
- * 3. If not found, find a similar model in the same family
- * 4. If not found, find any free model with similar capability
- */
-
-import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
-import type { Model } from "@mariozechner/pi-ai";
-import { createLogger } from "../lib/logger.ts";
-import {
-	detectModelFamily,
-	normalizeModelName,
-	toModelInfo,
-	type ModelInfo,
-} from "../lib/model-detection.ts";
-import { getHardcodedScore } from "./hardcoded-benchmarks.ts";
-
-const _logger = createLogger("auto-switch");
-
-export interface AutoSwitchConfig {
-	/** Whether to enable auto-switching (can be disabled by user) */
-	enabled: boolean;
-	/** Maximum CI score degradation allowed (e.g., 10 = can drop up to 10 points) */
-	maxCIScoreDrop: number;
-	/** Provider priority for fallback (preferred first) */
-	providerPriority: string[];
-}
-
-const DEFAULT_CONFIG: AutoSwitchConfig = {
-	enabled: true,
-	maxCIScoreDrop: 15,
-	providerPriority: ["zen", "go", "kilo", "openrouter", "nvidia", "fireworks", "mistral", "ollama", "cline"],
-};
-
-export interface AutoSwitchResult {
-	success: boolean;
-	switched: boolean;
-	message: string;
-	fallbackModel?: ModelInfo;
-}
-
-interface CandidateModel {
-	model: Model<any>;
-	modelInfo: ModelInfo;
-	ciScore: number;
-	normalizedName: string;
-	family: string;
-}
-
-/**
- * Find a fallback model when the current provider fails.
- * 
- * Priority order:
- * 1. Same model name from different provider (best match)
- * 2. Same model family, prefer free models
- * 3. Any free model with similar CI score
- */
-export async function findFallbackModel(
-	failedModel: Model<any>,
-	availableModels: Model<any>[],
-	config: Partial<AutoSwitchConfig> = {},
-): Promise<CandidateModel | null> {
-	const fullConfig = { ...DEFAULT_CONFIG, ...config };
-
-	// Convert to ModelInfo for internal processing
-	const failedModelInfo = toModelInfo(failedModel);
-	const failedFamily = detectModelFamily(failedModelInfo);
-	const failedNormalizedName = normalizeModelName(failedModelInfo.name || failedModelInfo.id);
-	const failedCIScore = getHardcodedScore(failedModel.name || "", failedModel.id) ?? 20;
-
-	_logger.info("Finding fallback model", {
-		failedModel: failedModel.id,
-		failedProvider: failedModel.provider,
-		failedFamily: failedFamily?.familyId,
-		failedNormalizedName: failedNormalizedName,
-		failedCIScore,
-	});
-
-	// Build candidate list
-	const candidates: CandidateModel[] = [];
-
-	for (const candidate of availableModels) {
-		// Skip the same provider
-		if (candidate.provider === failedModel.provider) continue;
-
-		// Skip if no auth configured for this provider
-		// (We'll assume available models have auth, but check anyway)
-		if (!candidate.baseUrl) continue;
-
-		const modelInfo = toModelInfo(candidate);
-		const family = detectModelFamily(modelInfo);
-		const normalizedName = normalizeModelName(modelInfo.name || modelInfo.id);
-		const ciScore = getHardcodedScore(candidate.name || "", candidate.id) ?? 20;
-
-		candidates.push({
-			model: candidate,
-			modelInfo,
-			ciScore,
-			normalizedName,
-			family: family?.familyId ?? "other",
-		});
-	}
-
-	if (candidates.length === 0) {
-		_logger.info("No candidate models found");
-		return null;
-	}
-
-	// Priority 1: Same model name (different provider)
-	// e.g., "minimax-m2.5" on zen → "minimax-m2.5" on openrouter
-	const sameName = candidates.find(
-		(c) => c.normalizedName === failedNormalizedName && c.model.provider !== failedModel.provider,
-	);
-	if (sameName) {
-		_logger.info("Found exact model match", {
-			provider: sameName.model.provider,
-			model: sameName.model.id,
-		});
-		return sameName;
-	}
-
-	// Priority 2: Same model family, prefer free models
-	const sameFamily = candidates
-		.filter((c) => c.family === failedFamily?.familyId && c.model.provider !== failedModel.provider)
-		.sort((a, b) => {
-			// Prefer free models
-			if (a.modelInfo.isFree !== b.modelInfo.isFree) {
-				return a.modelInfo.isFree ? -1 : 1;
-			}
-			// Then by CI score
-			return b.ciScore - a.ciScore;
-		});
-
-	if (sameFamily.length > 0) {
-		const best = sameFamily[0]!;
-		_logger.info("Found same family model", {
-			provider: best.model.provider,
-			model: best.model.id,
-			family: best.family,
-			isFree: best.modelInfo.isFree,
-			ciScore: best.ciScore,
-		});
-		return best;
-	}
-
-	// Priority 3: Any free model with similar CI score
-	// Check CI score degradation limit
-	const freeCandidates = candidates
-		.filter((c) => {
-			// Must be free
-			if (!c.modelInfo.isFree) return false;
-			// Must not drop CI score too much
-			const ciDrop = failedCIScore - c.ciScore;
-			return ciDrop <= fullConfig.maxCIScoreDrop;
-		})
-		.sort((a, b) => {
-			// Sort by CI score (closest to failed model first)
-			return b.ciScore - a.ciScore;
-		});
-
-	if (freeCandidates.length > 0) {
-		const best = freeCandidates[0]!;
-		_logger.info("Found free fallback model", {
-			provider: best.model.provider,
-			model: best.model.id,
-			ciScore: best.ciScore,
-			ciDrop: failedCIScore - best.ciScore,
-		});
-		return best;
-	}
-
-	// Priority 4: Any free model (no CI limit)
-	const anyFree = candidates
-		.filter((c) => c.modelInfo.isFree)
-		.sort((a, b) => b.ciScore - a.ciScore);
-
-	if (anyFree.length > 0) {
-		const best = anyFree[0]!;
-		_logger.info("Found any free fallback", {
-			provider: best.model.provider,
-			model: best.model.id,
-			ciScore: best.ciScore,
-		});
-		return best;
-	}
-
-	// Priority 5: Any model with similar CI score
-	const similarCI = candidates
-		.filter((c) => {
-			const ciDrop = failedCIScore - c.ciScore;
-			return ciDrop <= fullConfig.maxCIScoreDrop;
-		})
-		.sort((a, b) => {
-			// Prefer providers in priority order
-			const aPriority = fullConfig.providerPriority.indexOf(a.model.provider);
-			const bPriority = fullConfig.providerPriority.indexOf(b.model.provider);
-			if (aPriority !== bPriority && aPriority >= 0 && bPriority >= 0) {
-				return aPriority - bPriority;
-			}
-			// Then by CI score
-			return b.ciScore - a.ciScore;
-		});
-
-	if (similarCI.length > 0) {
-		const best = similarCI[0]!;
-		_logger.info("Found similar CI fallback", {
-			provider: best.model.provider,
-			model: best.model.id,
-			ciScore: best.ciScore,
-		});
-		return best;
-	}
-
-	// Last resort: any model, prefer by provider priority
-	const anyModel = candidates.sort((a, b) => {
-		const aPriority = fullConfig.providerPriority.indexOf(a.model.provider);
-		const bPriority = fullConfig.providerPriority.indexOf(b.model.provider);
-		if (aPriority !== bPriority && aPriority >= 0 && bPriority >= 0) {
-			return aPriority - bPriority;
-		}
-		return b.ciScore - a.ciScore;
-	});
-
-	if (anyModel.length > 0) {
-		const best = anyModel[0]!;
-		_logger.info("Found any fallback", {
-			provider: best.model.provider,
-			model: best.model.id,
-			ciScore: best.ciScore,
-		});
-		return best;
-	}
-
-	return null;
-}
-
-/**
- * Perform automatic failover when a provider hits an error.
- * Returns the result of the switch attempt.
- */
-export async function autoFailover(
-	_errorMessage: string,
-	failedModel: Model<any>,
-	pi: ExtensionAPI,
-	ctx: ExtensionContext,
-	config: Partial<AutoSwitchConfig> = {},
-): Promise<AutoSwitchResult> {
-	const fullConfig = { ...DEFAULT_CONFIG, ...config };
-	if (!fullConfig.enabled) {
-		return {
-			success: false,
-			switched: false,
-			message: "Auto-switch disabled",
-		};
-	}
-
-	// Get all available models
-	const availableModels = ctx.modelRegistry.getAvailable();
-
-	if (availableModels.length === 0) {
-		return {
-			success: false,
-			switched: false,
-			message: "No alternative models available",
-		};
-	}
-
-	// Find fallback model
-	const fallback = await findFallbackModel(
-		failedModel,
-		availableModels,
-		fullConfig,
-	);
-
-	if (!fallback) {
-		return {
-			success: false,
-			switched: false,
-			message: `No fallback model found for ${failedModel.provider}/${failedModel.id}`,
-		};
-	}
-
-	// Attempt to switch
-	const success = await pi.setModel(fallback.model);
-
-	if (success) {
-		const freeStatus = fallback.modelInfo.isFree ? " (free)" : "";
-		return {
-			success: true,
-			switched: true,
-			message: `Switched from ${failedModel.provider} to ${fallback.model.provider}/${fallback.model.id}${freeStatus}`,
-			fallbackModel: fallback.modelInfo,
-		};
-	} else {
-		return {
-			success: false,
-			switched: false,
-			message: `Failed to switch to ${fallback.model.provider}/${fallback.model.id} (no API key?)`,
-			fallbackModel: fallback.modelInfo,
-		};
-	}
-}
-
-/**
- * Check if a model is available from multiple providers
- */
-export function getModelAvailability(
-	modelId: string,
-	availableModels: Model<any>[],
-): string[] {
-	const normalizedName = normalizeModelName(modelId);
-
-	return availableModels
-		.filter((m) => {
-			const mNormalized = normalizeModelName(m.name || m.id);
-			return mNormalized === normalizedName;
-		})
-		.map((m) => m.provider);
-}
-
-/**
- * Get a summary of available models grouped by family
- */
-export function getModelAvailabilitySummary(
-	availableModels: Model<any>[],
-): Map<string, string[]> {
-	const families = new Map<string, string[]>();
-
-	for (const model of availableModels) {
-		const modelInfo = toModelInfo(model);
-		const family = detectModelFamily(modelInfo);
-
-		if (!family) continue;
-
-		const existing = families.get(family.familyId) ?? [];
-		if (!existing.includes(model.provider)) {
-			existing.push(model.provider);
-		}
-		families.set(family.familyId, existing);
-	}
-
-	return families;
-}