diff --git a/CHANGELOG.md b/CHANGELOG.md index 697197a9f..dbbbc99a6 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -9,6 +9,9 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0 ### Added +- Added NVIDIA Riva's `FastPitchTTSService` and `ParakeetSTTService`. + (see https://www.nvidia.com/en-us/ai-data-science/products/riva/) + - Added `IdentityFilter`. This is the simplest frame filter that lets through all incoming frames. diff --git a/examples/foundational/01c-fastpitch.py b/examples/foundational/01c-fastpitch.py index 49499cdfc..b041e6c7b 100644 --- a/examples/foundational/01c-fastpitch.py +++ b/examples/foundational/01c-fastpitch.py @@ -13,7 +13,7 @@ from pipecat.frames.frames import EndFrame, TTSSpeakFrame from pipecat.pipeline.pipeline import Pipeline from pipecat.pipeline.task import PipelineTask from pipecat.pipeline.runner import PipelineRunner -from pipecat.services.riva import FastpitchTTSService +from pipecat.services.riva import FastPitchTTSService from pipecat.transports.services.daily import DailyParams, DailyTransport from runner import configure @@ -36,7 +36,7 @@ async def main(): room_url, None, "Say One Thing", DailyParams(audio_out_enabled=True) ) - tts = FastpitchTTSService(api_key=os.getenv("NVIDIA_API_KEY")) + tts = FastPitchTTSService(api_key=os.getenv("NVIDIA_API_KEY")) runner = PipelineRunner() diff --git a/examples/foundational/07r-interruptible-riva-nim.py b/examples/foundational/07r-interruptible-riva-nim.py index 09d261bc3..1e02f82c0 100644 --- a/examples/foundational/07r-interruptible-riva-nim.py +++ b/examples/foundational/07r-interruptible-riva-nim.py @@ -20,7 +20,7 @@ from pipecat.pipeline.runner import PipelineRunner from pipecat.pipeline.task import PipelineParams, PipelineTask from pipecat.processors.aggregators.openai_llm_context import OpenAILLMContext from pipecat.services.nim import NimLLMService -from pipecat.services.riva import FastpitchTTSService, ParakeetSTTService +from pipecat.services.riva import FastPitchTTSService, ParakeetSTTService from pipecat.transports.services.daily import DailyParams, DailyTransport load_dotenv(override=True) @@ -51,7 +51,7 @@ async def main(): api_key=os.getenv("NVIDIA_API_KEY"), model="meta/llama-3.1-405b-instruct" ) - tts = FastpitchTTSService(api_key=os.getenv("NVIDIA_API_KEY")) + tts = FastPitchTTSService(api_key=os.getenv("NVIDIA_API_KEY")) messages = [ { diff --git a/src/pipecat/services/riva.py b/src/pipecat/services/riva.py index 0bbaee0ef..f57372775 100644 --- a/src/pipecat/services/riva.py +++ b/src/pipecat/services/riva.py @@ -36,7 +36,7 @@ except ModuleNotFoundError as e: raise Exception(f"Missing module: {e}") -class FastpitchTTSService(TTSService): +class FastPitchTTSService(TTSService): class InputParams(BaseModel): language: Optional[Language] = Language.EN_US quality: Optional[int] = 20