From 3f9d39329cb7945c0be49674e79a99a60ffb82d4 Mon Sep 17 00:00:00 2001
From: vipyne <vipyne@gmail.com>
Date: Mon, 9 Dec 2024 20:57:16 -0600
Subject: [PATCH] services(riva): model -> function_id

---
 src/pipecat/services/riva.py | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/src/pipecat/services/riva.py b/src/pipecat/services/riva.py
index 23c6b83a4..6652b5298 100644
--- a/src/pipecat/services/riva.py
+++ b/src/pipecat/services/riva.py
@@ -48,8 +48,7 @@ class FastpitchTTSService(TTSService):
         server: str = "grpc.nvcf.nvidia.com:443",
         voice_id: str = "English-US.Female-1",
         sample_rate: int = 24000,
-        # nvidia riva calls this 'function-id'
-        model: str = "0149dedb-2be8-4195-b9a0-e57e0e14f972",
+        function_id: str = "0149dedb-2be8-4195-b9a0-e57e0e14f972",
         params: InputParams = InputParams(),
         **kwargs,
     ):
@@ -64,7 +63,7 @@ class FastpitchTTSService(TTSService):
         self.set_voice(voice_id)
 
         metadata = [
-            ["function-id", model],
+            ["function-id", function_id],
             ["authorization", f"Bearer {api_key}"],
         ]
         auth = riva.client.Auth(None, True, server, metadata)
@@ -118,8 +117,7 @@ class ParakeetSTTService(STTService):
         *,
         api_key: str,
         server: str = "grpc.nvcf.nvidia.com:443",
-        # nvidia calls this 'function-id'
-        model: str = "1598d209-5e27-4d3c-8079-4751568b1081",
+        function_id: str = "1598d209-5e27-4d3c-8079-4751568b1081",
         params: InputParams = InputParams(),
         **kwargs,
     ):
@@ -143,7 +141,7 @@ class ParakeetSTTService(STTService):
         self.set_model_name("parakeet-ctc-1.1b-asr")
 
         metadata = [
-            ["function-id", model],
+            ["function-id", function_id],
             ["authorization", f"Bearer {api_key}"],
         ]
         auth = riva.client.Auth(None, True, server, metadata)