用node来连接星火认知大模型

最新推荐文章于 2024-07-19 22:46:50 发布

小王同学要好学

最新推荐文章于 2024-07-19 22:46:50 发布

阅读量215

点赞数 2

文章标签： node.js 人工智能

本文链接：https://blog.csdn.net/m0_63084496/article/details/137697291

版权

本文介绍了如何在Node.js环境中利用WebSocket与星火认知大模型进行通信，包括构建WebSocket连接、生成鉴权URL并处理鉴权过程。作者分享了解决权限问题和文档查找困难的经验。

摘要由CSDN通过智能技术生成

一、为什么使用

为什么使用星火认知大模型，因为他有免费的。。。。

二、代码

const CryptoJs = require('crypto-js');
const Base64 = require('crypto-js/enc-base64');
const WebSocket = require('ws'); // 引入ws模块来创建WebSocket连接

const requestObj = {
    APPID: '',//注册官网提供的
    APISecret: '',注册官网提供的
    APIKey: '',注册官网提供的
    Uid: "",//随你添加
}

let  resultMessage=''
// 鉴权url地址
const getWebsocketUrl = () => {
    let url = "wss://spark-api.xf-yun.com/v3.5/chat";
    let host = "spark-api.xf-yun.com";
    let apiKeyName = "api_key";
    let date = new Date().toGMTString()
    let algorithm = "hmac-sha256";
    let headers = "host date request-line";
    let signatureOrigin = `host: ${host}\ndate: ${date}\nGET /v3.5/chat HTTP/1.1`;//和URL的版本对应v3.5就填v3.5,不然会出现权限不行
    const hmac = CryptoJs.HmacSHA256(signatureOrigin,requestObj.APISecret)
    const signature = Base64.stringify(hmac);
    let authorizationOrigin = `${apiKeyName}="${requestObj.APIKey}", algorithm="${algorithm}", headers="${headers}", signature="${signature}"`;
   
    const wordArray =CryptoJs.enc.Utf8.parse(authorizationOrigin);
    let authorization = Base64.stringify(wordArray);
    // 不需要编码URL
    url = `${url}?authorization=${authorization}&date=${encodeURIComponent(date)}&host=${host}`;
    return url;
}

// 获取请求地址
let myUrl = getWebsocketUrl();
// 创建WebSocket连接
let socket = new WebSocket(myUrl);
// 监听WebSocket连接事件

socket.on('open', () => {
    console.log('WebSocket连接已经建立');
    
    // 构造发送的消息
    let params = {
        "header": {
            "app_id": requestObj.APPID,
            "uid": requestObj.Uid
        },
        "parameter": {
            "chat": {
                "domain": "generalv3.5",//对应版本
                "temperature": 0.5,
                "max_tokens": 1024,
            }
        },
        "payload": {
            "message": {
                "text": [
                    { "role": "user", "content": "你是谁" }, //# 用户的历史问题
                 //   { "role": "assistant", "content": "我是AI助手" },  //# AI的历史回答结果
                 //   { "role": "user", "content": 'nihao' },  //# 最新的一条问题，如无需上下文，可只传最新一条问题
                ]
            }
        }
    };

    // 发送消息
    socket.send(JSON.stringify(params));
});

// 监听WebSocket消息事件
socket.on('message', (data) => {
    // 解析消息
    let jsonData = JSON.parse(data);
    
    // 处理消息
    resultMessage+= jsonData.payload.choices.text[0].content;
    // 判断消息是否出错
    if (jsonData.header.code !== 0) {
        console.log("出错了", jsonData.header.code, ":", jsonData.header.message);
        // 出错了手动关闭连接
        socket.close(); 
    }
});

// 监听WebSocket连接关闭事件
socket.on('close', (event) => {
    console.log("机器人的回复:",resultMessage)
    console.log('连接关闭！！', event);
});

// 监听WebSocket连接错误事件
socket.on('error', (error) => {
    console.log('连接发送错误！！', error);
});