@hyav/pi-provider 0.1.3 → 0.1.4

package/status/openrouter.ts

@@ -0,0 +1,172 @@
+import type { StatusAdapter, StatusEntry, StatusSnapshot } from "@hyav/pi-provider";
+import { defineStatusExtension, ProviderDataError, parseRetryAfter } from "@hyav/pi-provider";
+
+export const OPENROUTER_KEY_URL = "https://openrouter.ai/api/v1/auth/key";
+export const OPENROUTER_CREDITS_URL = "https://openrouter.ai/api/v1/credits";
+
+function isRecord(value: unknown): value is Record<string, unknown> {
+	return value !== null && typeof value === "object" && !Array.isArray(value);
+}
+
+function safeNumber(value: unknown): number | undefined {
+	return typeof value === "number" && Number.isFinite(value) ? value : undefined;
+}
+
+function isErrorPayload(value: Record<string, unknown>): boolean {
+	return isRecord(value.error) && (typeof value.error.code === "number" || typeof value.error.message === "string");
+}
+
+interface OpenRouterKey {
+	label: string;
+	usage: number;
+	limit: number | null;
+	isFreeTier: boolean;
+}
+
+function parseKeyPayload(value: unknown): OpenRouterKey {
+	if (!isRecord(value) || !isRecord(value.data)) {
+		throw new ProviderDataError("OpenRouter status returned an invalid key response", "badjson");
+	}
+	const data = value.data;
+	const usage = safeNumber(data.usage);
+	if (
+		typeof data.label !== "string" ||
+		typeof data.is_free_tier !== "boolean" ||
+		usage === undefined ||
+		(data.limit !== null && safeNumber(data.limit) === undefined)
+	) {
+		throw new ProviderDataError("OpenRouter status returned an invalid key response", "badjson");
+	}
+	return {
+		label: data.label.replace(/[\u0000-\u001f\u007f]/g, "").trim() || "API key",
+		usage,
+		limit: data.limit === null ? null : (safeNumber(data.limit) as number),
+		isFreeTier: data.is_free_tier,
+	};
+}
+
+async function readJson(response: Response, providerName: string): Promise<unknown> {
+	let payload: unknown;
+	try {
+		payload = await response.json();
+	} catch {
+		throw new ProviderDataError(`${providerName} status returned invalid JSON`, "badjson");
+	}
+	return payload;
+}
+
+async function fetchWithAuth(
+	context: Parameters<StatusAdapter["fetch"]>[0],
+	key: string,
+	url: string,
+	providerName: string,
+): Promise<{ response: Response; payload: unknown }> {
+	const response = await context.fetch(url, {
+		headers: {
+			Accept: "application/json",
+			"Accept-Encoding": "identity",
+			Authorization: `Bearer ${key}`,
+			"User-Agent": "@hyav/pi-provider",
+		},
+		signal: context.signal,
+	});
+	if (!response.ok) {
+		throw new ProviderDataError(
+			`${providerName} status failed: HTTP ${response.status}`,
+			`http${response.status}`,
+			parseRetryAfter(response.headers.get("retry-after"), context.now()),
+			response.status,
+		);
+	}
+	return { response, payload: await readJson(response, providerName) };
+}
+
+function keyEntries(
+	key: OpenRouterKey,
+	limitEntry: StatusEntry | undefined,
+	freeTierEntry: StatusEntry,
+): StatusEntry[] {
+	const usageEntry: StatusEntry = {
+		kind: "amount",
+		id: "credits-used",
+		label: "Credits used",
+		value: key.usage,
+		unit: "USD",
+	};
+	const entries: StatusEntry[] = [{ kind: "text", id: "key", label: "Key", value: key.label }, usageEntry];
+	if (limitEntry) entries.push(limitEntry);
+	entries.push(freeTierEntry);
+	return entries;
+}
+
+function freeTierEntry(value: boolean): StatusEntry {
+	return { kind: "text", id: "account-tier", label: "Account", value: value ? "Free tier" : "Paid" };
+}
+
+export const openRouterStatusAdapter: StatusAdapter = {
+	id: "openrouter-status",
+	providerId: "openrouter",
+	name: "OpenRouter",
+	cacheTtlMs: 60_000,
+	requestTimeoutMs: 8_000,
+	async fetch(context): Promise<StatusSnapshot> {
+		const key = await context.getApiKey();
+		if (!key || key === "proxy-managed") {
+			throw new ProviderDataError("OpenRouter status requires an API key", "auth");
+		}
+		const keyResult = await fetchWithAuth(context, key, OPENROUTER_KEY_URL, "OpenRouter");
+		const keyPayload = keyResult.payload;
+		if (isRecord(keyPayload) && isErrorPayload(keyPayload)) {
+			throw new ProviderDataError(
+				"OpenRouter status failed: invalid API key response",
+				"auth",
+				undefined,
+				keyResult.response.status,
+			);
+		}
+		const openRouterKey = parseKeyPayload(keyPayload);
+
+		// /credits requires a management key; only show the limit when resolved.
+		let limitEntry: StatusEntry | undefined;
+		try {
+			const creditsResult = await fetchWithAuth(context, key, OPENROUTER_CREDITS_URL, "OpenRouter");
+			const payload = creditsResult.payload;
+			if (!isRecord(payload) || !isRecord(payload.data)) {
+				throw new ProviderDataError("OpenRouter status returned an invalid credits response", "badjson");
+			}
+			const totalCredits = safeNumber(payload.data.total_credits);
+			const totalUsage = safeNumber(payload.data.total_usage);
+			if (totalCredits === undefined || totalUsage === undefined) {
+				throw new ProviderDataError("OpenRouter status returned an invalid credits response", "badjson");
+			}
+			const remaining = Math.max(0, totalCredits - totalUsage);
+			limitEntry = {
+				kind: "amount",
+				id: "credits-remaining",
+				label: "Credits remaining",
+				value: remaining,
+				unit: "USD",
+			};
+		} catch (error) {
+			// Safe fallback: key credits, free-tier, and key-level limit still display.
+			if (!(error instanceof ProviderDataError)) throw error;
+		}
+
+		return {
+			entries: keyEntries(openRouterKey, limitEntry, freeTierEntry(openRouterKey.isFreeTier)),
+			updatedAt: context.now(),
+		};
+	},
+};
+
+export function createOpenRouterStatusAdapter(requestTimeoutMs: number): StatusAdapter {
+	return { ...openRouterStatusAdapter, requestTimeoutMs };
+}
+
+const openRouterStatusExtension = defineStatusExtension({
+	id: "openrouter-status",
+	providerId: "openrouter",
+	create: ({ statusRequestTimeoutMs }) => createOpenRouterStatusAdapter(statusRequestTimeoutMs),
+});
+
+export default openRouterStatusExtension;

