使用 wxauto 结合 AI 大模型实现微信聊天机器人

发送图片/视频/文件消息 SendFiles 参数说明：

from wxauto import WeChat
wx = WeChat()
content = 'hello world'
who = '文件传输助手'
file = [r'D:\软件\图片\荒.png', r'D:\C 语言学习资料\高质量的 C-C++编程.pdf']
wx.SendFiles(filepath=file, who=who)

聊天窗口消息获取 默认为当前聊天窗口

from wxauto import WeChat
wx = WeChat()
# 获取当前聊天窗口消息
msgs = wx.GetAllMessage()
# 输出消息内容
for msg in msgs:
    if msg.type == 'sys':
        print(f'【系统消息】{msg.content}')
    elif msg.type == 'friend':
        sender = msg.sender_remark # 获取备注名
        print(f'{sender.rjust(20)}：{msg.content}')
    elif msg.type == 'self':
        print(f'{msg.sender.ljust(20)}：{msg.content}')
    elif msg.type == 'time':
        print(f'\n【时间消息】{msg.time}')
    elif msg.type == 'recall':
        print(f'【撤回消息】{msg.content}')

另外 LoadMoreMessage 方法用于加载更多历史消息，配合 GetAllMessage 方法使用，实现获取更多历史消息。 注意：LoadMoreMessage 方法加载更多历史消息时，需要保证当前聊天窗口有历史消息，否则没有效果，即触发一次'查看更多消息'。

from wxauto import WeChat
wx = WeChat()
# 加载更多历史消息
wx.LoadMoreMessage()
# 获取当前聊天窗口消息
msgs = wx.GetAllMessage()
# 消息处理逻辑代码。。。

微信添加好友 AddNewFriend 方法用于发起好友申请。 **注意：**微信有一定的限制，如果频繁添加好友，可能会被限制添加好友的权限，请谨慎使用，切勿滥用！！！

from wxauto import WeChat
wx = WeChat()
keywords = 's15576806087' # 对方的微信号、手机号、QQ号
addmsg = '你好，我是 xxxx' # 添加好友的消息
remark = '备注名字' # 备注名，没有则不用设置
tags = ['朋友', '同事'] # 标签列表
# 发起好友申请
wx.AddNewFriend(keywords, addmsg=addmsg, remark=remark, tags=tags)

获取好友信息

from wxauto import WeChat
wx = WeChat()
friend_infos = wx.GetAllFriends() # 获取好友信息
print(friend_infos)

3. wxauto 的消息对象

这个很重要，下面结合大模型时会用到以下的消息对象。

好友消息

支持属性：

msgs = wx.GetAllMessage()
for msg in msgs:
    if msg.type == 'friend':
        # 消息类型
        sender = msg.sender_remark # 获取备注名
        print(f'{sender}：{msg.content}')

自己的消息

支持属性：

msgs = wx.GetAllMessage()
for msg in msgs:
    if msg.type == 'self':
        # 消息类型
        print(f'{msg.sender}：{msg.content}')

wxauto 结合大模型实现简单的聊天机器人

这里选用的是百度的千帆大模型，首先登录进去之后点击模型广场，选择免费模型即可。

选择好模型之后，点进去，点击开通付费（免费的，不要钱，放心点击），提交订单就开通成功啦。

返回到主页，点击应用接入，记住这里的 API Key 和 Secret Key，点击创建应用。

填写好应用名称和应用描述，点击确定。

返回主页，点击模型广场，点击你之前选择的模型，点击 API 文档。

往下翻找到对应的请求示例的 Python 代码，复制那段代码。

复制好代码后，将对应的 API Key 和 Secret Key 给添加上去。

运行一下代码可以看到，result 就是大模型根据我们的问题给出的结果，现在我们只需要将 content 改成微信中好友发送过来的消息作为问题给大模型，然后将大模型给出的结果中的 result 提取出来作为内容发送给好友，这样，一个简单的微信聊天机器人就完成了。

完整代码

import requests
import json
from wxauto import WeChat

def get_access_token():
    """
    使用 API Key，Secret Key 获取 access_token，替换下列示例中的应用 API Key、应用 Secret Key
    """
    url = "https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=[应用 API Key]&client_secret=[应用 Secret Key]"
    payload = json.dumps("")
    headers = {'Content-Type':'application/json','Accept':'application/json'}
    response = requests.request("POST", url, headers=headers, data=payload)
    return response.json().get("access_token")

def main(wx1, msg1):
    url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie-speed-128k?access_token=" + get_access_token()
    payload = json.dumps({"messages":[{"role":"user","content": msg1 }]})
    headers = {'Content-Type':'application/json'}
    response = requests.request("POST", url, headers=headers, data=payload)
    json_result = json.loads(response.text)
    print(json_result)
    # print(response.text)
    wx.SendMsg(msg=json_result['result']+"--此内容为 AI 生成", who="你要发送的人")

if __name__ == '__main__':
    wx = WeChat()
    while True:
        msgs = wx.GetAllMessage()
        if msgs:
            if msgs[-1].type == "friend":
                main(wx, msgs[-1].content)