livekit-plugins-fal 1.0.0.dev5__py3-none-any.whl → 1.0.0rc2__py3-none-any.whl

livekit/plugins/fal/stt.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-import dataclasses
 import os
 from dataclasses import dataclass
 
@@ -9,66 +8,49 @@ import fal_client
 from livekit import rtc
 from livekit.agents import APIConnectionError, APIConnectOptions, stt
 from livekit.agents.stt import SpeechEventType, STTCapabilities
-from livekit.agents.utils import AudioBuffer
+from livekit.agents.types import (
+    NOT_GIVEN,
+    NotGivenOr,
+)
+from livekit.agents.utils import AudioBuffer, is_given
 
 
 @dataclass
 class _STTOptions:
-    language: str
-    task: str
-    chunk_level: str
-    version: str
+    language: str = "en"
+    task: str = "transcribe"
+    chunk_level: str = "segment"
+    version: str = "3"
 
 
 class WizperSTT(stt.STT):
     def __init__(
         self,
         *,
-        language: str | None = "en",
-        task: str | None = "transcribe",
-        chunk_level: str | None = "segment",
-        version: str | None = "3",
+        language: NotGivenOr[str] = NOT_GIVEN,
+        api_key: NotGivenOr[str] = NOT_GIVEN,
     ):
         super().__init__(capabilities=STTCapabilities(streaming=False, interim_results=True))
-        self._api_key = os.getenv("FAL_KEY")
-        self._opts = _STTOptions(
-            language=language or "en",
-            task=task or "transcribe",
-            chunk_level=chunk_level or "segment",
-            version=version or "3",
-        )
-        self._fal_client = fal_client.AsyncClient()
-
+        self._api_key = api_key if is_given(api_key) else os.getenv("FAL_KEY")
         if not self._api_key:
             raise ValueError("fal AI API key is required. It should be set with env FAL_KEY")
+        self._opts = _STTOptions(language=language)
+        self._fal_client = fal_client.AsyncClient(key=self._api_key)
 
-    def update_options(self, *, language: str | None = None) -> None:
-        self._opts.language = language or self._opts.language
-
-    def _sanitize_options(
-        self,
-        *,
-        language: str | None = None,
-        task: str | None = None,
-        chunk_level: str | None = None,
-        version: str | None = None,
-    ) -> _STTOptions:
-        config = dataclasses.replace(self._opts)
-        config.language = language or config.language
-        config.task = task or config.task
-        config.chunk_level = chunk_level or config.chunk_level
-        config.version = version or config.version
-        return config
+    def update_options(self, *, language: NotGivenOr[str] = NOT_GIVEN) -> None:
+        if is_given(language):
+            self._opts.language = language
 
     async def _recognize_impl(
         self,
         buffer: AudioBuffer,
         *,
-        language: str | None,
+        language: NotGivenOr[str] = NOT_GIVEN,
         conn_options: APIConnectOptions,
     ) -> stt.SpeechEvent:
         try:
-            config = self._sanitize_options(language=language)
+            if is_given(language):
+                self._opts.language = language
             data_uri = fal_client.encode(
                 rtc.combine_audio_frames(buffer).to_wav_bytes(), "audio/x-wav"
             )
@@ -76,10 +58,10 @@ class WizperSTT(stt.STT):
                 "fal-ai/wizper",
                 arguments={
                     "audio_url": data_uri,
-                    "task": config.task,
-                    "language": config.language,
-                    "chunk_level": config.chunk_level,
-                    "version": config.version,
+                    "task": self._opts.task,
+                    "language": self._opts.language,
+                    "chunk_level": self._opts.chunk_level,
+                    "version": self._opts.version,
                 },
                 timeout=conn_options.timeout,
             )
@@ -88,11 +70,9 @@ class WizperSTT(stt.STT):
         except fal_client.client.FalClientError as e:
             raise APIConnectionError() from e
 
-    def _transcription_to_speech_event(
-        self, event_type=SpeechEventType.FINAL_TRANSCRIPT, text=None
-    ) -> stt.SpeechEvent:
+    def _transcription_to_speech_event(self, text: str) -> stt.SpeechEvent:
         return stt.SpeechEvent(
-            type=event_type,
+            type=SpeechEventType.FINAL_TRANSCRIPT,
             alternatives=[stt.SpeechData(text=text, language=self._opts.language)],
         )
 

livekit/plugins/fal/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "1.0.0.dev5"
+__version__ = '1.0.0.rc2'

{livekit_plugins_fal-1.0.0.dev5.dist-info → livekit_plugins_fal-1.0.0rc2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: livekit-plugins-fal
-Version: 1.0.0.dev5
+Version: 1.0.0rc2
 Summary: fal plugin template for LiveKit Agents
 Project-URL: Documentation, https://docs.livekit.io
 Project-URL: Website, https://livekit.io/
@@ -19,7 +19,7 @@ Classifier: Topic :: Multimedia :: Video
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.9.0
 Requires-Dist: fal-client
-Requires-Dist: livekit-agents>=1.0.0.dev5
+Requires-Dist: livekit-agents>=1.0.0.rc2
 Description-Content-Type: text/markdown
 
 # LiveKit Plugins fal

livekit_plugins_fal-1.0.0rc2.dist-info/RECORD

@@ -0,0 +1,8 @@
+livekit/plugins/fal/__init__.py,sha256=2yFGbnRBL846WEDry3a4CTTwWSiLSI4Otqhlo6g4Lf4,1476
+livekit/plugins/fal/log.py,sha256=E03cKaUVgXY2O7RpX33IQIaRCpFJ_nJMSkRCvCxX5fc,66
+livekit/plugins/fal/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+livekit/plugins/fal/stt.py,sha256=i7kWKvftNbOZFaJzsrTz3gcO06m1dUnfQaWG8vAGSBo,2688
+livekit/plugins/fal/version.py,sha256=9dPToS7uQhNVGiqP6fLx_l0Akub-11rRUoyDaiS4_CU,603
+livekit_plugins_fal-1.0.0rc2.dist-info/METADATA,sha256=fr1nbE2tmZA8MMtz191NrEyvVI9pi-PhrIY9DLEGDj4,1286
+livekit_plugins_fal-1.0.0rc2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+livekit_plugins_fal-1.0.0rc2.dist-info/RECORD,,

livekit_plugins_fal-1.0.0.dev5.dist-info/RECORD

@@ -1,8 +0,0 @@
-livekit/plugins/fal/__init__.py,sha256=2yFGbnRBL846WEDry3a4CTTwWSiLSI4Otqhlo6g4Lf4,1476
-livekit/plugins/fal/log.py,sha256=E03cKaUVgXY2O7RpX33IQIaRCpFJ_nJMSkRCvCxX5fc,66
-livekit/plugins/fal/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-livekit/plugins/fal/stt.py,sha256=PsLrehk5tunahzxAzOFA1OmpBHuV72ESCZE4MoqKCAw,3225
-livekit/plugins/fal/version.py,sha256=_S-ahwrVT8tkQeIdVxA-cTboUyLFxWwbotIZ6BTvw6E,604
-livekit_plugins_fal-1.0.0.dev5.dist-info/METADATA,sha256=Hb3vG82vMBctBao94XtlmUFUPvj8OlOoph3TgMdWr4s,1289
-livekit_plugins_fal-1.0.0.dev5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-livekit_plugins_fal-1.0.0.dev5.dist-info/RECORD,,