foliko 1.0.3 → 1.0.5

package/.claude/settings.local.json

@@ -26,7 +26,9 @@
       "Bash(node -e \":*)",
       "Bash(node -e \"\nconst { loadAgentConfig } = require\\('./plugins/default-plugins'\\);\nconst config = loadAgentConfig\\('D:/Date/20260321/app/.agent'\\);\nconsole.log\\('skillsDirs:', config.skillsDirs\\);\n\")",
       "Bash(cd D:/Code/vb-agent && pnpm install --shamefully-hoist 2>&1 | head -20)",
-      "Bash(cd D:/Code/vb-agent && rm -rf node_modules && pnpm install)"
+      "Bash(cd D:/Code/vb-agent && rm -rf node_modules && pnpm install)",
+      "Bash(cd D:/Code/vb-agent && timeout 8 node test-tg.js 2>&1 || true)",
+      "Bash(cd D:/Code/vb-agent && timeout 10 node test-tg.js 2>&1 || true)"
     ]
   }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "foliko",
-  "version": "1.0.3",
+  "version": "1.0.5",
   "description": "简约的插件化 Agent 框架",
   "main": "src/index.js",
   "bin": {
@@ -31,6 +31,7 @@
     "marked": "^11.2.0",
     "marked-terminal": "6",
     "node-cron": "^4.2.1",
+    "node-telegram-bot-api": "^0.67.0",
     "zod": "^3.24.0"
   }
 }

package/plugins/ai-plugin.js

