手把手教你搭建微信聊天机器人系列（五）：对接文心一言4.0

        之前的章节算是已经比较完整的实现了聊天功能，本来打算直接结束的。前几天听说文心一言4.0开放了公测，所以也去申请了一下，2天后就收到了通过短信。通过以后，可以在之前申请的应用列表中，点击“详情”，在“服务配置”的最后一行可以看到ERNIE-Bot 4.0

目前一直到10月底，4.0接口是免费的，大家抓紧时间体验。

        我们看一下API文档，发现调用方式和之前的类似，我们可以调整一下代码底层，将其配置化，以方便我们在不同的模型间进行切换。

config/config.default.js调整代码如下：

config.ernie = { client_id: '填入您的API Key', //API Key client_secret: '填入您的Secret Key',//Secret Key access_token: '', //先置空，后续由程序填充 expire_day: 30, //access_token过期时长（天） gpt_model: 'ERNIE-Bot-4', //此处值对应gpt_list中的type gpt_list: [{ type: 'ERNIE-Bot-4', //文心一言4.0 url: 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions_pro' }, { type: 'ERNIE-Bot', //文心一言 url: 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions' }, { type: 'ERNIE-Bot-turbo', //文心一言轻量版 url: 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/eb-instant' }] }

app/service/ernie.js全部代码如下：

const { Service } = require('egg'); const moment = require('moment'); 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) { console.log('access_token', ctx.app.config.ernie.access_token) return res.data.access_token; } } catch (error) { console.log(error) } return null; } async sendMsg(msg) { console.log('===================ErnieService sendMsg====================='); console.log(JSON.stringify(msg)); let ctx = this.ctx; let gpt_url; for (let i of ctx.app.config.ernie.gpt_list) { if (i.type === ctx.app.config.ernie.gpt_model) { gpt_url = i.url; break; } } if (!gpt_url) { console.log('您设置的gpt_model值不存在'); return null; } try { const res = await ctx.curl( `${gpt_url}?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) { if (res.data.error_code == '111') { //token过期 await ctx.service.ernie.checkAccessToken(true); } return res.data; } return null; } catch (error) { console.log(error) return null; } } async checkAccessToken(forceReflesh) { //forceReflesh强制刷新 console.log('===================ErnieService checkAccessToken====================='); let ctx = this.ctx; const query = {}; const accessToken = await ctx.model.AccessToken.findOne(query); if (accessToken) { console.log('accessToken', accessToken.access_token, accessToken.updated_at); if (moment(new Date()).diff(moment(accessToken.updated_at), 'days') >= ctx.app.config.ernie.expire_day - 1 || forceReflesh) { //提前一天刷新 console.log('accessToken已失效，重新获取中') const accessTokenStr = await ctx.service.ernie.getAccessToken(); if (accessTokenStr) { await accessToken.update({ accessToken: accessTokenStr, updated_at: moment().format('YYYY-MM-DD HH:mm:ss') }); ctx.app.config.ernie.access_token = accessTokenStr; console.log('accessToken更新成功'); } else { console.log('accessToken获取失败'); } } else { ctx.app.config.ernie.access_token = accessToken.access_token; console.log('accessToken在有效期内'); } } else { console.log('accessToken不存在'); const accessTokenStr = await ctx.service.ernie.getAccessToken(); if (accessTokenStr) { const queryData = await ctx.model.AccessToken.create({ access_token: accessTokenStr }); ctx.app.config.ernie.access_token = accessTokenStr; console.log('accessToken更新成功'); } else { console.log('accessToken获取失败'); } } } } module.exports = ErnieService;

最近发现access_token有时候莫名其妙的就过期，所以我们加入了强制刷新的逻辑。我们在app.js中也可以加入强制刷新，让服务一启动就刷新access_token。

app.js部分代码：

app.ready(async () => { console.log("==app ready=="); let ctx = app.createAnonymousContext(); await ctx.service.ernie.checkAccessToken(true); //强制刷新AccessToken await ctx.service.wechat.startBot(); //初始化BOT })

至此，我们已经可以在三种不同的模型直接快速切换，仅需要调整config.ernie.gpt_model配置项。

 ​ 本章完整代码在这里下载。运行前请先申请文心一言4.0测试资格，配置好config/config.default.js里面config.ernie下的client_id和client_secret配置项。