@jeffreycao/copilot-api 2.7.39 → 2.7.41

package/dist/{server-Jvc5DYJV.js → server-h60svXN0.js}

@@ -1,14 +1,15 @@
 import { D as state, E as isGitHubCopilotEnabled, O as HTTPError, S as PATHS, a as getAnthropicApiKey, c as getConfig, d as getUpstreamTransportConfig, g as isResponsesApiWebSocketEnabled, h as isResponsesApiWebSearchEnabled, i as getAlphaSearchModel, j as forwardError, l as getMessageApiWebSearchModel, m as isMessagesApiEnabled, o as getClaudeAutoModel, p as isAlphaSearchCodexPriorityEnabled, s as getClaudeTokenMultiplier, u as getOpencodeModelContextWindow, w as GITHUB_COPILOT_PROVIDER, y as reloadConfig } from "./config-store-BsluQttR.js";
 import { A as resolveMappedModel, C as getModelResponsesApiCompactThreshold$1, D as isContextManagementEnabledForMessages, E as getSmallModelForProvider, O as isContextManagementEnabledForResponses, S as getModelMappings, T as getSmallModel, _ as getModelsDevModelMaxOutputTokens, a as normalizeProviderBaseUrl, d as isProviderModelVisible, f as resolveProviderModelsDevId, g as getModelsDevModel, h as getBuiltinProviderModelRecords, i as listEnabledProviders, j as setModelMappings, k as isGpt56OrAbove, l as getLocalProviderModelRecords, m as builtinProviderModelRegistry, n as getRawProviderConfig, o as resolveEffectiveProviderType, p as usesModelsDevModelCatalog, s as resolveProviderConfigForModel, t as getProviderConfig, u as getProviderModelConfig, v as getModelsDevModelPricing, w as getReasoningEffortForModel, x as getExtraPromptForModel } from "./config-BqSuE5h5.js";
-import { $ as forwardCodexResponses, B as getXaiAccessToken, C as copilotHeaders, D as prepareForCompact, E as isOpencodeOauthApp, F as compactTextOnlyGuard, I as createAuthMiddleware, J as XAI_API_BASE_URL, K as readGitHubToken, L as getConfiguredAdminApiKeys, M as compactMessageSections, O as prepareInteractionHeaders, P as compactSystemPromptStarts, Q as buildCodexRequestHeaders, S as copilotBaseUrl, T as copilotWebSocketHeaders, Z as CODEX_API_BASE_URL, _ as isAsyncIterable, at as createResponsesHttpEventStream, b as parseUserIdMetadata, ct as getResponsesStreamTerminalDisposition, d as generateRequestIdFromPayload, et as generateTraceId, f as getPromptCacheKey, ft as stripInternalRequestHeaders, g as getUUID, h as getSessionAffinityFromHeaders, it as fetchUpstreamWithLifecycle, j as compactAutoContinuePromptStarts, k as prepareMessageProxyHeaders, lt as createPooledWebSocketStream, m as getRootSessionIdFromHeaders, n as invalidateCodexRuntime, nt as resolveTraceId$1, ot as createResponsesSafeStream, p as getRootSessionId, q as readGitHubTokenFromEnv, rt as setPluginRequestActualModel, s as setupCodexToken, st as encodePoolKeyPart, tt as requestContext, u as stopCopilotRefreshLoop, ut as createWebSocketUrl, v as isNullish, x as getCopilotUsage, y as isResponsesStream } from "./token-_1CBLbqH.js";
+import { $ as buildCodexRequestHeaders, B as getXaiAccessToken, C as copilotHeaders, D as prepareForCompact, E as isOpencodeOauthApp, F as compactTextOnlyGuard, I as createAuthMiddleware, J as readGitHubTokenFromEnv, L as getConfiguredAdminApiKeys, M as compactMessageSections, O as prepareInteractionHeaders, P as compactSystemPromptStarts, Q as CODEX_API_BASE_URL, S as copilotBaseUrl, T as copilotWebSocketHeaders, U as readXaiCredentials, Y as XAI_API_BASE_URL, _ as isAsyncIterable, at as fetchUpstreamWithLifecycle, b as parseUserIdMetadata, ct as encodePoolKeyPart, d as generateRequestIdFromPayload, dt as createWebSocketUrl, et as forwardCodexResponses, f as getPromptCacheKey, g as getUUID, h as getSessionAffinityFromHeaders, it as setPluginRequestActualModel, j as compactAutoContinuePromptStarts, k as prepareMessageProxyHeaders, lt as getResponsesStreamTerminalDisposition, m as getRootSessionIdFromHeaders, n as invalidateCodexRuntime, nt as requestContext, ot as createResponsesHttpEventStream, p as getRootSessionId, pt as stripInternalRequestHeaders, q as readGitHubToken, rt as resolveTraceId$1, s as setupCodexToken, st as createResponsesSafeStream, tt as generateTraceId, u as stopCopilotRefreshLoop, ut as createPooledWebSocketStream, v as isNullish, x as getCopilotUsage, y as isResponsesStream } from "./token-CZmr-geT.js";
 import { a as isDeferredToolName, c as parseMcpToolSearchSentinel, d as shouldEnableResponsesToolSearch, i as isBridgeToolSearchName, l as resolveBridgeToolSearchName, o as listDeferredToolNames, r as formatToolSearchBridgeArguments, s as normalizeToolSearchBridgeArguments, t as BRIDGE_TOOL_SEARCH_NAME, u as selectDeferredToolsByNames } from "./tool-search-DzipwtCX.js";
 import { n as isProviderAgentModelVisible, t as getProviderAgentModels } from "./provider-management-vyEQj9RR.js";
