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

services(deepgram): add url to DeepgramSTTService

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-06-21 16:52:28 -07:00
parent dc64e57f63
commit adf414e40f
2 changed files with 5 additions and 1 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
- Allow specifying a `DeepgramSTTService` url which allows using on-prem
Deepgram.
- Added new `FastAPIWebsocketTransport`. This is a new websocket transport that - Added new `FastAPIWebsocketTransport`. This is a new websocket transport that
can be integrated with FastAPI websockets. can be integrated with FastAPI websockets.

3
src/pipecat/services/deepgram.py
View File

@@ -89,6 +89,7 @@ class DeepgramTTSService(TTSService):
class DeepgramSTTService(AIService): class DeepgramSTTService(AIService):
def __init__(self, def __init__(self,
api_key: str, api_key: str,
url: str = "",
live_options: LiveOptions = LiveOptions( live_options: LiveOptions = LiveOptions(
encoding="linear16", encoding="linear16",
language="en-US", language="en-US",
@@ -104,7 +105,7 @@ class DeepgramSTTService(AIService):
self._live_options = live_options self._live_options = live_options
self._client = DeepgramClient( self._client = DeepgramClient(
api_key, config=DeepgramClientOptions(options={"keepalive": "true"})) api_key, config=DeepgramClientOptions(url=url, options={"keepalive": "true"}))
self._connection = self._client.listen.asynclive.v("1") self._connection = self._client.listen.asynclive.v("1")
self._connection.on(LiveTranscriptionEvents.Transcript, self._on_message) self._connection.on(LiveTranscriptionEvents.Transcript, self._on_message)