livekit-plugins-silero 0.6.0.dev1__tar.gz → 0.6.0.dev7__tar.gz

{livekit_plugins_silero-0.6.0.dev1 → livekit_plugins_silero-0.6.0.dev7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: livekit-plugins-silero
-Version: 0.6.0.dev1
+Version: 0.6.0.dev7
 Summary: Agent Framework Plugin for Silero
 Home-page: https://github.com/livekit/agents
 License: Apache-2.0
@@ -19,7 +19,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3 :: Only
 Requires-Python: >=3.9.0
 Description-Content-Type: text/markdown
-Requires-Dist: livekit-agents~=0.7
+Requires-Dist: livekit-agents>=0.8.0.dev0
 Requires-Dist: onnxruntime>=1.18
 Requires-Dist: numpy>=1.26
 

{livekit_plugins_silero-0.6.0.dev1 → livekit_plugins_silero-0.6.0.dev7}/livekit/plugins/silero/vad.py

@@ -19,6 +19,7 @@ import time
 from dataclasses import dataclass
 
 import numpy as np
+import onnxruntime  # type: ignore
 from livekit import agents, rtc
 from livekit.agents import utils
 
@@ -37,8 +38,9 @@ class _VADOptions:
 
 
 class VAD(agents.vad.VAD):
-    def __init__(
-        self,
+    @classmethod
+    def load(
+        cls,
         *,
         min_speech_duration: float = 0.05,
         min_silence_duration: float = 0.1,
@@ -47,7 +49,7 @@ class VAD(agents.vad.VAD):
         activation_threshold: float = 0.25,
         sample_rate: int = 16000,
         force_cpu: bool = True,
-    ) -> None:
+    ) -> "VAD":
         """
         Initialize the Silero VAD with the given options.
         The options are already set to strong defaults.
@@ -61,12 +63,11 @@ class VAD(agents.vad.VAD):
             sample_rate: sample rate for the inference (only 8KHz and 16KHz are supported)
             force_cpu: force to use CPU for inference
         """
-
         if sample_rate not in onnx_model.SUPPORTED_SAMPLE_RATES:
             raise ValueError("Silero VAD only supports 8KHz and 16KHz sample rates")
 
-        self._onnx_session = onnx_model.new_inference_session(force_cpu)
-        self._opts = _VADOptions(
+        session = onnx_model.new_inference_session(force_cpu)
+        opts = _VADOptions(
             min_speech_duration=min_speech_duration,
             min_silence_duration=min_silence_duration,
             padding_duration=padding_duration,
@@ -74,6 +75,17 @@ class VAD(agents.vad.VAD):
             activation_threshold=activation_threshold,
             sample_rate=sample_rate,
         )
+        return cls(session=session, opts=opts)
+
+    def __init__(
+        self,
+        *,
+        session: onnxruntime.InferenceSession,
+        opts: _VADOptions,
+    ) -> None:
+        super().__init__(capabilities=agents.vad.VADCapabilities(update_interval=0.032))
+        self._onnx_session = session
+        self._opts = opts
 
     def stream(self) -> "VADStream":
         return VADStream(
@@ -177,8 +189,12 @@ class VADStream(agents.vad.VADStream):
         may_start_at_sample = -1
         may_end_at_sample = -1
 
-        min_speech_samples = self._opts.min_speech_duration * self._opts.sample_rate
-        min_silence_samples = self._opts.min_silence_duration * self._opts.sample_rate
+        min_speech_samples = int(
+            self._opts.min_speech_duration * self._opts.sample_rate
+        )
+        min_silence_samples = int(
+            self._opts.min_silence_duration * self._opts.sample_rate
+        )
 
         current_sample = 0
 
@@ -204,6 +220,7 @@ class VADStream(agents.vad.VADStream):
             else:
                 max_data_s += self._opts.max_buffered_speech
 
+            assert self._original_sample_rate is not None
             cl = int(max_data_s) * self._original_sample_rate
             if len(pub_speech_buf) > cl:
                 pub_speech_buf = pub_speech_buf[-cl:]

{livekit_plugins_silero-0.6.0.dev1 → livekit_plugins_silero-0.6.0.dev7}/livekit/plugins/silero/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "0.6.0-dev.1"
+__version__ = "0.6.0-dev.7"

{livekit_plugins_silero-0.6.0.dev1 → livekit_plugins_silero-0.6.0.dev7}/livekit_plugins_silero.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: livekit-plugins-silero
-Version: 0.6.0.dev1
+Version: 0.6.0.dev7
 Summary: Agent Framework Plugin for Silero
 Home-page: https://github.com/livekit/agents
 License: Apache-2.0
@@ -19,7 +19,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3 :: Only
 Requires-Python: >=3.9.0
 Description-Content-Type: text/markdown
-Requires-Dist: livekit-agents~=0.7
+Requires-Dist: livekit-agents>=0.8.0.dev0
 Requires-Dist: onnxruntime>=1.18
 Requires-Dist: numpy>=1.26
 

{livekit_plugins_silero-0.6.0.dev1 → livekit_plugins_silero-0.6.0.dev7}/livekit_plugins_silero.egg-info/requires.txt

@@ -1,3 +1,3 @@
-livekit-agents~=0.7
+livekit-agents>=0.8.0.dev0
 onnxruntime>=1.18
 numpy>=1.26

{livekit_plugins_silero-0.6.0.dev1 → livekit_plugins_silero-0.6.0.dev7}/setup.py

@@ -47,7 +47,7 @@ setuptools.setup(
     license="Apache-2.0",
     packages=setuptools.find_namespace_packages(include=["livekit.*"]),
     python_requires=">=3.9.0",
-    install_requires=["livekit-agents~=0.7", "onnxruntime>=1.18", "numpy>=1.26"],
+    install_requires=["livekit-agents>=0.8.0.dev0", "onnxruntime>=1.18", "numpy>=1.26"],
     package_data={
         "livekit.plugins.silero.resources": ["silero_vad.onnx"],
         "livekit.plugins.silero": ["py.typed"],

livekit_plugins_silero-0.6.0.dev1/livekit/plugins/silero/resources/silero_vad.onnx

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:6b99cbfd39246b6706f98ec13c7c50c6b299181f2474fa05cbc8046acc274396
-size 2313101