kie-ai-cli 1.0.0

package/dist/commands/config.js

@@ -0,0 +1,61 @@
+import chalk from 'chalk';
+import { CONFIG_KEY_IMAGE_MODEL, formatModelLabel, getChatModel, getImageModel, maskSecret, printChatModels, promptSelectChatModel, setChatModel, } from '../models.js';
+import { getImageAssistModel, setImageAssistModel, } from '../image-assist.js';
+import { getActiveSessionId, listSessions, printSessionList, } from '../session.js';
+export async function configAction(options, config) {
+    if (options.setApiKey) {
+        config.set('apiKey', options.setApiKey);
+        console.log(chalk.green('✅ API Key 配置已保存。'));
+    }
+    if (options.setBaseUrl) {
+        config.set('baseUrl', options.setBaseUrl);
+        console.log(chalk.green('✅ Base URL 配置已保存。'));
+    }
+    if (options.setModel) {
+        setChatModel(config, options.setModel);
+        console.log(chalk.green(`✅ 默认对话模型已设为: ${options.setModel}`));
+    }
+    if (options.setImageModel) {
+        config.set(CONFIG_KEY_IMAGE_MODEL, options.setImageModel);
+        console.log(chalk.green(`✅ 默认图片模型已设为: ${options.setImageModel}`));
+    }
+    if (options.setImageAssistModel) {
+        setImageAssistModel(config, options.setImageAssistModel);
+        console.log(chalk.green(`✅ 图片助手模型已设为: ${formatModelLabel(options.setImageAssistModel)}`));
+    }
+    if (!options.setApiKey &&
+        !options.setBaseUrl &&
+        !options.setModel &&
+        !options.setImageModel &&
+        !options.setImageAssistModel) {
+        const apiKey = config.get('apiKey');
+        console.log(chalk.yellow('当前配置信息:'));
+        console.log(chalk.cyan(`API Key: ${apiKey ? maskSecret(apiKey) : '未设置'}`));
+        console.log(chalk.cyan(`Base URL: ${config.get('baseUrl') || 'https://api.kie.ai (默认)'}`));
+        console.log(chalk.cyan(`对话模型: ${formatModelLabel(getChatModel(config))}`));
+        console.log(chalk.cyan(`图片模型: ${getImageModel(config)}`));
+        console.log(chalk.cyan(`图片助手: ${formatModelLabel(getImageAssistModel(config))}`));
+    }
+}
+export async function sessionsAction(config) {
+    const activeId = getActiveSessionId(config);
+    printSessionList(listSessions(config), activeId);
+    console.log(chalk.gray(`切换会话: kie chat --resume <id>  或交互中 /sessions\n`));
+}
+export async function modelsAction(options, config) {
+    const current = getChatModel(config);
+    if (!process.stdin.isTTY || options.list) {
+        printChatModels(current);
+        return;
+    }
+    const picked = await promptSelectChatModel(current);
+    if (options.set) {
+        setChatModel(config, picked);
+        console.log(chalk.green(`\n✅ 默认模型已设为: ${formatModelLabel(picked)}`));
+    }
+    else {
+        console.log(chalk.cyan(`\n已选择: ${formatModelLabel(picked)}`));
+        console.log(chalk.gray(`保存为默认: kie config --set-model ${picked}`));
+        console.log(chalk.gray(`本次对话使用: kie chat --model ${picked}`));
+    }
+}

package/dist/commands/image.js

