foliko 1.0.74 → 1.0.76

package/cli/src/ui/chat-ui.js

@@ -1,199 +1,201 @@
-/**
- * 聊天界面组件
- * 使用 readline question() 实现多行输入
- */
-
-const readline = require('readline')
-const { CLEAR_LINE, CYAN, DIM, GREEN, RED, YELLOW, colored } = require('../utils/ansi')
-const { renderLine } = require('../utils/markdown')
-
-class ChatUI {
-  constructor(agent, options = {}) {
-    this.agent = agent
-    this.rl = null
-    this.lines = []  // 多行输入的累积
-
-    // 会话管理
-    this.sessionId = options.sessionId || 'cli_default'
-    this.sessionPlugin = null
-
-    // 如果 agent 有 framework，获取 SessionPlugin
-    if (agent.framework) {
-      this.sessionPlugin = agent.framework.pluginManager.get('session')
-      // 确保会话存在
-      if (this.sessionPlugin) {
-        this.sessionPlugin.getOrCreateSession(this.sessionId, {
-          metadata: { platform: 'cli' }
-        })
-      }
-    }
-  }
-
-  /**
-   * 启动聊天界面
-   */
-  start() {
-    this.rl = readline.createInterface({
-      input: process.stdin,
-      output: process.stdout,
-      crlfDelay: Infinity
-    })
-
-    this.printWelcome()
-    this.promptUser()
-  }
-
-  /**
-   * 打印欢迎信息
-   */
-  printWelcome() {
-    console.log(`${colored('Foliko', CYAN)} - 持续对话聊天`)
-    console.log(`${colored('Ctrl+C', DIM)} 退出 | ${colored('连续两次回车', DIM)} 发送多行 | ${colored('!!', DIM)} 立即发送\n`)
-  }
-
-  /**
-   * 获取多行输入
-   */
-  getMultilineInput() {
-    return new Promise((resolve) => {
-      const lines = []
-
-      const question = (isFirst) => {
-        const prompt = isFirst ? colored('> ', GREEN) : colored('- ', DIM)
-        this.rl.question(prompt, (input) => {
-          // 输入 !! 立即结束
-          if (input.trim() === '!!') {
-            const result = lines.join('\n').trim()
-            resolve(result)
-            return
-          }
-
-          // 空行：结束输入并发送（第一次空行就发送）
-          if (input.trim() === '') {
-            const result = lines.join('\n').trim()
-            resolve(result)
-            return
-          }
-
-          // 非空行：添加到 lines，继续输入
-          lines.push(input)
-          question(false)
-        })
-      }
-
-      question(true)
-    })
-  }
-
-  /**
-   * 提示用户输入
-   */
-  async promptUser() {
-    try {
-      const input = await this.getMultilineInput()
-
-      if (!input) {
-        await this.promptUser()
-        return
-      }
-
-      // 退出命令
-      if (input.toLowerCase() === 'exit' || input.toLowerCase() === 'quit') {
-        console.log('再见！')
-        this.rl.close()
-        return
-      }
-
-      // 发送消息
-      await this.sendMessage(input)
-
-      // 继续等待下一条消息
-      await this.promptUser()
-    } catch (err) {
-      console.error(`\n${colored('[错误]', RED)} ${err.message}`)
-      await this.promptUser()
-    }
-  }
-
-  /**
-   * 发送消息并显示响应
-   */
-  async sendMessage(message) {
-    console.log()
-
-    // 添加用户消息到会话历史
-    if (this.sessionPlugin) {
-      this.sessionPlugin.addMessage(this.sessionId, { role: 'user', content: message })
-    }
-
-    // 用于打断的标志
-    let interrupted = false
-
-    // 设置打断处理
-    const interruptHandler = () => {
-      if (!interrupted) {
-        interrupted = true
-        console.log(`\n${colored('[中断]', RED)} 输出已中断`)
-      }
-    }
-
-    // 监听 Ctrl+C
-    process.on('SIGINT', interruptHandler)
-
-    let fullResponse = ''
-
-    try {
-      let lineBuffer = ''
-      const renderState = { inThink: false, inCodeBlock: false }
-
-      console.log(colored('● ', GREEN))
-      // console.log()
-
-      const runWithContext = this.agent.framework?.runWithContext.bind(this.agent.framework)
-      const { sessionId } = this
-
-      await runWithContext({ sessionId }, async () => {
-        for await (const chunk of this.agent.chatStream(message, { sessionId })) {
-        if (interrupted) break
-
-        if (chunk.type === 'text') {
-          fullResponse += chunk.text
-          lineBuffer += chunk.text
-
-          while (lineBuffer.includes('\n')) {
-            if (interrupted) break
-
-            const nlIndex = lineBuffer.indexOf('\n')
-            const line = lineBuffer.substring(0, nlIndex)
-            lineBuffer = lineBuffer.substring(nlIndex + 1)
-            if (line.trim()) {
-              console.log(renderLine(line, renderState))
-            }
-          }
-        } else if (chunk.type === 'tool-call') {
-          console.log(`\n${colored('[工具调用]', YELLOW)} ${chunk.toolName}`)
-        } else if (chunk.type === 'error') {
-          console.error(`\n${colored('[错误]', RED)} ${chunk.error}`)
-        }
-        }
-      })
-
-      if (lineBuffer.trim() && !interrupted) {
-        console.log(renderLine(lineBuffer, renderState))
-      }
-    } catch (err) {
-      if (!interrupted) {
-        console.error(`\n${colored('[错误]', RED)} ${err.message}\n`)
-      }
-    } finally {
-      process.removeListener('SIGINT', interruptHandler)
-
-      // 添加助手回复到会话历史
-      if (this.sessionPlugin && fullResponse) {
-        this.sessionPlugin.addMessage(this.sessionId, { role: 'assistant', content: fullResponse })
-      }
-    }
-  }
-}
-
-module.exports = { ChatUI }
+/**
+ * 聊天界面组件
+ * 使用 readline question() 实现多行输入
+ */
+
+const readline = require('readline');
+const { CLEAR_LINE, CYAN, DIM, GREEN, RED, YELLOW, colored } = require('../utils/ansi');
+const { renderLine } = require('../utils/markdown');
+
+class ChatUI {
+  constructor(agent, options = {}) {
+    this.agent = agent;
+    this.rl = null;
+    this.lines = []; // 多行输入的累积
+
+    // 会话管理
+    this.sessionId = options.sessionId || 'cli_default';
+    this.sessionPlugin = null;
+
+    // 如果 agent 有 framework，获取 SessionPlugin
+    if (agent.framework) {
+      this.sessionPlugin = agent.framework.pluginManager.get('session');
+      // 确保会话存在
+      if (this.sessionPlugin) {
+        this.sessionPlugin.getOrCreateSession(this.sessionId, {
+          metadata: { platform: 'cli' },
+        });
+      }
+    }
+  }
+
+  /**
+   * 启动聊天界面
+   */
+  start() {
+    this.rl = readline.createInterface({
+      input: process.stdin,
+      output: process.stdout,
+      crlfDelay: Infinity,
+    });
+
+    this.printWelcome();
+    this.promptUser();
+  }
+
+  /**
+   * 打印欢迎信息
+   */
+  printWelcome() {
+    console.log(`${colored('Foliko', CYAN)} - 持续对话聊天`);
+    console.log(
+      `${colored('Ctrl+C', DIM)} 退出 | ${colored('连续两次回车', DIM)} 发送多行 | ${colored('!!', DIM)} 立即发送\n`
+    );
+  }
+
+  /**
+   * 获取多行输入
+   */
+  getMultilineInput() {
+    return new Promise((resolve) => {
+      const lines = [];
+
+      const question = (isFirst) => {
+        const prompt = isFirst ? colored('> ', GREEN) : colored('- ', DIM);
+        this.rl.question(prompt, (input) => {
+          // 输入 !! 立即结束
+          if (input.trim() === '!!') {
+            const result = lines.join('\n').trim();
+            resolve(result);
+            return;
+          }
+
+          // 空行：结束输入并发送（第一次空行就发送）
+          if (input.trim() === '') {
+            const result = lines.join('\n').trim();
+            resolve(result);
+            return;
+          }
+
+          // 非空行：添加到 lines，继续输入
+          lines.push(input);
+          question(false);
+        });
+      };
+
+      question(true);
+    });
+  }
+
+  /**
+   * 提示用户输入
+   */
+  async promptUser() {
+    try {
+      const input = await this.getMultilineInput();
+
+      if (!input) {
+        await this.promptUser();
+        return;
+      }
+
+      // 退出命令
+      if (input.toLowerCase() === 'exit' || input.toLowerCase() === 'quit') {
+        console.log('再见！');
+        this.rl.close();
+        return;
+      }
+
+      // 发送消息
+      await this.sendMessage(input);
+
+      // 继续等待下一条消息
+      await this.promptUser();
+    } catch (err) {
+      console.error(`\n${colored('[错误]', RED)} ${err.message}`);
+      await this.promptUser();
+    }
+  }
+
+  /**
+   * 发送消息并显示响应
+   */
+  async sendMessage(message) {
+    console.log();
+
+    // 添加用户消息到会话历史
+    if (this.sessionPlugin) {
+      this.sessionPlugin.addMessage(this.sessionId, { role: 'user', content: message });
+    }
+
+    // 用于打断的标志
+    let interrupted = false;
+
+    // 设置打断处理
+    const interruptHandler = () => {
+      if (!interrupted) {
+        interrupted = true;
+        console.log(`\n${colored('[中断]', RED)} 输出已中断`);
+      }
+    };
+
+    // 监听 Ctrl+C
+    process.on('SIGINT', interruptHandler);
+
+    let fullResponse = '';
+
+    try {
+      let lineBuffer = '';
+      const renderState = { inThink: false, inCodeBlock: false };
+
+      console.log(colored('● ', GREEN));
+      // console.log()
+
+      const runWithContext = this.agent.framework?.runWithContext.bind(this.agent.framework);
+      const { sessionId } = this;
+
+      await runWithContext({ sessionId }, async () => {
+        for await (const chunk of this.agent.chatStream(message, { sessionId })) {
+          if (interrupted) break;
+
+          if (chunk.type === 'text') {
+            fullResponse += chunk.text;
+            lineBuffer += chunk.text;
+
+            while (lineBuffer.includes('\n')) {
+              if (interrupted) break;
+
+              const nlIndex = lineBuffer.indexOf('\n');
+              const line = lineBuffer.substring(0, nlIndex);
+              lineBuffer = lineBuffer.substring(nlIndex + 1);
+              if (line.trim()) {
+                console.log(renderLine(line, renderState));
+              }
+            }
+          } else if (chunk.type === 'tool-call') {
+            console.log(`\n${colored('[工具调用]', YELLOW)} ${chunk.toolName}`);
+          } else if (chunk.type === 'error') {
+            console.error(`\n${colored('[错误]', RED)} ${chunk.error}`);
+          }
+        }
+      });
+
+      if (lineBuffer.trim() && !interrupted) {
+        console.log(renderLine(lineBuffer, renderState));
+      }
+    } catch (err) {
+      if (!interrupted) {
+        console.error(`\n${colored('[错误]', RED)} ${err.message}\n`);
+      }
+    } finally {
+      process.removeListener('SIGINT', interruptHandler);
+
+      // 添加助手回复到会话历史
+      if (this.sessionPlugin && fullResponse) {
+        this.sessionPlugin.addMessage(this.sessionId, { role: 'assistant', content: fullResponse });
+      }
+    }
+  }
+}
+
+module.exports = { ChatUI };

