wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

riva: rename FastpitchTTSService to FastPitchTTSService

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-12-10 13:30:43 -08:00
parent 1db624575f
commit d2dabe4358
4 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
CHANGELOG.md
View File

@@ -9,6 +9,9 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
### Added ### 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 - Added `IdentityFilter`. This is the simplest frame filter that lets through
all incoming frames. all incoming frames.

4
examples/foundational/01c-fastpitch.py
View File

@@ -13,7 +13,7 @@ from pipecat.frames.frames import EndFrame, TTSSpeakFrame
from pipecat.pipeline.pipeline import Pipeline from pipecat.pipeline.pipeline import Pipeline
from pipecat.pipeline.task import PipelineTask from pipecat.pipeline.task import PipelineTask
from pipecat.pipeline.runner import PipelineRunner 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 pipecat.transports.services.daily import DailyParams, DailyTransport
from runner import configure from runner import configure
@@ -36,7 +36,7 @@ async def main():
room_url, None, "Say One Thing", DailyParams(audio_out_enabled=True) 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() runner = PipelineRunner()

4
examples/foundational/07r-interruptible-riva-nim.py
View File

@@ -20,7 +20,7 @@ from pipecat.pipeline.runner import PipelineRunner
from pipecat.pipeline.task import PipelineParams, PipelineTask from pipecat.pipeline.task import PipelineParams, PipelineTask
from pipecat.processors.aggregators.openai_llm_context import OpenAILLMContext from pipecat.processors.aggregators.openai_llm_context import OpenAILLMContext
from pipecat.services.nim import NimLLMService 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 from pipecat.transports.services.daily import DailyParams, DailyTransport
load_dotenv(override=True) 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" 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 = [ messages = [
{ {

2
src/pipecat/services/riva.py
View File

@@ -36,7 +36,7 @@ except ModuleNotFoundError as e:
raise Exception(f"Missing module: {e}") raise Exception(f"Missing module: {e}")
class FastpitchTTSService(TTSService): class FastPitchTTSService(TTSService):
class InputParams(BaseModel): class InputParams(BaseModel):
language: Optional[Language] = Language.EN_US language: Optional[Language] = Language.EN_US
quality: Optional[int] = 20 quality: Optional[int] = 20