M5Module-LLM Arduino API

M5Module-LLM Arduino驱动库API文档。

M5ModuleLLM Class

M5ModuleLLM用于初始化LLM Module, 并且提供内部成员用于快速初始化LLM的各个单元, 方便根据自己的需求构建应用。

class M5ModuleLLM {
public:
    bool begin(Stream* targetPort);
    bool checkConnection();
    void update();

    m5_module_llm::ApiSys sys;
    m5_module_llm::ApiLlm llm;
    m5_module_llm::ApiAudio audio;
    m5_module_llm::ApiTts tts;
    m5_module_llm::ApiKws kws;
    m5_module_llm::ApiAsr asr;
    m5_module_llm::ModuleMsg msg;
    m5_module_llm::ModuleComm comm;
private:
};

begin

函数原型:

bool begin(Stream* targetPort);

功能说明:

• 初始化LLM Module UART接口配置

传入参数:

• Stream* targetPort:
  • 传入Serial指针

返回值:

• bool:
  • true: 初始化成功
  • false: 初始化失败

checkConnection

函数原型:

bool checkConnection();

功能说明:

• 发送sys.ping指令, 检查LLM Module连接状态

传入参数:

• null

返回值:

• bool:
  • true: 模组响应
  • false: 模组无响应

update

函数原型:

void update();

功能说明:

• 拉取LLM Module UART响应数据, 该API需包含在Loop中循环执行。

传入参数:

• null

返回值:

• null

ApiSys Class

M5ModuleLLM的内部成员ApiSys sys用于控制SYS单元实现系统复位等操作。

ping

函数原型:

int ping();

功能说明:

• 发送sys.ping指令, 检查LLM Module连接状态

传入参数:

• null

返回值:

• int:
  • MODULE_LLM_OK / Error Code

reset

函数原型:

int reset(bool waitResetFinish = true);

功能说明:

• 发送sys.reset指令, 复位软件服务。

传入参数:

• bool waitResetFinish:
  • true:阻塞等待复位
  • false:非阻塞执行复位

返回值:

• int:
  • MODULE_LLM_OK / Error Code

reboot

函数原型:

int reboot();

功能说明:

• 发送sys.reboot指令, 复位系统。

传入参数:

• null

返回值:

• int:
  • MODULE_LLM_OK / Error Code

ApiAudio Class

M5ModuleLLM的内部成员ApiAudio audio用于控制AUDIO单元的初始化和配置。

setup

函数原型:

String setup(ApiAudioSetupConfig_t config = ApiAudioSetupConfig_t(), String request_id = "audio_setup");

功能说明:

• 初始化Audio单元, 开启系统声卡。(使用KWS和TTS前需开启该功能)

传入参数:

ApiAudioSetupConfig_t config:

• LLM单元初始化配置:
• String request_id:
  • 会话id, 使用默认即可。

struct ApiAudioSetupConfig_t {
    int capcard      = 0;
    int capdevice    = 0;
    float capVolume  = 0.5;
    int playcard     = 0;
    int playdevice   = 1;
    float playVolume = 0.15;
};

返回值:

• String:
  • audio_work_id: audio单元work_id

ApiKws Class

M5ModuleLLM的内部成员ApiKws kws用于控制KWS单元的初始化和配置。

setup

函数原型:

String setup(ApiKwsSetupConfig_t config = ApiKwsSetupConfig_t(), String request_id = "kws_setup");

功能说明:

• 初始化KWS单元, 并配置唤醒关键字。

传入参数:

ApiKwsSetupConfig_t config:

• KWS单元初始化配置:
• String request_id:
  • 会话id, 使用默认即可。

struct ApiKwsSetupConfig_t {
    String kws             = "HELLO";
    String model           = "sherpa-onnx-kws-zipformer-gigaspeech-3.3M-2024-01-01";
    String response_format = "kws.bool";
    String input           = "sys.pcm";
    bool enoutput          = true;
};

返回值:

• String:
  • kws_work_id: kws单元work_id

ApiAsr Class

M5ModuleLLM的内部成员ApiAsr asr用于控制ASR单元的初始化和配置。

setup

函数原型:

String setup(ApiAsrSetupConfig_t config = ApiAsrSetupConfig_t(), String request_id = "asr_setup");

功能说明:

• 初始化ASR单元, 开启语音转文本功能。

传入参数:

ApiAsrSetupConfig_t config:

• ASR单元初始化配置:
• String request_id:
  • 会话id, 使用默认即可。

