@neta-art/generation 0.1.24 → 0.1.26

package/dist/{export-config-D47KgbO7.js → export-config-B3sM1Bxd.js}

@@ -1,4 +1,4 @@
-import { a as compactArray, c as slugifyFileName, i as cloneJson, l as GENERATION_MODEL_CATEGORIES, n as getBuiltinGenerationModel, o as compactObject, r as listBuiltinGenerationModels, s as getBlockMeta, t as builtinGenerationModels, u as MODEL_SCHEMA } from "./builtins-Bn6iOBYm.js";
+import { a as compactArray, c as slugifyFileName, d as MODEL_SCHEMA, i as cloneJson, l as GENERATION_MODEL_CATEGORIES, n as getBuiltinGenerationModel, o as compactObject, r as listBuiltinGenerationModels, s as getBlockMeta, t as builtinGenerationModels, u as GENERATION_PRICING_UNITS } from "./builtins-Cvj5PK_9.js";
 import { mkdir, readFile, readdir, writeFile } from "node:fs/promises";
 import { extname, join } from "node:path";
 import { parse, stringify } from "yaml";
@@ -223,6 +223,23 @@ function isRecord$7(value) {
 	return typeof value === "object" && value !== null;
 }
 
+//#endregion
+//#region src/task-poll.ts
+function sleep(ms) {
+	return new Promise((resolve) => {
+		setTimeout(resolve, Math.max(0, ms));
+	});
+}
+/** Yield immediately, then wait `intervalMs` between later ticks until `maxWaitMs`. */
+async function* taskPollTicks(maxWaitMs, intervalMs) {
+	const startedAt = Date.now();
+	while (Date.now() - startedAt <= maxWaitMs) {
+		yield;
+		if (Date.now() - startedAt > maxWaitMs) return;
+		await sleep(intervalMs);
+	}
+}
+
 //#endregion
 //#region src/validation.ts
 function specsByType(specs) {
@@ -359,11 +376,8 @@ function mergeTextBlocks(declaration, content) {
 //#endregion
 //#region src/adapters/ark-video-generations.ts
 const REQUEST_TIMEOUT_MS$8 = 186e4;
-const DEFAULT_POLL_INTERVAL_SEC$4 = 2;
+const DEFAULT_POLL_INTERVAL_SEC$4 = 1;
 const DEFAULT_MAX_WAIT_SEC$4 = 600;
-function sleep$4(ms) {
-	return new Promise((resolve) => setTimeout(resolve, ms));
-}
 function asString$4(value) {
 	return typeof value === "string" && value ? value : void 0;
 }
@@ -550,9 +564,7 @@ async function arkVideoGenerationsAdapter(input) {
 		method: "POST",
 		body: JSON.stringify(payload)
 	}));
-	const startedAt = Date.now();
-	while (Date.now() - startedAt <= maxWaitSec * 1e3) {
-		await sleep$4(pollIntervalSec * 1e3);
+	for await (const _ of taskPollTicks(maxWaitSec * 1e3, pollIntervalSec * 1e3)) {
 		const rawStatus = await requestJson$4(input, `/v1/video/generations/${encodeURIComponent(taskId)}`, { method: "GET" });
 		const status = normalizeTaskStatus(rawStatus);
 		if (status.status === "succeeded") {
@@ -954,9 +966,6 @@ const KLING_MODELS = {
 		imageMode: "multi"
 	}
 };
-function sleep$3(ms) {
-	return new Promise((resolve) => setTimeout(resolve, ms));
-}
 function asString$3(value) {
 	return typeof value === "string" && value.trim() ? value.trim() : void 0;
 }
@@ -1132,9 +1141,7 @@ async function klingVideoGenerationsAdapter(input) {
 	}));
 	const pollIntervalSec = asInteger$2(input.parameters.poll_interval, DEFAULT_POLL_INTERVAL_SEC$3);
 	const maxWaitSec = asInteger$2(input.parameters.max_wait, DEFAULT_MAX_WAIT_SEC$3);
