@xiaoso/dsh-tool-plus 0.1.0-beta.2 → 0.1.0-beta.3

package/lib/api-key-login-CpzSezsi.mjs

@@ -0,0 +1,157 @@
+import { installBunShim as __installBunShim } from "./bun-shim.mjs";
+__installBunShim();
+import { V as ProviderHttpError, _ as OnPromptRequiredError, h as LoginCancelledError, m as ApiKeyRequiredError } from "./error-B3bevhrs.mjs";
+//#region node_modules/.pnpm/@oh-my-pi+pi-ai@17.3.5/node_modules/@oh-my-pi/pi-ai/src/registry/api-key-validation.ts
+const VALIDATION_TIMEOUT_MS = 15e3;
+function normalizeAnthropicCompatibleBaseUrl(baseUrl) {
+	const trimmed = baseUrl.trim().replace(/\/+$/, "");
+	return trimmed.endsWith("/v1") ? trimmed.slice(0, -3) : trimmed;
+}
+function resolveValidationHeaders(headers) {
+	return typeof headers === "function" ? headers() : headers;
+}
+async function createApiKeyValidationError(provider, response) {
+	let details = "";
+	try {
+		details = (await response.text()).trim();
+	} catch {}
+	const message = details ? `${provider} API key validation failed (${response.status}): ${details}` : `${provider} API key validation failed (${response.status})`;
+	return new ProviderHttpError(message, response.status, { headers: response.headers });
+}
+/**
+* Validate an API key against an OpenAI-compatible chat completions endpoint.
+*
+* Performs a minimal request to verify credentials and endpoint access.
+*/
+async function validateOpenAICompatibleApiKey(options) {
+	const timeoutSignal = AbortSignal.timeout(VALIDATION_TIMEOUT_MS);
+	const signal = options.signal ? AbortSignal.any([options.signal, timeoutSignal]) : timeoutSignal;
+	const response = await (options.fetch ?? fetch)(`${options.baseUrl}/chat/completions`, {
+		method: "POST",
+		headers: {
+			"Content-Type": "application/json",
+			Authorization: `Bearer ${options.apiKey}`
+		},
+		body: JSON.stringify({
+			model: options.model,
+			messages: [{
+				role: "user",
+				content: "ping"
+			}],
+			max_tokens: 1,
+			temperature: 0
+		}),
+		signal
+	});
+	if (response.ok) return;
+	throw await createApiKeyValidationError(options.provider, response);
+}
+/**
+* Validate an API key against an Anthropic-compatible messages endpoint.
+*/
+async function validateAnthropicCompatibleApiKey(options) {
+	const timeoutSignal = AbortSignal.timeout(VALIDATION_TIMEOUT_MS);
+	const signal = options.signal ? AbortSignal.any([options.signal, timeoutSignal]) : timeoutSignal;
+	const baseUrl = normalizeAnthropicCompatibleBaseUrl(options.baseUrl);
+	const response = await (options.fetch ?? fetch)(`${baseUrl}/v1/messages`, {
+		method: "POST",
+		headers: {
+			"Content-Type": "application/json",
+			"anthropic-version": "2023-06-01",
+			"x-api-key": options.apiKey
+		},
+		body: JSON.stringify({
+			model: options.model,
+			messages: [{
+				role: "user",
+				content: "ping"
+			}],
+			max_tokens: 1
+		}),
+		signal
+	});
+	if (response.ok) return;
+	throw await createApiKeyValidationError(options.provider, response);
+}
+/**
+* Validate an API key against a provider models endpoint.
+*
+* Useful for providers where access to specific models may vary by plan and
+* should not block key validation.
+*/
+async function validateApiKeyAgainstModelsEndpoint(options) {
+	const timeoutSignal = AbortSignal.timeout(VALIDATION_TIMEOUT_MS);
+	const signal = options.signal ? AbortSignal.any([options.signal, timeoutSignal]) : timeoutSignal;
+	const response = await (options.fetch ?? fetch)(options.modelsUrl, {
+		method: "GET",
+		headers: {
+			...resolveValidationHeaders(options.headers) ?? {},
+			Authorization: `Bearer ${options.apiKey}`
+		},
+		signal
+	});
+	if (response.ok) return;
+	throw await createApiKeyValidationError(options.provider, response);
+}
+//#endregion
+//#region node_modules/.pnpm/@oh-my-pi+pi-ai@17.3.5/node_modules/@oh-my-pi/pi-ai/src/registry/api-key-login.ts
+/**
+* Shared factory for API-key-paste "login" flows.
+*
+* Several providers (Cerebras, Synthetic, Moonshot, Together, NanoGPT, ZenMux)
+* don't actually implement OAuth — they just ask the user to paste an API key,
+* optionally validate it, and return the trimmed key.
+*/
+function createApiKeyLogin(config) {
+	return async function login(options) {
+		if (!options.onPrompt) throw new OnPromptRequiredError(config.providerLabel);
+		if (config.authUrl && config.instructions) options.onAuth?.({
+			url: config.authUrl,
+			instructions: config.instructions
+		});
+		const apiKey = config.emptyKeyFallback === void 0 ? await options.onPrompt({
+			message: config.promptMessage,
+			placeholder: config.placeholder
+		}) : await options.onPrompt({
+			message: config.promptMessage,
+			placeholder: config.placeholder,
+			allowEmpty: true
+		});
+		if (options.signal?.aborted) throw new LoginCancelledError();
+		const trimmed = apiKey.trim();
+		if (!trimmed) {
+			if (config.emptyKeyFallback !== void 0) return config.emptyKeyFallback;
+			throw new ApiKeyRequiredError();
+		}
+		if (config.validation) {
+			options.onProgress?.("Validating API key...");
+			if (config.validation.kind === "chat-completions") await validateOpenAICompatibleApiKey({
+				provider: config.validation.provider,
+				apiKey: trimmed,
+				baseUrl: config.validation.baseUrl,
+				model: config.validation.model,
+				signal: options.signal,
+				fetch: options.fetch
+			});
+			else if (config.validation.kind === "anthropic-messages") await validateAnthropicCompatibleApiKey({
+				provider: config.validation.provider,
+				apiKey: trimmed,
+				baseUrl: config.validation.baseUrl,
+				model: config.validation.model,
+				signal: options.signal,
+				fetch: options.fetch
+			});
+			else await validateApiKeyAgainstModelsEndpoint({
+				provider: config.validation.provider,
+				apiKey: trimmed,
+				modelsUrl: typeof config.validation.modelsUrl === "function" ? config.validation.modelsUrl() : config.validation.modelsUrl,
+				headers: config.validation.headers,
+				signal: options.signal,
+				fetch: options.fetch
+			});
+		}
+		return trimmed;
+	};
+}
+//#endregion
+export { validateApiKeyAgainstModelsEndpoint as n, validateOpenAICompatibleApiKey as r, createApiKeyLogin as t };