foliko 1.0.40 → 1.0.41

package/plugins/telegram-plugin.js

@@ -1,5 +1,5 @@
 /**
- * Telegram 插件 v2
+ * Telegram 插件
  * 支持绑定主Agent进行持续对话
  *
  * 配置：
@@ -15,615 +15,400 @@ 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
-      // 默认不启用，需要在 plugins.json 中设置 enabled: true
-      this.enabled = false
-      this.systemPrompt='你是一个有帮助的AI助手。回复内容不要使用markdown格式文本。'
-
-      this.config = {
-        botToken: config.botToken || process.env.TELEGRAM_BOT_TOKEN,
-        allowedChats: config.allowedChats || [],
-        groupMode: config.groupMode || false,
-        prefix: config.prefix || '/'
-      }
-
-      this._framework = null
-      this._bot = null
-      this._sessionPlugin = null
-      this._sessionDeleteHandler = null
-      this._sessionAgents = new Map()  // chatId -> Agent
+class TelegramPlugin extends Plugin {
+  constructor(config = {}) {
+    super()
+    this.name = 'telegram'
+    this.version = '2.1.0'
+    this.description = 'Telegram 对话插件，绑定主Agent进行持续对话'
+    this.priority = 80
+    this.enabled = false
+    this.systemPrompt = '你是一个有帮助的AI助手。回复内容不要使用markdown格式文本。'
+
+    this.config = {
+      botToken: config.botToken || process.env.TELEGRAM_BOT_TOKEN,
+      allowedChats: config.allowedChats || [],
+      groupMode: config.groupMode || false,
+      prefix: config.prefix || '/'
     }
 
-    install(framework) {
-      this._framework = framework
+    this._framework = null
+    this._bot = null
+    this._sessionPlugin = null
+    this._sessionDeleteHandler = null
+    this._sessionAgents = new Map()
+  }
+
+  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
     }
 
-    start(framework) {
-      if (!this.config.botToken) {
-        console.warn('[Telegram] No bot token. Set TELEGRAM_BOT_TOKEN env var.')
-        return this
-      }
-
-      // 获取 SessionPlugin 引用
-      this._sessionPlugin = framework.pluginManager.get('session')
+    this._framework = framework
+    this._sessionPlugin = framework.pluginManager.get('session')
 
-      // 监听 SessionPlugin 的会话删除事件
-      if (this._sessionPlugin) {
-        this._sessionDeleteHandler = (sessionId) => {
-          // 只处理 telegram 会话的删除
-          if (sessionId.startsWith('telegram_')) {
-            console.log(`[Telegram] Session deleted: ${sessionId}`)
-          }
+    if (this._sessionPlugin) {
+      this._sessionDeleteHandler = (sessionId) => {
+        if (sessionId.startsWith('telegram_')) {
+          console.log(`[Telegram] Session deleted: ${sessionId}`)
         }
-        this._sessionPlugin.on('session:deleted', this._sessionDeleteHandler)
       }
-
-      this._initBot()
-      return this
+      this._sessionPlugin.on('session:deleted', this._sessionDeleteHandler)
     }
 
-    _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._initBot()
+    return this
+  }
 
-        // 监听消息
-        this._bot.on('message', (msg) => this._handleMessage(msg))
+  _initBot() {
+    try {
+      const TelegramBot = require('node-telegram-bot-api')
+      this._bot = new TelegramBot(this.config.botToken, { polling: true })
 
-        // 监听编辑消息
-        this._bot.on('edit_message', (msg) => this._handleEdit(msg))
+      console.log('[Telegram] Bot started successfully')
 
-        // 监听回调
-        this._bot.on('callback_query', (query) => this._handleCallback(query))
+      this._bot.setMyCommands([
+        { command: 'start', description: '显示帮助信息' },
+        { command: 'clear', description: '清除对话历史' },
+        { command: 'history', description: '查看对话状态' }
+      ]).catch((err) => {
+        console.error('[Telegram] Failed to register commands:', err.message)
+      })
 
-        // 监听错误
-        this._bot.on('polling_error', (err) => {
-          console.error('[Telegram] Polling error:', 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))
 
-        this._bot.on('error', (err) => {
-          console.error('[Telegram] Bot error:', err.message)
+      if (this._framework) {
+        this._framework.on('agent:created', (agent) => {
+          console.log('[Telegram] New agent created:', agent.name)
+        })
+        this._framework.on('scheduler:reminder', async (data) => {
+          await this._handleScheduledReminder(data)
         })
+      }
+    } catch (err) {
+      console.error('[Telegram] Failed to initialize bot:', err.message)
+    }
+  }
 
-        // 监听 Agent 创建事件，确保在 Agent 创建后能正确获取
-        if (this._framework) {
-          this._framework.on('agent:created', (agent) => {
-            console.log('[Telegram] New agent created:', agent.name)
-          })
-
-          // 监听定时提醒事件
-          this._framework.on('scheduler:reminder', async (data) => {
-            console.log('[Telegram] Received scheduler reminder:', data)
-            await this._handleScheduledReminder(data)
-          })
-        }
+  async _handleScheduledReminder(data) {
+    const { taskName, message, sessionId } = data
+    const reminderText = `🔔 [${taskName}]\n\n${message}`
 
+    if (sessionId && sessionId.startsWith('telegram_')) {
+      const chatId = sessionId.replace('telegram_', '')
+      try {
+        await this._bot.sendMessage(chatId, reminderText)
+        console.log(`[Telegram] Reminder sent to chat ${chatId}`)
       } catch (err) {
-        console.error('[Telegram] Failed to initialize bot:', err.message)
+        console.error(`[Telegram] Failed to send reminder:`, err.message)
       }
+      return
     }
 
-    /**
-     * 处理定时提醒
-     */
-    async _handleScheduledReminder(data) {
-      const { taskName, message, sessionId } = data
-
-      // 构建提醒消息
-      const reminderText = `🔔 [${taskName}]\n\n${message}`
+    if (this._sessionPlugin) {
+      const sessions = this._sessionPlugin.listSessions()
+        .filter(s => s.id.startsWith('telegram_'))
+        .sort((a, b) => new Date(b.lastActive).getTime() - new Date(a.lastActive).getTime())
 
-      // 如果有 sessionId 是 telegram 类型的，发送到对应 chat
-      if (sessionId && sessionId.startsWith('telegram_')) {
-        const chatId = sessionId.replace('telegram_', '')
+      if (sessions.length > 0) {
+        const chatId = sessions[0].id.replace('telegram_', '')
         try {
           await this._bot.sendMessage(chatId, reminderText)
-          console.log(`[Telegram] Reminder sent to chat ${chatId}`)
+          console.log(`[Telegram] Reminder sent to recent chat ${chatId}`)
         } catch (err) {
           console.error(`[Telegram] Failed to send reminder:`, err.message)
         }
-        return
-      }
-
-      // 其他情况（包括 null 或其他 sessionId），发送到最近的 Telegram 会话
-      if (this._sessionPlugin) {
-        const allSessions = this._sessionPlugin.listSessions()
-        const telegramSessions = allSessions
-          .filter(s => s.id.startsWith('telegram_'))
-          .sort((a, b) => new Date(b.lastActive).getTime() - new Date(a.lastActive).getTime())
-
-        if (telegramSessions.length > 0) {
-          const chatId = telegramSessions[0].id.replace('telegram_', '')
-          try {
-            await this._bot.sendMessage(chatId, reminderText)
-            console.log(`[Telegram] Reminder sent to chat ${chatId}`)
-          } catch (err) {
-            console.error(`[Telegram] Failed to send reminder to ${chatId}:`, err.message)
-          }
-        } else {
-          console.log('[Telegram] No active Telegram sessions to send reminder')
-        }
-      } else {
-        console.log('[Telegram] No active Telegram sessions to send reminder')
       }
     }
+  }
 
-    /**
-     * 获取主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
+  _getMainAgent() {
+    if (this._framework._mainAgent) return this._framework._mainAgent
+    const agents = this._framework._agents || []
+    return agents.length > 0 ? agents[agents.length - 1] : null
+  }
+
+  _getSessionAgent(chatId) {
+    if (this._sessionAgents.has(chatId)) {
+      return { agent: this._sessionAgents.get(chatId), sessionId: `telegram_${chatId}` }
     }
 
-    /**
-     * 获取或创建会话Agent
-     * 使用 SessionPlugin 统一管理会话历史
-     */
-    _getSessionAgent(chatId) {
-      console.log('[Telegram] _getSessionAgent called for chatId:', chatId)
-
-      // 检查缓存
-      if (this._sessionAgents.has(chatId)) {
-        const agent = this._sessionAgents.get(chatId)
-        const sessionId = `telegram_${chatId}`
-        console.log('[Telegram] Reusing cached session agent for chatId:', chatId)
-        return { agent, sessionId }
-      }
+    const agent = this._framework.createSessionAgent(`telegram_${chatId}`, {
+      systemPrompt: this.systemPrompt
+    })
+    this._sessionAgents.set(chatId, agent)
 
-      // 创建新 agent
-      const agent = this._framework.createSessionAgent(`telegram_${chatId}`, {
-        systemPrompt: this.systemPrompt
+    if (this._sessionPlugin) {
+      this._sessionPlugin.getOrCreateSession(`telegram_${chatId}`, {
+        metadata: { platform: 'telegram', chatId }
       })
-      this._sessionAgents.set(chatId, agent)
-      console.log('[Telegram] Created new session agent for chatId:', chatId)
-
-      // 使用 SessionPlugin 管理会话历史
-      if (this._sessionPlugin) {
-        const sessionId = `telegram_${chatId}`
-        this._sessionPlugin.getOrCreateSession(sessionId, {
-          metadata: { platform: 'telegram', chatId }
-        })
-      }
-
-      console.log(`[Telegram] Session ready for chatId: ${chatId}`)
-      return { agent, sessionId: `telegram_${chatId}` }
     }
 
-    /**
-     * 处理消息
-     */
-    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
+    return { agent, sessionId: `telegram_${chatId}` }
+  }
 
-      const text = msg.text.trim()
+  async _handleMessage(msg) {
+    if (!msg || !msg.chat) return
+    const chatId = msg.chat.id.toString()
 
-      // 命令处理
-      if (text.startsWith(this.config.prefix)) {
-        await this._handleCommand(msg)
-        return
-      }
+    if (msg.photo) { await this._handlePhoto(msg); return }
+    if (msg.document) { await this._handleDocument(msg); return }
+    if (!msg.text) return
 
-      // 群组模式检查
-      if (msg.chat.type === 'group' || msg.chat.type === 'supergroup') {
-        if (!this.config.groupMode) return
-      }
+    const text = msg.text.trim()
+    if (text.startsWith(this.config.prefix)) {
+      await this._handleCommand(msg)
+      return
+    }
 
-      // 权限检查
-      if (!this._checkPermission(chatId)) {
-        await this._sendMessage(chatId, '抱歉，您没有权限使用此 Bot。', msg.message_id)
-        return
-      }
+    if (msg.chat.type === 'group' || msg.chat.type === 'supergroup') {
+      if (!this.config.groupMode) return
+    }
 
-      // 处理对话
-      await this._processChat(chatId, text, msg.message_id)
+    if (!this._checkPermission(chatId)) {
+      await this._sendMessage(chatId, '抱歉，您没有权限使用此 Bot。', msg.message_id)
+      return
     }
 
-    /**
-     * 处理命令
-     */
-    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._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':
+        if (this._sessionPlugin) {
+          const session = this._sessionPlugin.getSession(`telegram_${chatId}`)
+          const sessions = this._sessionPlugin.listSessions().filter(s => s.id.startsWith('telegram_'))
           await this._sendMessage(chatId,
-            '👋 欢迎使用 AI 助手！\n\n' +
-            '直接发送消息即可与我对话，我会尽力帮助您。\n\n' +
-            '可用命令：\n' +
-            '/start - 显示帮助\n' +
-            '/clear - 清除对话历史\n' +
-            '/history - 查看历史消息数',
+            `📊 当前状态\n\n会话数：${sessions.length}\n历史消息：${session?.messages?.length || 0}\n最后活跃：${session?.lastActive?.toLocaleString() || '无'}`,
             msg.message_id)
-          break
-
-        case 'clear':
-          this._clearSession(chatId)
-          await this._sendMessage(chatId, '✅ 对话历史已清除', msg.message_id)
-          break
-
-        case 'history':
-          if (this._sessionPlugin) {
-            const sessionId = `telegram_${chatId}`
-            const session = this._sessionPlugin.getSession(sessionId)
-            const allSessions = this._sessionPlugin.listSessions()
-            const telegramSessionCount = allSessions.filter(s => s.id.startsWith('telegram_')).length
-            await this._sendMessage(chatId,
-              `📊 当前状态\n\n会话数：${telegramSessionCount}\n` +
-              `历史消息：${session?.messages.length || 0}\n` +
-              `最后活跃：${session?.lastActive?.toLocaleString() || '无'}`,
-              msg.message_id)
-          } else {
-            await this._sendMessage(chatId, '❌ 会话服务不可用', msg.message_id)
-          }
-          break
-
-        default:
-          // 未识别命令，作为普通消息处理
-          await this._processChat(chatId, text, msg.message_id)
-      }
+        } else {
+          await this._sendMessage(chatId, '❌ 会话服务不可用', 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 sessionInfo = this._getSessionAgent(chatId)
-      console.log('[Telegram] session:', !!sessionInfo)
-      if (!sessionInfo) {
-        await this._sendMessage(chatId, '❌ AI 服务未初始化', replyToMessageId)
-        return
-      }
+  async _processChat(chatId, text, replyToMessageId) {
+    const sessionInfo = this._getSessionAgent(chatId)
+    if (!sessionInfo) {
+      await this._sendMessage(chatId, '❌ AI 服务未初始化', replyToMessageId)
+      return
+    }
 
-      const { agent, sessionId } = sessionInfo
+    const { agent, sessionId } = sessionInfo
 
-      // 使用 SessionPlugin 添加用户消息到历史
-      if (this._sessionPlugin) {
-        this._sessionPlugin.addMessage(sessionId, { role: 'user', content: text })
-      }
+    if (this._sessionPlugin) {
+      this._sessionPlugin.addMessage(sessionId, { role: 'user', content: text })
+    }
 
-      // 发送思考中消息
-      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
-      }
+    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 = ''
-
-        // 使用流式响应
-        for await (const chunk of agent.chatStream(text, {
-          sessionId: sessionId
-        })) {
-          // 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(`📝 ${escapeMarkdown(fullResponse)}▌`, {
-                  chat_id: chatId,
-                  message_id: thinkingMsg.message_id
-                })
-              } catch (e) {
-                // 忽略编辑错误
-              }
-            }
+    try {
+      let fullResponse = ''
+      for await (const chunk of agent.chatStream(text, { sessionId })) {
+        if (chunk.type === 'text' && chunk.text) {
+          fullResponse += chunk.text
+          if (fullResponse.length % 100 === 0) {
+            try {
+              await this._bot.editMessageText(`📝 ${escapeMarkdown(fullResponse)}▌`, {
+                chat_id: chatId,
+                message_id: thinkingMsg.message_id
+              })
+            } catch (e) { /* ignore */ }
           }
         }
-
-        // 保存助手回复到历史（使用 SessionPlugin）
-        if (this._sessionPlugin) {
-          this._sessionPlugin.addMessage(sessionId, { role: 'assistant', content: fullResponse })
-        }
-
-        // 发送最终回复（转义 Markdown 特殊字符）
-        const safeResponse = escapeMarkdown(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(escapeMarkdown(`❌ 发生错误：${err.message}`), {
-          chat_id: chatId,
-          message_id: thinkingMsg.message_id
-        })
       }
-    }
 
-    /**
-     * 处理编辑消息
-     */
-    async _handleEdit(msg) {
-      // 支持编辑后重新处理
-    }
+      if (this._sessionPlugin) {
+        this._sessionPlugin.addMessage(sessionId, { role: 'assistant', content: fullResponse })
+      }
 
-    /**
-     * 处理回调查询
-     */
-    async _handleCallback(query) {
-      await this._bot.answerCallbackQuery(query.id, {
-        text: '处理中...'
+      const safeResponse = escapeMarkdown(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(escapeMarkdown(`❌ 发生错误：${err.message}`), {
+        chat_id: chatId,
+        message_id: thinkingMsg.message_id
       })
     }
+  }
 
-    /**
-     * 权限检查
-     */
-    _checkPermission(chatId) {
-      if (this.config.allowedChats.length === 0) return true
-      return this.config.allowedChats.includes(chatId)
-    }
+  async _handleEdit(msg) { /* 支持编辑后重新处理 */ }
 
-    /**
-     * 处理图片消息
-     */
-    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
-      }
+  async _handleCallback(query) {
+    await this._bot.answerCallbackQuery(query.id, { text: '处理中...' })
+  }
 
-      // 获取最大尺寸的图片
-      const photo = msg.photo[msg.photo.length - 1]
-      console.log(`[Telegram] Photo received, file_id: ${photo.file_id}`)
+  _checkPermission(chatId) {
+    return this.config.allowedChats.length === 0 || this.config.allowedChats.includes(chatId)
+  }
 
-      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)
-        if(caption){
-          const message=`图片：${filePath}, ${caption}`
-          await this._processChat(chatId, message, msg.message_id)
-        }
-        // TODO: 调用 AI 视觉能力分析图片
-        // await this._analyzeImage(chatId, filePath, capti on)
-      } catch (err) {
-        console.error('[Telegram] Failed to save photo:', err.message)
-        await this._sendMessage(chatId, '图片保存失败: ' + err.message, msg.message_id)
-      }
+  async _handlePhoto(msg) {
+    const chatId = msg.chat.id.toString()
+    if (!this._checkPermission(chatId)) {
+      await this._sendMessage(chatId, '抱歉，您没有权限使用此 Bot。', msg.message_id)
+      return
     }
 
-    /**
-     * 处理文档消息
-     */
-    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}`)
+    const caption = msg.caption?.trim() || ''
+    const photo = msg.photo[msg.photo.length - 1]
 
-      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)
-        if(caption){
-          const message=`文件：${filePath}, ${caption}`
-          await this._processChat(chatId, message, msg.message_id)
-        }
-      } catch (err) {
-        console.error('[Telegram] Failed to save document:', err.message)
-        await this._sendMessage(chatId, '文件保存失败: ' + err.message, msg.message_id)
+    try {
+      const filePath = await this._downloadFile(photo.file_id, '.jpg', 'telegram_images')
+      await this._sendMessage(chatId, `收到图片${caption ? ': ' + caption : ''}\n已保存至: ${filePath}`, msg.message_id)
+      if (caption) {
+        await this._processChat(chatId, `图片：${filePath}, ${caption}`, msg.message_id)
       }
+    } catch (err) {
+      console.error('[Telegram] Failed to save photo:', 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(filePath)
-          })
-          .catch(reject)
-      })
+  async _handleDocument(msg) {
+    const chatId = msg.chat.id.toString()
+    if (!this._checkPermission(chatId)) {
+      await this._sendMessage(chatId, '抱歉，您没有权限使用此 Bot。', msg.message_id)
+      return
     }
 
-    /**
-     * 清除会话
-     */
-    _clearSession(chatId) {
-      // 清除 SessionPlugin 中的会话（会触发 session:deleted 事件）
-      if (this._sessionPlugin) {
-        const sessionId = `telegram_${chatId}`
-        this._sessionPlugin.deleteSession(sessionId)
+    const fileName = msg.document.file_name || '未命名文件'
+    const caption = msg.caption?.trim() || ''
+    const ext = fileName.includes('.') ? fileName.split('.').pop() : 'bin'
+
+    try {
+      const filePath = await this._downloadFile(msg.document.file_id, ext, 'telegram_documents')
+      await this._sendMessage(chatId, `收到文件: ${fileName}\n已保存至: ${filePath}${caption ? '\n\n说明: ' + caption : ''}`, msg.message_id)
+      if (caption) {
+        await this._processChat(chatId, `文件：${filePath}, ${caption}`, msg.message_id)
       }
+    } catch (err) {
+      console.error('[Telegram] Failed to save document:', err.message)
+      await this._sendMessage(chatId, '文件保存失败: ' + err.message, msg.message_id)
     }
+  }
 
-    /**
-     * 发送消息
-     */
-    async _sendMessage(chatId, text, replyToMessageId = null) {
-      if (!this._bot) return
-      return this._bot.sendMessage(chatId, text, {
-        reply_to_message_id: replyToMessageId,
-              })
-    }
+  async _downloadFile(fileId, ext, subDir) {
+    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 })
 
-    /**
-     * 获取插件状态
-     */
-    getStatus() {
-      // 从 SessionPlugin 获取 Telegram 相关会话
-      let sessions = []
-      if (this._sessionPlugin) {
-        const allSessions = this._sessionPlugin.listSessions()
-        sessions = allSessions
-          .filter(s => s.id.startsWith('telegram_'))
-          .map(s => ({
-            chatId: s.id.replace('telegram_', ''),
-            historyLength: s.messageCount,
-            lastActive: s.lastActive
-          }))
-      }
+    const fileName = `${Date.now()}_${Math.random().toString(36).substring(7)}.${ext}`
+    const filePath = path.join(saveDir, fileName)
 
-      return {
-        connected: !!this._bot,
-        sessionCount: sessions.length,
-        sessions,
-        config: {
-          groupMode: this.config.groupMode,
-          prefix: this.config.prefix,
-          allowedChatsCount: this.config.allowedChats.length
-        }
-      }
-    }
+    const savedPath = await this._bot.downloadFile(fileId, saveDir)
+    if (savedPath !== filePath) fs.renameSync(savedPath, filePath)
+    return filePath
+  }
 
-    /**
-     * 停止 Bot
-     */
-    stopBot() {
-      if (this._bot) {
-        this._bot.stopPolling()
-        this._bot = null
-        console.log('[Telegram] Bot stopped')
-      }
+  _clearSession(chatId) {
+    if (this._sessionPlugin) {
+      this._sessionPlugin.deleteSession(`telegram_${chatId}`)
     }
+  }
 
-    reload(framework) {
-      this._framework = framework
-      if (this.config.botToken) {
-        this.stopBot()
-        this._initBot()
-      }
-    }
+  async _sendMessage(chatId, text, replyToMessageId = null) {
+    if (!this._bot) return
+    return this._bot.sendMessage(chatId, text, { reply_to_message_id: replyToMessageId })
+  }
 
-    /**
-     * 启动 Bot（用于重新启用时）
-     */
-    start(framework) {
-      if (this.config.botToken && !this._bot) {
-        this._initBot()
+  getStatus() {
+    let sessions = []
+    if (this._sessionPlugin) {
+      sessions = this._sessionPlugin.listSessions()
+        .filter(s => s.id.startsWith('telegram_'))
+        .map(s => ({
+          chatId: s.id.replace('telegram_', ''),
+          historyLength: s.messageCount,
+          lastActive: s.lastActive
+        }))
+    }
+    return {
+      connected: !!this._bot,
+      sessionCount: sessions.length,
+      sessions,
+      config: {
+        groupMode: this.config.groupMode,
+        prefix: this.config.prefix,
+        allowedChatsCount: this.config.allowedChats.length
       }
     }
+  }
 
-    uninstall(framework) {
-      // 销毁所有 session agents
-      for (const agent of this._sessionAgents.values()) {
-        agent.destroy()
-      }
-      this._sessionAgents.clear()
+  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()
-      if (this._sessionPlugin && this._sessionDeleteHandler) {
-        this._sessionPlugin.off('session:deleted', this._sessionDeleteHandler)
-        this._sessionDeleteHandler = null
-      }
-      this._sessionPlugin = null
-      this._framework = null
+      this._initBot()
+    }
+  }
+
+  uninstall() {
+    for (const agent of this._sessionAgents.values()) {
+      agent.destroy()
+    }
+    this._sessionAgents.clear()
+    this.stopBot()
+    if (this._sessionPlugin && this._sessionDeleteHandler) {
+      this._sessionPlugin.off('session:deleted', this._sessionDeleteHandler)
+      this._sessionDeleteHandler = null
     }
+    this._sessionPlugin = null
+    this._framework = null
   }
 }
+
+module.exports = { TelegramPlugin }