chatgpt-plugin/index.js

import plugin from '../../lib/plugins/plugin.js'
import { ChatGPTAPI } from 'chatgpt'
import _ from 'lodash'
const SESSION_TOKEN=''

/**
 * 每个对话保留的时长。单个对话内ai是保留上下文的。超时后销毁对话，再次对话创建新的对话。
 * 单位：秒
 * @type {number}
 */
const CONVERSATION_PRESERVE_TIME = 600
export class chatgpt extends plugin {
  constructor () {
    super({
      /** 功能名称 */
      name: 'chatgpt',
      /** 功能描述 */
      dsc: 'chatgpt from openai',
      /** https://oicqjs.github.io/oicq/#events */
      event: 'message',
      /** 优先级，数字越小等级越高 */
      priority: 5000,
      rule: [
        {
          /** 命令正则匹配 */
          reg: '^#chatgpt([\s\S]*)',
          /** 执行方法 */
          fnc: 'chatgpt'
        },
        {
          reg: 'chatgpt对话列表',
          fnc: 'getConversations',
          permission: 'master'
        },
        {
          reg: '^#结束对话([\s\S]*)',
          fnc: 'destroyConversations'
        },
        {
          reg: 'chatgpt帮助',
          fnc: 'help'
        }
      ]
    })
    this.chatGPTApi = new ChatGPTAPI({
      sessionToken: SESSION_TOKEN,
      markdown: true,
      userAgent: 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36'
    })
  }

  /**
   * 获取chatgpt当前对话列表
   * @param e
   * @returns {Promise<void>}
   */
  async getConversations (e) {
    let keys = await redis.keys('CHATGPT:CONVERSATIONS:*')
        if (!keys || keys.length === 0) {
            await this.reply('当前没有人正在与机器人对话', true)
        } else {
            let response = '当前对话列表：(格式为【开始时间 ｜ qq昵称 ｜ 对话长度 ｜ 最后活跃时间】)\n'
            await Promise.all(keys.map(async (key) => {
                let conversation = await redis.get(key)
                if (conversation) {
                    conversation = JSON.parse(conversation)
                    response += `${conversation.ctime} ｜ ${conversation.sender.nickname} ｜ ${conversation.num} ｜ ${conversation.utime} \n`
                }
            }))
            await this.reply(`${response}`, true)
        }
  }

  /**
     * 销毁指定人的对话
     * @param e
     * @returns {Promise<void>}
     */
  async destroyConversations (e) {
    let ats = e.message.filter(m => m.type === 'at')
    if (ats.length === 0) {
      let c = await redis.get(`CHATGPT:CONVERSATIONS:${e.sender.user_id}`)
      if (!c) {
        await this.reply('当前没有开启对话', true)
      } else {
        await redis.del(`CHATGPT:CONVERSATIONS:${e.sender.user_id}`)
        await this.reply('已结束当前对话，请使用#chatgpt进行聊天以开启新的对话', true)
      }
    } else {
      let at = ats[0]
      let qq = at.qq
      let atUser = _.trimStart(at.text, '@')
      let c = await redis.get(`CHATGPT:CONVERSATIONS:${qq}`)
      if (!c) {
        await this.reply(`当前${atUser}没有开启对话`, true)
      } else {
        await redis.del(`CHATGPT:CONVERSATIONS:${qq}`)
        await this.reply(`已结束${atUser}的对话，他仍可以使用#chatgpt进行聊天以开启新的对话`, true)
      }
    }
  }

  async help (e) {
    let response = 'chatgpt-plugin使用帮助文字版\n' +
            '#chatgpt+聊天内容: 发起对话与AI进行聊天\n' +
            'chatgpt对话列表: 查看当前发起的对话\n' +
            '#结束对话: 结束自己或@用户的对话\n' +
            'chatgpt帮助: 查看本帮助\n' + 
            '源代码：https://github.com/ikechan8370/chatgpt-plugin'
        await this.reply(response)
  }

  /**
   * #chatgpt
   * @param e oicq传递的事件参数e
   */
  async chatgpt (e) {
    let question = _.trimStart(e.msg, '#chatgpt')
    question = question.trimStart()
    try {
      await this.chatGPTApi.ensureAuth()
    } catch (e) {
      logger.error(e)
      await this.reply(`OpenAI认证失败，请检查Token：${e}`, true)
    }
    let c
    logger.info(`chatgpt question: ${question}`)
    let previousConversation = await redis.get(`CHATGPT:CONVERSATIONS:${e.sender.user_id}`)
    if (!previousConversation) {
      c = this.chatGPTApi.getConversation()
      let ctime = new Date()
      previousConversation = {
        sender: e.sender,
        conversation: c,
        ctime,
        utime: ctime,
        num: 0
      }
      await redis.set(`CHATGPT:CONVERSATIONS:${e.sender.user_id}`, JSON.stringify(previousConversation), { EX: CONVERSATION_PRESERVE_TIME })
    } else {
      previousConversation = JSON.parse(previousConversation)
      c = this.chatGPTApi.getConversation({
          conversationId: previousConversation.conversation.conversationId,
          parentMessageId: previousConversation.conversation.parentMessageId
      })
    }
    try {
      // console.log({ c })
      const response = await c.sendMessage(question)
      logger.info(response)
      // 更新redis中的conversation对象，因为send后c已经被自动更新了
      await redis.set(`CHATGPT:CONVERSATIONS:${e.sender.user_id}`, JSON.stringify({
        sender: e.sender,
        conversation: c,
        ctime: previousConversation.ctime,
        utime: new Date(),
        num: previousConversation.num + 1
      }), { EX: CONVERSATION_PRESERVE_TIME })
      /** 最后回复消息 */
      await this.reply(`${response}`, e.isGroup)
    } catch (e) {
      logger.error(e)
      await this.reply(`与OpenAI通信异常，请稍后重试：${e}`, e.isGroup, { recallMsg: e.isGroup ? 10 : 0 })
    }
  }
}