-import { i as registerProcessCleanup, n as stopModelsRefreshLoop, r as EventBus, t as setupCopilotRuntime } from "./copilot-runtime-0CujVONa.js";
+import { i as registerProcessCleanup, n as stopModelsRefreshLoop, r as EventBus, t as setupCopilotRuntime } from "./copilot-runtime-CjQPxPeV.js";
 import consola from "consola";
 import { createHash, randomUUID } from "node:crypto";
 import fs, { readFileSync } from "node:fs";
 import path from "node:path";
 import fs$1 from "node:fs/promises";
+import os from "node:os";
 import { fetch as fetch$1, getGlobalDispatcher } from "undici";
 import { events } from "fetch-event-stream";
 import { z } from "zod";
@@ -2063,14 +2064,16 @@ const createResponsesWebSocketStreamChunk = (data) => {
 	try {
 		const parsed = JSON.parse(data);
 		if (parsed.type === "response.completed") logCopilotQuotaSnapshots(parsed.copilot_quota_snapshots);
+		let forwardedData = data;
 		if (parsed.type === "error" && parsed.error) {
 			consola.warn("Copilot responses websocket stream error:", parsed.error);
 			parsed.code = parsed.error.code;
 			parsed.message = parsed.error.message;
+			forwardedData = JSON.stringify(parsed);
 		}
 		return {
 			event: typeof parsed.type === "string" ? parsed.type : void 0,
-			data: JSON.stringify(parsed),
+			data: forwardedData,
 			id: typeof parsed.id === "string" ? parsed.id : void 0
 		};
 	} catch {
@@ -9828,11 +9831,11 @@ const handleWithMessagesApi = async (c, anthropicPayload, options) => {
 					recordAnthropicFirstToken(parsedEvent);
 					if (parsedEvent?.type === "message_start") usage = mergeAnthropicUsage(usage, {
 						...normalizeAnthropicUsage(parsedEvent.message.usage),
-						...normalizeCopilotUsage(parsedEvent.message.copilot_usage)
+						...normalizeCopilotUsage$1(parsedEvent.message.copilot_usage)
 					});
 					else if (parsedEvent?.type === "message_delta") usage = mergeAnthropicUsage(usage, {
 						...normalizeAnthropicUsage(parsedEvent.usage),
-						...normalizeCopilotUsage(parsedEvent.copilot_usage)
+						...normalizeCopilotUsage$1(parsedEvent.copilot_usage)
 					});
 					if (parsedEvent?.type === "message_stop" || eventName === "message_stop") messageStopSeen = true;
 					else if (parsedEvent?.type === "error" || eventName === "error") errorSeen = true;
@@ -9861,7 +9864,7 @@ const handleWithMessagesApi = async (c, anthropicPayload, options) => {
 	});
 	recordUsage({
 		...normalizeAnthropicUsage(response.usage),
-		...normalizeCopilotUsage(response.copilot_usage)
+		...normalizeCopilotUsage$1(response.copilot_usage)
 	});
 	return c.json(response);
 };
@@ -9893,7 +9896,7 @@ const createCopilotUsageRecorder = (options) => createCopilotTokenUsageRecorder(
 	sessionId: getMetadataSessionId(options.payload)
 });
 const getMetadataSessionId = (payload) => parseUserIdMetadata(payload.metadata?.user_id).sessionId;
