@hyav/pi-provider 0.1.2 → 0.1.4

package/preflight/mistral.ts

@@ -0,0 +1,27 @@
+import type { PreflightAdapter } from "@hyav/pi-provider";
+import { definePreflightExtension } from "@hyav/pi-provider";
+import { createCatalogPreflightAdapter } from "../core/catalog-preflight.ts";
+
+export const MISTRAL_MODELS_URL = "https://api.mistral.ai/v1/models";
+
+export const mistralPreflightAdapter: PreflightAdapter = createCatalogPreflightAdapter(
+	{
+		id: "mistral-preflight",
+		providerId: "mistral",
+		name: "Mistral",
+		modelsUrl: MISTRAL_MODELS_URL,
+	},
+	8_000,
+);
+
+export function createMistralPreflightAdapter(requestTimeoutMs: number): PreflightAdapter {
+	return { ...mistralPreflightAdapter, requestTimeoutMs };
+}
+
+const mistralPreflightExtension = definePreflightExtension({
+	id: "mistral-preflight",
+	providerId: "mistral",
+	create: ({ statusRequestTimeoutMs }) => createMistralPreflightAdapter(statusRequestTimeoutMs),
+});
+
+export default mistralPreflightExtension;

package/preflight/moonshotai-cn.ts

@@ -0,0 +1,27 @@
+import type { PreflightAdapter } from "@hyav/pi-provider";
+import { definePreflightExtension } from "@hyav/pi-provider";
+import { createMoonshotPreflightAdapter, MOONSHOT_CN_MODELS_URL } from "./moonshotai.ts";
+
+export const moonshotaiCnPreflightAdapter: PreflightAdapter = createMoonshotPreflightAdapter(
+	MOONSHOT_CN_MODELS_URL,
+	8_000,
+	"moonshotai-cn-preflight",
+	"moonshotai-cn",
+);
+
+export function createMoonshotaiCnPreflightAdapter(requestTimeoutMs: number): PreflightAdapter {
+	return createMoonshotPreflightAdapter(
+		MOONSHOT_CN_MODELS_URL,
+		requestTimeoutMs,
+		"moonshotai-cn-preflight",
+		"moonshotai-cn",
+	);
+}
+
+const moonshotaiCnPreflightExtension = definePreflightExtension({
+	id: "moonshotai-cn-preflight",
+	providerId: "moonshotai-cn",
+	create: ({ statusRequestTimeoutMs }) => createMoonshotaiCnPreflightAdapter(statusRequestTimeoutMs),
+});
+
+export default moonshotaiCnPreflightExtension;

package/preflight/moonshotai.ts

@@ -0,0 +1,37 @@
+import type { PreflightAdapter } from "@hyav/pi-provider";
+import { definePreflightExtension } from "@hyav/pi-provider";
+import { createCatalogPreflightAdapter } from "../core/catalog-preflight.ts";
+
+export const MOONSHOT_MODELS_URL = "https://api.moonshot.ai/v1/models";
+export const MOONSHOT_CN_MODELS_URL = "https://api.moonshot.cn/v1/models";
+
+export function createMoonshotPreflightAdapter(
+	modelsUrl: string,
+	requestTimeoutMs: number,
+	id: "moonshotai-preflight" | "moonshotai-cn-preflight" = "moonshotai-preflight",
+	providerId: "moonshotai" | "moonshotai-cn" = "moonshotai",
+): PreflightAdapter {
+	return createCatalogPreflightAdapter(
+		{
+			id,
+			providerId,
+			name: providerId === "moonshotai-cn" ? "Moonshot CN (Kimi)" : "Moonshot (Kimi)",
+			modelsUrl,
+		},
+		requestTimeoutMs,
+	);
+}
+
+export const moonshotaiPreflightAdapter = createMoonshotPreflightAdapter(MOONSHOT_MODELS_URL, 8_000);
+
+export function createMoonshotaiPreflightAdapter(requestTimeoutMs: number): PreflightAdapter {
+	return { ...moonshotaiPreflightAdapter, requestTimeoutMs };
+}
+
+const moonshotaiPreflightExtension = definePreflightExtension({
+	id: "moonshotai-preflight",
+	providerId: "moonshotai",
+	create: ({ statusRequestTimeoutMs }) => createMoonshotPreflightAdapter(MOONSHOT_MODELS_URL, statusRequestTimeoutMs),
+});
+
+export default moonshotaiPreflightExtension;

