EvanAgent/dsh-desktop/app/plugins/sherpa-onnx-node/spoken-language-identification.js

/** @typedef {import('./types').SpokenLanguageIdentificationConfig} SpokenLanguageIdentificationConfig */
/** @typedef {import('./types').SpokenLanguageIdentificationHandle} SpokenLanguageIdentificationHandle */
/** @typedef {import('./non-streaming-asr').OfflineStream} OfflineStream */

const addon = require('./addon.js');
const non_streaming_asr = require('./non-streaming-asr.js');

class SpokenLanguageIdentification {
  /**
   * @param {SpokenLanguageIdentificationConfig} config
   */
  constructor(config) {
    this.handle = addon.createSpokenLanguageIdentification(config);
    this.config = config;
  }

  /**
   * @returns {OfflineStream}
   */
  createStream() {
    return new non_streaming_asr.OfflineStream(
        addon.createSpokenLanguageIdentificationOfflineStream(this.handle));
  }

  /**
   * Return a 2-letter language code, e.g. 'en', 'de', 'fr', 'es', 'zh'
   * @param {OfflineStream} stream
   * @returns {string}
   */
  compute(stream) {
    return addon.spokenLanguageIdentificationCompute(
        this.handle, stream.handle);
  }
}

module.exports = {
  SpokenLanguageIdentification,
}