居然只用 20 多行代码就可以实现微信聊天机器人

查看 73|回复 5
作者:461229187   
用了一年多 chatGPT 和白嫖了 vscode 的 copilot ,现在感觉已经离不开了,所以也想把这么好用的东西推荐给自己的家人和朋友。但是由于网络的限制,想让他们体验到聊天机器人的神奇是很困难的。当然现在百度阿里等巨头都已经发布了他们的产品,但需要下载相应的 app ,还要登录等一些列操作,稍微繁琐一点都会让他们失去兴趣。所以我打算开发一个微信聊天机器人,他们只需要加微信好友即可体验到。
在编写代码前,先要对以下几个东西做一些了解,简单且好用。
GPT API
在国内访问 chatGPT 是比较麻烦的,搞不好还容易封号。所以这里我推荐一下国内可用的一个接口,GPT-API-free,看名字就知道这是个免费的,虽然免费,但是有调用频率的限制,60 请求/小时/IP&Key ,可以通过 github 领取。理论上自己用的话免费的就可以,如果想调用 GPT4 ,每天也可以使用 3 次。
如果你访问频率高,可以购买付费服务,如果 gpt3.5 的话,我预测 30 块用一年应该没什么问题,GPT4 的话就比较贵了。
  • Github: https://github.com/chatanywhere/GPT_API_free
  • 接口文档: https://chatanywhere.apifox.cn/

    wechaty

    Wechaty 是一款用于构建聊天机器人的开源程序,他运行于 node(v16+) 环境,它提供了几乎微信所有操作接口,例如登录、接发消息、好友操作、备注、群组等等,这样我们就可以通过 Wechaty + GPT API 的组合实现微信聊天机器人了。
    对了,你最好再申请个微信号,不然封号就得不偿失了。我已经用了半个多月,没被封号。
    需要注意的是还需要了解一个库配合 wechaty 实现在命令行中扫码登录微信,qrcode-terminal ,他用法非常简单,无需去看文档。

  • Wechaty: https://github.com/wechaty/wechaty
  • qrcode-terminal: https://github.com/gtanner/qrcode-terminal

    代码实现
    由于功能很简单,我也懒得去搞部署之类的东西,所以我直接在服务器上写的代码...
    安装依赖
    首先创建一个 node 项目,npm init
    安装依赖:
    npm install wechaty qrcode-terminal axios
    代码不多,创建一个 index.js 即可,首先引入即将使用到的方法,并创建聊天机器人示例:
    import axios from 'axios';
    import { ScanStatus, WechatyBuilder, log } from 'wechaty'
    import qrcodeTerminal from 'qrcode-terminal'
    const bot = WechatyBuilder.build({
      name: 'codexu-chat-bot', // 名字随意
    })
    const sk = '' // 顺便把接口 sk 定义上
    bot.start();

    如果你将代码放在 git 仓库,建议创建环境变量等方式去获取 sk ,保证安全,不要泄漏。

    扫码登录
    随后先实现扫码登录功能,bot 实例提供了很多事件监听,这里我们监听 scan 事件,它提供了两个参数,二维码和状态,这里我们需要对二维码做一下转换,并通过 qrcodeTerminal 将二维码展示在命令行中。
    function onScan(qrcode, status) {
      if (status === ScanStatus.Waiting || status === ScanStatus.Timeout) {
        const qrcodeImageUrl = [
          'https://wechaty.js.org/qrcode/',
          encodeURIComponent(qrcode),
        ].join('')
        log.info('StarterBot', 'onScan: %s(%s) - %s', ScanStatus[status], status, qrcodeImageUrl)
        qrcodeTerminal.generate(qrcode, { small: true })  // show qrcode on console
      } else {
        log.info('StarterBot', 'onScan: %s(%s)', ScanStatus[status], status)
      }
    }
    bot.on('scan', onScan)

    与登录相关的,bot.on 还可以监听到 login 和 logout 事件,请自行添加。

    处理聊天信息
    然后就是关键的处理聊天消息,这里我使用 axios 去调用接口,需要在 header 中传入 sk ,并按照接口文档传递模型和消息记录:
    async function onMessage(msg) {
      const text = msg.text()
      let cacheMessagesItem = cacheMessages.find(item => item.name === talker.name);
      const message = {
        role: "user",
        content: text
      }
      const data = {
        model: 'gpt-3.5-turbo-1106',
        messages: [message]
      }
      const result = await axios.post(`https://api.chatanywhere.com.cn/v1/chat/completions`, data, {
        headers: {
          Authorization: `Bearer ${sk}`,
          'Content-Type': 'application/json'
        }
      })
      const resultContent = result.data.choices[0].message.content;
      await msg.say(resultContent)
    }
    bot.on('message', onMessage)
    然后通过 node index.js 运行即可,如果在服务器上推荐使用 pm2 去管理程序,不然报错可能就停止了。
    至此就已经实现了微信聊天机器人的基本功能,快去体验一下吧。
    优化
    通过 20 多行代码就实现了聊天机器人是不是很简单,但是它现在还傻傻的不太好用,我们可以对他进行一些优化。
    机器人人设
    当我把它介绍给亲人朋友时,神奇的是他们都会问这么几个问题:
  • 你是谁?
  • 你认识李旭吗?(我本人)

    但是 chatGPT 只能傻傻的回答他是什么语言大模型之类的,毫无新意,这样的话我们可以为他建立人设。


    可以看到调用 GPT 接口时,messages 参数是一个数组,它是连续聊天的关键,也就是说你把聊天记录都传进来,这样 GPT 就可以在回答时参考上下文来回答。
    所以我们只需要在每次调用接口时,告诉 GPT 它要去扮演李旭的机器人这个角色,并告诉他我是谁就可以了。
    const message = {
      role: "user",
      content: `本次对话你扮演的是李旭的机器人的角色,是微信的聊天助手,你的微信名是李旭的机器人,李旭是一个非常帅的人并且有才华的人,今年 18 岁。`
    }
    同理,如果你想实现连续聊天,把聊天记录传到 messages 中即可,当然调用接口是按照字数收费,并且数据有最大限制,所以我们尽量传 10 条以内的聊天记录也就够了,这里大家自行实现列队功能吧。
    自动添加好友
    当我把机器人名片推给别人时,还需要手动去微信添加好友,这就很麻烦了,尤其是朋友再推给其他人时,还需要让我去同意添加,这样比较麻烦。
    幸运的是 wechaty 还提供了对添加好友的操作。
    bot.on('friendship', async friendship => {
      try {
        switch (friendship.type()) {
          case 2:
            await friendship.accept()
            break
        }
      } catch (e) {
        console.error(e)
      }
    })
    这样我们就实现了自动添加好友的功能。
    其他
    我目前可以想到的:
  • 年龄大的人比较爱发语音,并且眼神不好,语音的体验方式更适合,可以借助转录将语音转成文字,再将结果文字转换成语音发送给使用者。
  • GPT4 接口较贵,所以默认使用 GPT3 即可,加入判断,在聊天文字前输入 GPT4 ,再去调用 GPT4 模型,这样可以根据需要使用不同模型。
  • 可以做一些画图的功能或识别图像的功能。

    总结
    当我产生了做这个机器人的想法到实现,我都没有想到居然如此简单,看似复杂的功能,已经被 wechaty 做成了非常简单的操作。
    另外其他人在使用的时候,你是可以看到他们的聊天记录的,我建议你应该告诉使用者你可以看到他们的消息。
    我没跟我妈说能看到他的问题,我看到她在问失眠怎么办?哎...

    const, 机器人, 聊天, wechaty

  • TenProX   
    支持大佬的作品!膜拜!
    sekiseki   
    wechaty 两年前已经看过了,但是要扫码。还有更方便的。
    winglight2016   
    感谢大佬无私分享!
    我之前也考虑做个类似的小工具,但是发现 github 上的几个微信库都非常陈旧,包括这个 wechaty ,lz 确定能兼容最新版的微信吗?
    461229187
    OP
      
    @sekiseki 还有高手?叫什么我去研究一下
    461229187
    OP
      
    @winglight2016 没深入研究,反正目前我这点功能足够用了,运行正常
    您需要登录后才可以回帖 登录 | 立即注册

    返回顶部