package/preflight/nvidia.ts

@@ -0,0 +1,27 @@
+import type { PreflightAdapter } from "@hyav/pi-provider";
+import { definePreflightExtension } from "@hyav/pi-provider";
+import { createCatalogPreflightAdapter } from "../core/catalog-preflight.ts";
+
+export const NVIDIA_MODELS_URL = "https://integrate.api.nvidia.com/v1/models";
+
+export const nvidiaPreflightAdapter: PreflightAdapter = createCatalogPreflightAdapter(
+	{
+		id: "nvidia-preflight",
+		providerId: "nvidia",
+		name: "NVIDIA NIM",
+		modelsUrl: NVIDIA_MODELS_URL,
+	},
+	8_000,
+);
+
+export function createNvidiaPreflightAdapter(requestTimeoutMs: number): PreflightAdapter {
+	return { ...nvidiaPreflightAdapter, requestTimeoutMs };
+}
+
+const nvidiaPreflightExtension = definePreflightExtension({
+	id: "nvidia-preflight",
+	providerId: "nvidia",
+	create: ({ statusRequestTimeoutMs }) => createNvidiaPreflightAdapter(statusRequestTimeoutMs),
+});
+
+export default nvidiaPreflightExtension;

package/preflight/openai-codex.ts

@@ -1,7 +1,5 @@
-import { definePreflightExtension } from "../core/adapter-extensions.ts";
-import { ProviderDataError } from "../core/errors.ts";
-import type { PreflightAdapter } from "../core/preflight-manager.ts";
-import { parseRetryAfter } from "../core/retry-after.ts";
+import type { PreflightAdapter } from "@hyav/pi-provider";
+import { definePreflightExtension, ProviderDataError, parseRetryAfter } from "@hyav/pi-provider";
 import { extractCodexAccountId } from "../status/openai-codex.ts";
 
 export const CODEX_MODELS_URL = "https://chatgpt.com/backend-api/codex/models";

package/preflight/openai.ts

@@ -0,0 +1,27 @@
+import type { PreflightAdapter } from "@hyav/pi-provider";
+import { definePreflightExtension } from "@hyav/pi-provider";
+import { createCatalogPreflightAdapter } from "../core/catalog-preflight.ts";
+
+export const OPENAI_MODELS_URL = "https://api.openai.com/v1/models";
+
+export const openAIPreflightAdapter: PreflightAdapter = createCatalogPreflightAdapter(
+	{
+		id: "openai-preflight",
+		providerId: "openai",
+		name: "OpenAI",
+		modelsUrl: OPENAI_MODELS_URL,
+	},
+	8_000,
+);
+
+export function createOpenAIPreflightAdapter(requestTimeoutMs: number): PreflightAdapter {
+	return { ...openAIPreflightAdapter, requestTimeoutMs };
+}
+
+const openAIPreflightExtension = definePreflightExtension({
+	id: "openai-preflight",
+	providerId: "openai",
+	create: ({ statusRequestTimeoutMs }) => createOpenAIPreflightAdapter(statusRequestTimeoutMs),
+});
+
+export default openAIPreflightExtension;

package/preflight/opencode-go.ts

@@ -1,6 +1,5 @@
-import { definePreflightExtension } from "../core/adapter-extensions.ts";
-import { createOpenCodeCatalogPreflightAdapter } from "../core/opencode-preflight.ts";
-import type { PreflightAdapter } from "../core/preflight-manager.ts";
+import type { PreflightAdapter } from "@hyav/pi-provider";
+import { createOpenCodeCatalogPreflightAdapter, definePreflightExtension } from "@hyav/pi-provider";
 
 export const OPENCODE_GO_MODELS_URL = "https://opencode.ai/zen/go/v1/models";
 

