kie-ai-cli 1.0.0

package/dist/commands/code-command.js

@@ -0,0 +1,1025 @@
+import fs from 'fs';
+import path from 'path';
+import readline from 'readline';
+import chalk from 'chalk';
+import ora from 'ora';
+import { getChatCompletion, sendChatMessage } from '../api.js';
+import { buildCodeContext, buildCodeAgentSystemPrompt, createCodeSessionId, buildDefaultRoutingConfig, initCodeWorkspace, initCodeWorkspaceProfile, isDangerousCommand, listCodeSessions, loadCodeModelMetrics, loadCodeSession, loadCodeWorkspaceProfile, recordCodeModelMetric, removeWeightedModel, saveCodeModelMetrics, saveCodeWorkspaceConfig, selectModelFromRouting, parseAgentResponse, truncateCommandOutput, buildCodeSystemPrompt, buildCodeUserPrompt, loadCodeWorkspaceConfig, resolveCodeRoute, saveCodeSession, upsertWeightedModel, trimConversation, } from '../code.js';
+import { isAbortError } from '../kie-http.js';
+import { formatModelLabel, resolveChatModel, resolveModelPickInput, promptSelectChatModel, } from '../models.js';
+import { initRenderer, marked } from '../utils/renderer.js';
+function parsePositiveInt(raw, fallback) {
+    const n = Number.parseInt(raw, 10);
+    if (Number.isNaN(n) || n <= 0)
+        return fallback;
+    return n;
+}
+function printCodeReplHelp() {
+    console.log(chalk.cyan('\nCode 模式可用指令:\n'));
+    const rows = [
+        ['/help', '显示帮助'],
+        ['/clear', '清空当前 code 会话上下文'],
+        ['/cwd', '显示当前 code 工作目录'],
+        ['/init', '在当前工作目录生成 .kie/code.json'],
+        ['/model [id|序号]', '查看或切换当前会话模型'],
+        ['/route [auto|quality|balanced|fast]', '查看或设置路由策略'],
+        ['/models [plan|execute|fix]', '交互式配置分阶段模型池与权重（亦可 add/rm/strategy）'],
+        ['/approve on|off', '设置命令执行是否自动确认'],
+        ['/sessions [id|序号]', '列出或切换 code 会话'],
+        ['/files', '查看当前注入的文件列表'],
+        ['/file <路径>', '追加文件到上下文'],
+        ['/run <命令>', '在 code 工作目录执行命令并注入输出'],
+        ['/tree on|off', '开关项目结构注入'],
+        ['/status', '查看当前 code 配置'],
+        ['exit / quit', '退出 code 对话'],
+    ];
+    for (const [cmd, desc] of rows) {
+        console.log(`  ${chalk.white(cmd.padEnd(18))} ${chalk.gray(desc)}`);
+    }
+    console.log();
+}
+async function askForConfirmation(question) {
+    if (!process.stdin.isTTY)
+        return false;
+    const rl = readline.createInterface({
+        input: process.stdin,
+        output: process.stdout,
+    });
+    return new Promise((resolve) => {
+        rl.question(question, (answer) => {
+            rl.close();
+            const normalized = answer.trim().toLowerCase();
+            resolve(normalized === 'y' || normalized === 'yes');
+        });
+    });
+}
+function parsePhase(input) {
+    const normalized = input.trim().toLowerCase();
+    if (normalized === 'plan' || normalized === 'execute' || normalized === 'fix') {
+        return normalized;
+    }
+    return null;
+}
+function inferFallbackCommands(task) {
+    const commands = [];
+    const normalized = task.toLowerCase();
+    const quotedPath = task.match(/["']([^"']+\.[a-zA-Z0-9]+)["']/)?.[1] ||
+        task.match(/([A-Za-z0-9_./\\-]+\.[A-Za-z0-9]+)/)?.[1];
+    if (quotedPath && /(读取|查看|分析|read|inspect|summar)/i.test(task)) {
+        if (process.platform === 'win32') {
+            commands.push(`type "${quotedPath.replace(/\//g, '\\')}"`);
+        }
+        else {
+            commands.push(`cat "${quotedPath}"`);
+        }
+    }
+    if (commands.length === 0 && /(读取|查看|分析|代码|文件|module|模块)/i.test(task)) {
+        commands.push(process.platform === 'win32' ? 'dir' : 'ls');
+    }
+    if (commands.length === 0 && /(test|测试|lint|构建|build)/i.test(normalized)) {
+        commands.push('npm run test');
+    }
+    return commands.slice(0, 2);
+}
+function printRoutingTable(routing, routeMode, metrics) {
+    console.log(chalk.cyan(`\n当前路由: ${routeMode} | strategy: ${routing.strategy}`));
+    const phases = ['plan', 'execute', 'fix'];
+    phases.forEach((phase) => {
+        console.log(chalk.yellow(`\n[${phase}]`));
+        routing.phases[phase].forEach((entry, idx) => {
+            const m = metrics.metrics[entry.model];
+            const successRate = !m || m.runs === 0 ? '-' : `${Math.round((m.success / m.runs) * 100)}%`;
+            const latency = !m || m.runs === 0 ? '-' : `${m.avgLatencyMs}ms`;
+            console.log(`  ${idx + 1}. ${entry.model}  w=${entry.weight}  success=${successRate}  latency=${latency}`);
+        });
+    });
+    console.log();
+}
+export async function codeAction(task, options, config, readStdin) {
+    const apiKey = config.get('apiKey');
+    const baseUrl = config.get('baseUrl') || 'https://api.kie.ai';
+    const codeCwd = path.resolve(process.cwd(), options.cwd || '.');
+    if (!fs.existsSync(codeCwd) || !fs.statSync(codeCwd).isDirectory()) {
+        console.log(chalk.red(`❌ 无效目录: ${codeCwd}`));
+        process.exit(1);
+    }
+    const rawArgv = process.argv.slice(2);
+    const hasNoTree = rawArgv.includes('--no-tree');
+    const hasTreeLimit = rawArgv.some((a) => a === '--tree-limit' || a.startsWith('--tree-limit='));
+    const hasFileMaxChars = rawArgv.some((a) => a === '--file-max-chars' || a.startsWith('--file-max-chars='));
+    const workspace = await loadCodeWorkspaceConfig(codeCwd);
+    let workspaceProfile = await loadCodeWorkspaceProfile(codeCwd);
+    let modelMetrics = await loadCodeModelMetrics(codeCwd);
+    let includeTree = hasNoTree ? false : (workspace?.includeTree ?? options.tree);
+    let treeLimit = hasTreeLimit
+        ? parsePositiveInt(options.treeLimit, 120)
+        : (workspace?.treeLimit ?? parsePositiveInt(options.treeLimit, 120));
+    let fileMaxChars = hasFileMaxChars
+        ? parsePositiveInt(options.fileMaxChars, 8000)
+        : (workspace?.fileMaxChars ?? parsePositiveInt(options.fileMaxChars, 8000));
+    let codeModel = resolveChatModel(config, {
+        cli: options.model || workspace?.defaultModel,
+    });
+    let routeMode = resolveCodeRoute(options.route || workspace?.defaultRoute);
+    let routingConfig = workspace?.routing || buildDefaultRoutingConfig(codeModel);
+    const maxAgentSteps = parsePositiveInt(options.maxSteps, 25);
+    const useAgentMode = !options.chat;
+    let autoApprove = Boolean(options.yes);
+    let activeSessionId = options.newSession
+        ? createCodeSessionId()
+        : options.session || 'default';
+    if (options.init) {
+        const initResult = await initCodeWorkspace(codeCwd, {
+            includeTree,
+            treeLimit,
+            fileMaxChars,
+            defaultModel: codeModel,
+            defaultRoute: routeMode,
+            routing: routingConfig,
+        });
+        await saveCodeWorkspaceConfig(codeCwd, initResult.config);
+        const profileInit = await initCodeWorkspaceProfile(codeCwd);
+        workspaceProfile = profileInit.profile;
+        routingConfig = initResult.config.routing || buildDefaultRoutingConfig(codeModel);
+        modelMetrics = await loadCodeModelMetrics(codeCwd);
+        if (initResult.created) {
+            console.log(chalk.green(`✅ code 工作区初始化完成: ${initResult.configPath}`));
+        }
+        else {
+            console.log(chalk.yellow(`ℹ️ 已存在配置，未覆盖: ${initResult.configPath}`));
+        }
+        console.log(chalk.green(`✅ 工作区画像已刷新: ${profileInit.profilePath}`));
+        const p = profileInit.profile;
+        console.log(chalk.cyan('\n📋 工作区分析结果:'));
+        if (p.projectName)
+            console.log(`  项目名称: ${chalk.bold(p.projectName)}${p.projectDescription ? chalk.gray(` — ${p.projectDescription}`) : ''}`);
+        console.log(`  包管理器: ${p.packageManager}`);
+        if (p.gitBranch)
+            console.log(`  Git 分支: ${p.gitBranch}`);
+        if (p.nodeVersion)
+            console.log(`  Node 版本: ${p.nodeVersion}`);
+        if (p.frameworks?.length)
+            console.log(`  技术栈:   ${chalk.yellow(p.frameworks.join(', '))}`);
+        if (p.buildTool)
+            console.log(`  构建工具: ${p.buildTool}`);
+        if (p.testFramework)
+            console.log(`  测试框架: ${p.testFramework}`);
+        if (p.entryPoints?.length)
+            console.log(`  入口文件: ${p.entryPoints.join(', ')}`);
+        console.log(`  Scripts:  ${p.scripts.join(', ') || '(无)'}`);
+        console.log(`  语言分布: ${p.languageStats.slice(0, 6).map(l => `${l.ext}(${l.count})`).join(', ')}`);
+        console.log(`  目录结构: ${p.topLevelDirs.join(', ')}`);
+        if (p.configFiles?.length)
+            console.log(`  配置文件: ${p.configFiles.join(', ')}`);
+        if (p.dependencies?.length)
+            console.log(`  依赖(${p.dependencies.length}): ${p.dependencies.slice(0, 12).join(', ')}${p.dependencies.length > 12 ? ' ...' : ''}`);
+        if (p.devDependencies?.length)
+            console.log(`  开发依赖(${p.devDependencies.length}): ${p.devDependencies.slice(0, 8).join(', ')}${p.devDependencies.length > 8 ? ' ...' : ''}`);
+        if (p.keyFiles?.length)
+            console.log(`  关键文件: ${p.keyFiles.join(', ')}`);
+        if (p.totalFiles)
+            console.log(`  文件总数: ${p.totalFiles}`);
+        console.log();
+    }
+    if (options.init && !task) {
+        return;
+    }
+    const piped = await readStdin();
+    let finalTask = '';
+    if (piped && task)
+        finalTask = `${piped}\n\n${task}`;
+    else if (piped)
+        finalTask = piped;
+    else if (task)
+        finalTask = task;
+    if (!apiKey) {
+        console.log(chalk.red('❌ 缺少必要配置，请先运行以下命令配置 API Key:'));
+        console.log(chalk.yellow('kie config --set-api-key <KEY>'));
+        process.exit(1);
+    }
+    const loadedSession = await loadCodeSession(codeCwd, activeSessionId);
+    if (loadedSession && !options.newSession) {
+        codeModel = options.model || loadedSession.model || codeModel;
+        routeMode = resolveCodeRoute(options.route || loadedSession.route || routeMode);
+        autoApprove = Boolean(options.yes || loadedSession.autoApprove);
+    }
+    const persistWorkspaceConfig = async () => {
+        await saveCodeWorkspaceConfig(codeCwd, {
+            version: workspace?.version || 1,
+            createdAt: workspace?.createdAt || new Date().toISOString(),
+            root: codeCwd,
+            includeTree,
+            treeLimit,
+            fileMaxChars,
+            defaultModel: codeModel,
+            defaultRoute: routeMode,
+            routing: routingConfig,
+        });
+    };
+    const extraFiles = loadedSession ? [...loadedSession.extraFiles, ...options.file] : [...options.file];
+    const conversation = loadedSession?.conversation || [];
+    let activeAbort = null;
+    let activeChildProcess = null;
+    let interrupted = false;
+    const normalizeCommandForPlatform = (cmd) => {
+        if (process.platform !== 'win32')
+            return cmd;
+        const trimmed = cmd.trim();
+        if (/^cat\s+/i.test(trimmed)) {
+            return trimmed.replace(/^cat\s+/i, 'type ');
+        }
+        if (/^pwd$/i.test(trimmed)) {
+            return 'cd';
+        }
+        return cmd;
+    };
+    const persistCodeSession = async () => {
+        await saveCodeSession(codeCwd, {
+            id: activeSessionId,
+            updatedAt: new Date().toISOString(),
+            conversation,
+            extraFiles,
+            model: codeModel,
+            route: routeMode,
+            autoApprove,
+        });
+    };
+    const renderSessionList = async () => {
+        const sessions = await listCodeSessions(codeCwd);
+        if (sessions.length === 0) {
+            console.log(chalk.gray('暂无 code 会话。\n'));
+            return sessions;
+        }
+        console.log(chalk.cyan('\nCode 会话列表:'));
+        sessions.forEach((session, idx) => {
+            const current = session.id === activeSessionId ? chalk.green(' (当前)') : '';
+            const updated = session.updatedAt ? session.updatedAt.slice(0, 19).replace('T', ' ') : '-';
+            console.log(`  ${idx + 1}. ${session.id}${current} ${chalk.gray(`消息 ${session.messageCount}，更新 ${updated}`)}`);
+        });
+        console.log();
+        return sessions;
+    };
+    const switchSession = async (targetId) => {
+        const loaded = await loadCodeSession(codeCwd, targetId);
+        activeSessionId = targetId;
+        conversation.length = 0;
+        extraFiles.length = 0;
+        if (loaded) {
+            conversation.push(...loaded.conversation);
+            extraFiles.push(...loaded.extraFiles);
+            codeModel = loaded.model || codeModel;
+            routeMode = resolveCodeRoute(loaded.route || routeMode);
+            autoApprove = loaded.autoApprove;
+        }
+        await persistCodeSession();
+        console.log(chalk.green(`✅ 已切换 code 会话: ${activeSessionId}\n`));
+    };
+    if (!loadedSession) {
+        await persistCodeSession();
+    }
+    const COMMAND_TIMEOUT_MS = 30_000;
+    const executeCommandIntoContext = async (cmd) => {
+        const actualCmd = normalizeCommandForPlatform(cmd);
+        const separator = chalk.gray('─'.repeat(Math.min(process.stdout.columns || 50, 60)));
+        try {
+            console.log(chalk.bgGray.white(' ▸ ') + ' ' + chalk.dim(actualCmd));
+            const { exec } = await import('child_process');
+            let stdoutData = '';
+            let stderrData = '';
+            const p = exec(actualCmd, { cwd: codeCwd, encoding: 'utf-8' });
+            activeChildProcess = p;
+            p.stdout?.on('data', (d) => { stdoutData += d; process.stdout.write(chalk.dim(d)); });
+            p.stderr?.on('data', (d) => { stderrData += d; process.stderr.write(chalk.yellow(d)); });
+            await Promise.race([
+                new Promise((resolve, reject) => {
+                    p.on('close', (code) => {
+                        if (code === 0)
+                            resolve();
+                        else
+                            reject(new Error(`exit code ${code}\n${stderrData}`));
+                    });
+                    p.on('error', reject);
+                }),
+                new Promise((_, reject) => setTimeout(() => {
+                    p.kill();
+                    reject(new Error(`命令执行超时 (${COMMAND_TIMEOUT_MS / 1000}s)，已终止`));
+                }, COMMAND_TIMEOUT_MS)),
+            ]);
+            activeChildProcess = null;
+            const rawOutput = stdoutData.trim();
+            const output = truncateCommandOutput(rawOutput);
+            conversation.push({
+                role: 'user',
+                content: [
+                    `[System] Command executed in '${codeCwd}': ${actualCmd}`,
+                    '',
+                    'stdout:',
+                    output || '(empty)',
+                ].join('\n'),
+            });
+            console.log(chalk.green('  ✓ 执行成功，输出已注入上下文'));
+            console.log(separator);
+        }
+        catch (err) {
+            activeChildProcess = null;
+            const errOut = truncateCommandOutput(err?.stderr || err?.message || 'unknown error');
+            conversation.push({
+                role: 'user',
+                content: [
+                    `[System] Command FAILED in '${codeCwd}': ${actualCmd}`,
+                    '',
+                    'stderr:',
+                    errOut,
+                ].join('\n'),
+            });
+            console.log(chalk.red(`  ✗ 执行失败`));
+            console.log(separator);
+        }
+        await persistCodeSession();
+    };
+    for (const cmd of options.run) {
+        if (cmd.trim())
+            await executeCommandIntoContext(cmd.trim());
+    }
+    const pickModelForPhase = (phase) => selectModelFromRouting(routeMode, codeModel, phase, routingConfig, modelMetrics).model;
+    const runCodeTurn = async (userTask) => {
+        const context = await buildCodeContext({
+            cwd: codeCwd,
+            includeFiles: extraFiles,
+            includeTree,
+            treeMaxEntries: treeLimit,
+            maxFileChars: fileMaxChars,
+        });
+        const userPrompt = [`【代码工作目录】\n${codeCwd}`, buildCodeUserPrompt(userTask, context, workspaceProfile)].join('\n\n');
+        const modelForTurn = pickModelForPhase('execute');
+        const contextMessages = trimConversation(conversation);
+        const messages = [
+            { role: 'system', content: buildCodeSystemPrompt(workspaceProfile) },
+            ...contextMessages,
+            { role: 'user', content: userPrompt },
+        ];
+        const ac = new AbortController();
+        activeAbort = ac;
+        const startedAt = Date.now();
+        const spinner = ora({
+            text: `Code Agent 思考中 (${modelForTurn})...`,
+            discardStdin: true,
+        }).start();
+        let wroteHeader = false;
+        let fullResponse = '';
+        try {
+            await sendChatMessage({
+                apiKey,
+                baseUrl,
+                model: modelForTurn,
+                messages,
+                signal: ac.signal,
+                onChunk: (text) => {
+                    if (ac.signal.aborted || !text)
+                        return;
+                    if (!wroteHeader) {
+                        spinner.stop();
+                        process.stdout.write(chalk.green.bold('\nKie Code: '));
+                        wroteHeader = true;
+                    }
+                    process.stdout.write(chalk.white(text));
+                    fullResponse += text;
+                },
+            });
+            spinner.stop();
+            process.stdout.write('\n\n');
+            conversation.push({ role: 'user', content: userTask });
+            conversation.push({ role: 'assistant', content: fullResponse || '(空回复)' });
+            recordCodeModelMetric(modelMetrics, modelForTurn, true, Date.now() - startedAt);
+            await saveCodeModelMetrics(codeCwd, modelMetrics);
+            await persistCodeSession();
+        }
+        catch (error) {
+            recordCodeModelMetric(modelMetrics, modelForTurn, false, Date.now() - startedAt);
+            await saveCodeModelMetrics(codeCwd, modelMetrics);
+            throw error;
+        }
+        finally {
+            activeAbort = null;
+        }
+    };
+    const runCodeAgentLoop = async (initialTask, allowInteractiveConfirm) => {
+        let currentTask = initialTask;
+        // Build full context once — subsequent steps rely on conversation history
+        const fullContext = await buildCodeContext({
+            cwd: codeCwd,
+            includeFiles: extraFiles,
+            includeTree,
+            treeMaxEntries: treeLimit,
+            maxFileChars: fileMaxChars,
+        });
+        for (let step = 1; step <= maxAgentSteps; step += 1) {
+            if (interrupted) {
+                console.log(chalk.yellow('\n⚠️ 已中断 agent 执行。'));
+                return;
+            }
+            const userPrompt = step === 1
+                ? [
+                    `【代码工作目录】\n${codeCwd}`,
+                    `【当前步骤】\n第 ${step} / ${maxAgentSteps} 轮`,
+                    buildCodeUserPrompt(currentTask, fullContext, workspaceProfile),
+                ].join('\n\n')
+                : [
+                    `【代码工作目录】\n${codeCwd}`,
+                    `【当前步骤】\n第 ${step} / ${maxAgentSteps} 轮`,
+                    currentTask,
+                ].join('\n\n');
+            const modelForLoop = pickModelForPhase(step >= 2 ? 'fix' : 'plan');
+            const contextMessages = trimConversation(conversation);
+            const messages = [
+                { role: 'system', content: buildCodeAgentSystemPrompt(workspaceProfile) },
+                ...contextMessages,
+                { role: 'user', content: userPrompt },
+            ];
+            const ac = new AbortController();
+            activeAbort = ac;
+            const startedAt = Date.now();
+            const spinner = ora({
+                text: `Code Agent 执行中（第 ${step} 轮 / ${maxAgentSteps}，${modelForLoop}）...`,
+                discardStdin: true,
+            }).start();
+            let rawResponse = '';
+            try {
+                rawResponse = await getChatCompletion({
+                    apiKey,
+                    baseUrl,
+                    model: modelForLoop,
+                    messages,
+                    signal: ac.signal,
+                });
+                recordCodeModelMetric(modelMetrics, modelForLoop, true, Date.now() - startedAt);
+                await saveCodeModelMetrics(codeCwd, modelMetrics);
+            }
+            catch (error) {
+                recordCodeModelMetric(modelMetrics, modelForLoop, false, Date.now() - startedAt);
+                await saveCodeModelMetrics(codeCwd, modelMetrics);
+                throw error;
+            }
+            finally {
+                spinner.stop();
+                activeAbort = null;
+            }
+            const parsed = parseAgentResponse(rawResponse);
+            let effectiveParsed = parsed;
+            let assistantForHistory = rawResponse || '(空回复)';
+            if (!effectiveParsed.done && effectiveParsed.commands.length === 0) {
+                const fixPrompt = [
+                    '你上一次回复没有提供可执行命令，导致 agent 无法继续。',
+                    '请仅返回严格标签格式：<assistant_reply>...</assistant_reply><commands>...</commands><done>true|false</done>',
+                    '若任务未完成，commands 至少给 1 条可执行命令（例如查看文件、运行测试）。',
+                    `当前任务: ${currentTask}`,
+                    `你上一次回复: ${rawResponse ? rawResponse.slice(0, 800) : '(空)'}`,
+                ].join('\n');
+                // Use a lighter model for format correction to save tokens/time
+                const fixModel = pickModelForPhase('fix');
+                const fixMessages = trimConversation(conversation);
+                const corrected = await getChatCompletion({
+                    apiKey,
+                    baseUrl,
+                    model: fixModel,
+                    messages: [
+                        { role: 'system', content: buildCodeAgentSystemPrompt(workspaceProfile) },
+                        ...fixMessages,
+                        { role: 'user', content: fixPrompt },
+                    ],
+                });
+                effectiveParsed = parseAgentResponse(corrected);
+                if (!effectiveParsed.reply || effectiveParsed.reply === '(无文本回复)') {
+                    effectiveParsed.reply = '已请求模型按可执行格式重试。';
+                }
+                assistantForHistory = corrected || '(空回复)';
+            }
+            conversation.push({ role: 'user', content: currentTask });
+            conversation.push({ role: 'assistant', content: assistantForHistory });
+            await persistCodeSession();
+            console.log(chalk.green.bold('\nKie Code Agent:'));
+            initRenderer();
+            const renderedReply = marked.parse(effectiveParsed.reply || '(无文本回复)');
+            process.stdout.write(renderedReply.trimEnd() + '\n');
+            if (effectiveParsed.done && effectiveParsed.commands.length === 0) {
+                console.log(chalk.green(`✅ Agent 已在第 ${step} 轮完成任务。`));
+                return;
+            }
+            if (effectiveParsed.commands.length === 0) {
+                const fallbackCommands = inferFallbackCommands(currentTask);
+                if (fallbackCommands.length === 0) {
+                    console.log(chalk.yellow('⚠️ 模型仍未返回可执行命令，建议 /model 切换更强模型后重试。'));
+                    return;
+                }
+                console.log(chalk.yellow(`⚠️ 模型未给命令，已启用兜底命令: ${fallbackCommands.join(' ; ')}`));
+                effectiveParsed = {
+                    ...effectiveParsed,
+                    done: false,
+                    commands: fallbackCommands,
+                };
+            }
+            for (const cmd of effectiveParsed.commands) {
+                if (isDangerousCommand(cmd)) {
+                    console.log(chalk.red(`⛔ 已拦截高风险命令: ${cmd}`));
+                    conversation.push({
+                        role: 'user',
+                        content: `[System] Dangerous command blocked and NOT executed: ${cmd}`,
+                    });
+                    continue;
+                }
+                let approved = autoApprove;
+                if (!approved && allowInteractiveConfirm) {
+                    approved = await askForConfirmation(chalk.yellow(`执行命令? ${cmd} (y/N): `));
+                }
+                if (!approved) {
+                    console.log(chalk.yellow(`⏭️ 已跳过命令: ${cmd}`));
+                    conversation.push({
+                        role: 'user',
+                        content: `[System] Command skipped by user confirmation policy: ${cmd}`,
+                    });
+                    await persistCodeSession();
+                    continue;
+                }
+                await executeCommandIntoContext(cmd);
+            }
+            currentTask =
+                `原始任务: ${initialTask}\n\n请基于上方命令的执行结果继续推进。若已全部完成且验证通过，输出 done=true。`;
+        }
+        console.log(chalk.yellow(`⚠️ 已达到最大轮数 ${maxAgentSteps}，请根据当前结果决定是否继续。`));
+    };
+    if (finalTask.trim()) {
+        const onSigint = () => {
+            console.log(chalk.yellow('\n正在取消…'));
+            interrupted = true;
+            activeAbort?.abort();
+            if (activeChildProcess) {
+                try {
+                    activeChildProcess.kill();
+                }
+                catch { }
+                activeChildProcess = null;
+            }
+        };
+        process.on('SIGINT', onSigint);
+        try {
+            if (useAgentMode) {
+                await runCodeAgentLoop(finalTask, true);
+            }
+            else {
+                await runCodeTurn(finalTask);
+            }
+        }
+        catch (error) {
+            if (isAbortError(error)) {
+                console.log(chalk.yellow('\n(已取消)\n'));
+                process.exit(130);
+            }
+            const msg = error instanceof Error ? error.message : String(error);
+            console.error(chalk.red(`\n❌ ${msg}\n`));
+            process.exit(1);
+        }
+        finally {
+            process.off('SIGINT', onSigint);
+        }
+        return;
+    }
+    console.log(chalk.cyan('=================================================='));
+    console.log(chalk.cyan(' 🧠 Kie Code 工作台（交互模式）'));
+    console.log(chalk.cyan('=================================================='));
+    console.log(chalk.gray(`会话: ${activeSessionId}`));
+    console.log(chalk.gray(`工作目录: ${codeCwd}`));
+    console.log(chalk.gray(`模型: ${formatModelLabel(codeModel)}`));
+    console.log(chalk.gray(`路由: ${routeMode} (${routingConfig.strategy})`));
+    console.log(chalk.gray(`模式: ${useAgentMode ? 'agent 自动执行' : 'code 对话'}`));
+    if (useAgentMode && !autoApprove) {
+        console.log(chalk.yellow('提示: 当前为逐条确认命令执行（可 /approve on 开启自动确认）。'));
+    }
+    if (workspace) {
+        console.log(chalk.gray(`工作区配置: tree=${workspace.includeTree ? 'on' : 'off'}, treeLimit=${workspace.treeLimit}, fileMaxChars=${workspace.fileMaxChars}`));
+    }
+    else {
+        console.log(chalk.gray('未检测到 .kie/code.json，可输入 /init 初始化'));
+    }
+    if (workspaceProfile) {
+        const parts = [`pm=${workspaceProfile.packageManager}`];
+        if (workspaceProfile.frameworks?.length)
+            parts.push(`stack=${workspaceProfile.frameworks.join(',')}`);
+        if (workspaceProfile.buildTool)
+            parts.push(`build=${workspaceProfile.buildTool}`);
+        if (workspaceProfile.testFramework)
+            parts.push(`test=${workspaceProfile.testFramework}`);
+        parts.push(`files=${workspaceProfile.totalFiles || '?'}`);
+        console.log(chalk.gray(`工作区画像: ${parts.join(', ')}`));
+    }
+    console.log(chalk.gray('输入 /help 查看指令；/sessions 管理会话；exit 退出\n'));
+    return new Promise((resolve) => {
+        const startREPL = () => {
+            const rl = readline.createInterface({
+                input: process.stdin,
+                output: process.stdout,
+                terminal: true,
+                prompt: 'kie:code › ',
+            });
+            let exiting = false;
+            let running = false;
+            rl.on('close', () => {
+                if (exiting)
+                    resolve();
+            });
+            rl.on('SIGINT', () => {
+                if (activeAbort) {
+                    activeAbort.abort();
+                    return;
+                }
+                if (activeChildProcess) {
+                    try {
+                        activeChildProcess.kill();
+                    }
+                    catch { }
+                    activeChildProcess = null;
+                    console.log(chalk.yellow('\n✘ 已终止命令执行'));
+                    interrupted = true;
+                    return;
+                }
+                exiting = true;
+                console.log(chalk.gray('\n👋 已退出 code 对话'));
+                rl.close();
+            });
+            const showPrompt = () => rl.prompt();
+            rl.on('line', async (line) => {
+                let input = line.trim();
+                if (/^kie\s+code\b/i.test(input)) {
+                    input = input.replace(/^kie\s+code\b/i, '').trim();
+                    input = input.replace(/^-y\b/i, '').trim();
+                    if (!input) {
+                        console.log(chalk.gray('提示：你已在 code REPL 内，直接输入任务即可，无需再写 kie code。\n'));
+                        showPrompt();
+                        return;
+                    }
+                }
+                if (!input) {
+                    showPrompt();
+                    return;
+                }
+                if (running) {
+                    console.log(chalk.gray('请等待当前请求完成（Ctrl+C 可取消）\n'));
+                    showPrompt();
+                    return;
+                }
+                if (input === 'exit' || input === 'quit') {
+                    exiting = true;
+                    console.log(chalk.gray('👋 已退出 code 对话'));
+                    rl.close();
+                    return;
+                }
+                if (input === '/help') {
+                    printCodeReplHelp();
+                    showPrompt();
+                    return;
+                }
+                if (input === '/clear') {
+                    conversation.length = 0;
+                    await persistCodeSession();
+                    console.log(chalk.green('✅ code 会话上下文已清空。\n'));
+                    showPrompt();
+                    return;
+                }
+                if (input === '/cwd') {
+                    console.log(chalk.cyan(`${codeCwd}\n`));
+                    showPrompt();
+                    return;
+                }
+                if (input === '/files') {
+                    if (extraFiles.length === 0) {
+                        console.log(chalk.gray('当前没有附加文件。\n'));
+                    }
+                    else {
+                        console.log(chalk.cyan('\n附加文件列表:'));
+                        extraFiles.forEach((f, i) => console.log(`  ${i + 1}. ${f}`));
+                        console.log();
+                    }
+                    showPrompt();
+                    return;
+                }
+                if (input === '/model') {
+                    exiting = false;
+                    rl.close();
+                    setTimeout(async () => {
+                        try {
+                            const picked = await promptSelectChatModel(codeModel, '选择切换当前会话默认模型');
+                            if (picked) {
+                                codeModel = picked;
+                                await persistWorkspaceConfig();
+                                await persistCodeSession();
+                                console.log(chalk.green(`\n✅ 已切换模型: ${formatModelLabel(codeModel)}\n`));
+                            }
+                        }
+                        catch (err) {
+                            console.log(chalk.red(`\n❌ 模型切换已取消或出错: ${err.message || err}\n`));
+                        }
+                        finally {
+                            startREPL();
+                        }
+                    }, 100);
+                    return;
+                }
+                if (input.startsWith('/model ')) {
+                    const picked = resolveModelPickInput(input.slice(7).trim());
+                    if (!picked) {
+                        console.log(chalk.gray('已取消模型切换。\n'));
+                    }
+                    else {
+                        codeModel = picked;
+                        await persistWorkspaceConfig();
+                        await persistCodeSession();
+                        console.log(chalk.green(`✅ 已切换模型: ${formatModelLabel(codeModel)}\n`));
+                    }
+                    showPrompt();
+                    return;
+                }
+                if (input === '/route') {
+                    console.log(chalk.cyan(`当前路由: ${routeMode}\n`));
+                    showPrompt();
+                    return;
+                }
+                if (input.startsWith('/route ')) {
+                    routeMode = resolveCodeRoute(input.slice(7).trim());
+                    await persistWorkspaceConfig();
+                    await persistCodeSession();
+                    console.log(chalk.green(`✅ 已设置路由: ${routeMode} (${routingConfig.strategy})\n`));
+                    showPrompt();
+                    return;
+                }
+                if (input === '/models') {
+                    printRoutingTable(routingConfig, routeMode, modelMetrics);
+                    showPrompt();
+                    return;
+                }
+                const phaseMatch = input.match(/^\/models\s+(plan|execute|fix)$/i);
+                if (phaseMatch) {
+                    const phase = phaseMatch[1].toLowerCase();
+                    exiting = false;
+                    rl.close();
+                    setTimeout(async () => {
+                        try {
+                            const pickedModel = await promptSelectChatModel(routingConfig.phases[phase]?.[0]?.model || codeModel, `选择要在 [${phase}] 阶段加入的 Chat 模型`);
+                            if (pickedModel) {
+                                const { default: select } = await import('@inquirer/select');
+                                const weight = await select({
+                                    message: `设置模型 ${pickedModel} 在 [${phase}] 阶段的权重 (权重越高，被选中的概率越大)`,
+                                    choices: [
+                                        { name: '1 (默认)', value: 1 },
+                                        { name: '2', value: 2 },
+                                        { name: '3 (推荐偏好)', value: 3 },
+                                        { name: '5 (高优先级)', value: 5 },
+                                        { name: '10 (极高优先级)', value: 10 },
+                                    ],
+                                });
+                                routingConfig = upsertWeightedModel(routingConfig, phase, pickedModel, weight);
+                                await persistWorkspaceConfig();
+                                console.log(chalk.green(`\n✅ 已成功配置: phase=${phase}, model=${pickedModel}, weight=${weight}\n`));
+                            }
+                        }
+                        catch (err) {
+                            console.log(chalk.red(`\n❌ 选择取消或发生错误: ${err.message || err}\n`));
+                        }
+                        finally {
+                            startREPL();
+                        }
+                    }, 100);
+                    return;
+                }
+                if (input.startsWith('/models strategy ')) {
+                    const strategy = input.slice('/models strategy '.length).trim().toLowerCase();
+                    if (strategy !== 'argmax' && strategy !== 'weighted-random') {
+                        console.log(chalk.red('❌ strategy 仅支持 argmax 或 weighted-random\n'));
+                        showPrompt();
+                        return;
+                    }
+                    routingConfig = { ...routingConfig, strategy };
+                    await persistWorkspaceConfig();
+                    console.log(chalk.green(`✅ 已设置路由策略: ${strategy}\n`));
+                    showPrompt();
+                    return;
+                }
+                if (input.startsWith('/models add ')) {
+                    const tokens = input.slice('/models add '.length).trim().split(/\s+/);
+                    const phase = parsePhase(tokens[0] || '');
+                    const model = tokens[1] || '';
+                    const weight = parsePositiveInt(tokens[2] || '1', 1);
+                    if (!phase || !model) {
+                        console.log(chalk.red('❌ 用法: /models add <plan|execute|fix> <model> [weight]\n'));
+                        showPrompt();
+                        return;
+                    }
+                    routingConfig = upsertWeightedModel(routingConfig, phase, model, weight);
+                    await persistWorkspaceConfig();
+                    console.log(chalk.green(`✅ 已添加模型: phase=${phase}, model=${model}, weight=${weight}\n`));
+                    showPrompt();
+                    return;
+                }
+                if (input.startsWith('/models rm ')) {
+                    const tokens = input.slice('/models rm '.length).trim().split(/\s+/);
+                    const phase = parsePhase(tokens[0] || '');
+                    const model = tokens[1] || '';
+                    if (!phase || !model) {
+                        console.log(chalk.red('❌ 用法: /models rm <plan|execute|fix> <model>\n'));
+                        showPrompt();
+                        return;
+                    }
+                    routingConfig = removeWeightedModel(routingConfig, phase, model);
+                    await persistWorkspaceConfig();
+                    console.log(chalk.green(`✅ 已移除模型: phase=${phase}, model=${model}\n`));
+                    showPrompt();
+                    return;
+                }
+                if (input === '/approve on') {
+                    autoApprove = true;
+                    await persistCodeSession();
+                    console.log(chalk.green('✅ 已开启自动确认执行命令。\n'));
+                    showPrompt();
+                    return;
+                }
+                if (input === '/approve off') {
+                    autoApprove = false;
+                    await persistCodeSession();
+                    console.log(chalk.green('✅ 已关闭自动确认，改为逐条确认。\n'));
+                    showPrompt();
+                    return;
+                }
+                if (input === '/sessions') {
+                    await renderSessionList();
+                    showPrompt();
+                    return;
+                }
+                if (input.startsWith('/sessions ')) {
+                    const arg = input.slice(10).trim();
+                    const sessions = await renderSessionList();
+                    let picked = arg;
+                    const n = Number.parseInt(arg, 10);
+                    if (!Number.isNaN(n) && n >= 1 && n <= sessions.length) {
+                        picked = sessions[n - 1].id;
+                    }
+                    if (!picked) {
+                        console.log(chalk.gray('已取消。\n'));
+                    }
+                    else if (picked === activeSessionId) {
+                        console.log(chalk.gray('已在当前会话。\n'));
+                    }
+                    else {
+                        await switchSession(picked);
+                    }
+                    showPrompt();
+                    return;
+                }
+                if (input === '/init') {
+                    const initResult = await initCodeWorkspace(codeCwd, {
+                        includeTree,
+                        treeLimit,
+                        fileMaxChars,
+                        defaultModel: codeModel,
+                        defaultRoute: routeMode,
+                    });
+                    const profileInit = await initCodeWorkspaceProfile(codeCwd);
+                    workspaceProfile = profileInit.profile;
+                    if (initResult.created) {
+                        console.log(chalk.green(`✅ 初始化完成: ${initResult.configPath}\n`));
+                    }
+                    else {
+                        console.log(chalk.yellow(`ℹ️ 已存在: ${initResult.configPath}\n`));
+                    }
+                    console.log(chalk.green(`✅ 工作区画像已刷新: ${profileInit.profilePath}\n`));
+                    showPrompt();
+                    return;
+                }
+                if (input === '/status') {
+                    console.log(chalk.cyan('\n当前 code 状态:'));
+                    console.log(`  session: ${activeSessionId}`);
+                    console.log(`  cwd: ${codeCwd}`);
+                    console.log(`  model: ${codeModel}`);
+                    console.log(`  route: ${routeMode} (${routingConfig.strategy})`);
+                    console.log(`  autoApprove: ${autoApprove ? 'on' : 'off'}`);
+                    console.log(`  mode: ${useAgentMode ? 'agent' : 'chat'}`);
+                    console.log(`  tree: ${includeTree ? 'on' : 'off'} (limit=${treeLimit})`);
+                    console.log(`  fileMaxChars: ${fileMaxChars}`);
+                    console.log(`  files: ${extraFiles.length}`);
+                    console.log(`  metricsModels: ${Object.keys(modelMetrics.metrics).length}`);
+                    console.log();
+                    showPrompt();
+                    return;
+                }
+                if (input === '/tree on') {
+                    includeTree = true;
+                    await persistWorkspaceConfig();
+                    await persistCodeSession();
+                    console.log(chalk.green('✅ 已开启项目结构注入。\n'));
+                    showPrompt();
+                    return;
+                }
+                if (input === '/tree off') {
+                    includeTree = false;
+                    await persistWorkspaceConfig();
+                    await persistCodeSession();
+                    console.log(chalk.green('✅ 已关闭项目结构注入。\n'));
+                    showPrompt();
+                    return;
+                }
+                if (input.startsWith('/file ')) {
+                    const raw = input.slice(6).trim();
+                    const filePath = (raw.startsWith('"') && raw.endsWith('"')) ||
+                        (raw.startsWith("'") && raw.endsWith("'"))
+                        ? raw.slice(1, -1)
+                        : raw;
+                    if (!filePath) {
+                        console.log(chalk.yellow('⚠️ 请提供文件路径。\n'));
+                        showPrompt();
+                        return;
+                    }
+                    try {
+                        const fullPath = path.resolve(codeCwd, filePath);
+                        if (!fs.existsSync(fullPath)) {
+                            console.log(chalk.red(`❌ 文件不存在: ${fullPath}\n`));
+                        }
+                        else {
+                            if (!extraFiles.includes(filePath)) {
+                                extraFiles.push(filePath);
+                                await persistCodeSession();
+                                console.log(chalk.green(`✅ 已添加文件: ${filePath}\n`));
+                            }
+                            else {
+                                console.log(chalk.gray(`ℹ️ 文件已在列表中: ${filePath}\n`));
+                            }
+                        }
+                    }
+                    catch (err) {
+                        console.log(chalk.red(`❌ 错误: ${err.message}\n`));
+                    }
+                    showPrompt();
+                    return;
+                }
+                if (input.startsWith('/run ')) {
+                    const cmd = input.slice(5).trim();
+                    if (!cmd) {
+                        console.log(chalk.yellow('⚠️ 请提供要执行的命令。\n'));
+                    }
+                    else {
+                        await executeCommandIntoContext(cmd);
+                    }
+                    showPrompt();
+                    return;
+                }
+                running = true;
+                rl.pause();
+                try {
+                    if (useAgentMode) {
+                        await runCodeAgentLoop(input, true);
+                    }
+                    else {
+                        await runCodeTurn(input);
+                    }
+                }
+                catch (error) {
+                    if (isAbortError(error)) {
+                        console.log(chalk.yellow('\n(已取消)\n'));
+                    }
+                    else {
+                        const msg = error instanceof Error ? error.message : String(error);
+                        console.log(chalk.red(`\n❌ ${msg}\n`));
+                    }
+                }
+                finally {
+                    running = false;
+                    interrupted = false;
+                    rl.resume();
+                    showPrompt();
+                }
+            });
+            showPrompt();
+        };
+        startREPL();
+    });
+}
+export function registerCodeCommand(program, config, readStdin) {
+    program
+        .command('code')
+        .description('代码模式：支持单次任务、初始化与交互式对话')
+        .argument('[task]', '代码任务描述（不传则进入交互模式）')
+        .option('-m, --model <model>', '指定代码模型')
+        .option('--cwd <dir>', '代码工作目录')
+        .option('--init', '初始化当前 code 工作目录')
+        .option('--run <cmd>', '先执行命令并注入上下文', (val, prev) => [...prev, val], [])
+        .option('--agent', '启用自动执行代理循环')
+        .option('--chat', '切换为纯对话模式')
+        .option('--route <mode>', '模型路由')
+        .option('--max-steps <n>', 'agent 最大执行轮数', '25')
+        .option('-y, --yes', 'agent 模式自动确认执行命令')
+        .option('--session <id>', '使用指定 code 会话')
+        .option('--new-session', '创建新的 code 会话')
+        .option('-f, --file <path>', '附加文件内容', (val, prev) => [...prev, val], [])
+        .option('--no-tree', '不附带项目结构摘要')
+        .option('--tree-limit <n>', '项目结构最大条目数', '120')
+        .option('--file-max-chars <n>', '每个附加文件最大字符数', '8000')
+        .action(async (task, options) => {
+        await codeAction(task, options, config, readStdin);
+    });
+}