结合 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)

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}')

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

本示例选用百度千帆大模型。登录控制台进入模型广场，选择一个免费模型进行开通。 开通后，在应用接入页面记录 API Key 和 Secret Key，并创建应用。 返回主页，点击模型广场，选择已选模型，查看 API 文档。 参考文档中的 Python 请求示例，将 API Key 和 Secret Key 填入代码。 运行测试代码，确认 result 字段包含大模型生成的回复。 将微信好友消息作为 content 传入大模型接口，提取返回的 result 并通过 wxauto 发送回好友，即可完成聊天机器人闭环。

完整代码

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)