@softvisio/core/lib/api/whisper.js

Softisio core

import Api from "#lib/api";

export default class WhisperApi {
    #api;

    constructor ( url ) {
        this.#api = new Api( url );
    }

    // public
    async getModels () {
        return this.#api.call( "whisper/get-models" );
    }

    async getInstalledModels () {
        return this.#api.call( "whisper/get-installed-models" );
    }

    async getLoadedModels () {
        return this.#api.call( "whisper/get-loaded-models" );
    }

    async installModel ( model, { force } = {} ) {
        return this.#api.call( "whisper/install-model", model, { "force": Boolean( force ) } );
    }

    async unloadModel ( model ) {
        return this.#api.call( "whisper/unload-model", model );
    }

    async deleteModel ( model ) {
        return this.#api.call( "whisper/delete-model", model );
    }

    async detectLanguage ( audioFile, { model } = {} ) {
        return this.#api
            .upload( {
                "method": "whisper/detect-language",
                "args": [
                    audioFile,
                    {
                        model,
                    },
                ],
                "headersTimeout": 0,
            } )
            .start();
    }

    async transformSpeechToText ( audioFile, { model, language } = {} ) {
        return this.#api
            .upload( {
                "method": "whisper/transform-speech-to-text",
                "args": [
                    audioFile,
                    {
                        model,
                        language,
                    },
                ],
                "headersTimeout": 0,
            } )
            .start();
    }
}