esp-ai/src/index.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 log = require("./utils/log");
const package = require("../package.json");
const EspAiInstance = require("./instance");

global.G_config = {};
const LOGO = `  

 ********  ******** *******            **     **
/**/////  **////// /**////**          ****   /**
/**      /**       /**   /**         **//**  /**
/******* /*********/*******  *****  **  //** /**
/**////  ////////**/**////  /////  **********/**
/**             /**/**            /**//////**/**
/******** ******** /**            /**     /**/**
//////// ////////  //             //      // // 

服务端版本号：v${package.version}
`
const IS_DEV = process.argv[2] === "--dev";

// 判断nodejs版本，低于 18 进行报错
const nodeVersion = process.version.split('.')[0].replace('v', '');
if (parseInt(nodeVersion, 10) < 18) {
    log.error(`Node.js 版本必须至少为 18.x\n`);
    process.exit(1);
}

function main(config = {}) {
    try {
        log.info(LOGO);

        /**
         * [UUID, {
         *       ws:{},
         *       user_config: {  // 用户配置
         *
         *       },
         *       authed: boolean,                 // 已鉴权
         *       runing: boolean,                 // 小明同学运行中
         *       tts_list: new Map(ws),           // TTS 任务队列
         *       await_out_tts: [],               // 待播放任务列表
         *       await_out_tts_ing: boolean,      // 待播放任务正在执行
         *       await_out_tts_run: ()=> {},      // 执行播放
         *       iat_server_connect_ing: boolean,
         *       iat_server_connected: boolean,
         *       iat_ws: {},
         *       iat_end_queue: ()=> void,        // iat 结束后的任务
         *       llm_ws: {},
         *       llm_historys: [],                // 历史对话
         *       first_session: boolean,          // 第一次的 会话
         *       session_id: "",                  // 会话id
         *       client_out_audio_ing: boolean,   // 客户端是否还在播放音频
         *       iat_end_frame_timer: null,       // 最后一帧发送倒计时
         *       send_pcm: (pcm)=> void           // 音频发送函数
         *       start_iat: ()=> void             // 音频发送函数
         *       add_audio_out_over_queue: ()=> Promise<void>  // 音频流播放完毕的任务队列
         *       backlog_instruction: [],          // Backlog instruction
         *       stop_next_session: boolean        // 异步停止下一次会话
         *       client_available_audio: number,   //  客户端可用字节流
         *       abort_controllers: [AbortController],   //  所有的中断控制器
         *
         *       // 播放音频的参数
         *       play_audio_ing: boolean,
         *       play_audio_on_end: (event: string) => void,
         *       play_audio_seek: number,
         *
         *       // 指令相关参数
         *       intention_ing: boolean,
         *       stop_ack_fn: () => void,
         *
         *       // 上一次 LLM 的情绪
         *       prev_emo: string
         *
         * }]
        */
        global.G_devices = new Map();
        const Instance = new EspAiInstance()
        global.G_Instance = Instance;


        /**
         * [特定的数据帧]
         * 见：esp-ai/src/globals.cpp
        */
        global.G_session_ids = {
            cache_du: "1000",
            cache_hello: "1001",
            cache_connected: "0001",
            cache_sleep_reply: "1002",
            tts_fn: "0010",

            tts_session: "00",
            tts_chunk_end: "01",
            tts_all_end_align: "02",
            tts_all_end: "03",

            // task_id
            play_music: "play_music",
        }


        const init_server = require("./functions/init_server")
        const _config = IS_DEV ? require("./config_dev") : require("./config")


        global.G_ws_server = null;
        global.G_config = { ..._config, ...config };

        // 缓存 TTS
        global.G_cahce_TTS = new Map();
        global.G_cahce_TTS_number = (G_config.cache_TTS_number || G_config.cache_TTS_number === 0) ? G_config.cache_TTS_number : 1000; // 缓存 TTS 数量。
        global.G_get_cahce_TTS = (text) => G_cahce_TTS.get(text);
        global.G_set_cahce_TTS = (text, audio) => {
            if (G_cahce_TTS.size > G_cahce_TTS_number) return;
            G_cahce_TTS.set(text, audio);
        };

        log.info(`服务端口：${G_config.port}`);
        log.info(`服务插件：${G_config.plugins ? G_config.plugins.map(item => item.name).join(" | ") : "-"}`);
        G_ws_server = init_server();

        return Instance
    } catch (err) {
        console.log(err);
        log.error(`服务运行失败。`);
    }
}

IS_DEV && main({});

module.exports = main;