package/status/vercel-ai-gateway/constants.ts

@@ -0,0 +1,3 @@
+/** Shared identity constants for Vercel AI Gateway adapters. */
+
+export const VERCEL_PROVIDER_ID = "vercel-ai-gateway";

package/status/vercel-ai-gateway.ts

@@ -0,0 +1,94 @@
+import type { StatusAdapter, StatusSnapshot } from "@hyav/pi-provider";
+import { defineStatusExtension, ProviderDataError, parseRetryAfter } from "@hyav/pi-provider";
+import { VERCEL_PROVIDER_ID } from "./vercel-ai-gateway/constants.ts";
+
+export const VERCEL_CREDITS_URL = "https://ai-gateway.vercel.sh/v1/credits";
+
+interface VercelCredits {
+	balance: number;
+	totalUsed: number;
+}
+
+function isRecord(value: unknown): value is Record<string, unknown> {
+	return value !== null && typeof value === "object" && !Array.isArray(value);
+}
+
+function parseFiniteAmount(value: unknown): number | undefined {
+	if (typeof value === "number") return Number.isFinite(value) ? value : undefined;
+	if (typeof value !== "string" || value.trim() === "") return undefined;
+	const parsed = Number(value);
+	return Number.isFinite(parsed) ? parsed : undefined;
+}
+
+export function parseVercelCredits(payload: unknown): VercelCredits {
+	if (!isRecord(payload)) {
+		throw new ProviderDataError("Vercel AI Gateway status returned an invalid credits response", "badjson");
+	}
+
+	const balance = parseFiniteAmount(payload.balance);
+	const totalUsed = parseFiniteAmount(payload.total_used);
+	if (balance === undefined || totalUsed === undefined) {
+		throw new ProviderDataError("Vercel AI Gateway status returned an invalid credits response", "badjson");
+	}
+
+	return { balance, totalUsed };
+}
+
+export function createVercelAIGatewayStatusAdapter(requestTimeoutMs: number): StatusAdapter {
+	return {
+		id: "vercel-ai-gateway-status",
+		providerId: VERCEL_PROVIDER_ID,
+		name: "Vercel AI Gateway",
+		cacheTtlMs: 30_000,
+		requestTimeoutMs,
+		async fetch(context): Promise<StatusSnapshot> {
+			const key = await context.getApiKey();
+			if (!key || key === "proxy-managed") {
+				throw new ProviderDataError("Vercel AI Gateway status requires an API key", "auth");
+			}
+
+			const response = await context.fetch(VERCEL_CREDITS_URL, {
+				headers: {
+					Accept: "application/json",
+					"Accept-Encoding": "identity",
+					Authorization: `Bearer ${key}`,
+				},
+				signal: context.signal,
+			});
+			if (!response.ok) {
+				throw new ProviderDataError(
+					`Vercel AI Gateway status failed: HTTP ${response.status}`,
+					response.status === 401 ? "auth" : `http${response.status}`,
+					parseRetryAfter(response.headers.get("retry-after"), context.now()),
+					response.status,
+				);
+			}
+
+			let payload: unknown;
+			try {
+				payload = await response.json();
+			} catch {
+				throw new ProviderDataError("Vercel AI Gateway status returned invalid JSON", "badjson");
+			}
+
+			const credits = parseVercelCredits(payload);
+			return {
+				entries: [
+					{ kind: "amount", id: "balance", label: "Balance", value: credits.balance, unit: "USD" },
+					{ kind: "amount", id: "total-used", label: "Total Used", value: credits.totalUsed, unit: "USD" },
+				],
+				updatedAt: context.now(),
+			};
+		},
+	};
+}
+
+export const vercelAIGatewayStatusAdapter = createVercelAIGatewayStatusAdapter(8_000);
+
+const vercelAIGatewayStatusExtension = defineStatusExtension({
+	id: "vercel-ai-gateway-status",
+	providerId: VERCEL_PROVIDER_ID,
+	create: ({ statusRequestTimeoutMs }) => createVercelAIGatewayStatusAdapter(statusRequestTimeoutMs),
+});
+
+export default vercelAIGatewayStatusExtension;