@@ -0,0 +1,102 @@
+import chalk from 'chalk';
+import { buildImageTaskInput, printImageModels, promptSelectImageModel, resolveImageModel, setImageModel, } from '../image-models.js';
+import { createJob } from '../job.js';
+import { runImageRepl } from '../image-repl.js';
+import { waitImageTask, } from '../image.js';
+import { getImageModel } from '../models.js';
+import { kieUploadFile } from '../api.js';
+export async function imageModelsAction(options, config) {
+    const current = getImageModel(config);
+    if (!process.stdin.isTTY || options.list) {
+        printImageModels(current);
+        return;
+    }
+    const picked = await promptSelectImageModel(current);
+    if (options.set) {
+        setImageModel(config, picked);
+        console.log(chalk.green(`\n✅ 默认图片模型已设为: ${picked}`));
+    }
+    else {
+        console.log(chalk.cyan(`\n已选择: ${picked}`));
+        console.log(chalk.gray(`保存为默认: kie config --set-image-model ${picked}`));
+        console.log(chalk.gray(`本次生成: kie image -m ${picked} "提示词"`));
+    }
+}
+export async function imageAction(prompt, options, config) {
+    const apiKey = config.get('apiKey');
+    const baseUrl = config.get('baseUrl') || 'https://api.kie.ai';
+    if (!apiKey) {
+        console.log(chalk.red('❌ 缺少必要配置，请先运行以下命令配置 API Key:'));
+        console.log(chalk.yellow('kie config --set-api-key <KEY>'));
+        process.exit(1);
+    }
+    if (!prompt) {
+        await runImageRepl({ config, apiKey, baseUrl });
+        return;
+    }
+    const ora = (await import('ora')).default;
+    const model = resolveImageModel(config, options.model);
+    // Pre-process local reference images and upload if needed
+    let imageUrls = undefined;
+    if (options.imageUrl) {
+        imageUrls = [];
+        const fs = await import('fs');
+        const path = await import('path');
+        const rawUrls = Array.isArray(options.imageUrl) ? options.imageUrl : [options.imageUrl];
+        for (const item of rawUrls) {
+            if (item.startsWith('http://') || item.startsWith('https://')) {
+                imageUrls.push(item);
+            }
+            else {
+                const resolvedPath = path.resolve(item);
+                if (fs.existsSync(resolvedPath)) {
+                    const uploadSpinner = ora(`正在上传本地参考图: ${item}...`).start();
+                    try {
+                        const trimmedBaseUrl = baseUrl.replace(/\/+$/, '');
+                        const uploadUrl = `${trimmedBaseUrl}/api/file-stream-upload`;
+                        const uploadRes = await kieUploadFile(uploadUrl, {
+                            apiKey,
+                            filePath: resolvedPath,
+                        });
+                        if (uploadRes.success) {
+                            uploadSpinner.succeed(chalk.green(`本地参考图 ${item} 上传成功！`));
+                            imageUrls.push(uploadRes.data.downloadUrl);
+                        }
+                        else {
+                            uploadSpinner.fail(chalk.red(`本地参考图 ${item} 上传失败: ${uploadRes.msg}`));
+                            throw new Error(`参考图上传失败: ${uploadRes.msg}`);
+                        }
+                    }
+                    catch (uploadErr) {
+                        uploadSpinner.fail(chalk.red(`本地参考图 ${item} 上传出错: ${uploadErr.message}`));
+                        throw uploadErr;
+                    }
+                }
+                else {
+                    throw new Error(`图片路径既不是合法的 URL，也不是存在的本地文件: ${item}`);
+                }
+            }
+        }
+    }
+    const spinner = ora(`正在提交图片生成任务 (${model})...`).start();
+    try {
+        const input = buildImageTaskInput(model, {
+            prompt,
+            aspectRatio: options.ratio,
+            quality: options.quality,
+            imageUrls,
+            seed: options.seed ? parseInt(options.seed, 10) : undefined,
+        });
+        const taskId = await createJob({ apiKey, baseUrl, model, input });
+        spinner.succeed(chalk.green(`任务提交成功！Job ID: ${taskId}`));
+        if (options.wait !== false) {
+            await waitImageTask({ apiKey, baseUrl, taskId, download: options.download });
+        }
+        else {
+            console.log(chalk.gray(`查询进度: kie job ${taskId}`));
+        }
+    }
+    catch (err) {
+        spinner.fail(chalk.red(`任务提交失败: ${err.message}`));
+    }
+}

package/dist/commands/job.js

@@ -0,0 +1,25 @@
+import chalk from 'chalk';
+import { queryImageTask, printJobStatus } from '../image.js';
+export async function jobAction(jobId, options, config) {
+    const apiKey = config.get('apiKey');
+    const baseUrl = config.get('baseUrl') || 'https://api.kie.ai';
+    if (!apiKey) {
+        console.log(chalk.red('❌ 缺少必要配置，请运行 kie config --set-api-key 设置'));
+        process.exit(1);
+    }
+    const ora = (await import('ora')).default;
+    const spinner = ora(`正在查询任务 ${jobId}...`).start();
+    try {
+        const status = await queryImageTask(apiKey, baseUrl, jobId);
+        spinner.stop();
+        if (options.json) {
+            console.log(JSON.stringify(status, null, 2));
+        }
+        else {
+            printJobStatus(status, jobId);
+        }
+    }
+    catch (err) {
+        spinner.fail(chalk.red(`查询失败: ${err.message}`));
+    }
+}