package/cli/src/utils/ansi.js

@@ -1,40 +1,40 @@
-/**
- * ANSI 转义序列工具
- */
-
-const CLEAR = '\x1B[0m'
-const CLEAR_LINE = '\x1B[2K'
-
-// 颜色
-const RED = '\x1B[31m'
-const GREEN = '\x1B[32m'
-const YELLOW = '\x1B[33m'
-const BLUE = '\x1B[34m'
-const MAGENTA = '\x1B[35m'
-const CYAN = '\x1B[36m'
-const WHITE = '\x1B[37m'
-const DIM = '\x1B[2m'
-const BOLD = '\x1B[1m'
-
-/**
- * 创建带颜色的字符串
- */
-function colored(text, color) {
-  if (!color) return text
-  return `${color}${text}${CLEAR}`
-}
-
-module.exports = {
-  CLEAR,
-  CLEAR_LINE,
-  RED,
-  GREEN,
-  YELLOW,
-  BLUE,
-  MAGENTA,
-  CYAN,
-  WHITE,
-  DIM,
-  BOLD,
-  colored
-}
+/**
+ * ANSI 转义序列工具
+ */
+
+const CLEAR = '\x1B[0m';
+const CLEAR_LINE = '\x1B[2K';
+
+// 颜色
+const RED = '\x1B[31m';
+const GREEN = '\x1B[32m';
+const YELLOW = '\x1B[33m';
+const BLUE = '\x1B[34m';
+const MAGENTA = '\x1B[35m';
+const CYAN = '\x1B[36m';
+const WHITE = '\x1B[37m';
+const DIM = '\x1B[2m';
+const BOLD = '\x1B[1m';
+
+/**
+ * 创建带颜色的字符串
+ */
+function colored(text, color) {
+  if (!color) return text;
+  return `${color}${text}${CLEAR}`;
+}
+
+module.exports = {
+  CLEAR,
+  CLEAR_LINE,
+  RED,
+  GREEN,
+  YELLOW,
+  BLUE,
+  MAGENTA,
+  CYAN,
+  WHITE,
+  DIM,
+  BOLD,
+  colored,
+};