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

fix: Remove unused imports, use sample_rate from base class

Browse Source
This commit is contained in:
shreyas-sarvam
2025-10-31 17:31:59 +05:30
parent 09a7e08cbf
commit 2d03e51109
2 changed files with 6 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/foundational/07z-interruptible-sarvam.py
View File

@@ -27,7 +27,6 @@ from pipecat.runner.utils import create_transport
from pipecat.services.openai.llm import OpenAILLMService from pipecat.services.openai.llm import OpenAILLMService
from pipecat.services.sarvam.stt import SarvamSTTService from pipecat.services.sarvam.stt import SarvamSTTService
from pipecat.services.sarvam.tts import SarvamTTSService from pipecat.services.sarvam.tts import SarvamTTSService
from pipecat.transcriptions.language import Language
from pipecat.transports.base_transport import BaseTransport, TransportParams from pipecat.transports.base_transport import BaseTransport, TransportParams
from pipecat.transports.daily.transport import DailyParams from pipecat.transports.daily.transport import DailyParams
from pipecat.transports.websocket.fastapi import FastAPIWebsocketParams from pipecat.transports.websocket.fastapi import FastAPIWebsocketParams

15
src/pipecat/services/sarvam/stt.py
View File

@@ -107,8 +107,6 @@ class SarvamSTTService(STTService):
params: Configuration parameters for Sarvam STT service. params: Configuration parameters for Sarvam STT service.
**kwargs: Additional arguments passed to the parent STTService. **kwargs: Additional arguments passed to the parent STTService.
""" """
super().__init__(sample_rate=sample_rate, **kwargs)
params = params or SarvamSTTService.InputParams() params = params or SarvamSTTService.InputParams()
# Validate that saaras models don't accept language parameter # Validate that saaras models don't accept language parameter
@@ -127,6 +125,10 @@ class SarvamSTTService(STTService):
"Prompts are only supported for STT-Translate models" "Prompts are only supported for STT-Translate models"
) )
# Use sample_rate from params if provided, otherwise use the parameter
sample_rate = params.sample_rate if params.sample_rate is not None else sample_rate
super().__init__(sample_rate=sample_rate, **kwargs)
self.set_model_name(model) self.set_model_name(model)
self._api_key = api_key self._api_key = api_key
self._language_code = params.language self._language_code = params.language
@@ -140,7 +142,6 @@ class SarvamSTTService(STTService):
self._prompt = params.prompt self._prompt = params.prompt
# Store connection parameters # Store connection parameters
self._sample_rate = params.sample_rate
self._vad_signals = params.vad_signals self._vad_signals = params.vad_signals
self._high_vad_sensitivity = params.high_vad_sensitivity self._high_vad_sensitivity = params.high_vad_sensitivity
self._input_audio_codec = params.input_audio_codec self._input_audio_codec = params.input_audio_codec
@@ -268,10 +269,8 @@ class SarvamSTTService(STTService):
method_kwargs = { method_kwargs = {
"audio": audio_base64, "audio": audio_base64,
"encoding": encoding, "encoding": encoding,
"sample_rate": self.sample_rate,
} }
# Only include sample_rate if provided in params
if self._sample_rate is not None:
method_kwargs["sample_rate"] = self._sample_rate
# Use appropriate method based on service type # Use appropriate method based on service type
if "saarika" in self.model_name.lower(): if "saarika" in self.model_name.lower():
@@ -302,10 +301,8 @@ class SarvamSTTService(STTService):
"vad_signals": vad_signals_str, "vad_signals": vad_signals_str,
"high_vad_sensitivity": high_vad_sensitivity_str, "high_vad_sensitivity": high_vad_sensitivity_str,
"input_audio_codec": self._input_audio_codec, "input_audio_codec": self._input_audio_codec,
"sample_rate": str(self.sample_rate),
} }
# Only include sample_rate if provided in params
if self._sample_rate is not None:
connect_kwargs["sample_rate"] = str(self._sample_rate)
# Choose the appropriate service based on model # Choose the appropriate service based on model
if "saarika" in self.model_name.lower(): if "saarika" in self.model_name.lower():