-	const startedAt = Date.now();
-	while (Date.now() - startedAt <= maxWaitSec * 1e3) {
-		await sleep$3(pollIntervalSec * 1e3);
+	for await (const _ of taskPollTicks(maxWaitSec * 1e3, pollIntervalSec * 1e3)) {
 		const rawStatus = await requestJson$3(input, `${model.submitPath}/${encodeURIComponent(taskId)}`, { method: "GET" });
 		const status = extractStatus(rawStatus);
 		if (status.status === "succeeded") {
@@ -1168,10 +1175,8 @@ async function klingVideoGenerationsAdapter(input) {
 //#endregion
 //#region src/adapters/minimax-h3-video-generations.ts
 const REQUEST_TIMEOUT_MS$4 = 186e4;
-const DEFAULT_POLL_INTERVAL_SEC$2 = 2;
-const DEFAULT_MAX_WAIT_SEC$2 = 600;
-const MIN_DURATION_SEC = 4;
-const MAX_DURATION_SEC = 15;
+const DEFAULT_POLL_INTERVAL_SEC$2 = 1;
+const DEFAULT_MAX_WAIT_SEC$2 = 1800;
 const MAX_MEDIA_COUNT = 12;
 const MAX_REFERENCE_IMAGES = 9;
 const MAX_REFERENCE_VIDEOS = 3;
@@ -1187,9 +1192,6 @@ const H3_RATIOS = [
 	"3:4",
 	"9:16"
 ];
-function sleep$2(ms) {
-	return new Promise((resolve) => setTimeout(resolve, ms));
-}
 function asString$2(value) {
 	return typeof value === "string" && value.trim() ? value.trim() : void 0;
 }
@@ -1301,14 +1303,10 @@ function buildContent(prompt, media) {
 	})];
 }
 function resolveResolution(value) {
-	const resolution = (asString$2(value) ?? "768P").toUpperCase();
-	if (resolution !== "768P" && resolution !== "2K") throw new GenerationValidationError("MiniMax H3 resolution must be 768P or 2K");
-	return resolution;
+	return asString$2(value) ?? "768P";
 }
 function resolveDuration(value) {
-	const duration = asInteger$1(value, 5);
-	if (duration < MIN_DURATION_SEC || duration > MAX_DURATION_SEC) throw new GenerationValidationError("MiniMax H3 duration must be an integer between 4 and 15");
-	return duration;
+	return asInteger$1(value, 5);
 }
 function resolveRatio(value, mode) {
 	const requested = asString$2(value) ?? DEFAULT_RATIO;
@@ -1409,9 +1407,7 @@ async function minimaxH3VideoGenerationsAdapter(input) {
 		method: "POST",
 		body: JSON.stringify(payload)
 	}));
