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

Riva: refactor function_id and model_name

Browse Source
This commit is contained in:
vipyne
2025-05-02 11:34:40 -05:00
parent 6c428c303b
commit b2754117c8
3 changed files with 64 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -41,14 +41,14 @@ async def run_bot(webrtc_connection: SmallWebRTCConnection, _: argparse.Namespac
),
)
stt = RivaSTTService(api_key=os.getenv("NVIDIA_API_KEY"))
# stt = RivaSegmentedSTTService(api_key=os.getenv("NVIDIA_API_KEY"))
# stt = RivaSTTService(api_key=os.getenv("NVIDIA_API_KEY")) # parakeet og
stt = RivaSegmentedSTTService(api_key=os.getenv("NVIDIA_API_KEY")) # canary
# stt = ParakeetSTTService(api_key=os.getenv("NVIDIA_API_KEY"))
llm = NimLLMService(api_key=os.getenv("NVIDIA_API_KEY"), model="meta/llama-3.1-405b-instruct")
# tts = FastPitchTTSService(api_key=os.getenv("NVIDIA_API_KEY"))
tts = RivaTTSService(api_key=os.getenv("NVIDIA_API_KEY"))
tts = FastPitchTTSService(api_key=os.getenv("NVIDIA_API_KEY"))
# tts = RivaTTSService(api_key=os.getenv("NVIDIA_API_KEY"))
messages = [
{