背景介绍
在过去的几年中,人工智能技术尤其是大规模预训练模型(俗称AI大模型)取得了显著的进展。这些模型因其强大的学习能力和通用性,能够处理从自然语言处理到计算机视觉等多种任务,已被广泛应用于各个行业。
对于开发者而言,将AI大模型接入小程序能够显著提升应用的智能化水平,提高用户体验和业务价值,这次接入的是星火大模型
准备工作
1.注册星火账号
根据提示注册账号,领取免费试用流量
2.创建应用
创建完后获取appid***(重要,后面要用appid)***
--
接入AI大模型的步骤
接口鉴权
// httpUrl: "https://spark-api.xf-yun.com/v3.5/chat"
// 鉴权
getWebSocketUrl() {
console.log(this.httpUrl)
var httpUrlHost = (this.httpUrl).substring(8, 28);
var httpUrlPath = (this.httpUrl).substring(28);
console.log(httpUrlHost)
console.log(httpUrlPath)
switch (httpUrlPath) {
case "/v1.1/chat":
this.modelDomain = "general";
break;
case "/v2.1/chat":
this.modelDomain = "generalv2";
break;
case "/v3.1/chat":
this.modelDomain = "generalv3";
break;
case "/v3.5/chat":
this.modelDomain = "generalv3.5";
break;
}
console.log(this.modelDomain)
return new Promise((resolve, reject) => {
// https://spark-api.xf-yun.com/v1.1/chat V1.5 domain general
// https://spark-api.xf-yun.com/v2.1/chat V2.0 domain generalv2
var url = "wss://" + httpUrlHost + httpUrlPath;
var host = "spark-api.xf-yun.com";
var apiKeyName = "api_key";
var date = new Date().toGMTString();
var algorithm = "hmac-sha256";
var headers = "host date request-line";
var signatureOrigin = `host: ${host}\ndate: ${date}\nGET ${httpUrlPath} HTTP/1.1`;
var signatureSha = CryptoJS.HmacSHA256(signatureOrigin, this.APISecret);
var signature = CryptoJS.enc.Base64.stringify(signatureSha);
var authorizationOrigin =
`${apiKeyName}="${this.APIKey}", algorithm="${algorithm}", headers="${headers}", signature="${signature}"`;
var authorization = base64.encode(authorizationOrigin);
url = `${url}?authorization=${authorization}&date=${encodeURI(date)}&host=${host}`;
// console.log(url)
resolve(url); // 主要是返回地址
});
},
注:APIKey:填成您的实际 API 密钥; APISecret:填成您的实际 API 密钥密钥
API调用方法
async sendToSpark() {
let myUrl = await this.getWebSocketUrl();
this.tempRes = "";
this.sparkResult = "";
let realThis = this;
this.socketTask = uni.connectSocket({
//url: encodeURI(encodeURI(myUrl).replace(/\+/g, '%2B')),
url: myUrl,
method: 'GET',
success: res => {
console.log(res, "ws成功连接...", myUrl)
realThis.wsLiveFlag = true;
}
})
realThis.socketTask.onError((res) => {
console.log("连接发生错误,请检查appid是否填写", res)
})
realThis.socketTask.onOpen((res) => {
this.historyTextList.push({
"role": "user",
"content": "梦见" + this.TEXT + "意味着什么"
})
console.info("wss的onOpen成功执行...", res)
// 第一帧..........................................
console.log('open成功...')
let params = {
"header": {
"app_id": this.APPID,
"uid": "aef9f963-7"
},
"parameter": {
"chat": {
"domain": this.modelDomain,
"temperature": 0.5,
"max_tokens": 1024
}
},
"payload": {
"message": {
"text": this.historyTextList
}
}
};
console.log("请求的params:" + JSON.stringify(params))
// this.sparkResult = this.sparkResult + "\r\n梦境:" + this.TEXT + "\r\n"
this.sparkResult = this.sparkResult + "梦境解析:"
console.log("发送第一帧...", params)
realThis.socketTask.send({ // 发送消息,,都用uni的官方版本
data: JSON.stringify(params),
success() {
console.log('第一帧发送成功')
}
});
});
// 接受到消息时
realThis.socketTask.onMessage((res) => {
console.log('收到API返回的内容:', res.data);
let obj = JSON.parse(res.data)
// console.log("我打印的"+obj.payload);
let dataArray = obj.payload.choices.text;
for (let i = 0; i < dataArray.length; i++) {
realThis.sparkResult = realThis.sparkResult + dataArray[i].content
realThis.tempRes = realThis.tempRes + dataArray[i].content
}
// realThis.sparkResult =realThis.sparkResult+
let temp = JSON.parse(res.data)
// console.log("0726",temp.header.code)
if (temp.header.code !== 0) {
console.log(`${temp.header.code}:${temp.message}`);
realThis.socketTask.close({
success(res) {
console.log('关闭成功', res)
realThis.wsLiveFlag = false;
},
fail(err) {
console.log('关闭失败', err)
}
})
}
if (temp.header.code === 0) {
if (res.data && temp.header.status === 2) {
realThis.sparkResult = realThis.sparkResult +
"\r\n**********************************************"
this.historyTextList.push({
"role": "assistant",
"content": this.tempRes
})
/* let dataArray= obj.payload.choices.text;
for(let i=0;i<dataArray.length;i++){
realThis.sparkResult =realThis.sparkResult+ dataArray[i].content
} */
setTimeout(() => {
realThis.socketTask.close({
success(res) {
console.log('关闭成功', res)
},
fail(err) {
// console.log('关闭失败', err)
}
})
}, 1000)
}
}
})
},
注:APPID填你实际的id