修复多轮对话bug

src/ui/terminal.ts

@@ -1,18 +1,23 @@
 import * as readline from 'readline';
 import chalk from 'chalk';
-import ora from 'ora';
 import type { Agent } from '../core/agent.js';
 
 export class TerminalUI {
   private agent: Agent;
   private rl: readline.Interface;
-  private spinner = ora();
+  private isClosed = false;
 
   constructor(agent: Agent) {
     this.agent = agent;
     this.rl = readline.createInterface({
       input: process.stdin,
       output: process.stdout,
+      terminal: true,
+    });
+
+    // 监听关闭事件
+    this.rl.on('close', () => {
+      this.isClosed = true;
     });
   }
 
@@ -20,7 +25,7 @@ export class TerminalUI {
   private showWelcome(): void {
     console.log(chalk.cyan('\n╔════════════════════════════════════════╗'));
     console.log(chalk.cyan('║') + chalk.bold.white('     🤖 AI Terminal Assistant          ') + chalk.cyan('║'));
-    console.log(chalk.cyan('║') + chalk.gray('     Powered by Claude                 ') + chalk.cyan('║'));
+    console.log(chalk.cyan('║') + chalk.gray('     Powered by DeepSeek / Claude      ') + chalk.cyan('║'));
     console.log(chalk.cyan('╚════════════════════════════════════════╝\n'));
     console.log(chalk.gray('输入你的问题，或使用以下命令：'));
     console.log(chalk.yellow('  /help') + chalk.gray('  - 显示帮助'));
@@ -52,6 +57,7 @@ export class TerminalUI {
       case '/exit':
       case '/quit':
         console.log(chalk.cyan('\n👋 再见！\n'));
+        this.close();
         process.exit(0);
 
       default:
@@ -61,9 +67,14 @@ export class TerminalUI {
 
   // 提问并获取用户输入
   private prompt(): Promise<string> {
-    return new Promise((resolve) => {
+    return new Promise((resolve, reject) => {
+      if (this.isClosed) {
+        reject(new Error('readline closed'));
+        return;
+      }
+
       this.rl.question(chalk.green('You > '), (answer) => {
-        resolve(answer);
+        resolve(answer ?? '');
       });
     });
   }
@@ -72,7 +83,8 @@ export class TerminalUI {
   async start(): Promise<void> {
     this.showWelcome();
 
-    while (true) {
+    while (!this.isClosed) {
+      try {
         const input = await this.prompt();
 
         // 跳过空输入
@@ -88,14 +100,15 @@ export class TerminalUI {
         }
 
         // 发送给 AI
-      this.spinner.start(chalk.gray('思考中...'));
+        process.stdout.write(chalk.gray('思考中...'));
 
         try {
           let isFirstChunk = true;
 
           await this.agent.chat(input, (text) => {
             if (isFirstChunk) {
-            this.spinner.stop();
+              // 清除 "思考中..." 并显示 AI 前缀
+              process.stdout.write('\r' + ' '.repeat(20) + '\r');
               process.stdout.write(chalk.blue('AI > '));
               isFirstChunk = false;
             }
@@ -112,18 +125,28 @@ export class TerminalUI {
 
           console.log('\n');
         } catch (error) {
-        this.spinner.stop();
+          // 清除 "思考中..."
+          process.stdout.write('\r' + ' '.repeat(20) + '\r');
           console.log(
             chalk.red(
-            `\n❌ 错误: ${error instanceof Error ? error.message : String(error)}\n`
+              `❌ 错误: ${error instanceof Error ? error.message : String(error)}\n`
             )
           );
         }
+      } catch {
+        // readline 关闭，退出循环
+        break;
       }
     }
 
+    console.log(chalk.cyan('\n👋 再见！\n'));
+  }
+
   // 关闭
   close(): void {
+    if (!this.isClosed) {
+      this.isClosed = true;
       this.rl.close();
     }
   }
+}