struct ApiAsrSetupConfig_t {
    String model           = "sherpa-ncnn-streaming-zipformer-20M-2023-02-17";
    String response_format = "asr.utf-8.stream";
    String input           = "sys.pcm";
    bool enoutput          = true;
    bool enkws             = true;
    float rule1            = 2.4;
    float rule2            = 1.2;
    float rule3            = 30.0;
};

返回值:

• String:
  • asr_work_id: asr单元work_id

ApiLlm Class

M5ModuleLLM的内部成员ApiLlm llm用于控制LLM单元的初始化和配置。

setup

函数原型:

String setup(ApiLlmSetupConfig_t config = ApiLlmSetupConfig_t(), String request_id = "llm_setup");

功能说明:

• 初始化LLM单元, 支持配置LLM单元输入输出数据方式。

传入参数:

• ApiLlmSetupConfig_t config:
  • LLM单元初始化配置:
• String request_id:
  • 会话id, 使用默认即可。

struct ApiLlmSetupConfig_t {
    String prompt;
    String model           = "qwen2.5-0.5b";
    String response_format = "llm.utf-8.stream";
    String input           = "llm.utf-8.stream";
    bool enoutput          = true;
    bool enkws             = true;
    int max_token_len      = 127;
};

返回值:

• String:
  • llm_work_id: llm单元work_id

inference

函数原型:

int inference(String work_id, String input, String request_id = "llm_inference");

功能说明:

• 输入数据, 开始推理。返回结果内容将进入M5ModuleLLM.msg中的responseMsgList列表容器中。

传入参数:

• String work_id:
  • 调用的LLM单元work_id
• String input:
  • 输入文本
• String request_id:
  • 会话ID, 当同时存在多个会话的时候用于区分。

返回值:

• int:
  • MODULE_LLM_OK / Error Code

inferenceAndWaitResult

函数原型:

int inferenceAndWaitResult(String work_id, String input, std::function<void(String&)> onResult, uint32_t timeout = 5000, String request_id = "llm_inference");

功能说明:

• 输入数据, 开始推理。并阻塞等待返回结果, 然后调用callback函数。

传入参数:

• String work_id:
  • 调用的LLM单元work_id
• String input:
  • 输入文本
• void onResult(String&)
  • 推理结果callback函数
• uint32_t timeout:
  • 等待推理超时时间
• String request_id:
  • 会话ID, 当同时存在多个会话的时候用于区分。

返回值:

• int:
  • MODULE_LLM_OK / Error Code

ApiTts Class

M5ModuleLLM的内部成员ApiTts tts用于控制TTS单元的初始化和配置。

setup

函数原型:

String setup(ApiTtsSetupConfig_t config = ApiTtsSetupConfig_t(), String request_id = "tts_setup");

功能说明:

• 初始化TTS单元, 开启文本转语音功能。

传入参数:

ApiTtsSetupConfig_t config:

• LLM单元初始化配置:
• String request_id:
  • 会话id, 使用默认即可。

struct ApiTtsSetupConfig_t {
    String model           = "single_speaker_english_fast";
    String response_format = "tts.base64.wav";
    String input           = "tts.utf-8.stream";
    bool enoutput          = true;
    bool enkws             = true;
};

返回值:

• String:
  • tts_work_id: tts单元work_id

inference

函数原型:

int inference(String work_id, String input, uint32_t timeout = 0, String request_id = "tts_inference");

功能说明:

• 输入数据, 开始推理转换, 完成后将自动播放至扬声器。

传入参数:

• String work_id:
  • 调用的TTS单元work_id
• String input:
  • 输入文本
• uint32_t timeout:
  • 等待推理超时时间
• String request_id:
  • 会话ID, 当同时存在多个会话的时候用于区分。

返回值:

• int:
  • MODULE_LLM_OK / Error Code

ModuleMsg Class

M5ModuleLLM的内部成员ModuleMsg msg提供了responseMsgList容器用于用于缓存接收LLM Module返回的各种信息。参考以下案例，在主循环中遍历获取返回结果。

void loop()
{
    module_llm.update();

    // Handle response msg
    for (auto& msg : module_llm.msg.responseMsgList) {
        // KWS msg
        if (msg.work_id == kws_work_id) {
            Serial.printf(">> Keyword detected\n");
        }

        // ASR msg
        if (msg.work_id == asr_work_id) {
            if (msg.object == "asr.utf-8.stream") {
                // Parse and get asr result
                JsonDocument doc;
                deserializeJson(doc, msg.raw_msg);
                String asr_result = doc["data"]["delta"].as<String>();
                Serial.printf(">> %s\n", asr_result.c_str());
            }
        }
    }
    module_llm.msg.responseMsgList.clear();
}

