intellinode/wrappers/GoogleAIWrapper.js

Create AI agents using the latest models, including ChatGPT, Llama, Diffusion, Cohere, Gemini, and Hugging Face.

/*
Apache License
*/
const config = require('../config.json');
const connHelper = require('../utils/ConnHelper');
const FetchClient = require('../utils/FetchClient');

class GoogleAIWrapper {
  constructor(apiKey) {
    this.API_SPEECH_URL = config.url.google.base.replace(
      '{1}',
      config.url.google.speech.prefix
    );
    this.API_KEY = apiKey;

    this.client = new FetchClient({
      baseURL: this.API_SPEECH_URL,
      headers: {
        'Content-Type': 'application/json; charset=utf-8',
        'X-Goog-Api-Key': this.API_KEY
      }
    });
  }

  async generateSpeech(params) {
    const endpoint =
      config.url.google.speech.prefix +
      config.url.google.speech.synthesize.postfix;
    const url = this.API_SPEECH_URL + config.url.google.speech.synthesize.postfix;

    const json = this.getSynthesizeInput(params);
    try {
      return await this.client.post(url, JSON.parse(json));
    } catch (error) {
      throw new Error(connHelper.getErrorMessage(error));
    }
  }

  getSynthesizeInput(params) {
    const text = params.text;
    const languageCode = params.languageCode;
    const name = params.name;
    const ssmlGender = params.ssmlGender;

    const modelInput = {
      input: {
        text: text
      },
      voice: {
        languageCode: languageCode,
        name: name,
        ssmlGender: ssmlGender
      },
      audioConfig: {
        audioEncoding: 'MP3'
      }
    };

    return JSON.stringify(modelInput);
  }
}

module.exports = GoogleAIWrapper;