Simple Python function to connect with Riva

riva_test_client.py

import io
import os
import librosa
from time import time
import numpy as np
import grpc
import requests

# TTS proto
import riva_api.riva_tts_pb2 as rtts
import riva_api.riva_tts_pb2_grpc as rtts_srv
import riva_api.riva_audio_pb2 as ra

load_dotenv()

if __name__ == '__main__':
    import argparse

    parser = argparse.ArgumentParser(
        formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    parser.add_argument('text', type=str, help='text for speech synthesis')
    parser.add_argument('--host', type=str, default='0.0.0.0',
                        help='server IP address')
    args = parser.parse_args()
    
    channel = grpc.insecure_channel(f'{os.getenv(args.host)}:50051')
    riva_tts = rtts_srv.RivaSpeechSynthesisStub(channel)
    req = rtts.SynthesizeSpeechRequest()
    req.text = args.text
    req.language_code = "en-US"                    # currently required to be "en-US"
    req.encoding = ra.AudioEncoding.LINEAR_PCM     # Supports LINEAR_PCM, FLAC, MULAW and ALAW audio encodings
    req.sample_rate_hz = 22050                     # ignored, audio returned will be 22.05KHz
    req.voice_name = "ljspeech"                    # ignored
    
    resp = riva_tts.Synthesize(req)

The module 'riva_api.riva_tts_pb2' is missing please help.