livekit-plugins-deepgram 1.0.19__py3-none-any.whl → 1.0.21__py3-none-any.whl

livekit/plugins/deepgram/__init__.py

@@ -1,3 +1,24 @@
+# Copyright 2025 LiveKit, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Deepgram plugin for LiveKit Agents
+
+Support for speech-to-text with [Deepgram](https://deepgram.com/).
+
+See https://docs.livekit.io/agents/integrations/stt/deepgram/ for more information.
+"""
+
 from .stt import STT, AudioEnergyFilter, SpeechStream
 from .tts import TTS
 from .version import __version__

livekit/plugins/deepgram/_utils.py

@@ -1,5 +1,6 @@
 import time
 from typing import Callable, Generic, Optional, TypeVar
+from urllib.parse import urlencode
 
 T = TypeVar("T")
 
@@ -34,3 +35,23 @@ class PeriodicCollector(Generic[T]):
             self._callback(self._total)
             self._total = None
         self._last_flush_time = time.monotonic()
+
+
+def _to_deepgram_url(opts: dict, base_url: str, *, websocket: bool) -> str:
+    # don't modify the original opts
+    opts = opts.copy()
+    if opts.get("keywords"):
+        # convert keywords to a list of "keyword:intensifier"
+        opts["keywords"] = [
+            f"{keyword}:{intensifier}" for (keyword, intensifier) in opts["keywords"]
+        ]
+
+    # lowercase bools
+    opts = {k: str(v).lower() if isinstance(v, bool) else v for k, v in opts.items()}
+
+    if websocket and base_url.startswith("http"):
+        base_url = base_url.replace("http", "ws", 1)
+
+    elif not websocket and base_url.startswith("ws"):
+        base_url = base_url.replace("ws", "http", 1)
+    return f"{base_url}?{urlencode(opts, doseq=True)}"

livekit/plugins/deepgram/stt.py

@@ -22,7 +22,6 @@ import weakref
 from dataclasses import dataclass
 from enum import Enum
 from typing import Any
-from urllib.parse import urlencode
 
 import aiohttp
 import numpy as np
@@ -43,7 +42,7 @@ from livekit.agents.types import (
 )
 from livekit.agents.utils import AudioBuffer, is_given
 
-from ._utils import PeriodicCollector
+from ._utils import PeriodicCollector, _to_deepgram_url
 from .log import logger
 from .models import DeepgramLanguages, DeepgramModels
 
@@ -566,10 +565,11 @@ class SpeechStream(stt.SpeechStream):
                     asyncio.create_task(recv_task(ws)),
                     asyncio.create_task(keepalive_task(ws)),
                 ]
+                tasks_group = asyncio.gather(*tasks)
                 wait_reconnect_task = asyncio.create_task(self._reconnect_event.wait())
                 try:
                     done, _ = await asyncio.wait(
-                        [asyncio.gather(*tasks), wait_reconnect_task],
+                        [tasks_group, wait_reconnect_task],
                         return_when=asyncio.FIRST_COMPLETED,
                     )  # type: ignore
 
@@ -584,6 +584,7 @@ class SpeechStream(stt.SpeechStream):
                     self._reconnect_event.clear()
                 finally:
                     await utils.aio.gracefully_cancel(*tasks, wait_reconnect_task)
+                    await tasks_group
             finally:
                 if ws is not None:
                     await ws.close()
@@ -751,26 +752,6 @@ def prerecorded_transcription_to_speech_event(
     )
 
 
-def _to_deepgram_url(opts: dict, base_url: str, *, websocket: bool) -> str:
-    # don't modify the original opts
-    opts = opts.copy()
-    if opts.get("keywords"):
-        # convert keywords to a list of "keyword:intensifier"
-        opts["keywords"] = [
-            f"{keyword}:{intensifier}" for (keyword, intensifier) in opts["keywords"]
-        ]
-
-    # lowercase bools
-    opts = {k: str(v).lower() if isinstance(v, bool) else v for k, v in opts.items()}
-
-    if websocket and base_url.startswith("http"):
-        base_url = base_url.replace("http", "ws", 1)
-
-    elif not websocket and base_url.startswith("ws"):
-        base_url = base_url.replace("ws", "http", 1)
-    return f"{base_url}?{urlencode(opts, doseq=True)}"
-
-
 def _validate_model(
     model: DeepgramModels | str, language: NotGivenOr[DeepgramLanguages | str]
 ) -> DeepgramModels | str:

livekit/plugins/deepgram/tts.py

@@ -5,7 +5,6 @@ import json
 import os
 import weakref
 from dataclasses import dataclass
-from urllib.parse import urlencode
 
 import aiohttp
 
@@ -25,6 +24,7 @@ from livekit.agents.types import (
 )
 from livekit.agents.utils import is_given
 
+from ._utils import _to_deepgram_url
 from .log import logger
 
 BASE_URL = "https://api.deepgram.com/v1/speak"
@@ -44,7 +44,7 @@ class TTS(tts.TTS):
     def __init__(
         self,
         *,
-        model: str = "aura-asteria-en",
+        model: str = "aura-2-andromeda-en",
         encoding: str = "linear16",
         sample_rate: int = 24000,
         api_key: NotGivenOr[str] = NOT_GIVEN,
@@ -57,7 +57,7 @@ class TTS(tts.TTS):
         Create a new instance of Deepgram TTS.
 
         Args:
-            model (str): TTS model to use. Defaults to "aura-asteria-en".
+            model (str): TTS model to use. Defaults to "aura-2-andromeda-en".
             encoding (str): Audio encoding to use. Defaults to "linear16".
             sample_rate (int): Sample rate of audio. Defaults to 24000.
             api_key (str): Deepgram API key. If not provided, will look for DEEPGRAM_API_KEY in environment.
@@ -221,7 +221,7 @@ class ChunkedStream(tts.ChunkedStream):
                     "Content-Type": "application/json",
                 },
                 json={"text": self._input_text},
-                timeout=self._conn_options.timeout,
+                timeout=aiohttp.ClientTimeout(connect=self._conn_options.timeout, total=30),
             ) as res:
                 if res.status != 200:
                     raise APIStatusError(
@@ -436,18 +436,3 @@ class SynthesizeStream(tts.SynthesizeStream):
             finally:
                 if ws is not None and not ws.closed:
                     await ws.close()
-
-
-def _to_deepgram_url(
-    opts: dict,
-    base_url: str,
-    *,
-    websocket: bool,
-) -> str:
-    if websocket and base_url.startswith("http"):
-        base_url = base_url.replace("http", "ws", 1)
-
-    elif not websocket and base_url.startswith("ws"):
-        base_url = base_url.replace("ws", "http", 1)
-
-    return f"{base_url}?{urlencode(opts, doseq=True)}"

livekit/plugins/deepgram/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "1.0.19"
+__version__ = "1.0.21"

{livekit_plugins_deepgram-1.0.19.dist-info → livekit_plugins_deepgram-1.0.21.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: livekit-plugins-deepgram
-Version: 1.0.19
+Version: 1.0.21
 Summary: Agent Framework plugin for services using Deepgram's API.
 Project-URL: Documentation, https://docs.livekit.io
 Project-URL: Website, https://livekit.io/
@@ -18,13 +18,15 @@ Classifier: Topic :: Multimedia :: Sound/Audio
 Classifier: Topic :: Multimedia :: Video
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.9.0
-Requires-Dist: livekit-agents[codecs]>=1.0.19
+Requires-Dist: livekit-agents[codecs]>=1.0.21
 Requires-Dist: numpy>=1.26
 Description-Content-Type: text/markdown
 
-# LiveKit Plugins DeepGram
+# Deepgram plugin for LiveKit Agents
 
-Agent Framework plugin for speech-to-text with [DeepGram](https://deepgram.com/)'s API. Currently supports speech-to-text.
+Support for speech-to-text with [Deepgram](https://deepgram.com/).
+
+See [https://docs.livekit.io/agents/integrations/stt/deepgram/](https://docs.livekit.io/agents/integrations/stt/deepgram/) for more information.
 
 ## Installation
 

livekit_plugins_deepgram-1.0.21.dist-info/RECORD

@@ -0,0 +1,11 @@
+livekit/plugins/deepgram/__init__.py,sha256=7sCI3vSCelS8mjtjaIZP_tt61S8KKncEBnuf0urNejw,1358
+livekit/plugins/deepgram/_utils.py,sha256=NgeR4qKZOeqs1wr8v4G2Q_KPZ5xUSFDE4f2N6WXnZH4,2041
+livekit/plugins/deepgram/log.py,sha256=isjd2-ROJXiDFhRRnqRmYxv16U5H9dBV6ut2g5bU7q0,71
+livekit/plugins/deepgram/models.py,sha256=dVguYc9AfjlexreN_O1C0NxX3q-ZK9k8s5B3hWsbtZ0,1236
+livekit/plugins/deepgram/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+livekit/plugins/deepgram/stt.py,sha256=snDTVjK6r0QJraLL0HzyvCqcy-bAJwuMNWlbJMLFybs,32025
+livekit/plugins/deepgram/tts.py,sha256=PgQfOJFcBoWzOhTJZEnipGo0hxhBzZNqZhtVW9txmUw,15415
+livekit/plugins/deepgram/version.py,sha256=5lzQkS1jEPqreexacwMd18b2EOx7R5m8AQMKtQRBgC4,601
+livekit_plugins_deepgram-1.0.21.dist-info/METADATA,sha256=yt3zUCbuCeQvdAYHu5fFpRDyaLh7sv8H-Q0yl53ApRc,1450
+livekit_plugins_deepgram-1.0.21.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+livekit_plugins_deepgram-1.0.21.dist-info/RECORD,,

livekit_plugins_deepgram-1.0.19.dist-info/RECORD

@@ -1,11 +0,0 @@
-livekit/plugins/deepgram/__init__.py,sha256=o1BgJLSoxrTkSUHXkKrFMl9e0bF6am2TFrDMV4U6j7g,584
-livekit/plugins/deepgram/_utils.py,sha256=0O3O77qURXjYYCenl5McE85fcBApaQvQl5zKQzvYfR8,1276
-livekit/plugins/deepgram/log.py,sha256=isjd2-ROJXiDFhRRnqRmYxv16U5H9dBV6ut2g5bU7q0,71
-livekit/plugins/deepgram/models.py,sha256=dVguYc9AfjlexreN_O1C0NxX3q-ZK9k8s5B3hWsbtZ0,1236
-livekit/plugins/deepgram/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-livekit/plugins/deepgram/stt.py,sha256=1O1jbfQRRX9WrpL2sCuKNb5bKvuE8bVsd9goA0td_FE,32692
-livekit/plugins/deepgram/tts.py,sha256=SX7vdMz4AZoZv2o_0EVwym5WMbFzy44nNGEyw2s9XFA,15728
-livekit/plugins/deepgram/version.py,sha256=UDC8ahmGgRkv-qMQUY3QibuuVevGMQ9Fd4yIhcQBZwA,601
-livekit_plugins_deepgram-1.0.19.dist-info/METADATA,sha256=LOmgHOZx7QUh6LDSL3PhoBSY4P4-UdEQY4_1bHdVN1Q,1350
-livekit_plugins_deepgram-1.0.19.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-livekit_plugins_deepgram-1.0.19.dist-info/RECORD,,