nodejs文心一言API接入
需求
在nodejs里面接入文心一言API，官方调用步骤API介绍 - 千帆大模型平台 | 百度智能云文档
大致流程
创建应用——>API授权——>获取访问凭证——>调用接口
创建应用
注册账号创建应用
首先注册百度云智能账号，登录进入百度智能云千帆控制台 ，然后进入控制台创建应用 。
创建应用，我乱输入的。。。
获取AppID、API Key、Secret Key
API授权
应用创建成功后，千帆平台默认为应用开通所有API调用权限，无需申请授权。当然你也可以停止某些接口
怎么调用API
官方API文档API列表 - 千帆大模型平台 | 百度智能云文档，选择你想要调用的API，我这里以
ERNIE-Bot模型为例
选择合适的接口
官方文档的模型广场有，各模型的描述，然后选择合适的模型调用即可。注意这里使用付费模型的话需要保证账户有余额。刚注册官方好像会赠送优惠券，可以直接抵扣。后续需要自己充值。
在线测试
千帆平台提供在线测试功能，可以在线检验平台创建模型的精确性和反应效果。以下文档有详细介绍体验中心 - 千帆大模型平台 | 百度智能云文档，在体验中心选择你想要测试的模型即可
模型介绍
点击创建chat后可以看到调用接口的方式
在线调试平台示例代码
点击API在线调试平台-示例代码，选择你自己使用的语言调试即可
代码
以下封装的函数测试了3个模型Qianfan-Chinese-Llama-2-13B 模型 、 CodeLlama-7b-Instruct模型、ERNIE-Bot 模型，当然首先要获取access_token，具体获取方法我就不介绍了，参考官方API获取即可。
// 访问模型服务
const axios = require("axios");
const app_id = process.env.WX_APPID //文心一言appid,填你自己的appid
const secret_key = process.env.WX_SECRETKEY //文心一言secretKey,填你自己的secret_key
const api_key = process.env.WX_APIKEY //文心一言apiKey,填你自己的api_key
// 获取 access_token
function getAccessToken() {
return new Promise((resolve, reject) => {
axios({
method: 'POST',
url: 'https://aip.baidubce.com/oauth/2.0/token',
headers: {
'Content-Type': 'application/json',
},
params: {
'grant_type': 'client_credentials',
'client_id': api_key,
'client_secret': secret_key,
},
})
.then((response) => {
// 在 Promise 成功时返回 access_token
resolve(response.data.access_token);
})
.catch((error) => {
// 在 Promise 失败时返回错误
reject(error);
});
});
}
//文星一言 Qianfan-Chinese-Llama-2-13B 模型 免费
function llamaModel(contentValue,access_token){
return new Promise(async (resolve, reject) => {
try {
const response = await axios({
method: 'POST',
url: 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/qianfan_chinese_llama_2_13b',
headers: {
'Content-Type': 'application/json',
},
data: {
messages: [{
role: 'user',
content: contentValue,
}],
},
params: {
access_token: access_token,
},
});
console.log('啦啦啦', response.data.result);
resolve(response.data.result);
} catch (error) {
console.error('Error:', error.message);
reject(error);
}
});
}
//文心一言 CodeLlama-7b-Instruct 模型免费
function instructModel(contentValue,access_token){
return new Promise(async (resolve, reject) => {
try {
const res = await axios({
method: 'POST',
url: 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/completions/codellama_7b_instruct',
headers: {
'Content-Type': 'application/json',
},
params: {
access_token: access_token,
},
data: {
prompt: contentValue,
// Add other parameters as needed
},
});
console.log('啦啦啦', res.data.result);
resolve(res.data.result);
} catch (error) {
console.error('Error:', error.message);
reject(error);
}
});
}
//文星一言 ERNIE-Bot 模型 付费
function erniebotModel(contentValue,access_token){
return new Promise(async (resolve, reject) => {
try {
const res = await axios({
method: 'POST',
url: 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions',
headers: {
'Content-Type': 'application/json',
},
params: {
access_token: access_token,
},
data: {
messages: [{
role: 'user',
content: contentValue,
}],
},
});
console.log('啦啦啦', res.data.result);
resolve(res.data.result);
} catch (error) {
console.error('Error:', error.message);
reject(error);
}
});
}
// 访问讯飞星火 API 的方法
async function getWXApi(contentValue) {
try {
const access_token = await getAccessToken();
console.log('拿到token', access_token);
//使用 CodeLlama-7b-Instruct 模型
const result = await instructModel(contentValue, access_token);
//使用 Qianfan-Chinese-Llama-2-13B 模型
// const result = await llamaModel(contentValue, access_token);
//使用 ERNIE-Bot 模型 模型
// const result = await erniebotModel(contentValue, access_token);
} catch (error) {
console.error('Main Error:', error.message);
}
}
// 导出 initUrl 函数
module.exports = getWXApi;
记录一下。。。。。。。。