@@ -28,11 +28,12 @@ class AIPlugin extends Plugin {
 
   install(framework) {
     this._framework = framework
+    this._initAIClient()
     return this
   }
 
   start(framework) {
-    this._initAIClient()
+    // AI client 已在 install 中初始化
     return this
   }
 

package/plugins/default-plugins.js

@@ -212,6 +212,25 @@ async function bootstrapDefaults(framework, config = {}) {
     console.log('[Bootstrap] AI Plugin loaded')
   }
 
+  // 1.5 创建主 Agent（供 Telegram 等需要绑定 Agent 的插件使用）
+  if (!framework._mainAgent && aiConfig.provider) {
+    const { Agent } = require('../src/core/agent')
+    const aiPlugin = framework.pluginManager.get('ai')
+    const aiClient = aiPlugin ? aiPlugin.getAIClient() : null
+    console.log('[Bootstrap] Creating Main Agent - aiClient:', !!aiClient)
+
+    framework._mainAgent = new Agent(framework, {
+      name: 'MainAgent',
+      systemPrompt: '你是一个有帮助的AI助手。',
+      model: aiConfig.model,
+      provider: aiConfig.provider,
+      apiKey: aiConfig.apiKey,
+      baseURL: aiConfig.baseURL
+    })
+    framework._agents.push(framework._mainAgent)
+    console.log('[Bootstrap] Main Agent created, has _chatHandler:', !!framework._mainAgent._chatHandler)
+  }
+
   // 2. Storage 存储插件
   if (!framework.pluginManager.has('storage')) {
     const { StoragePlugin } = require('./storage-plugin')
@@ -351,6 +370,20 @@ async function bootstrapDefaults(framework, config = {}) {
     console.log('[Bootstrap] Python Plugin Loader already loaded, skipping')
   }
 
+  // 12.6 Telegram 插件（如果配置了Token）
+  if (!framework.pluginManager.has('telegram')) {
+    const telegramToken = process.env.TELEGRAM_BOT_TOKEN || agentConfig.telegram?.botToken
+    if (telegramToken) {
+      const { Plugin } = require('../src/core/plugin-base')
+      const createTelegramPlugin = require('./telegram-plugin')
+      const TelegramPlugin = createTelegramPlugin(Plugin)
+      await framework.loadPlugin(new TelegramPlugin({ botToken: telegramToken }))
+      console.log('[Bootstrap] Telegram Plugin loaded')
+    }
+  } else {
+    console.log('[Bootstrap] Telegram Plugin already loaded, skipping')
+  }
+
   // 13. 加载自定义插件
   await loadCustomPlugins(framework, agentConfig)
 

package/plugins/telegram-plugin.js

@@ -0,0 +1,510 @@
+/**
+ * Telegram 插件 v2
+ * 支持绑定主Agent进行持续对话
+ *
+ * 配置：
+ * - botToken: Telegram Bot Token (必需)
+ * - allowedChats: 允许的聊天ID数组
+ * - groupMode: 是否启用群组模式
+ * - prefix: 命令前缀
+ */
+
+const { Plugin } = require('../src/core/plugin-base')
+const { z } = require('zod')
+
+// 转义 MarkdownV2 特殊字符
+function escapeMarkdown(text) {
+  if (!text) return ''
+  // MarkdownV2 需要转义的字符
+  return text.replace(/([\_*\[\]()~`>#+\-=|{}.!])/g, '\\$1')
+}
+
+module.exports = function(Plugin) {
+  return class TelegramPlugin extends Plugin {
+    constructor(config = {}) {
+      super()
+      this.name = 'telegram'
+      this.version = '2.0.0'
+      this.description = 'Telegram 对话插件，绑定主Agent进行持续对话'
+      this.priority = 80
+
+      this.config = {
+        botToken: config.botToken || process.env.TELEGRAM_BOT_TOKEN,
+        allowedChats: config.allowedChats || [],
+        groupMode: config.groupMode || false,
+        prefix: config.prefix || '/',
+        systemPrompt: config.systemPrompt || '你是一个有帮助的AI助手。'
+      }
+
+      this._framework = null
+      this._bot = null
+      this._sessions = new Map() // chatId -> { agent, history, lastActive }
+      this._streamingMessages = new Map() // chatId -> messageId
+    }
+
+    install(framework) {
+      this._framework = framework
+      return this
+    }
+
+    start(framework) {
+      if (!this.config.botToken) {
+        console.warn('[Telegram] No bot token. Set TELEGRAM_BOT_TOKEN env var.')
+        return this
+      }
+
+      this._initBot()
+      return this
+    }
+
+    _initBot() {
+      try {
+        // 添加 .agent/node_modules 到搜索路径
+        // const path = require('path')
+        // const agentNodeModules = path.join(process.cwd(), '.agent', 'node_modules')
+        // if (!module.paths.includes(agentNodeModules)) {
+        //   module.paths.unshift(agentNodeModules)
+        // }
+
+        const TelegramBot = require('node-telegram-bot-api')
+        this._bot = new TelegramBot(this.config.botToken, { polling: true })
+
+        console.log('[Telegram] Bot started successfully')
+
+        // 注册菜单命令
+        this._bot.setMyCommands([
+          { command: 'start', description: '显示帮助信息' },
+          { command: 'clear', description: '清除对话历史' },
+          { command: 'history', description: '查看对话状态' }
+        ]).then(() => {
+          console.log('[Telegram] Commands registered')
+        }).catch((err) => {
+          console.error('[Telegram] Failed to register commands:', err.message)
+        })
+
+        // 监听消息
+        this._bot.on('message', (msg) => this._handleMessage(msg))
+
+        // 监听编辑消息
+        this._bot.on('edit_message', (msg) => this._handleEdit(msg))
+
+        // 监听回调
+        this._bot.on('callback_query', (query) => this._handleCallback(query))
+
+        // 监听错误
+        this._bot.on('polling_error', (err) => {
+          console.error('[Telegram] Polling error:', err.message)
+        })
+
+        this._bot.on('error', (err) => {
+          console.error('[Telegram] Bot error:', err.message)
+        })
+
+        // 监听 Agent 创建事件，确保在 Agent 创建后能正确获取
+        if (this._framework) {
+          this._framework.on('agent:created', (agent) => {
+            console.log('[Telegram] New agent created:', agent.name)
+          })
+        }
+
+      } catch (err) {
+        console.error('[Telegram] Failed to initialize bot:', err.message)
+      }
+    }
+
+    /**
+     * 获取主Agent
+     */
+    _getMainAgent() {
+      console.log('[Telegram] _getMainAgent called')
+      console.log('[Telegram]   _framework:', !!this._framework)
+      console.log('[Telegram]   _framework._mainAgent:', !!this._framework?._mainAgent)
+      console.log('[Telegram]   _framework._agents:', this._framework?._agents?.length || 0)
+
+      if (this._framework._mainAgent) {
+        console.log('[Telegram]   returning _mainAgent')
+        return this._framework._mainAgent
+      }
+      const agents = this._framework._agents || []
+      console.log('[Telegram]   returning last agent:', agents.length > 0 ? agents[agents.length - 1].name : 'none')
+      return agents.length > 0 ? agents[agents.length - 1] : null
+    }
+
+    /**
+     * 获取或创建会话Agent
+     */
+    _getSessionAgent(chatId) {
+      console.log('[Telegram] _getSessionAgent called for chatId:', chatId)
+      let session = this._sessions.get(chatId)
+
+      if (!session) {
+        const mainAgent = this._getMainAgent()
+        console.log('[Telegram] mainAgent:', !!mainAgent, mainAgent?.name)
+        if (!mainAgent) {
+          console.log('[Telegram] ERROR: mainAgent is null!')
+          return null
+        }
+
+        // 创建会话Agent（复用主Agent的配置）
+        session = {
+          agent: mainAgent,
+          history: [], // 对话历史
+          lastActive: new Date()
+        }
+        this._sessions.set(chatId, session)
+        console.log(`[Telegram] New session created for: ${chatId}`)
+      }
+
+      return session
+    }
+
+    /**
+     * 处理消息
+     */
+    async _handleMessage(msg) {
+      if (!msg || !msg.chat) return
+
+      const chatId = msg.chat.id.toString()
+
+      // 处理图片消息
+      if (msg.photo) {
+        await this._handlePhoto(msg)
+        return
+      }
+
+      // 处理文档消息
+      if (msg.document) {
+        await this._handleDocument(msg)
+        return
+      }
+
+      if (!msg.text) return
+
+      const text = msg.text.trim()
+
+      // 命令处理
+      if (text.startsWith(this.config.prefix)) {
+        await this._handleCommand(msg)
+        return
+      }
+
+      // 群组模式检查
+      if (msg.chat.type === 'group' || msg.chat.type === 'supergroup') {
+        if (!this.config.groupMode) return
+      }
+
+      // 权限检查
+      if (!this._checkPermission(chatId)) {
+        await this._sendMessage(chatId, '抱歉，您没有权限使用此 Bot。', msg.message_id)
+        return
+      }
+
+      // 处理对话
+      await this._processChat(chatId, text, msg.message_id)
+    }
+
+    /**
+     * 处理命令
+     */
+    async _handleCommand(msg) {
+      const chatId = msg.chat.id.toString()
+      const text = msg.text.trim()
+      const parts = text.split(' ')
+      const command = parts[0].substring(1)
+      const args = parts.slice(1).join(' ')
+
+      switch (command.toLowerCase()) {
+        case 'start':
+          await this._sendMessage(chatId,
+            '👋 欢迎使用 AI 助手！\n\n' +
+            '直接发送消息即可与我对话，我会尽力帮助您。\n\n' +
+            '可用命令：\n' +
+            '/start - 显示帮助\n' +
+            '/clear - 清除对话历史\n' +
+            '/history - 查看历史消息数',
+            msg.message_id)
+          break
+
+        case 'clear':
+          this._clearSession(chatId)
+          await this._sendMessage(chatId, '✅ 对话历史已清除', msg.message_id)
+          break
+
+        case 'history':
+          const session = this._sessions.get(chatId)
+          await this._sendMessage(chatId,
+            `📊 当前状态\n\n会话数：${this._sessions.size}\n` +
+            `历史消息：${session?.history.length || 0}\n` +
+            `最后活跃：${session?.lastActive?.toLocaleString() || '无'}`,
+            msg.message_id)
+          break
+
+        default:
+          // 未识别命令，作为普通消息处理
+          await this._processChat(chatId, text, msg.message_id)
+      }
+    }
+
+    /**
+     * 处理对话
+     */
+    async _processChat(chatId, text, replyToMessageId) {
+      console.log('[Telegram] _processChat called, chatId:', chatId, 'text:', text.substring(0, 50))
+      const session = this._getSessionAgent(chatId)
+      console.log('[Telegram] session:', !!session)
+      if (!session) {
+        await this._sendMessage(chatId, '❌ AI 服务未初始化', replyToMessageId)
+        return
+      }
+
+      // 添加用户消息到历史
+      session.history.push({ role: 'user', content: text })
+      session.lastActive = new Date()
+
+      // 发送思考中消息
+      let thinkingMsg
+      try {
+        thinkingMsg = await this._bot.sendMessage(chatId, '🤔 思考中...', {
+          reply_to_message_id: replyToMessageId
+        })
+      } catch (err) {
+        console.error('[Telegram] Send thinking error:', err.message)
+        return
+      }
+
+      try {
+        let fullResponse = ''
+        const chatIdStr = `telegram_${chatId}`
+
+        // 使用流式响应
+        for await (const chunk of session.agent.chatStream(text, {
+          sessionId: chatIdStr
+        })) {
+          // chatStream 返回 { type: 'text', text } 或 { type: 'tool-call', toolName, args }
+          if (chunk.type === 'text' && chunk.text) {
+            fullResponse += chunk.text
+
+            // 定期更新消息
+            if (fullResponse.length % 100 === 0) {
+              try {
+                await this._bot.editMessageText(`📝 ${fullResponse}▌`, {
+                  chat_id: chatId,
+                  message_id: thinkingMsg.message_id
+                })
+              } catch (e) {
+                // 忽略编辑错误
+              }
+            }
+          }
+        }
+
+        // 保存助手回复到历史
+        session.history.push({ role: 'assistant', content: fullResponse })
+
+        // 发送最终回复（转义 Markdown 特殊字符）
+        const safeResponse = fullResponse || '抱歉，我没有收到有效的回复。'
+        await this._bot.editMessageText(safeResponse, {
+          chat_id: chatId,
+          message_id: thinkingMsg.message_id,
+          parse_mode: 'MarkdownV2'
+        })
+
+      } catch (err) {
+        console.error('[Telegram] Chat error:', err)
+        await this._bot.editMessageText(`❌ 发生错误：${err.message}`, {
+          chat_id: chatId,
+          message_id: thinkingMsg.message_id
+        })
+      }
+    }
+
+    /**
+     * 处理编辑消息
+     */
+    async _handleEdit(msg) {
+      // 支持编辑后重新处理
+    }
+
+    /**
+     * 处理回调查询
+     */
+    async _handleCallback(query) {
+      await this._bot.answerCallbackQuery(query.id, {
+        text: '处理中...'
+      })
+    }
+
+    /**
+     * 权限检查
+     */
+    _checkPermission(chatId) {
+      if (this.config.allowedChats.length === 0) return true
+      return this.config.allowedChats.includes(chatId)
+    }
+
+    /**
+     * 处理图片消息
+     */
+    async _handlePhoto(msg) {
+      const chatId = msg.chat.id.toString()
+      const caption = msg.caption?.trim() || ''
+
+      // 权限检查
+      if (!this._checkPermission(chatId)) {
+        await this._sendMessage(chatId, '抱歉，您没有权限使用此 Bot。', msg.message_id)
+        return
+      }
+
+      // 获取最大尺寸的图片
+      const photo = msg.photo[msg.photo.length - 1]
+      console.log(`[Telegram] Photo received, file_id: ${photo.file_id}`)
+
+      try {
+        // 下载图片
+        const filePath = await this._downloadFile(photo.file_id, '.jpg', 'telegram_images')
+        console.log(`[Telegram] Photo saved: ${filePath}`)
+
+        const savedMsg = `收到图片${caption ? ': ' + caption : ''}\n已保存至: ${filePath}`
+        await this._sendMessage(chatId, savedMsg, msg.message_id)
+
+        // TODO: 调用 AI 视觉能力分析图片
+        // await this._analyzeImage(chatId, filePath, caption)
+      } catch (err) {
+        console.error('[Telegram] Failed to save photo:', err.message)
+        await this._sendMessage(chatId, '图片保存失败: ' + err.message, msg.message_id)
+      }
+    }
+
+    /**
+     * 处理文档消息
+     */
+    async _handleDocument(msg) {
+      const chatId = msg.chat.id.toString()
+      const fileName = msg.document.file_name || '未命名文件'
+      const caption = msg.caption?.trim() || ''
+
+      // 权限检查
+      if (!this._checkPermission(chatId)) {
+        await this._sendMessage(chatId, '抱歉，您没有权限使用此 Bot。', msg.message_id)
+        return
+      }
+
+      console.log(`[Telegram] Document received: ${fileName}, file_id: ${msg.document.file_id}`)
+
+      try {
+        // 根据文件扩展名确定保存目录和扩展名
+        const ext = fileName.includes('.') ? fileName.split('.').pop() : 'bin'
+        const filePath = await this._downloadFile(msg.document.file_id, ext, 'telegram_documents')
+        console.log(`[Telegram] Document saved: ${filePath}`)
+
+        const savedMsg = `收到文件: ${fileName}\n已保存至: ${filePath}${caption ? '\n\n说明: ' + caption : ''}`
+        await this._sendMessage(chatId, savedMsg, msg.message_id)
+      } catch (err) {
+        console.error('[Telegram] Failed to save document:', err.message)
+        await this._sendMessage(chatId, '文件保存失败: ' + err.message, msg.message_id)
+      }
+    }
+
+    /**
+     * 下载文件
+     * @param {string} fileId - Telegram 文件 ID
+     * @param {string} ext - 文件扩展名
+     * @param {string} subDir - 子目录名
+     * @returns {Promise<string>} 保存的文件路径
+     */
+    async _downloadFile(fileId, ext, subDir) {
+      return new Promise((resolve, reject) => {
+        const path = require('path')
+        const fs = require('fs')
+
+        // 创建保存目录
+        const saveDir = path.join(process.cwd(), '.agent', 'data', subDir)
+        if (!fs.existsSync(saveDir)) {
+          fs.mkdirSync(saveDir, { recursive: true })
+        }
+
+        // 生成文件名
+        const fileName = `${Date.now()}_${Math.random().toString(36).substring(7)}.${ext}`
+        const filePath = path.join(saveDir, fileName)
+
+        // 下载文件
+        this._bot.downloadFile(fileId, saveDir)
+          .then((savedPath) => {
+            // 重命名为期望的文件名
+            const finalPath = path.join(saveDir, fileName)
+            if (savedPath !== finalPath) {
+              fs.renameSync(savedPath, finalPath)
+            }
+            resolve(finalPath)
+          })
+          .catch(reject)
+      })
+    }
+
+    /**
+     * 清除会话
+     */
+    _clearSession(chatId) {
+      const session = this._sessions.get(chatId)
+      if (session && session.agent) {
+        session.agent.clearHistory?.()
+      }
+      this._sessions.delete(chatId)
+    }
+
+    /**
+     * 发送消息
+     */
+    async _sendMessage(chatId, text, replyToMessageId = null) {
+      if (!this._bot) return
+      return this._bot.sendMessage(chatId, text, {
+        reply_to_message_id: replyToMessageId,
+              })
+    }
+
+    /**
+     * 获取插件状态
+     */
+    getStatus() {
+      return {
+        connected: !!this._bot,
+        sessionCount: this._sessions.size,
+        sessions: Array.from(this._sessions.entries()).map(([chatId, session]) => ({
+          chatId,
+          historyLength: session.history.length,
+          lastActive: session.lastActive
+        })),
+        config: {
+          groupMode: this.config.groupMode,
+          prefix: this.config.prefix,
+          allowedChatsCount: this.config.allowedChats.length
+        }
+      }
+    }
+
+    /**
+     * 停止 Bot
+     */
+    stopBot() {
+      if (this._bot) {
+        this._bot.stopPolling()
+        this._bot = null
+        console.log('[Telegram] Bot stopped')
+      }
+    }
+
+    reload(framework) {
+      this._framework = framework
+      if (this.config.botToken) {
+        this.stopBot()
+        this._initBot()
+      }
+    }
+
+    uninstall(framework) {
+      this.stopBot()
+      this._sessions.clear()
+      this._framework = null
+    }
+  }
+}

package/test-tg-bot.js

@@ -0,0 +1,42 @@
+require('dotenv').config();
+const { Framework } = require('./src');
+
+async function main() {
+  console.log('Starting test...\n');
+
+  const framework = new Framework({ debug: false });
+
+  await framework.bootstrap({
+    agentDir: process.cwd() + '/.agent',
+    aiConfig: {
+      provider: process.env.FOLIKO_PROVIDER || 'minimax',
+      model: process.env.FOLIKO_MODEL || 'MiniMax-M2.7',
+      baseURL: process.env.FOLIKO_BASE_URL || 'https://api.minimaxi.com/v1',
+      apiKey: process.env.MINIMAX_API_KEY || process.env.DEEPSEEK_API_KEY
+    }
+  });
+
+  // 检查主 Agent
+  const mainAgent = framework._mainAgent;
+  console.log('Main Agent:', mainAgent ? mainAgent.name : 'NOT FOUND');
+  console.log('Agent apiKey:', mainAgent?.apiKey ? 'set' : 'NOT SET');
+  console.log('Agent _chatHandler:', mainAgent?._chatHandler ? 'exists' : 'NOT EXISTS');
+  console.log('Agent _chatHandler._aiClient:', mainAgent?._chatHandler?._aiClient ? 'exists' : 'NOT EXISTS');
+
+  // 检查 Telegram 插件
+  const tgPlugin = framework.pluginManager.get('telegram');
+  console.log('\nTelegram Plugin:', tgPlugin ? 'loaded' : 'NOT LOADED');
+
+  if (tgPlugin) {
+    console.log('Telegram _bot:', tgPlugin._bot ? 'exists' : 'NOT EXISTS');
+  }
+
+  console.log('\n========================================');
+  console.log('Bot is running. Send a message to test!');
+  console.log('========================================\n');
+
+  // 保持运行
+  await new Promise(() => {});
+}
+
+main().catch(console.error);