package/status/xai.ts

@@ -0,0 +1,73 @@
+import type { StatusAdapter, StatusEntry, StatusSnapshot } from "@hyav/pi-provider";
+import { defineStatusExtension, ProviderDataError, parseRetryAfter } from "@hyav/pi-provider";
+import { parseRateLimitWindows } from "../core/ratelimit-headers.ts";
+
+export const XAI_MODELS_URL = "https://api.x.ai/v1/models";
+
+function windowEntry(id: string, label: string, limit: number, remaining: number, resetAt?: number): StatusEntry {
+	const percent = (remaining / limit) * 100;
+	return {
+		kind: "window",
+		id,
+		label,
+		remainingPercent: Math.max(0, Math.min(100, percent)),
+		...(resetAt !== undefined ? { resetAt } : {}),
+	};
+}
+
+export const xaiStatusAdapter: StatusAdapter = {
+	id: "xai-status",
+	providerId: "xai",
+	name: "xAI",
+	cacheTtlMs: 60_000,
+	requestTimeoutMs: 8_000,
+	async fetch(context): Promise<StatusSnapshot> {
+		const key = await context.getApiKey();
+		if (!key || key === "proxy-managed") {
+			throw new ProviderDataError("xAI status requires Grok OAuth or an API key", "auth");
+		}
+		const response = await context.fetch(XAI_MODELS_URL, {
+			headers: {
+				Accept: "application/json",
+				"Accept-Encoding": "identity",
+				Authorization: `Bearer ${key}`,
+				"User-Agent": "@hyav/pi-provider",
+			},
+			signal: context.signal,
+		});
+		if (!response.ok) {
+			throw new ProviderDataError(
+				`xAI status failed: HTTP ${response.status}`,
+				`http${response.status}`,
+				parseRetryAfter(response.headers.get("retry-after"), context.now()),
+				response.status,
+			);
+		}
+
+		const now = context.now();
+		const { requests, tokens } = parseRateLimitWindows(response.headers, now);
+		const entries: StatusEntry[] = [];
+		if (requests) {
+			entries.push(windowEntry("requests-window", "Requests", requests.limit, requests.remaining, requests.resetAt));
+		}
+		if (tokens) {
+			entries.push(windowEntry("tokens-window", "Tokens", tokens.limit, tokens.remaining, tokens.resetAt));
+		}
+		if (entries.length === 0) {
+			entries.push({ kind: "text", id: "limits", label: "Limits", value: "not available" });
+		}
+		return { entries, updatedAt: now };
+	},
+};
+
+export function createXaiStatusAdapter(requestTimeoutMs: number): StatusAdapter {
+	return { ...xaiStatusAdapter, requestTimeoutMs };
+}
+
+const xaiStatusExtension = defineStatusExtension({
+	id: "xai-status",
+	providerId: "xai",
+	create: ({ statusRequestTimeoutMs }) => createXaiStatusAdapter(statusRequestTimeoutMs),
+});
+
+export default xaiStatusExtension;