手把手教你搭建微信聊天机器人系列（三）：搭建对接文心一言（ERNIE-Bot大模型）接口的微信聊天机器人-Toy模板网

这篇具有很好参考价值的文章主要介绍了手把手教你搭建微信聊天机器人系列（三）：搭建对接文心一言（ERNIE-Bot大模型）接口的微信聊天机器人。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

上一章介绍了eggjs+wechaty搭建一个简单的微信聊天机器人，只是实现了简单的回复，这一章我们将对接ERNIE-Bot接口，实现智能回复。

access_token的请求

config.default.js配置

我们在上一章代码基础上进行调整，首先我们要加入access_token的请求逻辑。第一章我们说过，请求需要使用应用对应的API Key，Secret Key作为入参。eggjs规定了配置的写法，我们可以将其配置在config/config.default.js里面。

自建微信聊天机器人,chatgpt,文心一言,机器人,node.js,chatgpt

//配置ernie
	config.ernie = {
		client_id: '填入您的API Key', //API Key
		client_secret: '填入您的Secret Key',//Secret Key
		access_token: '',//先置空，后续由程序填充
		expire_day: 30 //access_token过期时长（天）
	}

ernie.js定义getAccessToken

我们在app/service目录增加ernie.js。

自建微信聊天机器人,chatgpt,文心一言,机器人,node.js,chatgpt

ernie.js文件专门用来和ERNIE-Bot接口进行交互，代码如下：

const {
	Service
} = require('egg');


class ErnieService extends Service {
	async getAccessToken() {
		console.log('===================ErnieService getAccessToken=====================');
		let ctx = this.ctx;

		try {
			const res = await ctx.curl(
				`https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=${ctx.app.config.ernie.client_id}&client_secret=${ctx.app.config.ernie.client_secret}`, {
					method: 'GET',
					rejectUnauthorized: false,
					data: {},
					headers: {},
					timeout: 30000,
					contentType: 'json',
					dataType: 'json',
				})
			console.log(res)

			if (res.data.access_token) {
				ctx.app.config.ernie.access_token = res.data.access_token;
				console.log('access_token', ctx.app.config.ernie.access_token)
			}

		} catch (error) {
			console.log(error)
		}
	}
}

module.exports = ErnieService;

我们按eggjs的service的写法，定义了一个方法，用来请求access_token，但是这样写并不能执行，我们需要有个逻辑，在应用启动完毕后会调用这个方法。

app.js执行getAccessToken

我们在根目录增加app.js。

自建微信聊天机器人,chatgpt,文心一言,机器人,node.js,chatgpt

app.js代码如下：

module.exports = app => {

	app.beforeStart(async () => {
		// 应用会等待这个函数执行完成才启动
		console.log("==app beforeStart==");
	});

	app.ready(async () => {
		console.log("==app ready==");
		let ctx = app.createAnonymousContext();
		ctx.service.ernie.getAccessToken(); //请求AccessToken
	})

	app.beforeClose(async () => {
		console.log("==app beforeClose==");
	})
};

我们执行命令npm run dev启动项目，看到最终会打印出access_token，有了access_token我们继续开发接口请求函数sendMsg。

对话请求

ernie.js定义sendMsg

sendMsg函数用于调用ERNIE-Bot会话接口，传入对话内容并返回对话结果。在class ErnieService中加入以下代码：

	async sendMsg(msg) {
		console.log('===================ErnieService sendMsg=====================');
		let ctx = this.ctx;
		try {
			const res = await ctx.curl(
				`https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions?access_token=${ctx.app.config.ernie.access_token}`, {
					method: 'POST',
					rejectUnauthorized: false,
					data: {
						"messages": msg
					},
					timeout: 30000,
					contentType: 'json',
					dataType: 'json',
				})
			console.log(res)
	
			if (res.data) {
				return res.data;
			}
			return null;
		} catch (error) {
			console.log(error)
			return null;
		}
	}

wechat.js

我们在wechat.js中，将原先的逻辑调整一下，主要是改两个地方，

1、我们在之前的代码中经常有看到ctx这个变量，可以通过他获取配置文件的参数值。但Service文件中的ctx的获取只能通过app.js调用Service中函数的方式来获得。所以我们将机器人初始化的操作定义在startBot()方法中，由app.js来执行这个方法，就可以接收到ctx的值。

2、原先回复是直接扔回用户发的信息的，现在将其对接到ernie.js中的sendMsg方法。

wechat.js全部代码如下：

const {
	Service
} = require('egg');


const {
	WechatyBuilder,
	ScanStatus
} = require("wechaty");
const qrcode = require("qrcode-terminal");

let ctx;
let wechaty;
let startStatus = false;

const onMessage = async (message) => {
	console.log(`收到消息: ${message}`);
	if (message.type() === wechaty.Message.Type.Text) {
		const userMsg = await message.text();

		try {
			let msgRecord = [{
				"role": "user",
				"content": userMsg
			}];
			let res = await ctx.service.ernie.sendMsg(msgRecord);
			if (res) {
				if (res.error_code) {
					message.say(JSON.stringify(res));
					console.log(`报错: ${JSON.stringify(res)}`);
				} else {
					if (res.result) {
						message.say(res.result);
						console.log(`回复: ${res.result}`);
					}
				}
			}
		} catch (error) {
			console.log(error);
			message.say(JSON.stringify(error));
		}
	}
};

const onLogout = (user) => {
	console.log(`用户 ${user} 退出成功`);
};
const onLogin = async (user) => {
	console.log(`用户 ${user} 登录成功`);
};
const onError = console.error;
const onScan = (code, status) => {
	// status: 2代表链接等待调用，3代表链接已打开，这个链接实际上是提供一个登录的二维码供扫描
	if (status === ScanStatus.Waiting) {
		// status: 2代表等待，3代表扫码完成
		qrcode.generate(code, {
			small: true
		}, console.log)
	}
};

class WechatService extends Service {
	async startBot() {
		console.log('===================WechatService startBot=====================');
		ctx = this.ctx;
		if (startStatus && wechaty) {
			if (wechaty.isLoggedIn) {
				await wechaty.logout();
			}
			await wechaty.stop();
			startStatus = false;
			wechaty = null;
		}
		wechaty = await WechatyBuilder.build();
		wechaty
			.on("scan", onScan)
			.on("login", onLogin)
			.on("logout", onLogout)
			.on("error", onError)
			.on("message", onMessage);
		await wechaty.start();
		startStatus = true;
	}
}

module.exports = WechatService;

app.js执行startBot

在AccessToken请求后加入初始化BOT

app.ready(async () => {
		console.log("==app ready==");
		let ctx = app.createAnonymousContext();
		await ctx.service.ernie.getAccessToken(); //请求AccessToken
		await ctx.service.wechat.startBot(); //初始化BOT
	})

自建微信聊天机器人,chatgpt,文心一言,机器人,node.js,chatgpt