kie-ai-cli 1.0.0

package/dist/commands/chat.js

@@ -0,0 +1,552 @@
+import fs from 'fs';
+import path from 'path';
+import readline from 'readline';
+import chalk from 'chalk';
+import { initRenderer, marked } from '../utils/renderer.js';
+import { sendChatMessage } from '../api.js';
+import { readMultilineFromEditor } from '../editor.js';
+import { isAbortError } from '../kie-http.js';
+import { formatModelLabel, printModelPickerMenu, resolveChatModel, resolveModelPickInput, setChatModel, } from '../models.js';
+import { clearSession, canCompress, compressMessages, createSessionId, estimateSessionSize, getActiveSessionId, listSessions, loadSession, maybeCompressSession, printSessionList, resolveSessionPick, saveSession, setActiveSessionId, } from '../session.js';
+const SLASH_ALIASES = {
+    '/chear': '/clear',
+    '/clea': '/clear',
+    '/cleer': '/clear',
+    '/cls': '/clear',
+    '/models': '/model',
+    '/compresss': '/compress',
+    '/h': '/help',
+    '/?': '/help',
+    '/session': '/sessions',
+};
+function normalizeSlashInput(input) {
+    const space = input.indexOf(' ');
+    const cmd = (space === -1 ? input : input.slice(0, space)).toLowerCase();
+    const alias = SLASH_ALIASES[cmd];
+    if (!alias)
+        return input;
+    return alias + (space === -1 ? '' : input.slice(space));
+}
+function printReplHelp() {
+    console.log(chalk.cyan('\n可用指令:\n'));
+    const rows = [
+        ['/help', '显示此帮助'],
+        ['/new', '新建会话（保存当前会话并切换到新 ID）'],
+        ['/sessions [id|序号]', '列出或切换已保存会话'],
+        ['/model [id|序号]', '切换对话模型'],
+        ['/clear', '清空当前会话上下文'],
+        ['/compress', '手动压缩对话历史'],
+        ['/file <路径>', '读取文件加入上下文'],
+        ['/run <命令>', '执行命令并将输出加入上下文'],
+        ['/edit', '用编辑器编写多行消息（EDITOR 环境变量）'],
+        ['行末 \\', '多行输入：续行，空行发送'],
+        ['Ctrl+C', '取消进行中的 AI 请求；空闲时取消当前输入'],
+        ['exit / quit', '退出并保存会话'],
+    ];
+    for (const [cmd, desc] of rows) {
+        console.log(`  ${chalk.white(cmd.padEnd(22))} ${chalk.gray(desc)}`);
+    }
+    console.log();
+}
+export async function chatAction(message, options, config, readStdin) {
+    // Initialize renderer only when needed
+    initRenderer();
+    const ora = (await import('ora')).default;
+    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);
+    }
+    let sessionModel = resolveChatModel(config, { cli: options.model });
+    const pipedInput = await readStdin();
+    const isInteractive = !message && !pipedInput;
+    let sessionId = getActiveSessionId(config);
+    if (options.new) {
+        sessionId = createSessionId();
+    }
+    else if (options.resume !== undefined) {
+        sessionId =
+            typeof options.resume === 'string' && options.resume.length > 0
+                ? options.resume
+                : getActiveSessionId(config);
+    }
+    setActiveSessionId(config, sessionId);
+    const messagesHistory = isInteractive
+        ? options.new
+            ? []
+            : loadSession(config, sessionId)
+        : [];
+    const persistSession = () => {
+        if (isInteractive) {
+            saveSession(config, sessionId, messagesHistory);
+        }
+    };
+    const applyCompressed = (compressed) => {
+        const next = compressed.map((m) => ({ ...m }));
+        messagesHistory.length = 0;
+        messagesHistory.push(...next);
+    };
+    const persistAndMaybeCompress = async (quiet = false) => {
+        persistSession();
+        if (!isInteractive)
+            return;
+        const beforeSize = estimateSessionSize(messagesHistory);
+        const maxChars = config.get('sessionMaxChars') || 16_000;
+        if (beforeSize <= maxChars)
+            return;
+        const spinner = quiet ? null : ora('会话较长，正在自动压缩历史...').start();
+        try {
+            const compressed = await maybeCompressSession(messagesHistory, {
+                apiKey,
+                baseUrl,
+                model: sessionModel,
+                config,
+            });
+            applyCompressed(compressed);
+            persistSession();
+            spinner?.succeed(chalk.green(`会话已压缩：${beforeSize} → ${estimateSessionSize(messagesHistory)} 字符`));
+        }
+        catch (err) {
+            spinner?.fail(chalk.yellow(`会话压缩失败: ${err.message}`));
+        }
+    };
+    let chatInFlight = false;
+    const doChat = async (userText, opts) => {
+        if (chatInFlight)
+            return;
+        chatInFlight = true;
+        const userMsg = { role: 'user', content: userText };
+        messagesHistory.push(userMsg);
+        const spinner = ora({
+            text: `AI 思考中 (${sessionModel})...`,
+            discardStdin: true,
+        }).start();
+        let fullResponse = '';
+        let gotChunk = false;
+        let wroteHeader = false;
+        try {
+            await sendChatMessage({
+                apiKey,
+                baseUrl,
+                model: sessionModel,
+                messages: messagesHistory,
+                signal: opts?.signal,
+                onChunk: (text) => {
+                    if (opts?.signal?.aborted)
+                        return;
+                    if (!text)
+                        return;
+                    gotChunk = true;
+                    if (!wroteHeader) {
+                        spinner.stop();
+                        process.stdout.write(chalk.green.bold('\nKie AI: '));
+                        wroteHeader = true;
+                    }
+                    process.stdout.write(chalk.white(text));
+                    fullResponse += text;
+                },
+            });
+            if (!wroteHeader) {
+                spinner.stop();
+                if (!gotChunk) {
+                    console.log(chalk.yellow('\n(模型未返回文本内容)\n'));
+                }
+            }
+            else {
+                console.log('\n');
+            }
+            messagesHistory.push({ role: 'assistant', content: fullResponse });
+            if (isInteractive) {
+                await persistAndMaybeCompress(true);
+            }
+            if (!isInteractive) {
+                console.log(chalk.gray('\n====== 格式化结果 ======'));
+                console.log(marked.parse(fullResponse || '(空)'));
+            }
+        }
+        catch (error) {
+            if (isAbortError(error)) {
+                spinner.stop();
+                const last = messagesHistory[messagesHistory.length - 1];
+                if (gotChunk && fullResponse.trim()) {
+                    messagesHistory.push({ role: 'assistant', content: fullResponse });
+                    if (isInteractive)
+                        persistSession();
+                    console.log(chalk.yellow('\n\n(已取消，已保留部分回复)\n'));
+                }
+                else if (last === userMsg) {
+                    messagesHistory.pop();
+                    console.log(chalk.yellow('\n(已取消)\n'));
+                }
+                else {
+                    console.log(chalk.yellow('\n(已取消)\n'));
+                }
+                return;
+            }
+            const last = messagesHistory[messagesHistory.length - 1];
+            if (last === userMsg) {
+                messagesHistory.pop();
+            }
+            spinner.fail(chalk.red('请求失败'));
+            const msg = error instanceof Error ? error.message : '未知错误';
+            console.error(chalk.red(`\n❌ ${msg}\n`));
+            if (!isInteractive)
+                process.exit(1);
+        }
+        finally {
+            spinner.stop();
+            chatInFlight = false;
+        }
+    };
+    if (!isInteractive) {
+        let finalMessage = '';
+        if (pipedInput && message)
+            finalMessage = `${pipedInput}\n\n${message}`;
+        else if (pipedInput)
+            finalMessage = pipedInput;
+        else if (message)
+            finalMessage = message;
+        await doChat(finalMessage);
+        return;
+    }
+    console.log(chalk.cyan('=================================================='));
+    console.log(chalk.cyan(' 🚀 欢迎进入 Kie 沉浸式对话模式！(输入 exit 退出) '));
+    console.log(chalk.cyan('=================================================='));
+    console.log(chalk.gray(`会话 ID: ${sessionId}`));
+    console.log(chalk.gray(`对话模型: ${formatModelLabel(sessionModel)}`));
+    if (messagesHistory.length > 0) {
+        console.log(chalk.gray(`已恢复 ${messagesHistory.length} 条历史消息（约 ${estimateSessionSize(messagesHistory)} 字符）`));
+    }
+    console.log(chalk.gray('指令: /help 查看全部 | /new 新会话 | /sessions 切换 | 行末 \\ 多行 | Ctrl+C 取消\n'));
+    return new Promise((resolve) => {
+        let exiting = false;
+        let isProcessing = false;
+        let replMode = 'chat';
+        let multilineBuffer = [];
+        let activeAbort = null;
+        let lastIdleSigintAt = 0;
+        const PROMPT_CHAT = '(You) > ';
+        const PROMPT_PICK = '序号 > ';
+        const PROMPT_SESSION = '会话序号 > ';
+        const PROMPT_MULTILINE = '... > ';
+        const rl = readline.createInterface({
+            input: process.stdin,
+            output: process.stdout,
+            terminal: true,
+            prompt: PROMPT_CHAT,
+        });
+        const showPrompt = () => {
+            rl.prompt();
+        };
+        const resetReplInputMode = () => {
+            replMode = 'chat';
+            multilineBuffer = [];
+            rl.setPrompt(PROMPT_CHAT);
+        };
+        const switchToSession = (targetId) => {
+            persistSession();
+            sessionId = targetId;
+            setActiveSessionId(config, sessionId);
+            messagesHistory.length = 0;
+            messagesHistory.push(...loadSession(config, sessionId));
+            console.log(chalk.green(`✅ 已切换到会话: ${sessionId}（${messagesHistory.length} 条消息，约 ${estimateSessionSize(messagesHistory)} 字符）\n`));
+        };
+        const startNewSession = () => {
+            persistSession();
+            switchToSession(createSessionId());
+        };
+        const submitChatText = async (text) => {
+            if (!acquireProcessing()) {
+                showPrompt();
+                return;
+            }
+            const ac = new AbortController();
+            activeAbort = ac;
+            try {
+                await doChat(text, { signal: ac.signal });
+            }
+            finally {
+                activeAbort = null;
+                isProcessing = false;
+                showPrompt();
+            }
+        };
+        rl.on('close', () => {
+            if (exiting)
+                resolve();
+        });
+        rl.on('SIGINT', () => {
+            if (activeAbort) {
+                activeAbort.abort();
+                return;
+            }
+            if (replMode !== 'chat') {
+                resetReplInputMode();
+                console.log(chalk.gray('\n已取消当前输入。\n'));
+                showPrompt();
+                return;
+            }
+            const now = Date.now();
+            if (now - lastIdleSigintAt < 1500) {
+                exiting = true;
+                persistSession();
+                console.log(chalk.gray(`\n👋 再见！会话已保存 (${sessionId})`));
+                rl.close();
+                return;
+            }
+            lastIdleSigintAt = now;
+            console.log(chalk.gray('\n再按一次 Ctrl+C 退出，或输入 exit\n'));
+            showPrompt();
+        });
+        const acquireProcessing = () => {
+            if (isProcessing)
+                return false;
+            isProcessing = true;
+            return true;
+        };
+        rl.on('line', async (line) => {
+            const input = normalizeSlashInput(line.trim());
+            if (isProcessing) {
+                if (input) {
+                    console.log(chalk.gray('\n请等待当前操作完成（Ctrl+C 可取消）...\n'));
+                }
+                showPrompt();
+                return;
+            }
+            if (replMode === 'multiline') {
+                if (!input) {
+                    const text = multilineBuffer.join('\n').trim();
+                    resetReplInputMode();
+                    if (!text) {
+                        console.log(chalk.gray('已取消多行输入。\n'));
+                        showPrompt();
+                        return;
+                    }
+                    await submitChatText(text);
+                    return;
+                }
+                if (input.endsWith('\\') && input.length > 1) {
+                    multilineBuffer.push(input.slice(0, -1));
+                }
+                else {
+                    multilineBuffer.push(input);
+                }
+                showPrompt();
+                return;
+            }
+            if (replMode === 'pick_session') {
+                replMode = 'chat';
+                rl.setPrompt(PROMPT_CHAT);
+                const sessions = listSessions(config);
+                const picked = resolveSessionPick(input, sessions);
+                if (!picked) {
+                    console.log(chalk.gray('已取消。\n'));
+                }
+                else if (picked === sessionId) {
+                    console.log(chalk.gray('已在当前会话。\n'));
+                }
+                else {
+                    switchToSession(picked);
+                }
+                showPrompt();
+                return;
+            }
+            if (replMode === 'pick_model') {
+                replMode = 'chat';
+                rl.setPrompt(PROMPT_CHAT);
+                const picked = resolveModelPickInput(input);
+                if (!picked) {
+                    console.log(chalk.gray('已取消。\n'));
+                }
+                else {
+                    sessionModel = picked;
+                    setChatModel(config, picked);
+                    console.log(chalk.green(`✅ 已切换模型: ${formatModelLabel(picked)}\n`));
+                }
+                showPrompt();
+                return;
+            }
+            if (!input) {
+                showPrompt();
+                return;
+            }
+            if (input.toLowerCase() === 'exit' || input.toLowerCase() === 'quit') {
+                exiting = true;
+                persistSession();
+                console.log(chalk.gray(`👋 再见！会话已保存 (${sessionId})`));
+                rl.close();
+                return;
+            }
+            if (input.endsWith('\\') && input.length > 1) {
+                multilineBuffer = [input.slice(0, -1)];
+                replMode = 'multiline';
+                rl.setPrompt(PROMPT_MULTILINE);
+                console.log(chalk.gray('多行模式：继续输入，空行发送，Ctrl+C 取消\n'));
+                showPrompt();
+                return;
+            }
+            if (input === '/help') {
+                printReplHelp();
+                showPrompt();
+                return;
+            }
+            if (input === '/new') {
+                startNewSession();
+                showPrompt();
+                return;
+            }
+            if (input === '/edit') {
+                const text = readMultilineFromEditor();
+                if (!text) {
+                    console.log(chalk.gray('已取消编辑。\n'));
+                    showPrompt();
+                    return;
+                }
+                await submitChatText(text);
+                return;
+            }
+            if (input === '/sessions' || input.startsWith('/sessions ')) {
+                const arg = input.slice(9).trim();
+                const sessions = listSessions(config);
+                if (!arg) {
+                    printSessionList(sessions, sessionId);
+                    if (sessions.length > 0) {
+                        replMode = 'pick_session';
+                        rl.setPrompt(PROMPT_SESSION);
+                    }
+                    showPrompt();
+                    return;
+                }
+                const picked = resolveSessionPick(arg, sessions);
+                if (!picked) {
+                    console.log(chalk.red(`❌ 未找到会话: ${arg}\n`));
+                }
+                else if (picked === sessionId) {
+                    console.log(chalk.gray('已在当前会话。\n'));
+                }
+                else {
+                    switchToSession(picked);
+                }
+                showPrompt();
+                return;
+            }
+            if (input === '/model' || input.startsWith('/model ')) {
+                const arg = input.slice(6).trim();
+                if (!arg) {
+                    printModelPickerMenu(sessionModel);
+                    replMode = 'pick_model';
+                    rl.setPrompt(PROMPT_PICK);
+                    showPrompt();
+                    return;
+                }
+                sessionModel = arg;
+                setChatModel(config, arg);
+                console.log(chalk.green(`✅ 已切换模型: ${formatModelLabel(arg)}\n`));
+                showPrompt();
+                return;
+            }
+            if (input === '/clear') {
+                messagesHistory.length = 0;
+                clearSession(config, sessionId);
+                console.log(chalk.green('✅ 当前会话上下文已清空。\n'));
+                showPrompt();
+                return;
+            }
+            if (input === '/compress') {
+                if (!acquireProcessing()) {
+                    showPrompt();
+                    return;
+                }
+                const ac = new AbortController();
+                activeAbort = ac;
+                try {
+                    if (messagesHistory.length === 0) {
+                        console.log(chalk.yellow('⚠️ 当前没有可压缩的对话历史。\n'));
+                    }
+                    else if (!canCompress(messagesHistory, true)) {
+                        console.log(chalk.yellow('⚠️ 对话太短（不足 2 轮），无需压缩。\n'));
+                    }
+                    else {
+                        const spinner = ora('正在压缩会话...').start();
+                        try {
+                            const beforeCount = messagesHistory.length;
+                            const beforeSize = estimateSessionSize(messagesHistory);
+                            const compressed = await compressMessages(messagesHistory, {
+                                apiKey,
+                                baseUrl,
+                                model: sessionModel,
+                                signal: ac.signal,
+                                force: true,
+                                maxChars: config.get('sessionMaxChars') || 16_000,
+                                keepRecentPairs: config.get('sessionKeepRecent') || 6,
+                            });
+                            applyCompressed(compressed);
+                            persistSession();
+                            spinner.succeed(chalk.green(`压缩完成：${beforeCount} → ${messagesHistory.length} 条消息，${beforeSize} → ${estimateSessionSize(messagesHistory)} 字符`));
+                        }
+                        catch (err) {
+                            if (isAbortError(err)) {
+                                spinner.fail(chalk.yellow('压缩已取消'));
+                            }
+                            else {
+                                const msg = err instanceof Error ? err.message : String(err);
+                                spinner.fail(chalk.red(`压缩失败: ${msg}`));
+                            }
+                        }
+                    }
+                }
+                finally {
+                    activeAbort = null;
+                    isProcessing = false;
+                    console.log();
+                    showPrompt();
+                }
+                return;
+            }
+            if (input.startsWith('/file ')) {
+                let filePath = input.slice(6).trim();
+                if (filePath.startsWith('"') && filePath.endsWith('"')) {
+                    filePath = filePath.slice(1, -1);
+                }
+                else if (filePath.startsWith("'") && filePath.endsWith("'")) {
+                    filePath = filePath.slice(1, -1);
+                }
+                try {
+                    const content = fs.readFileSync(path.resolve(process.cwd(), filePath), 'utf-8');
+                    messagesHistory.push({
+                        role: 'user',
+                        content: `[System] The user has provided the contents of the file '${filePath}':\n\n${content}`,
+                    });
+                    console.log(chalk.green(`✅ 已读取文件 ${filePath} 并加入记忆上下文。你可以针对它进行提问了。\n`));
+                    persistSession();
+                }
+                catch (err) {
+                    console.log(chalk.red(`❌ 读取文件失败: ${err.message}\n`));
+                }
+                showPrompt();
+                return;
+            }
+            if (input.startsWith('/run ')) {
+                const cmd = input.slice(5).trim();
+                try {
+                    const { execSync } = await import('child_process');
+                    const output = execSync(cmd, { encoding: 'utf-8' });
+                    messagesHistory.push({
+                        role: 'user',
+                        content: `[System] The user executed the command '${cmd}' and the output was:\n\n${output}`,
+                    });
+                    console.log(chalk.green(`✅ 已执行命令并将其输出加入记忆上下文。\n`));
+                    persistSession();
+                }
+                catch (err) {
+                    console.log(chalk.red(`❌ 执行命令失败: ${err.message}\n`));
+                }
+                showPrompt();
+                return;
+            }
+            await submitChatText(line);
+        });
+    });
+}