langchain搭建微信机器人

环境准备

准备OpenAI账号

项目默认使用OpenAI接口，需前往OpenAI注册页面创建账号，创建完账号则前往API管理页面创建一个 API Key 并保存下来，接口需要海外网络访问及绑定信用卡支付。

将API Key添加到环境变量中

export OPENAI_API_KEY="sk-xxxxxxxxxxxxxxxxxxxxxx"

安装Python 代码通过Python3.10进行开发测试，需要安装python 3.10，下载后点击安装傻瓜式下一步。

项目搭建

创建项目demo-bot

mkdir demo-bot
cd demo-bot

安装依赖包

pip install langchain langchain_openai loguru qrcode

itchat.zip

下载解压到当前目录

该文件是基于[ItChat]https://github.com/littlecodersh/ItChat项目二次封装的微信机器人框架

创建main文件，开始写代码

此时的目录结构应类似于

.
├── itchat
├── itchat.zip
└── main.py

写代码

将下列代码复制到main.py文件中即可

import itchat

from itchat.content import INCOME_MSG

from loguru import logger
from qrcode.main import QRCode

from langchain_openai import ChatOpenAI

from langchain_core.messages import HumanMessage

chat = ChatOpenAI(model="gpt-3.5-turbo-1106", temperature=0.2)


@itchat.msg_register(INCOME_MSG)
def handle_single(msg):
    """
    监听私聊消息
    :param msg:
    :return:
    """
    logger.info(f"Human Message:{msg.Type} {msg.content}")
    if msg.Type == "Text":
        res = chat.invoke([HumanMessage(content=msg.content)])
        logger.info(f"AI message: {msg.content}")
        itchat.send(res.content, toUserName=msg["FromUserName"])


def qrcode_callback(uuid, status, qr_code):
    """
    二维码回调
    :param uuid:
    :param status:
    :param qr_code:
    :return:
    """
    url = f"https://login.weixin.qq.com/l/{uuid}"

    if status == "0":
        logger.info("QR code scanned")
        qr = QRCode(border=1)
        qr.add_data(url)
        qr.make(fit=True)
        qr.print_ascii(invert=True)
    elif status == "200":
        logger.info("QR code confirmed")
    elif status == "201":
        logger.info("QR code scanned, waiting for confirmation")
    else:
        logger.info(f"QR code status: {status}")


def hello_bot():
    """
    机器人启动
    :return:
    """
    itchat.auto_login(
        enableCmdQR=2,
        qrCallback=qrcode_callback,
    )
    itchat.run()


if __name__ == '__main__':
    hello_bot()

运行代码

python main.py

代码运行成功之后会在日志中打印一个登录二维码，扫描登录，然后就可以跟他愉快的聊天啦！

一个最简单的微信AI机器人到此就搭建结束啦！

写在最后

想尝试和了解更多功能？！

欢迎围观我的个人项目 https://github.com/ma-pony/langchain-wechat