esp-ai/src/functions/getServerURL.js

Provide a complete set of AI dialogue solutions for your development board, including but not limited to the IAT+LLM+TTS integration solution for the ESP32 series development board. | 为你的开发板提供全套的AI对话方案，包括但不限于 `ESP32` 系列开发板的 `IAT+LLM+TTS` 集成方案。

/**
 * Copyright (c) 2024 小明IO
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * Commercial use of this software requires prior written authorization from the Licensor.
 * 请注意：将 ESP-AI 代码用于商业用途需要事先获得许可方的授权。
 * 删除与修改版权属于侵权行为，请尊重作者版权，避免产生不必要的纠纷。
 * 
 * @author 小明IO   
 * @email  1746809408@qq.com
 * @github https://github.com/wangzongming/esp-ai
 * @websit https://espai.fun
 */
const CryptoJS = require('crypto-js')

/**
 * @param {"IAT" | "TTS" | "LLM" } type 服务类型
*/
function getServerURL(type, config) { 
    const { iat_server, llm_server, tts_server } = config; 
    switch (type) {
        case "IAT":
            return getServersURL[iat_server](type, config);
        case "TTS":
            return getServersURL[tts_server](type, config);
        case "LLM":
            return getServersURL[llm_server](type, config);
        default:
            break;
    }
}

const getServersURL = {
    xun_fei,
    bai_du,
}
function xun_fei(type, _config) { 
    const { apiSecret, apiKey, llm = "v3.5" } = _config;
    let url = "";
    // 获取当前时间 RFC1123格式
    const date = (new Date().toUTCString()) 
    const config = {
        // IAT
        hostUrl: "wss://iat-api.xfyun.cn/v2/iat",
        host: "iat-api.xfyun.cn",
        uri: "/v2/iat",
        // TTS
        tts_hostUrl: "wss://tts-api.xfyun.cn/v2/tts",
        tts_hostUrlhost: "tts-api.xfyun.cn",
        tts_uri: "/v2/tts",
        // LLM
        llm_hostUrl: `wss://spark-api.xf-yun.com/${llm}/chat`,
        llm_host: "spark-api.xf-yun.com",
        llm_uri: `/${llm}/chat`,
        apiKey: apiKey,
        apiSecret: apiSecret,
    }
    function getAuthStr({ host, date, uri }) {
        let signatureOrigin = `host: ${host}\ndate: ${date}\nGET ${uri} HTTP/1.1`
        let signatureSha = CryptoJS.HmacSHA256(signatureOrigin, config.apiSecret)
        let signature = CryptoJS.enc.Base64.stringify(signatureSha)
        let authorizationOrigin = `api_key="${config.apiKey}", algorithm="hmac-sha256", headers="host date request-line", signature="${signature}"`
        let authStr = CryptoJS.enc.Base64.stringify(CryptoJS.enc.Utf8.parse(authorizationOrigin))
        return authStr
    }

    switch (type) {
        case "IAT":
            url = `${config.hostUrl}?authorization=${getAuthStr({ host: config.host, date, uri: config.uri })}&date=${date}&host=${config.host}`;
            break;
        case "TTS":
            url = `${config.tts_hostUrl}?authorization=${getAuthStr({ host: config.tts_host, date, uri: config.tts_uri })}&date=${date}&host=${config.tts_host}`;
            break;
        case "LLM":
            url = `${config.llm_hostUrl}?authorization=${getAuthStr({ host: config.llm_host, date, uri: config.llm_uri })}&date=${date}&host=${config.llm_host}`;
            break;
        default:
            break;
    }

    return url;
}

function bai_du(type) {
    // ing...
}

module.exports = getServerURL;