VoiceAssistant Class

M5ModuleLLM_VoiceAssistant用于快速创建LLM语音助手实例, 快速实现KWS(语音唤醒)->ASR(语音转文本)->LLM(大模型推理)->TTS(文本转语音)。

• 初始化时候只需要将M5ModuleLLM实例传入构造函数, 并注册对应事件的回调函数即可完成语音助手创建。

/*
 * SPDX-FileCopyrightText: 2024 M5Stack Technology CO LTD
 *
 * SPDX-License-Identifier: MIT
 */
#include <Arduino.h>
#include <M5Unified.h>
#include <M5ModuleLLM.h>

M5ModuleLLM module_llm;
M5ModuleLLM_VoiceAssistant voice_assistant(&module_llm);

/* On ASR data callback */
void on_asr_data_input(String data, bool isFinish, int index)
{
    M5.Display.setTextColor(TFT_GREEN, TFT_BLACK);
    M5.Display.printf(">> %s\n", data.c_str());

    /* If ASR data is finish */
    if (isFinish) {
        M5.Display.setTextColor(TFT_YELLOW, TFT_BLACK);
        M5.Display.print(">> ");
    }
};

/* On LLM data callback */
void on_llm_data_input(String data, bool isFinish, int index)
{
    M5.Display.print(data);

    /* If LLM data is finish */
    if (isFinish) {
        M5.Display.print("\n");
    }
};

void setup()
{
    M5.begin();
    M5.Display.setTextSize(2);
    M5.Display.setTextScroll(true);

    /* Init module serial port */
    Serial2.begin(115200, SERIAL_8N1, 16, 17);  // Basic
    // Serial2.begin(115200, SERIAL_8N1, 13, 14);  // Core2
    // Serial2.begin(115200, SERIAL_8N1, 18, 17);  // CoreS3

    /* Init module */
    module_llm.begin(&Serial2);

    /* Make sure module is connected */
    M5.Display.printf(">> Check ModuleLLM connection..\n");
    while (1) {
        if (module_llm.checkConnection()) {
            break;
        }
    }

    /* Begin voice assistant preset */
    M5.Display.printf(">> Begin voice assistant..\n");
    int ret = voice_assistant.begin("HELLO");
    if (ret != MODULE_LLM_OK) {
        while (1) {
            M5.Display.setTextColor(TFT_RED);
            M5.Display.printf(">> Begin voice assistant failed\n");
        }
    }

    /* Register on ASR data callback function */
    voice_assistant.onAsrDataInput(on_asr_data_input);

    /* Register on LLM data callback function */
    voice_assistant.onLlmDataInput(on_llm_data_input);

    M5.Display.printf(">> Voice assistant ready\n");
}

void loop()
{
    /* Keep voice assistant preset update */
    voice_assistant.update();
}

Error Code

enum ModuleLLMErrorCode_t {
    MODULE_LLM_OK                              = 0,
    MODULE_LLM_RESET_WARN                      = -1,
    MODULE_LLM_JSON_FORMAT_ERROR               = -2,
    MODULE_LLM_ACTION_MATCH_FAILED             = -3,
    MODULE_LLM_INFERENCE_DATA_PUSH_FAILED      = -4,
    MODULE_LLM_MODEL_LOADING_FAILED            = -5,
    MODULE_LLM_UNIT_NOT_EXIST                  = -6,
    MODULE_LLM_UNKNOWN_OPERATION               = -7,
    MODULE_LLM_UNIT_RESOURCE_ALLOCATION_FAILED = -8,
    MODULE_LLM_UNIT_CALL_FAILED                = -9,
    MODULE_LLM_MODEL_INIT_FAILED               = -10,
    MODULE_LLM_MODEL_RUN_FAILED                = -11,
    MODULE_LLM_MODULE_NOT_INITIALISED          = -12,
    MODULE_LLM_MODULE_ALREADY_WORKING          = -13,
    MODULE_LLM_MODULE_NOT_WORKING              = -14,
    MODULE_LLM_NO_UPDATEABLE_MODULES           = -15,
    MODULE_LLM_NO_MODULES_AVAILABLE_FOR_UPDATE = -16,
    MODULE_LLM_FILE_OPEN_FAILED                = -17,
    MODULE_LLM_WAIT_RESPONSE_TIMEOUT           = -97,
    MODULE_LLM_RESPONSE_PARSE_FAILED           = -98,
    MODULE_LLM_ERROR_NONE                      = -99,
};