package/preflight/opencode.ts

@@ -1,6 +1,5 @@
-import { definePreflightExtension } from "../core/adapter-extensions.ts";
-import { createOpenCodeCatalogPreflightAdapter } from "../core/opencode-preflight.ts";
-import type { PreflightAdapter } from "../core/preflight-manager.ts";
+import type { PreflightAdapter } from "@hyav/pi-provider";
+import { createOpenCodeCatalogPreflightAdapter, definePreflightExtension } from "@hyav/pi-provider";
 
 export const OPENCODE_MODELS_URL = "https://opencode.ai/zen/v1/models";
 

package/preflight/openrouter.ts

@@ -0,0 +1,111 @@
+import type { PreflightAdapter } from "@hyav/pi-provider";
+import { definePreflightExtension, ProviderDataError, parseRetryAfter } from "@hyav/pi-provider";
+import { OPENROUTER_KEY_URL } from "../status/openrouter.ts";
+
+export const OPENROUTER_MODELS_URL = "https://openrouter.ai/api/v1/models";
+
+function isRecord(value: unknown): value is Record<string, unknown> {
+	return value !== null && typeof value === "object" && !Array.isArray(value);
+}
+
+async function collectModelIds(context: Parameters<PreflightAdapter["fetch"]>[0]): Promise<Set<string>> {
+	const response = await context.fetch(OPENROUTER_MODELS_URL, {
+		headers: {
+			Accept: "application/json",
+			"Accept-Encoding": "identity",
+			"User-Agent": "@hyav/pi-provider",
+		},
+		signal: context.signal,
+	});
+	if (!response.ok) {
+		throw new ProviderDataError(
+			`OpenRouter preflight failed: HTTP ${response.status}`,
+			`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("OpenRouter preflight returned invalid JSON", "badjson");
+	}
+	if (!isRecord(payload) || !Array.isArray(payload.data)) {
+		throw new ProviderDataError("OpenRouter preflight returned invalid catalog data", "badjson");
+	}
+	return new Set(
+		payload.data
+			.filter(isRecord)
+			.map((model) => (typeof model.id === "string" ? model.id.trim() : undefined))
+			.filter((id): id is string => id !== undefined && id !== ""),
+	);
+}
+
+async function checkCredential(context: Parameters<PreflightAdapter["fetch"]>[0], apiKey: string): Promise<number> {
+	const response = await context.fetch(OPENROUTER_KEY_URL, {
+		headers: {
+			Accept: "application/json",
+			"Accept-Encoding": "identity",
+			Authorization: `Bearer ${apiKey}`,
+			"User-Agent": "@hyav/pi-provider",
+		},
+		signal: context.signal,
+	});
+	if (response.status === 404) return response.status;
+	if (!response.ok) {
+		throw new ProviderDataError(
+			`OpenRouter preflight failed: HTTP ${response.status}`,
+			`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("OpenRouter preflight returned invalid JSON", "badjson");
+	}
+	if (!isRecord(payload) || !isRecord(payload.data) || typeof payload.data.label !== "string") {
+		throw new ProviderDataError("OpenRouter preflight returned an invalid key response", "badjson");
+	}
+	return response.status;
+}
+
+export const openRouterPreflightAdapter: PreflightAdapter = {
+	id: "openrouter-preflight",
+	providerId: "openrouter",
+	name: "OpenRouter",
+	cacheTtlMs: 30_000,
+	requestTimeoutMs: 8_000,
+	async fetch(context) {
+		const modelIds = await collectModelIds(context);
+		const apiKey = await context.getApiKey();
+		const checks: string[] = ["endpoint", "catalog"];
+		if (!apiKey || apiKey === "proxy-managed") {
+			return { passed: modelIds.has(context.model.id), checks: [...checks, "auth"], updatedAt: context.now() };
+		}
+		// Management keys use /api/v1/credits as the only key endpoint and get 404 here.
+		const authStatus = await checkCredential(context, apiKey);
+		const authPassed = authStatus !== 404;
+		checks.push("auth");
+		return {
+			passed: modelIds.has(context.model.id) && authPassed,
+			checks,
+			updatedAt: context.now(),
+		};
+	},
+};
+
+export function createOpenRouterPreflightAdapter(requestTimeoutMs: number): PreflightAdapter {
+	return { ...openRouterPreflightAdapter, requestTimeoutMs };
+}
+
+const openRouterPreflightExtension = definePreflightExtension({
+	id: "openrouter-preflight",
+	providerId: "openrouter",
+	create: ({ statusRequestTimeoutMs }) => createOpenRouterPreflightAdapter(statusRequestTimeoutMs),
+});
+
+export default openRouterPreflightExtension;

package/preflight/vercel-ai-gateway.ts

@@ -0,0 +1,86 @@
+import type { PreflightAdapter, PreflightSnapshot } from "@hyav/pi-provider";
+import { definePreflightExtension, ProviderDataError, parseRetryAfter } from "@hyav/pi-provider";
+import { VERCEL_PROVIDER_ID } from "../status/vercel-ai-gateway/constants.ts";
+
+export const VERCEL_MODELS_URL = "https://ai-gateway.vercel.sh/v1/models";
+
+function isRecord(value: unknown): value is Record<string, unknown> {
+	return value !== null && typeof value === "object" && !Array.isArray(value);
+}
+
+export function parseVercelModelIds(payload: unknown): Set<string> {
+	if (!isRecord(payload) || !Array.isArray(payload.data)) {
+		throw new ProviderDataError("Vercel AI Gateway preflight returned invalid catalog data", "badjson");
+	}
+
+	const modelIds = new Set<string>();
+	for (const value of payload.data) {
+		if (!isRecord(value)) continue;
+		// Mixed-type gateway catalogs can include non-language entries.
+		if (value.type !== undefined && value.type !== "language") continue;
+		if (typeof value.id !== "string") continue;
+		const id = value.id.trim();
+		if (id !== "") modelIds.add(id);
+	}
+
+	if (modelIds.size === 0) {
+		throw new ProviderDataError("Vercel AI Gateway preflight returned an empty catalog", "badjson");
+	}
+	return modelIds;
+}
+
+export function createVercelAIGatewayPreflightAdapter(requestTimeoutMs: number): PreflightAdapter {
+	return {
+		id: "vercel-ai-gateway-preflight",
+		providerId: VERCEL_PROVIDER_ID,
+		name: "Vercel AI Gateway",
+		cacheTtlMs: 30_000,
+		requestTimeoutMs,
+		async fetch(context): Promise<PreflightSnapshot> {
+			const key = await context.getApiKey();
+			if (!key || key === "proxy-managed") {
+				return { passed: false, checks: ["auth"], updatedAt: context.now() };
+			}
+
+			const response = await context.fetch(VERCEL_MODELS_URL, {
+				headers: {
+					Accept: "application/json",
+					"Accept-Encoding": "identity",
+				},
+				signal: context.signal,
+			});
+			if (!response.ok) {
+				throw new ProviderDataError(
+					`Vercel AI Gateway preflight failed: HTTP ${response.status}`,
+					`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 preflight returned invalid JSON", "badjson");
+			}
+
+			return {
+				passed: parseVercelModelIds(payload).has(context.model.id),
+				checks: ["endpoint", "auth", "catalog"],
+				updatedAt: context.now(),
+				httpStatus: response.status,
+			};
+		},
+	};
+}
+
+export const vercelAIGatewayPreflightAdapter = createVercelAIGatewayPreflightAdapter(8_000);
+
+const vercelAIGatewayPreflightExtension = definePreflightExtension({
+	id: "vercel-ai-gateway-preflight",
+	providerId: VERCEL_PROVIDER_ID,
+	create: ({ statusRequestTimeoutMs }) => createVercelAIGatewayPreflightAdapter(statusRequestTimeoutMs),
+});
+
+export default vercelAIGatewayPreflightExtension;

package/preflight/xai.ts

@@ -0,0 +1,70 @@
+import type { PreflightAdapter } from "@hyav/pi-provider";
+import { definePreflightExtension, ProviderDataError, parseRetryAfter } from "@hyav/pi-provider";
+import { XAI_MODELS_URL } from "../status/xai.ts";
+
+function isRecord(value: unknown): value is Record<string, unknown> {
+	return value !== null && typeof value === "object" && !Array.isArray(value);
+}
+
+export const xaiPreflightAdapter: PreflightAdapter = {
+	id: "xai-preflight",
+	providerId: "xai",
+	name: "xAI",
+	cacheTtlMs: 30_000,
+	requestTimeoutMs: 8_000,
+	async fetch(context) {
+		const apiKey = await context.getApiKey();
+		const authHeaders: Record<string, string> = {
+			Accept: "application/json",
+			"Accept-Encoding": "identity",
+			"User-Agent": "@hyav/pi-provider",
+		};
+		if (apiKey && apiKey !== "proxy-managed") authHeaders.Authorization = `Bearer ${apiKey}`;
+		const response = await context.fetch(XAI_MODELS_URL, {
+			headers: authHeaders,
+			signal: context.signal,
+		});
+		if (!response.ok) {
+			throw new ProviderDataError(
+				`xAI preflight failed: HTTP ${response.status}`,
+				`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("xAI preflight returned invalid JSON", "badjson");
+		}
+		if (!isRecord(payload) || !Array.isArray(payload.data)) {
+			throw new ProviderDataError("xAI preflight returned invalid catalog data", "badjson");
+		}
+		const modelIds = new Set(
+			payload.data
+				.filter(isRecord)
+				.map((model) => (typeof model.id === "string" ? model.id.trim() : undefined))
+				.filter((id): id is string => id !== undefined && id !== ""),
+		);
+		const checks = apiKey && apiKey !== "proxy-managed" ? ["endpoint", "catalog", "auth"] : ["endpoint", "catalog"];
+		return {
+			passed: modelIds.has(context.model.id),
+			checks,
+			updatedAt: context.now(),
+			httpStatus: response.status,
+		};
+	},
+};
+
+export function createXaiPreflightAdapter(requestTimeoutMs: number): PreflightAdapter {
+	return { ...xaiPreflightAdapter, requestTimeoutMs };
+}
+
+const xaiPreflightExtension = definePreflightExtension({
+	id: "xai-preflight",
+	providerId: "xai",
+	create: ({ statusRequestTimeoutMs }) => createXaiPreflightAdapter(statusRequestTimeoutMs),
+});
+
+export default xaiPreflightExtension;

package/providers/charm-hyper.ts

@@ -1,7 +1,3 @@
-import { defineProviderExtension } from "../core/adapter-extensions.ts";
-import { withDeadline } from "../core/deadline.ts";
-import { isProviderDataError, ProviderDataError } from "../core/errors.ts";
-import { normalizeProviderModels } from "../core/provider-registration.ts";
 import type {
 	ModelCatalogStatus,
 	ProviderAdapter,
@@ -9,7 +5,14 @@ import type {
 	ProviderModelDraft,
 	ProviderRefreshContext,
 	ThinkingLevel,
-} from "../core/types.ts";
+} from "@hyav/pi-provider";
+import {
+	defineProviderExtension,
+	isProviderDataError,
+	normalizeProviderModels,
+	ProviderDataError,
+	withDeadline,
+} from "@hyav/pi-provider";
 import { HYPER_BASE_URL, HYPER_USER_AGENT, hyperJsonHeaders } from "./charm-hyper/constants.ts";
 import { createCharmHyperOAuth } from "./charm-hyper/oauth.ts";