【WebSocket】微信小程序原生组件使用SocketTask 调用星火认知大模型

直接上代码

微信开发者工具-调试器-终端-新建终端 进行依赖安装

npm install base-64

npm install crypto-js

然后顶部工具栏依次点击 工具-构建npm

// index.js
const defaultAvatarUrl = 'https://mmbiz.qpic.cn/mmbiz/icTdbqWNOwNRna42FI242Lcia07jQodd2FJGIYQfG0LAJGFxM4FbnQP6yfMxBgJ0F3YRqJCJ1aPAK2dQagdusBZg/0'
const Base64 = require('base-64');
const CryptoJS = require('crypto-js')
let socket = ""
Page({data: {APPID: "",//从开放平台控制台中获取APIKEY: "",//从开放平台控制台中获取APISecret: "",//从开放平台控制台中获取motto: 'Hello World',historyTextList: [],userInfo: {avatarUrl: defaultAvatarUrl,nickName: '',},hasUserInfo: false,canIUseGetUserProfile: wx.canIUse('getUserProfile'),canIUseNicknameComp: wx.canIUse('input.type.nickname'),TEXT: '你好，我的名字叫大王',sparkResult: '',tempRes: '' // 临时答复保存},bindViewTap() {wx.navigateTo({url: '../logs/logs'})},onLoad() {this.connectWebsocket()},getWebsocketUrl() {var _this2 = this;var url = "wss://spark-api.xf-yun.com/v3.5/chat";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: ".concat(host, "\ndate: ").concat(date, "\nGET /v3.5/chat HTTP/1.1");var signatureSha = CryptoJS.HmacSHA256(signatureOrigin, _this2.data.APISecret);var signature = CryptoJS.enc.Base64.stringify(signatureSha);var authorizationOrigin = "".concat(apiKeyName, "=\"").concat(_this2.data.APIKEY, "\",algorithm=\"").concat(algorithm, "\",headers=\"").concat(headers, "\",signature=\"").concat(signature, "\"");var authorization = Base64.encode(authorizationOrigin);url = "".concat(url, "?authorization=").concat(authorization, "&date=").concat(encodeURI(date), "&host=")//进行|不进行URRLENCODE均没报错.concat(host);return url;},connectWebsocket() {var url = this.getWebsocketUrl();socket = wx.connectSocket({url: url,success(res) {console.log('WebSocket 连接成功: ', res)},fail(err) {console.log('WebSocket 连接失败: ', err)}})// onOpensocket.onOpen((res) => {var _this = this;console.log('WebSocket 已连接',res)_this.data.historyTextList.push({"role": "user","content": _this.data.TEXT});var params = {"header": {"app_id": _this.data.APPID,"uid": "123456789"},"parameter": {"chat": {"domain": "generalv3.5","random_threshold": 0,"max_tokens": 1024}},"payload": {"message": {"text": _this.data.historyTextList}}};_this.sparkResult = _this.sparkResult + "\r\n我：" + _this.TEXT + "\r\n";_this.sparkResult = _this.sparkResult + "大模型：";console.log("发送第一帧...", JSON.stringify(params));socket.send({data:JSON.stringify(params),fail(err) {console.log('WebSocket 消息发送失败', err)}})})// onErrorsocket.onError((err) => {console.log('WebSocket 连接失败：', err)})// onClosesocket.onClose((ret) => {console.log('断开 WebSocket 连接', ret)})//监听接收到的消息socket.onMessage((res) => {let msg = res.dataconsole.info("接收到的服务器消息--原始:")console.info(res)if (typeof msg === "string" && msg.startsWith("{")) {msg = JSON.parse(msg)}console.log("接收到的服务器消息", msg)})},
})