328 lines
12 KiB
JavaScript
328 lines
12 KiB
JavaScript
|
#!/usr/bin/env node
|
|||
|
|
/**
|
|||
|
|
* Kling AI image generation — text-to-image, image-to-image, 4K, series, subject
|
|||
|
|
* Node.js 18+, zero external deps
|
|||
|
|
*/
|
|||
|
|
import { submitTask, queryTask, pollTask, downloadFile } from './shared/task.mjs';
|
|||
|
|
import { resolve, join } from 'node:path';
|
|||
|
|
import { fileURLToPath } from 'node:url';
|
|||
|
|
import { parseArgs, getTokenOrExit, readMediaAsValue, resolveAllowedOutputDir } from './shared/args.mjs';
|
|||
|
|
|
|||
|
|
const API_GEN = '/v1/images/generations';
|
|||
|
|
const API_OMNI = '/v1/images/omni-image';
|
|||
|
|
|
|||
|
|
function normalizeModelName(v) {
|
|||
|
|
return String(v || '').trim();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function normalizeAliasKey(v) {
|
|||
|
|
return String(v || '').trim().toLowerCase().replace(/[\s_]+/g, '-');
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function getImageModelAliasTarget(v) {
|
|||
|
|
const key = normalizeAliasKey(v);
|
|||
|
|
const aliasMap = new Map([
|
|||
|
|
['omni3', 'kling-v3-omni'],
|
|||
|
|
['omni-3', 'kling-v3-omni'],
|
|||
|
|
['omni-v3', 'kling-v3-omni'],
|
|||
|
|
['v3-omni', 'kling-v3-omni'],
|
|||
|
|
['o3', 'kling-v3-omni'],
|
|||
|
|
['O3', 'kling-v3-omni'],
|
|||
|
|
['kling-image-o3', 'kling-v3-omni'],
|
|||
|
|
['kling-o3', 'kling-v3-omni'],
|
|||
|
|
['omni1', 'kling-image-o1'],
|
|||
|
|
['omni-1', 'kling-image-o1'],
|
|||
|
|
['o1', 'kling-image-o1'],
|
|||
|
|
['kling-o1', 'kling-image-o1'],
|
|||
|
|
]);
|
|||
|
|
return aliasMap.get(key) || '';
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function validateModelAliasInput(rawModel) {
|
|||
|
|
if (!rawModel) return;
|
|||
|
|
const model = normalizeModelName(rawModel).toLowerCase();
|
|||
|
|
const target = getImageModelAliasTarget(rawModel);
|
|||
|
|
if (!target || model === target) return;
|
|||
|
|
throw new Error(
|
|||
|
|
`Invalid --model alias / --model 使用了别名: ${rawModel}\n`
|
|||
|
|
+ `Use canonical name / 请改用标准名: ${target}\n`
|
|||
|
|
+ 'Alias mapping / 别名映射: omni3 | omni v3 | o3 -> kling-v3-omni; image o1/omni1 -> kling-image-o1',
|
|||
|
|
);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function validateModelForRoute(apiPath, args) {
|
|||
|
|
validateModelAliasInput(args.model);
|
|||
|
|
const model = normalizeModelName(args.model);
|
|||
|
|
if (!model) return;
|
|||
|
|
|
|||
|
|
// We only validate what we can be sure about from public enums.
|
|||
|
|
// - omni-image: only kling-v3-omni / kling-image-o1
|
|||
|
|
// - generations: must not use omni-only models
|
|||
|
|
if (apiPath === API_OMNI) {
|
|||
|
|
const allowed = new Set(['kling-v3-omni', 'kling-image-o1']);
|
|||
|
|
if (!allowed.has(model)) {
|
|||
|
|
throw new Error(
|
|||
|
|
`Invalid --model for omni-image / omni-image 不支持该模型: ${model}\n`
|
|||
|
|
+ `Allowed / 允许: kling-v3-omni, kling-image-o1`,
|
|||
|
|
);
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
const forbidden = new Set(['kling-v3-omni', 'kling-image-o1', 'kling-video-o1']);
|
|||
|
|
if (forbidden.has(model)) {
|
|||
|
|
throw new Error(
|
|||
|
|
`Invalid --model for generations / generations 不支持该模型: ${model}\n`
|
|||
|
|
+ `Hint / 提示: remove --model or use kling-v3`,
|
|||
|
|
);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function parseImageInputs(rawImageArg) {
|
|||
|
|
if (!rawImageArg) return [];
|
|||
|
|
const parts = String(rawImageArg).split(',').map(s => s.trim());
|
|||
|
|
if (parts.some(p => !p)) {
|
|||
|
|
throw new Error(
|
|||
|
|
'Invalid --image list / --image 列表中存在空值；请移除空项并确保每个 image 非空。',
|
|||
|
|
);
|
|||
|
|
}
|
|||
|
|
return parts;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function parseElementIds(rawElementIdsArg) {
|
|||
|
|
if (!rawElementIdsArg) return [];
|
|||
|
|
const parts = String(rawElementIdsArg).split(',').map(s => s.trim());
|
|||
|
|
if (parts.some(p => !p)) {
|
|||
|
|
throw new Error(
|
|||
|
|
'Invalid --element_ids list / --element_ids 列表中存在空值；请移除空项并确保每个 element_id 非空。',
|
|||
|
|
);
|
|||
|
|
}
|
|||
|
|
return parts;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function validateOmniRefCount(imageInputs, elementIds) {
|
|||
|
|
const totalRefs = imageInputs.length + elementIds.length;
|
|||
|
|
if (totalRefs > 10) {
|
|||
|
|
throw new Error(
|
|||
|
|
`Too many refs for omni-image / omni-image 参考图与主体总数超限: max 10 (current ${totalRefs})`,
|
|||
|
|
);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function printHelp() {
|
|||
|
|
console.log(`Kling AI image generation
|
|||
|
|
|
|||
|
|
Usage:
|
|||
|
|
node kling.mjs image --prompt <text> [options] # Text/image-to-image
|
|||
|
|
node kling.mjs image --prompt "..." [--resolution 4k] # 4K / series / subject → Omni
|
|||
|
|
node kling.mjs image --model kling-v3-omni --prompt "..." # explicit Omni model → omni-image (t2i / i2i)
|
|||
|
|
node kling.mjs image --task_id <id> [--download] # Query/download
|
|||
|
|
|
|||
|
|
Submit (common):
|
|||
|
|
--prompt Image description (required). Omni: <<<image_1>>> / <<<element_1>>>
|
|||
|
|
--resolution 1k / 2k / 4k (4k uses Omni)
|
|||
|
|
--aspect_ratio Aspect ratio (default: 16:9 basic, auto for Omni)
|
|||
|
|
--n Number of images 1-9
|
|||
|
|
--output_dir Output dir (default: ./output)
|
|||
|
|
--no-wait Submit only, do not wait
|
|||
|
|
--wait Wait for completion (default)
|
|||
|
|
|
|||
|
|
Basic API:
|
|||
|
|
--negative_prompt Negative prompt
|
|||
|
|
--model Model (default: kling-v3)
|
|||
|
|
|
|||
|
|
Omni (4K/series/subject):
|
|||
|
|
--model kling-v3-omni / kling-image-o1
|
|||
|
|
--result_type single / series (default: single)
|
|||
|
|
--series_amount Series count 2-9 (when result_type=series)
|
|||
|
|
--image Reference image path or URL, comma-separated for multiple
|
|||
|
|
--element_ids Subject IDs, comma-separated
|
|||
|
|
(omni refs) image count + element count <= 10
|
|||
|
|
|
|||
|
|
Query/download:
|
|||
|
|
--task_id Task ID
|
|||
|
|
--download Download if task succeeded
|
|||
|
|
|
|||
|
|
Env:
|
|||
|
|
credentials file ~/.config/kling/.credentials (access_key_id, secret_access_key)
|
|||
|
|
KLING_TOKEN Session-only Bearer (optional override)
|
|||
|
|
KLING_MEDIA_ROOTS Comma-separated extra dirs for --image / --output_dir (default: cwd only)
|
|||
|
|
KLING_ALLOW_ABSOLUTE_PATHS=1 Allow any local path (e.g. WSL downloads)`);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function useOmniApi(args) {
|
|||
|
|
// Match video.mjs chooseApiPath: explicit Omni image models → omni-image (incl. plain text-to-image).
|
|||
|
|
const m = normalizeModelName(args.model).toLowerCase();
|
|||
|
|
if (m === 'kling-v3-omni' || m === 'kling-image-o1') return true;
|
|||
|
|
if (args.element_ids) return true;
|
|||
|
|
if (args.result_type === 'series') return true;
|
|||
|
|
if ((args.resolution || '').toLowerCase() === '4k') return true;
|
|||
|
|
if ((args.aspect_ratio || '').toLowerCase() === 'auto') return true;
|
|||
|
|
if (args.image && args.image.includes(',')) return true;
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async function queryTaskAnyPath(taskId, token) {
|
|||
|
|
for (const apiPath of [API_OMNI, API_GEN]) {
|
|||
|
|
try {
|
|||
|
|
const data = await queryTask(apiPath, taskId, token);
|
|||
|
|
if (data && (data.task_status === 'succeed' || data.task_status === 'failed' || data.task_status === 'processing' || data.task_status === 'submitted')) {
|
|||
|
|
return { apiPath, data };
|
|||
|
|
}
|
|||
|
|
} catch (_) { /* try next */ }
|
|||
|
|
}
|
|||
|
|
throw new Error(`Task not found / 未找到任务: ${taskId}`);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function collectImageUrls(taskResult) {
|
|||
|
|
const urls = [];
|
|||
|
|
const append = (list) => {
|
|||
|
|
if (!Array.isArray(list)) return;
|
|||
|
|
for (const item of list) {
|
|||
|
|
if (item?.url) urls.push(item.url);
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
append(taskResult?.images);
|
|||
|
|
append(taskResult?.series_images);
|
|||
|
|
if (urls.length === 0 && taskResult?.url) urls.push(taskResult.url);
|
|||
|
|
return urls;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async function pollAndDownloadImages(apiPath, taskId, outputDir, opts = {}) {
|
|||
|
|
const data = await pollTask(apiPath, taskId, opts);
|
|||
|
|
const urls = collectImageUrls(data?.task_result || {});
|
|||
|
|
if (urls.length === 0) {
|
|||
|
|
throw new Error('Task succeeded but missing image urls / 任务成功但未返回图片 URL');
|
|||
|
|
}
|
|||
|
|
const outPaths = [];
|
|||
|
|
for (let i = 0; i < urls.length; i++) {
|
|||
|
|
const outPath = join(outputDir, urls.length === 1 ? `${taskId}.png` : `${taskId}_${i}.png`);
|
|||
|
|
await downloadFile(urls[i], outPath);
|
|||
|
|
outPaths.push(outPath);
|
|||
|
|
}
|
|||
|
|
return outPaths;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export async function main() {
|
|||
|
|
const args = parseArgs(process.argv);
|
|||
|
|
if (args.help) { printHelp(); return; }
|
|||
|
|
validateModelAliasInput(args.model);
|
|||
|
|
|
|||
|
|
const token = await getTokenOrExit();
|
|||
|
|
const outputDir = resolveAllowedOutputDir(args.output_dir || './output');
|
|||
|
|
const queryHint = `node kling.mjs image --task_id`;
|
|||
|
|
|
|||
|
|
if (args.task_id && !args.prompt) {
|
|||
|
|
try {
|
|||
|
|
const { apiPath, data } = await queryTaskAnyPath(args.task_id, token);
|
|||
|
|
console.log(`Task ID / 任务 ID: ${args.task_id}`);
|
|||
|
|
console.log(`Status / 状态: ${data?.task_status || 'unknown'}`);
|
|||
|
|
const result = data?.task_result || {};
|
|||
|
|
const imageUrls = collectImageUrls(result);
|
|||
|
|
imageUrls.forEach((url, i) => {
|
|||
|
|
console.log(`Image / 图片[${i}]: ${url}`);
|
|||
|
|
});
|
|||
|
|
if (args.download && imageUrls.length > 0) {
|
|||
|
|
const { mkdir } = await import('node:fs/promises');
|
|||
|
|
await mkdir(outputDir, { recursive: true });
|
|||
|
|
for (let i = 0; i < imageUrls.length; i++) {
|
|||
|
|
const outPath = join(outputDir, imageUrls.length === 1 ? `${args.task_id}.png` : `${args.task_id}_${i}.png`);
|
|||
|
|
await downloadFile(imageUrls[i], outPath);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
} catch (e) {
|
|||
|
|
console.error(`Error / 错误: ${e.message}`);
|
|||
|
|
process.exit(1);
|
|||
|
|
}
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (!args.prompt) {
|
|||
|
|
console.error('Error / 错误: --prompt or --task_id required');
|
|||
|
|
console.error('Use --help / 使用 --help 查看帮助');
|
|||
|
|
process.exit(1);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const apiPath = useOmniApi(args) ? API_OMNI : API_GEN;
|
|||
|
|
const imageInputs = parseImageInputs(args.image);
|
|||
|
|
const elementIds = parseElementIds(args.element_ids);
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
validateModelForRoute(apiPath, args);
|
|||
|
|
|
|||
|
|
if (apiPath === API_GEN) {
|
|||
|
|
const payload = {
|
|||
|
|
model_name: args.model || 'kling-v3',
|
|||
|
|
prompt: args.prompt,
|
|||
|
|
negative_prompt: args.negative_prompt || '',
|
|||
|
|
n: parseInt(args.n || '1', 10),
|
|||
|
|
aspect_ratio: args.aspect_ratio || '16:9',
|
|||
|
|
resolution: args.resolution || '1k',
|
|||
|
|
callback_url: '',
|
|||
|
|
};
|
|||
|
|
if (imageInputs.length > 0) {
|
|||
|
|
payload.image = await readMediaAsValue(imageInputs[0]);
|
|||
|
|
}
|
|||
|
|
const result = await submitTask(API_GEN, payload, token);
|
|||
|
|
console.log(`\nTask ID / 任务 ID: ${result.taskId}`);
|
|||
|
|
console.log(`Query / 查询: ${queryHint} ${result.taskId} [--download]`);
|
|||
|
|
if (args.wait !== false) {
|
|||
|
|
console.log();
|
|||
|
|
const outPaths = await pollAndDownloadImages(API_GEN, result.taskId, outputDir, { token });
|
|||
|
|
console.log(`\n✓ Done / 完成: ${outPaths.length} image(s)`);
|
|||
|
|
outPaths.forEach((p) => console.log(` - ${p}`));
|
|||
|
|
}
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const payload = {
|
|||
|
|
model_name: args.model || 'kling-v3-omni',
|
|||
|
|
prompt: args.prompt,
|
|||
|
|
resolution: (args.resolution || '1k').toLowerCase(),
|
|||
|
|
aspect_ratio: (args.aspect_ratio || 'auto').toLowerCase(),
|
|||
|
|
result_type: args.result_type || 'single',
|
|||
|
|
callback_url: '',
|
|||
|
|
};
|
|||
|
|
if (payload.result_type === 'series') {
|
|||
|
|
if (imageInputs.length === 0) {
|
|||
|
|
throw new Error(
|
|||
|
|
'Invalid --result_type series without --image / 组图仅支持 i2i，请提供 --image（t2i 不支持 series）。',
|
|||
|
|
);
|
|||
|
|
}
|
|||
|
|
payload.series_amount = parseInt(args.series_amount || '4', 10);
|
|||
|
|
} else {
|
|||
|
|
payload.n = parseInt(args.n || '1', 10);
|
|||
|
|
}
|
|||
|
|
validateOmniRefCount(imageInputs, elementIds);
|
|||
|
|
if (imageInputs.length > 0) {
|
|||
|
|
payload.image_list = [];
|
|||
|
|
for (const img of imageInputs) {
|
|||
|
|
payload.image_list.push({ image: await readMediaAsValue(img) });
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if (elementIds.length > 0) {
|
|||
|
|
payload.element_list = elementIds.map(id => ({ element_id: id }));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const result = await submitTask(API_OMNI, payload, token);
|
|||
|
|
console.log(`\nTask ID / 任务 ID: ${result.taskId}`);
|
|||
|
|
console.log(`Query / 查询: ${queryHint} ${result.taskId} [--download]`);
|
|||
|
|
if (args.wait !== false) {
|
|||
|
|
console.log();
|
|||
|
|
const outPaths = await pollAndDownloadImages(API_OMNI, result.taskId, outputDir, { token });
|
|||
|
|
console.log(`\n✓ Done / 完成: ${outPaths.length} image(s)`);
|
|||
|
|
outPaths.forEach((p) => console.log(` - ${p}`));
|
|||
|
|
}
|
|||
|
|
} catch (e) {
|
|||
|
|
console.error(`Error / 错误: ${e.message}`);
|
|||
|
|
process.exit(1);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const __filename = fileURLToPath(import.meta.url);
|
|||
|
|
if (process.argv[1] && resolve(__filename) === resolve(process.argv[1])) {
|
|||
|
|
main().catch((e) => {
|
|||
|
|
console.error(`Error / 错误: ${e?.message || e}`);
|
|||
|
|
process.exit(1);
|
|||
|
|
});
|
|||
|
|
}
|