@onkernel/cua-cli 0.1.0

package/dist/harness-models-GT8Ke1vt.js

@@ -0,0 +1,106 @@
+import { formatCuaModelRef, getCuaModel, isCuaProvider, listCuaModels, parseCuaModelRef } from "@onkernel/cua-ai";
+import Kernel, { NotFoundError } from "@onkernel/sdk";
+//#region src/harness-browser.ts
+const CUID2_LENGTH = 24;
+const CUID2_PATTERN = /^[a-z][a-z0-9]{23}$/;
+function looksLikeProfileId(selector) {
+	const trimmed = selector.trim();
+	return trimmed.length === CUID2_LENGTH && CUID2_PATTERN.test(trimmed);
+}
+/**
+* Resolve a `--profile <name|id>` selector to a concrete profile id.
+* Looks up by id first; if the API reports not-found and the selector does
+* not look like a CUID2 id, the profile is created with that name.
+*/
+async function resolveProfileId(client, selector) {
+	const trimmed = selector.trim();
+	if (!trimmed) throw new Error("profile selector is empty");
+	try {
+		return (await client.profiles.retrieve(trimmed)).id;
+	} catch (err) {
+		if (!(err instanceof NotFoundError)) throw new Error(`looking up browser profile "${trimmed}": ${err.message}`, { cause: err });
+		if (looksLikeProfileId(trimmed)) throw new Error(`browser profile "${trimmed}" was not found`);
+		return (await client.profiles.create({ name: trimmed })).id;
+	}
+}
+/** Create a Kernel SDK client with the supplied auth. */
+function createKernelClient(apiKey, baseUrl) {
+	return new Kernel({
+		apiKey,
+		...baseUrl ? { baseURL: baseUrl } : {}
+	});
+}
+/** Provision a fresh Kernel cloud browser session and return a handle. */
+async function provisionBrowser(opts) {
+	const client = createKernelClient(opts.apiKey, opts.baseUrl);
+	const timeoutSeconds = opts.timeoutSeconds && opts.timeoutSeconds > 0 ? opts.timeoutSeconds : 300;
+	let profileId = (opts.profileId ?? "").trim();
+	if (!profileId && opts.profileSelector && opts.profileSelector.trim()) profileId = await resolveProfileId(client, opts.profileSelector);
+	const params = {
+		stealth: true,
+		timeout_seconds: timeoutSeconds
+	};
+	if (profileId) params.profile = {
+		id: profileId,
+		save_changes: opts.saveChanges ?? false
+	};
+	const browser = await client.browsers.create(params);
+	return {
+		client,
+		browser,
+		profileId: profileId || void 0,
+		async close() {
+			await client.browsers.deleteByID(browser.session_id);
+		}
+	};
+}
+/**
+* Capture a screenshot through the SDK. Falls back to undefined when the
+* call fails — first-prompt images are best-effort.
+*/
+async function captureScreenshot(client, sessionId) {
+	try {
+		const arrayBuffer = await (await client.browsers.computer.captureScreenshot(sessionId)).arrayBuffer();
+		return Buffer.from(arrayBuffer);
+	} catch {
+		return;
+	}
+}
+//#endregion
+//#region src/harness-models.ts
+/** Default model used by the new harness wiring. */
+const DEFAULT_CUA_MODEL_REF = "openai:gpt-5.5";
+/**
+* Resolve a model ref from CLI input. Accepts either a provider-qualified
+* `provider:model` ref or a bare model id when it matches exactly one
+* catalog entry. Throws when bare ids are ambiguous or unknown.
+*/
+function resolveCuaModelRef(input) {
+	if (!input || !input.trim()) return DEFAULT_CUA_MODEL_REF;
+	const value = input.trim();
+	if (value.includes(":")) {
+		const { provider, model } = parseCuaModelRef(value);
+		const ref = formatCuaModelRef(provider, model);
+		getCuaModel(ref);
+		return ref;
+	}
+	const matches = listCuaModels().filter((m) => m.model === value);
+	if (matches.length === 0) throw new Error(`unknown model "${value}" (run \`cua models\` to list supported -m/--model values)`);
+	if (matches.length > 1) {
+		const refs = matches.map((m) => m.ref).join(", ");
+		throw new Error(`ambiguous model "${value}" (matches: ${refs}); pass a provider-qualified ref like "openai:${value}"`);
+	}
+	return matches[0].ref;
+}
+/**
+* List supported models, optionally filtered to a provider. Accepts either
+* the canonical `"google"` or the CLI-friendly `"gemini"` alias.
+*/
+function listSupportedModels(provider) {
+	if (!provider) return listCuaModels();
+	const normalized = provider === "gemini" ? "google" : provider;
+	if (!isCuaProvider(normalized)) throw new Error(`unknown provider "${provider}"`);
+	return listCuaModels(normalized);
+}
+//#endregion
+export { createKernelClient as a, captureScreenshot as i, listSupportedModels as n, provisionBrowser as o, resolveCuaModelRef as r, resolveProfileId as s, DEFAULT_CUA_MODEL_REF as t };