-	const startedAt = Date.now();
-	while (Date.now() - startedAt <= maxWaitSec * 1e3) {
-		await sleep$2(pollIntervalSec * 1e3);
+	for await (const _ of taskPollTicks(maxWaitSec * 1e3, pollIntervalSec * 1e3)) {
 		const rawStatus = await requestJson$2(input, `/v1/video/generations/${encodeURIComponent(taskId)}`, { method: "GET" });
 		const status = extractTaskStatus$1(rawStatus);
 		if (status.status === "succeeded") {
@@ -1635,9 +1631,6 @@ const FINAL_FAILURE_STATUSES = new Set([
 	"canceled",
 	"expired"
 ]);
-function sleep$1(ms) {
-	return new Promise((resolve) => setTimeout(resolve, ms));
-}
 function isRecord$3(value) {
 	return !!value && typeof value === "object" && !Array.isArray(value);
 }
@@ -1905,9 +1898,7 @@ function buildResult(operation, task, raw) {
 	});
 }
 async function pollSunoTask(input, operation, taskId, pollIntervalSec, maxWaitSec) {
-	const startedAt = Date.now();
-	while (Date.now() - startedAt <= maxWaitSec * 1e3) {
-		await sleep$1(pollIntervalSec * 1e3);
+	for await (const _ of taskPollTicks(maxWaitSec * 1e3, pollIntervalSec * 1e3)) {
 		let raw;
 		try {
 			raw = await requestJson$1(input, `/suno/fetch/${encodeURIComponent(taskId)}`, { method: "GET" });
@@ -1950,11 +1941,8 @@ async function sunoTasksAdapter(input) {
 //#endregion
 //#region src/adapters/video-upscale-native.ts
 const REQUEST_TIMEOUT_MS = 186e4;
-const DEFAULT_POLL_INTERVAL_SEC = 2;
+const DEFAULT_POLL_INTERVAL_SEC = 1;
 const DEFAULT_MAX_WAIT_SEC = 600;
-function sleep(ms) {
-	return new Promise((resolve) => setTimeout(resolve, ms));
-}
 function isRecord$2(value) {
 	return !!value && typeof value === "object" && !Array.isArray(value);
 }
@@ -2077,9 +2065,7 @@ async function videoUpscaleNativeAdapter(input) {
 			video_url: videoUrl
 		})
 	}));
-	const startedAt = Date.now();
-	while (Date.now() - startedAt <= maxWaitSec * 1e3) {
-		await sleep(pollIntervalSec * 1e3);
+	for await (const _ of taskPollTicks(maxWaitSec * 1e3, pollIntervalSec * 1e3)) {
 		const rawStatus = await requestJson(input, `/v1/video/generations/${encodeURIComponent(taskId)}`, { method: "GET" });
 		const status = extractTaskStatus(rawStatus);
 		if (status.status === "succeeded") {
@@ -2173,6 +2159,19 @@ function isMetaTaskVariantSpec(value) {
 function isGenerationModelCategory(value) {
 	return typeof value === "string" && GENERATION_MODEL_CATEGORIES.includes(value);
 }
+function isFiniteNonNegative(value) {
+	return typeof value === "number" && Number.isFinite(value) && value >= 0;
+}
+function isGenerationModelPricing(value) {
+	if (!isRecord$1(value)) return false;
+	if (typeof value.unit !== "string" || !GENERATION_PRICING_UNITS.includes(value.unit)) return false;
+	if (value.note !== void 0 && typeof value.note !== "string") return false;
+	if (value.amount !== void 0) {
+		if (value.min !== void 0 || value.max !== void 0) return false;
+		return isFiniteNonNegative(value.amount);
+	}
+	return isFiniteNonNegative(value.min) && isFiniteNonNegative(value.max) && value.min <= value.max;
+}
 function isMetaSpec(value) {
 	if (!isRecord$1(value)) return false;
 	return (value.fields === void 0 || isRecord$1(value.fields) && Object.values(value.fields).every((field) => isMetaFieldSpec(field))) && (value.taskField === void 0 || typeof value.taskField === "string") && (value.taskVariants === void 0 || isRecord$1(value.taskVariants) && Object.values(value.taskVariants).every((variant) => isMetaTaskVariantSpec(variant)));
@@ -2184,7 +2183,7 @@ function isGenerationModelDeclaration(value) {
 	const parameters = value.parameters;
 	const meta = value.meta;
 	const examples = value.examples;
-	return value.schema === MODEL_SCHEMA && typeof value.model === "string" && value.model.trim().length > 0 && (value.category === void 0 || isGenerationModelCategory(value.category)) && (value.hidden === void 0 || typeof value.hidden === "boolean") && (value.allowUnknownParameters === void 0 || typeof value.allowUnknownParameters === "boolean") && isRecord$1(adapter) && typeof adapter.type === "string" && isRecord$1(content) && Array.isArray(content.input) && (parameters === void 0 || isRecord$1(parameters) && Object.values(parameters).every(isParameterSpec)) && (meta === void 0 || isMetaSpec(meta)) && (examples === void 0 || Array.isArray(examples));
+	return value.schema === MODEL_SCHEMA && typeof value.model === "string" && value.model.trim().length > 0 && (value.category === void 0 || isGenerationModelCategory(value.category)) && (value.pricing === void 0 || isGenerationModelPricing(value.pricing)) && (value.hidden === void 0 || typeof value.hidden === "boolean") && (value.allowUnknownParameters === void 0 || typeof value.allowUnknownParameters === "boolean") && isRecord$1(adapter) && typeof adapter.type === "string" && isRecord$1(content) && Array.isArray(content.input) && (parameters === void 0 || isRecord$1(parameters) && Object.values(parameters).every(isParameterSpec)) && (meta === void 0 || isMetaSpec(meta)) && (examples === void 0 || Array.isArray(examples));
 }
 function parseGenerationModelDeclaration(rawText, filePath = "model.yaml") {
 	const parsed = extname(filePath) === ".json" ? JSON.parse(rawText) : parse(rawText);
@@ -2475,4 +2474,4 @@ async function exportBuiltinModelConfigs(directory) {
 
 //#endregion
 export { mergeTextBlocks as A, GenerationValidationError as B, openAiImageEditsAdapter as C, audioSpeechAdapter as D, geminiGenerateContentAdapter as E, GenerationError as F, GenerationProviderError as I, GenerationTimeoutError as L, resolveGenerationParameters as M, validateGenerationContent as N, arkVideoGenerationsAdapter as O, GenerationConfigError as P, GenerationTransportError as R, openAiImagesAdapter as S, klingVideoGenerationsAdapter as T, writeGenerationModelDeclarations as _, createGenerationClientFromDirectory as a, videoUpscaleNativeAdapter as b, defaultGenerationSourceResolver as c, parseGenerationModelDeclaration as d, readGenerationModelDeclaration as f, writeGenerationModelDeclaration as g, stringifyGenerationModelDeclaration as h, createGenerationClient as i, resolveGenerationMeta as j, mergeGenerationMeta as k, isGenerationModelDeclaration as l, readGenerationModelDeclarationsFromFiles as m, exportBuiltinModelConfigs as n, createGenerationClientFromFile as o, readGenerationModelDeclarationsFromDirectory as p, stringifyBuiltinModelConfig as r, createGenerationClientFromFiles as s, exportBuiltinModelConfig as t, mergeGenerationModelDeclarations as u, builtinGenerationAdapters as v, minimaxH3VideoGenerationsAdapter as w, sunoTasksAdapter as x, getGenerationAdapter as y, GenerationUnsupportedAdapterError as z };
-//# sourceMappingURL=export-config-D47KgbO7.js.map
+//# sourceMappingURL=export-config-B3sM1Bxd.js.map