livekit-plugins-soniox 1.2.15__py3-none-any.whl → 1.2.17__py3-none-any.whl

livekit/plugins/soniox/__init__.py

@@ -17,10 +17,17 @@
 See https://docs.livekit.io/agents/integrations/stt/soniox/ for more information.
 """
 
-from .stt import STT, STTOptions
+from .stt import STT, ContextGeneralItem, ContextObject, ContextTranslationTerm, STTOptions
 from .version import __version__
 
-__all__ = ["STT", "STTOptions", "__version__"]
+__all__ = [
+    "STT",
+    "STTOptions",
+    "ContextObject",
+    "ContextGeneralItem",
+    "ContextTranslationTerm",
+    "__version__",
+]
 
 
 from livekit.agents import Plugin

livekit/plugins/soniox/stt.py

@@ -18,7 +18,7 @@ import asyncio
 import json
 import os
 import time
-from dataclasses import dataclass
+from dataclasses import asdict, dataclass
 
 import aiohttp
 
@@ -56,22 +56,47 @@ def is_end_token(token: dict) -> bool:
     return token.get("text") in (END_TOKEN, FINALIZED_TOKEN)
 
 
+@dataclass
+class ContextGeneralItem:
+    key: str
+    value: str
+
+
+@dataclass
+class ContextTranslationTerm:
+    source: str
+    target: str
+
+
+@dataclass
+class ContextObject:
+    """Context object for models with context_version 2, for Soniox stt-rt-v3-preview and higher.
+
+    Learn more about context in the documentation:
+    https://soniox.com/docs/stt/concepts/context
+    """
+
+    general: list[ContextGeneralItem] | None = None
+    text: str | None = None
+    terms: list[str] | None = None
+    translation_terms: list[ContextTranslationTerm] | None = None
+
+
 @dataclass
 class STTOptions:
     """Configuration options for Soniox Speech-to-Text service."""
 
     model: str | None = "stt-rt-preview"
+
     language_hints: list[str] | None = None
-    context: str | None = None
+    context: ContextObject | str | None = None
 
     num_channels: int = 1
     sample_rate: int = 16000
 
+    enable_speaker_diarization: bool = False
     enable_language_identification: bool = True
 
-    enable_non_final_tokens: bool = True
-    max_non_final_tokens_duration_ms: int | None = None
-
     client_reference_id: str | None = None
 
 
@@ -176,6 +201,10 @@ class SpeechStream(stt.SpeechStream):
         # If VAD was passed, disable endpoint detection, otherwise enable it.
         enable_endpoint_detection = not self._stt._vad_stream
 
+        context = self._stt._params.context
+        if isinstance(context, ContextObject):
+            context = asdict(context)
+
         # Create initial config object.
         config = {
             "api_key": self._stt._api_key,
@@ -185,9 +214,8 @@ class SpeechStream(stt.SpeechStream):
             "enable_endpoint_detection": enable_endpoint_detection,
             "sample_rate": self._stt._params.sample_rate,
             "language_hints": self._stt._params.language_hints,
-            "context": self._stt._params.context,
-            "enable_non_final_tokens": self._stt._params.enable_non_final_tokens,
-            "max_non_final_tokens_duration_ms": self._stt._params.max_non_final_tokens_duration_ms,
+            "context": context,
+            "enable_speaker_diarization": self._stt._params.enable_speaker_diarization,
             "enable_language_identification": self._stt._params.enable_language_identification,
             "client_reference_id": self._stt._params.client_reference_id,
         }

livekit/plugins/soniox/version.py

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

{livekit_plugins_soniox-1.2.15.dist-info → livekit_plugins_soniox-1.2.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: livekit-plugins-soniox
-Version: 1.2.15
+Version: 1.2.17
 Summary: Agent Framework plugin for services using Soniox's API.
 Project-URL: Documentation, https://docs.livekit.io
 Project-URL: Website, https://livekit.io/
@@ -17,7 +17,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Topic :: Multimedia :: Sound/Audio
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.9.0
-Requires-Dist: livekit-agents>=1.2.15
+Requires-Dist: livekit-agents>=1.2.17
 Description-Content-Type: text/markdown
 
 # Soniox plugin for LiveKit Agents

livekit_plugins_soniox-1.2.17.dist-info/RECORD

@@ -0,0 +1,8 @@
+livekit/plugins/soniox/__init__.py,sha256=xCvIeDKtCnq40LelQq4HiB4OG_dpEs_YHsSqFe6qLh0,1361
+livekit/plugins/soniox/log.py,sha256=GNs8rdX3HEj24K-AeVEglYPEXkXTYh0aV_cL0JqVI2c,69
+livekit/plugins/soniox/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+livekit/plugins/soniox/stt.py,sha256=Emv5Bvuh9c93fQREaASm_nI5VIJmLirTtZYUdqW24n0,18626
+livekit/plugins/soniox/version.py,sha256=SQ8mRFoSQ7YNG0lOmbx6EypVTyCiYuT7rUNMG_zJJqY,601
+livekit_plugins_soniox-1.2.17.dist-info/METADATA,sha256=5nsDFzInGa1MHA_ZtPdiRDuaObQRISVYjY-xnbqxHhM,2106
+livekit_plugins_soniox-1.2.17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+livekit_plugins_soniox-1.2.17.dist-info/RECORD,,

livekit_plugins_soniox-1.2.15.dist-info/RECORD

@@ -1,8 +0,0 @@
-livekit/plugins/soniox/__init__.py,sha256=mLCOj2GQSIiKql8Nv_8O1vn1fhoKHXVWH2CZ-4qZivU,1210
-livekit/plugins/soniox/log.py,sha256=GNs8rdX3HEj24K-AeVEglYPEXkXTYh0aV_cL0JqVI2c,69
-livekit/plugins/soniox/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-livekit/plugins/soniox/stt.py,sha256=SRfAAN7drYZKRz6HVYin4KTwS78p1eGf8rE4UJhsLtA,18071
-livekit/plugins/soniox/version.py,sha256=PzihL71bIGfvVnuJ06sJkqOP2JIhHRl7e-vbgCcuf_M,601
-livekit_plugins_soniox-1.2.15.dist-info/METADATA,sha256=mqa95UNI4lpZO05EKeegwPWv5_5Ar0sAzYNEoiRCpT4,2106
-livekit_plugins_soniox-1.2.15.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-livekit_plugins_soniox-1.2.15.dist-info/RECORD,,