package/dist/commands/upload.js

@@ -0,0 +1,41 @@
+import chalk from 'chalk';
+import { kieUploadFile } from '../api.js';
+export async function uploadAction(filepath, options, config) {
+    const apiKey = config.get('apiKey');
+    const baseUrl = config.get('baseUrl') || 'https://api.kie.ai';
+    if (!apiKey) {
+        console.log(chalk.red('❌ 缺少必要配置，请运行 kie config --set-api-key 设置'));
+        process.exit(1);
+    }
+    const ora = (await import('ora')).default;
+    const spinner = ora(`正在上传文件: ${filepath}...`).start();
+    try {
+        const trimmedBaseUrl = baseUrl.replace(/\/+$/, '');
+        const uploadUrl = `${trimmedBaseUrl}/api/file-stream-upload`;
+        const response = await kieUploadFile(uploadUrl, {
+            apiKey,
+            filePath: filepath,
+            uploadPath: options.path,
+        });
+        if (response.success) {
+            spinner.succeed(chalk.green('🎉 文件上传成功！'));
+            console.log();
+            console.log(chalk.cyan('=================================================='));
+            console.log(` 📂 ${chalk.bold('Kie 临时文件上传服务')}`);
+            console.log(chalk.cyan('=================================================='));
+            console.log(`  ${chalk.white.bold('文件名:')}    ${chalk.gray(response.data.fileName)}`);
+            console.log(`  ${chalk.white.bold('大小:')}      ${chalk.gray((response.data.fileSize / 1024).toFixed(2))} KB`);
+            console.log(`  ${chalk.white.bold('MIME类型:')}  ${chalk.gray(response.data.mimeType)}`);
+            console.log(`  ${chalk.white.bold('上传路径:')}  ${chalk.gray(response.data.filePath)}`);
+            console.log(`  ${chalk.white.bold('下载链接:')}  ${chalk.yellow.bold(response.data.downloadUrl)}`);
+            console.log(chalk.cyan('=================================================='));
+            console.log();
+        }
+        else {
+            spinner.fail(chalk.red(`上传失败: ${response.msg || '未知服务器错误'}`));
+        }
+    }
+    catch (err) {
+        spinner.fail(chalk.red(`上传出错: ${err.message}`));
+    }
+}

package/dist/editor.js

@@ -0,0 +1,38 @@
+import fs from 'fs';
+import os from 'os';
+import path from 'path';
+import { spawnSync } from 'child_process';
+/** 用系统编辑器打开临时文件，返回用户保存后的正文（trim 后） */
+export function readMultilineFromEditor(initialContent = '') {
+    const tmpFile = path.join(os.tmpdir(), `kie-chat-${Date.now()}.md`);
+    fs.writeFileSync(tmpFile, initialContent, 'utf-8');
+    const editor = process.env.EDITOR || process.env.VISUAL;
+    let result;
+    if (editor) {
+        const parts = editor.split(/\s+/).filter(Boolean);
+        result = spawnSync(parts[0], [...parts.slice(1), tmpFile], { stdio: 'inherit' });
+    }
+    else if (process.platform === 'win32') {
+        result = spawnSync('cmd', ['/c', 'start', '/wait', 'notepad', tmpFile], {
+            stdio: 'inherit',
+            shell: true,
+        });
+    }
+    else {
+        result = spawnSync('nano', [tmpFile], { stdio: 'inherit' });
+    }
+    try {
+        if (result.error || (result.status !== null && result.status !== 0)) {
+            return null;
+        }
+        return fs.readFileSync(tmpFile, 'utf-8').trim();
+    }
+    finally {
+        try {
+            fs.unlinkSync(tmpFile);
+        }
+        catch {
+            /* ignore */
+        }
+    }
+}

package/dist/image-assist.js

