videosdk-plugins-assemblyai 0.0.34__tar.gz → 0.0.36__tar.gz

{videosdk_plugins_assemblyai-0.0.34 → videosdk_plugins_assemblyai-0.0.36}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: videosdk-plugins-assemblyai
-Version: 0.0.34
+Version: 0.0.36
 Summary: VideoSDK Agent Framework plugin for AssemblyAI
 Author: videosdk
 License-Expression: Apache-2.0
@@ -13,7 +13,7 @@ Classifier: Topic :: Multimedia :: Video
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.11
 Requires-Dist: aiohttp
-Requires-Dist: videosdk-agents>=0.0.34
+Requires-Dist: videosdk-agents>=0.0.36
 Description-Content-Type: text/markdown
 
 # VideoSDK Assembly AI Plugin

{videosdk_plugins_assemblyai-0.0.34 → videosdk_plugins_assemblyai-0.0.36}/pyproject.toml

@@ -20,7 +20,7 @@ classifiers = [
     "Topic :: Multimedia :: Video",
     "Topic :: Scientific/Engineering :: Artificial Intelligence",
 ]
-dependencies = ["videosdk-agents>=0.0.34","aiohttp"]
+dependencies = ["videosdk-agents>=0.0.36","aiohttp"]
 
 [tool.hatch.version]
 path = "videosdk/plugins/assemblyai/version.py"

{videosdk_plugins_assemblyai-0.0.34 → videosdk_plugins_assemblyai-0.0.36}/videosdk/plugins/assemblyai/stt.py

@@ -28,7 +28,7 @@ class AssemblyAISTT(BaseSTT):
         input_sample_rate: int = 48000,
         target_sample_rate: int = 16000,
         format_turns: bool = True,
-        word_boost: list[str] | None = None,
+        keyterms_prompt: list[str] | None = None,
         end_of_turn_confidence_threshold: float = 0.5,
         min_end_of_turn_silence_when_confident: int = 800,
         max_turn_silence: int = 2000,
@@ -40,7 +40,7 @@ class AssemblyAISTT(BaseSTT):
             input_sample_rate (int): The input sample rate to use for the STT plugin. Defaults to 48000.
             target_sample_rate (int): The target sample rate to use for the STT plugin. Defaults to 16000.
             format_turns (bool): Whether to format turns. Defaults to True.
-            word_boost (list[str] | None, optional): The word boost to use for the STT plugin. Defaults to None.
+            keyterms_prompt (list[str] | None, optional): The word boost list to use for the STT plugin. Defaults to None.
             end_of_turn_confidence_threshold (float): The end of turn confidence threshold to use for the STT plugin. Defaults to 0.5.
             min_end_of_turn_silence_when_confident (int): The minimum end of turn silence when confident to use for the STT plugin. Defaults to 800.
             max_turn_silence (int): The maximum turn silence to use for the STT plugin. Defaults to 2000.
@@ -59,7 +59,7 @@ class AssemblyAISTT(BaseSTT):
         self.input_sample_rate = input_sample_rate
         self.target_sample_rate = target_sample_rate
         self.format_turns = format_turns
-        self.word_boost = word_boost or []
+        self.keyterms_prompt = keyterms_prompt or []
         self.end_of_turn_confidence_threshold = end_of_turn_confidence_threshold
         self.min_end_of_turn_silence_when_confident = min_end_of_turn_silence_when_confident
         self.max_turn_silence = max_turn_silence
@@ -77,8 +77,8 @@ class AssemblyAISTT(BaseSTT):
         if self.max_turn_silence != 3000:
             connection_params["max_turn_silence"] = self.max_turn_silence
         
-        if self.word_boost:
-            connection_params["word_boost"] = json.dumps(self.word_boost)
+        if self.keyterms_prompt:
+            connection_params["keyterms_prompt"] = json.dumps(self.keyterms_prompt)
 
         self.ws_url = f"wss://streaming.assemblyai.com/v3/ws?{urlencode(connection_params)}"
         logger.info(f"[AssemblyAI] WebSocket URL: {self.ws_url}")
@@ -162,6 +162,7 @@ class AssemblyAISTT(BaseSTT):
             
         headers = {
             "Authorization": self.api_key,
+            "User-Agent": "AssemblyAI/1.0 (integration=VideoSDK)"
         }
         
         try:
@@ -293,4 +294,4 @@ class AssemblyAISTT(BaseSTT):
             
         if self._session:
             await self._session.close()
-            self._session = None
+            self._session = None

videosdk_plugins_assemblyai-0.0.36/videosdk/plugins/assemblyai/version.py

@@ -0,0 +1 @@
+__version__ = "0.0.36"

videosdk_plugins_assemblyai-0.0.34/videosdk/plugins/assemblyai/version.py

@@ -1 +0,0 @@
-__version__ = "0.0.34"