当前位置:AIGC资讯 > AIGC > 正文

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;
  

记录一下。。。。。。。。

更新时间 2023-12-31