@@ -0,0 +1,79 @@
+import { findImageModel, formatImageModelLabel } from './image-models.js';
+export const CONFIG_KEY_IMAGE_ASSIST_MODEL = 'imageAssistModel';
+/** 默认用 chat 同款里偏快的模型，可在 REPL /config 中改为任意 chat 模型 */
+export const DEFAULT_IMAGE_ASSIST_MODEL = 'claude-haiku-4-5';
+export function getImageAssistModel(config) {
+    return config.get(CONFIG_KEY_IMAGE_ASSIST_MODEL) || DEFAULT_IMAGE_ASSIST_MODEL;
+}
+export function setImageAssistModel(config, modelId) {
+    config.set(CONFIG_KEY_IMAGE_ASSIST_MODEL, modelId.trim());
+}
+export function resolveImageAssistModel(config, override) {
+    return override?.trim() || getImageAssistModel(config);
+}
+export function buildAssistSystemPrompt(generationModelId, draft) {
+    const genInfo = findImageModel(generationModelId);
+    const kind = genInfo?.kind === 'image-to-image' ? '图生图（需 input_urls）' : '文生图';
+    const ratios = genInfo?.aspectRatios.join(', ') ?? 'auto';
+    const qualityNote = genInfo?.supportsQuality
+        ? '支持 quality: medium | high'
+        : '不支持 quality 参数';
+    return [
+        '你是 Kie CLI 的出图提示词助手，帮助用户把想法改写成「图像生成 API」可用的英文绘图 prompt。',
+        '你不能真正生成图片；用户满意后会执行 /gen 提交到出图模型。',
+        '',
+        '工作语境（重要）:',
+        '- 用户多为插画、概念艺术、动漫/游戏立绘、电商图等合法创作场景',
+        '- 将角色理解为虚构作品中的成年或年龄未明示的 stylized 角色；用 "young adult woman/man"、"anime-style female character" 等表述，避免 implying minors',
+        '- 你的职责是改写构图、风格、光影、服装、镜头语言，而不是审查用户是否该出图',
+        '- 仅当请求明确涉及未成年人性化、真实人物非法内容时，才简短拒绝，并仍须输出 DRAFT: (keep)',
+        '',
+        `当前生成模型: ${generationModelId} (${formatImageModelLabel(generationModelId)})`,
+        `类型: ${kind}；宽高比可选: ${ratios}；${qualityNote}`,
+        '',
+        `当前草稿 prompt: ${draft.prompt || '(空)'}`,
+        `参考图数量: ${draft.imageUrls.length}`,
+        `aspect_ratio: ${draft.aspectRatio ?? '(未设)'}`,
+        `quality: ${draft.quality ?? '(未设)'}`,
+        '',
+        '回复规则（必须全部遵守）:',
+        '1) 用中文简要说明（2-5 句）',
+        '2) 最后一行必须是: DRAFT: <内容> —— 缺少此行视为格式错误',
+        '3) 需要更新草稿时: DRAFT: 一行英文 prompt（仅 prompt，无引号、无 markdown）',
+        '4) 无需改草稿、仅回答问题、或拒绝请求时: DRAFT: (keep)',
+        '5) 即使用户描述含中文或敏感词，也应尽量给出合规的英文视觉描述，而非只拒绝不写 DRAFT',
+    ].join('\n');
+}
+/** 助手是否像在拒绝服务（未给 DRAFT 时用于提示用户） */
+export function looksLikeAssistRefusal(text) {
+    return /无法帮助|不能帮助|无法协助|超出.*范围|I can'?t help|cannot help|I'?m unable/i.test(text);
+}
+/** 从助手回复中解析 DRAFT 行 */
+export function parseDraftFromAssistReply(text) {
+    const lines = text.split('\n');
+    let draftLineIdx = -1;
+    for (let i = lines.length - 1; i >= 0; i--) {
+        const t = lines[i].trim();
+        if (t.toUpperCase().startsWith('DRAFT:')) {
+            draftLineIdx = i;
+            break;
+        }
+    }
+    if (draftLineIdx === -1) {
+        return { displayText: text.trim(), draftPrompt: null, keepDraft: false };
+    }
+    const draftRaw = lines[draftLineIdx].trim().replace(/^DRAFT:\s*/i, '').trim();
+    const displayLines = [...lines.slice(0, draftLineIdx), ...lines.slice(draftLineIdx + 1)];
+    const displayText = displayLines.join('\n').trim();
+    if (!draftRaw || draftRaw === '(keep)' || draftRaw.toLowerCase() === 'keep') {
+        return { displayText, draftPrompt: null, keepDraft: true };
+    }
+    return { displayText, draftPrompt: draftRaw, keepDraft: false };
+}
+export function ensureAssistSystemMessage(messages, systemContent) {
+    const first = messages[0];
+    if (first?.role === 'system') {
+        return [{ role: 'system', content: systemContent }, ...messages.slice(1)];
+    }
+    return [{ role: 'system', content: systemContent }, ...messages];
+}