4

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

 7 months ago
source link: https://www.v2ex.com/t/1013267
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.
neoserver,ios ssh client

V2EX  ›  程序员

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

  461229187 · 9 小时 7 分钟前 · 5422 次点击

用了一年多 chatGPT 和白嫖了 vscode 的 copilot ,现在感觉已经离不开了,所以也想把这么好用的东西推荐给自己的家人和朋友。但是由于网络的限制,想让他们体验到聊天机器人的神奇是很困难的。当然现在百度阿里等巨头都已经发布了他们的产品,但需要下载相应的 app ,还要登录等一些列操作,稍微繁琐一点都会让他们失去兴趣。所以我打算开发一个微信聊天机器人,他们只需要加微信好友即可体验到。

在编写代码前,先要对以下几个东西做一些了解,简单且好用。

GPT API

在国内访问 chatGPT 是比较麻烦的,搞不好还容易封号。所以这里我推荐一下国内可用的一个接口,GPT-API-free,看名字就知道这是个免费的,虽然免费,但是有调用频率的限制,60 请求/小时/IP&Key ,可以通过 github 领取。理论上自己用的话免费的就可以,如果想调用 GPT4 ,每天也可以使用 3 次。

如果你访问频率高,可以购买付费服务,如果 gpt3.5 的话,我预测 30 块用一年应该没什么问题,GPT4 的话就比较贵了。

wechaty

42b9dbbb34054c76b24a2e00a818f8e0~tplv-k3u1fbpfcp-image.image#?w=1920&h=540&s=7006&e=svg&a=1&b=45a83a

Wechaty 是一款用于构建聊天机器人的开源程序,他运行于 node(v16+) 环境,它提供了几乎微信所有操作接口,例如登录、接发消息、好友操作、备注、群组等等,这样我们就可以通过 Wechaty + GPT API 的组合实现微信聊天机器人了。

对了,你最好再申请个微信号,不然封号就得不偿失了。我已经用了半个多月,没被封号。

需要注意的是还需要了解一个库配合 wechaty 实现在命令行中扫码登录微信,qrcode-terminal ,他用法非常简单,无需去看文档。

a6bc398b115f40038e8169bedadc928a~tplv-k3u1fbpfcp-jj-mark:0:0:0:0:q75.image#?w=844&h=701&s=46426&e=png&a=1&b=000000

由于功能很简单,我也懒得去搞部署之类的东西,所以我直接在服务器上写的代码...

首先创建一个 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 还可以监听到 loginlogout 事件,请自行添加。

处理聊天信息

然后就是关键的处理聊天消息,这里我使用 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 只能傻傻的回答他是什么语言大模型之类的,毫无新意,这样的话我们可以为他建立人设。

image.png
image.png

可以看到调用 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 做成了非常简单的操作。

另外其他人在使用的时候,你是可以看到他们的聊天记录的,我建议你应该告诉使用者你可以看到他们的消息。

我没跟我妈说能看到他的问题,我看到她在问失眠怎么办?哎...

第 1 条附言  ·  3 小时 26 分钟前

记录一下,备用机几天没充电了,发现服务器掉了,报错
Error: 状态同步超过 297.966s 未响应,5s 后尝试重启
所以这个微信机器人还得长期手机端在线才行...

About Joyk


Aggregate valuable and interesting links.
Joyk means Joy of geeK