调用大模型实现微信自动回复新年祝福类信息

调用大模型实现微信自动回复新年祝福类信息

一、实现功能

1. 通过uiautomation实现自动读取微信（电脑版）未读消息
2. 根据关键字判断是否是新春祝福类信息，如果是，调用智谱AI模型，根据接收到的消息，利用模型生成回复内容
3. 自动将回复内容发送出去 

二、实现效果

 三、实现代码

from uiautomation import WindowControl,MenuControl
from zhipuai import ZhipuAI
# 绑定微信主窗口
weixin = WindowControl(Name='微信',)
print('微信主窗口信息：---',weixin,'---')
# 切换窗口
weixin.SwitchToThisWindow()
# 寻找聊天框绑定
wx_chat = weixin.ListControl(Name='会话')
print("聊天框窗口信息：--- ",wx_chat,' ---')
# 调用智谱AI大模型根据接收信息获取自动回复的内容
def get_response(last_msg):
    client = ZhipuAI(api_key="***************************")
    response = client.chat.completions.create(
        model="chatglm_std",  # 填写需要调用的模型名称，根据效果选择不同的模型
        messages=[
            {"role": "user", "content": last_msg}
        ],
    )
    return response.choices[0].message.content
# 接收并回复消息
while True:
    unreadmsg = wx_chat.TextControl(searchDepth=4)
    # 死循环，没有超时报错
    while not unreadmsg.Exists(0):
        pass
    print('查询到的未读消息:--- ',unreadmsg,' ----')
    if unreadmsg.Name:
        # 点击未读消息
        unreadmsg.Click(simulateMove=False)
        # 读取最后一条
        last_msg = weixin.ListControl(Name='消息').GetChildren()[-1].Name
        print('读取到的最后一条消息:--- ',last_msg,' ---')
        keyword = ['拜年','龙年','新年','新春','新的一年','佳节','阖家幸福','祝您','健康','快乐','如意']
        for i in keyword:
            if i in last_msg:
                # weixin.SendKeys(ar[0].replace('{br}', '{Shift}{Enter}'), waitTime=0)
                response_msg = get_response(last_msg)
                print('自动回复的消息 --- ',response_msg)
                response_msg = response_msg + '【自动回复测试】'
                weixin.SendKeys( response_msg, waitTime=0)
                weixin.SendKeys('{Enter}', waitTime=3)
                break
    #如果一直循环，这里不要break
    break