-const normalizeCopilotUsage = (copilotUsage) => ({ total_nano_aiu: normalizeOptionalToken(copilotUsage?.total_nano_aiu) });
+const normalizeCopilotUsage$1 = (copilotUsage) => ({ total_nano_aiu: normalizeOptionalToken(copilotUsage?.total_nano_aiu) });
 const parseAnthropicStreamEvent = (data) => {
 	try {
 		return JSON.parse(data);
@@ -13593,9 +13596,8 @@ const streamProviderResponses = async (c, upstreamResponse, options) => {
 					normalizeCodex: options.normalizeCodex,
 					provider: options.provider
 				});
-				if (event && options.normalizeCodex) responseChunk = {
+				if (event && options.normalizeCodex && chunk.event !== event.type) responseChunk = {
 					...chunk,
-					data: JSON.stringify(event),
 					event: event.type
 				};
 			}
@@ -13947,8 +13949,329 @@ function parsePageQuery(c) {
 }
 tokenUsageRoute.get("/events", async (c) => c.json(await getTokenUsageEventsPage(parsePageQuery(c))));
 tokenUsageRoute.get("/sessions", async (c) => c.json(await getTokenUsageSessionsPage(parsePageQuery(c))));
+// src/services/subscription-usage.ts
+const CODEX_CLI_VERSION = "0.164.0";
+const amount = z.union([z.number().finite(), z.string().trim().min(1).refine((value) => Number.isFinite(Number(value)))]).transform(Number);
+const number = z.number().finite();
+const codexWindow = z.object({
+	used_percent: number,
+	limit_window_seconds: number.nullish(),
+	reset_at: number.nullish(),
+	reset_after_seconds: number.nullish()
+});
+const codexRateLimit = z.object({
+	allowed: z.boolean().nullish(),
+	limit_reached: z.boolean().nullish(),
+	primary_window: codexWindow.nullish(),
+	secondary_window: codexWindow.nullish()
+});
+const codexUsage = z.object({
+	plan_type: z.string(),
+	rate_limit: codexRateLimit.nullish(),
+	additional_rate_limits: z.array(z.object({
+		limit_name: z.string(),
+		metered_feature: z.string(),
+		rate_limit: codexRateLimit.nullish()
+	})).nullish(),
+	credits: z.object({
+		has_credits: z.boolean(),
+		unlimited: z.boolean(),
+		balance: amount.nullish()
+	}).nullish(),
+	spend_control: z.object({
+		reached: z.boolean(),
+		individual_limit: z.object({
+			limit: amount,
+			used: amount,
+			used_percent: number,
+			reset_at: number.nullish(),
+			reset_after_seconds: number.nullish()
+		}).nullish()
+	}).nullish(),
+	rate_limit_reached_type: z.object({ type: z.string() }).nullish()
+});
+const cents = z.object({ val: amount.default(0) }).nullish();
+const xaiUsage = z.object({
+	config: z.object({
+		creditUsagePercent: number.nullish(),
+		currentPeriod: z.object({
+			type: z.string().nullish(),
+			end: z.string().nullish()
+		}).nullish(),
+		monthlyLimit: cents,
+		used: cents,
+		onDemandCap: cents,
+		onDemandUsed: cents,
+		prepaidBalance: cents,
+		billingPeriodEnd: z.string().nullish()
+	}).nullish(),
+	subscription_tier: z.string().nullish()
+});
+const copilotQuota = z.object({
+	entitlement: number.nullish(),
+	remaining: number.nullish(),
+	quota_remaining: number.nullish(),
+	percent_remaining: number.nullish(),
+	unlimited: z.boolean().nullish()
+});
+const copilotUsage = z.object({
+	copilot_plan: z.string().nullish(),
+	quota_reset_date: z.string().nullish(),
+	quota_snapshots: z.object({
+		premium_interactions: copilotQuota.nullish(),
+		chat: copilotQuota.nullish(),
+		completions: copilotQuota.nullish()
+	}).nullish()
+});
+const opencodeGoWindow = z.object({
+	status: z.enum(["ok", "rate-limited"]),
+	percent: number,
+	resetsAt: z.string().nullish()
+});
+const opencodeGoUsage = z.object({ usage: z.object({
+	rolling: opencodeGoWindow.nullish(),
+	weekly: opencodeGoWindow.nullish(),
+	monthly: opencodeGoWindow.nullish()
+}) });
+function normalizeOpencodeGoUsage(value) {
+	const { usage } = opencodeGoUsage.parse(value);
+	const windows = [];
+	for (const scope of [
+		"rolling",
+		"weekly",
+		"monthly"
+	]) {
+		const window = usage[scope];
+		if (!window) continue;
+		windows.push({
+			id: `opencode-go:${scope}`,
+			kind: "usage",
+			windowSeconds: scope === "rolling" ? 18e3 : void 0,
+			period: scope === "rolling" ? void 0 : scope,
+			usedPercent: window.percent,
+			resetAt: parseResetDate(window.resetsAt),
+			limitReached: window.status === "rate-limited" || window.percent >= 100
+		});
+	}
+	return {
+		provider: "opencode-go",
+		windows,
+		limitReached: windows.some((window) => window.limitReached)
+	};
+}
+function normalizeCopilotUsage(value) {
+	const usage = copilotUsage.parse(value);
+	const windows = [];
+	const labels = {
+		premium_interactions: "Premium",
+		chat: "Chat",
+		completions: "Completions"
+	};
+	for (const id of [
+		"premium_interactions",
+		"chat",
+		"completions"
+	]) {
+		const quota = usage.quota_snapshots?.[id];
+		if (!quota) continue;
+		const remaining = quota.remaining ?? quota.quota_remaining;
+		const limit = quota.entitlement ?? void 0;
+		const remainingPercent = quota.percent_remaining ?? (limit !== void 0 && limit > 0 && remaining != null ? remaining / limit * 100 : void 0);
+		if (!quota.unlimited && remainingPercent == null && remaining == null && limit === void 0) continue;
+		windows.push({
+			id: `copilot:${id}`,
+			kind: "usage",
+			name: labels[id],
+			unlimited: quota.unlimited ?? false,
+			usedPercent: quota.unlimited || remainingPercent == null ? void 0 : 100 - remainingPercent,
+			used: limit !== void 0 && remaining != null ? Math.max(0, limit - remaining) : void 0,
+			limit,
+			unit: "requests",
+			resetDate: usage.quota_reset_date ?? void 0,
+			limitReached: !quota.unlimited && remainingPercent != null && remainingPercent <= 0
+		});
+	}
+	return {
+		provider: "github-copilot",
+		planType: usage.copilot_plan ?? void 0,
+		windows,
+		limitReached: windows.some((window) => window.limitReached)
+	};
+}
+function resetAt(window, now) {
+	return window.reset_at ?? (window.reset_after_seconds == null ? void 0 : Math.floor(now / 1e3) + window.reset_after_seconds);
+}
+function normalizeCodexUsage(value, now = Date.now()) {
+	const usage = codexUsage.parse(value);
+	const windows = [];
+	const addWindows = (id, limits, name) => {
+		for (const scope of ["primary_window", "secondary_window"]) {
+			const window = limits?.[scope];
+			if (!window) continue;
+			windows.push({
+				id: `${id}:${scope}`,
+				kind: "usage",
+				name,
+				usedPercent: window.used_percent,
+				windowSeconds: window.limit_window_seconds ?? void 0,
+				resetAt: resetAt(window, now),
+				limitReached: limits?.limit_reached === true || limits?.allowed === false
+			});
+		}
+	};
+	addWindows("codex", usage.rate_limit);
+	for (const [index, limit] of (usage.additional_rate_limits ?? []).entries()) addWindows(`${limit.metered_feature}:${index}`, limit.rate_limit, limit.limit_name);
+	const spend = usage.spend_control?.individual_limit;
+	if (spend) windows.push({
+		id: "individual-limit",
+		kind: "spend",
+		period: "monthly",
+		usedPercent: spend.used_percent,
+		used: spend.used,
+		limit: spend.limit,
+		unit: "credits",
+		resetAt: resetAt(spend, now),
+		limitReached: usage.spend_control?.reached
+	});
+	return {
+		provider: "codex",
+		planType: usage.plan_type,
+		windows,
+		credits: usage.credits ? {
+			balance: usage.credits.balance ?? void 0,
+			unlimited: usage.credits.unlimited,
+			hasCredits: usage.credits.has_credits,
+			unit: "credits"
+		} : void 0,
+		limitReached: usage.rate_limit?.limit_reached === true || usage.rate_limit?.allowed === false || usage.spend_control?.reached === true || usage.rate_limit_reached_type != null && usage.rate_limit_reached_type.type !== "unknown",
+		limitReason: usage.rate_limit_reached_type?.type
+	};
+}
+function parseResetDate(value) {
+	const milliseconds = value ? Date.parse(value) : NaN;
+	return Number.isFinite(milliseconds) ? milliseconds / 1e3 : void 0;
+}
+function normalizeXaiUsage(value) {
+	const usage = xaiUsage.parse(value);
+	const config = usage.config;
+	const windows = [];
+	if (config) {
+		const limit = config.monthlyLimit?.val;
+		const used = config.used?.val;
+		const usedPercent = config.creditUsagePercent ?? (limit !== void 0 && limit > 0 && used !== void 0 ? used / limit * 100 : void 0);
+		if (usedPercent != null || limit !== void 0 || used !== void 0) windows.push({
+			id: "included",
+			kind: "usage",
+			period: config.currentPeriod?.type === "USAGE_PERIOD_TYPE_WEEKLY" ? "weekly" : config.currentPeriod?.type === "USAGE_PERIOD_TYPE_MONTHLY" || config.monthlyLimit ? "monthly" : void 0,
+			usedPercent,
+			used: config.creditUsagePercent == null && used !== void 0 ? used / 100 : void 0,
+			limit: config.creditUsagePercent == null && limit !== void 0 ? limit / 100 : void 0,
+			unit: "USD",
+			resetAt: parseResetDate(config.currentPeriod?.end ?? config.billingPeriodEnd)
+		});
+		if (config.onDemandCap || config.onDemandUsed) {
+			const cap = Math.abs(config.onDemandCap?.val ?? 0);
+			const spent = Math.abs(config.onDemandUsed?.val ?? 0);
+			windows.push({
+				id: "on-demand",
+				kind: "on-demand",
+				period: "monthly",
+				unit: "USD",
+				used: spent / 100,
+				limit: cap / 100,
+				usedPercent: cap > 0 ? spent / cap * 100 : void 0
+			});
+		}
+	}
+	return {
+		provider: "xai",
+		planType: usage.subscription_tier ?? void 0,
+		windows,
+		credits: config?.prepaidBalance ? {
+			balance: Math.abs(config.prepaidBalance.val) / 100,
+			unlimited: false,
+			hasCredits: config.prepaidBalance.val !== 0,
+			unit: "USD"
+		} : void 0,
+		limitReached: (windows.find((window) => window.id === "included")?.usedPercent ?? 0) >= 100
+	};
+}
+async function getSubscriptionUsage(provider, signal) {
+	if (provider === "github-copilot") {
+		if (!isGitHubCopilotEnabled()) return null;
+		const timeout = AbortSignal.timeout(15e3);
+		const usage = await getCopilotUsage(void 0, signal ? AbortSignal.any([signal, timeout]) : timeout);
+		return usage ? normalizeCopilotUsage(usage) : null;
+	}
+	const config = await resolveProviderConfig(provider);
+	if (!config) return null;
+	if (provider !== "opencode-go" && config.authType !== "oauth2") return null;
+	let headers;
+	let url;
+	if (provider === "opencode-go") {
+		url = resolveProviderEndpointUrl(config, "usage");
+		headers = new Headers({
+			Authorization: `Bearer ${config.apiKey}`,
+			Accept: "application/json"
+		});
+	} else if (provider === "codex") {
+		const platform = process.platform === "win32" ? "Windows" : process.platform === "darwin" ? "Mac OS" : os.type();
+		const arch = process.arch === "x64" ? "x86_64" : process.arch === "arm64" ? "aarch64" : process.arch;
+		headers = buildCodexRequestHeaders(new Headers({
+			"user-agent": `codex_cli_rs/${CODEX_CLI_VERSION} (${platform} ${os.release()}; ${arch}) unknown`,
+			originator: "codex_cli_rs",
+			accept: "application/json"
+		}));
+		url = `${CODEX_API_BASE_URL}/wham/usage`;
+	} else {
+		const credentials = await readXaiCredentials();
+		if (!credentials) return null;
+		if (credentials.accessToken !== config.apiKey) throw new Error("Subscription account changed during the usage query");
+		const version = await getGrokBuildVersion();
+		const platform = process.platform === "win32" ? "windows" : process.platform === "darwin" ? "macos" : process.platform;
+		const arch = process.arch === "x64" ? "x86_64" : process.arch === "arm64" ? "aarch64" : process.arch;
+		headers = new Headers({
+			Authorization: `Bearer ${config.apiKey}`,
+			"X-XAI-Token-Auth": "xai-grok-cli",
+			"x-userid": credentials.accountId,
+			"x-grok-client-version": version,
+			"x-grok-client-mode": "headless",
+			"User-Agent": `grok-shell/${version} (${platform}; ${arch})`,
+			Accept: "application/json"
+		});
+		url = `${XAI_API_BASE_URL}/v1/billing?format=credits`;
+	}
+	const timeout = AbortSignal.timeout(15e3);
+	const response = await fetch(url, {
+		headers,
+		signal: signal ? AbortSignal.any([signal, timeout]) : timeout,
+		redirect: "error"
+	});
+	if (provider === "opencode-go" && response.status === 403) {
+		await response.body?.cancel();
+		return null;
+	}
+	if (!response.ok) {
+		await response.body?.cancel();
+		throw new Error(`Subscription usage request failed (HTTP ${response.status})`);
+	}
+	const value = await response.json();
+	if (provider === "opencode-go") return normalizeOpencodeGoUsage(value);
+	return provider === "codex" ? normalizeCodexUsage(value) : normalizeXaiUsage(value);
+}
 // src/routes/usage/route.ts
 const usageRoute = new Hono();
+usageRoute.get("/:provider", async (c) => {
+	const provider = c.req.param("provider");
+	c.header("Cache-Control", "no-store");
+	if (provider !== "codex" && provider !== "xai" && provider !== "github-copilot" && provider !== "opencode-go") return c.json({ error: "Unsupported subscription provider" }, 404);
+	try {
+		const usage = await getSubscriptionUsage(provider, c.req.raw.signal);
+		return usage ? c.json(usage) : c.json({ error: "Subscription account is not available" }, 404);
+	} catch {
+		return c.json({ error: "Failed to fetch subscription usage" }, 502);
+	}
+});
 usageRoute.get("/", async (c) => {
 	try {
 		const usage = await getCopilotUsage();
@@ -14109,4 +14432,4 @@ function createServer(options = {}) {
 createServer();
 export { createServer };
 
-//# sourceMappingURL=server-Jvc5DYJV.js.map
+//# sourceMappingURL=server-h60svXN0.js.map