videosdk-plugins-assemblyai 0.0.33__tar.gz → 0.0.35__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of videosdk-plugins-assemblyai might be problematic. Click here for more details.
- {videosdk_plugins_assemblyai-0.0.33 → videosdk_plugins_assemblyai-0.0.35}/PKG-INFO +2 -2
- {videosdk_plugins_assemblyai-0.0.33 → videosdk_plugins_assemblyai-0.0.35}/pyproject.toml +1 -1
- {videosdk_plugins_assemblyai-0.0.33 → videosdk_plugins_assemblyai-0.0.35}/videosdk/plugins/assemblyai/stt.py +7 -6
- videosdk_plugins_assemblyai-0.0.35/videosdk/plugins/assemblyai/version.py +1 -0
- videosdk_plugins_assemblyai-0.0.33/videosdk/plugins/assemblyai/version.py +0 -1
- {videosdk_plugins_assemblyai-0.0.33 → videosdk_plugins_assemblyai-0.0.35}/.gitignore +0 -0
- {videosdk_plugins_assemblyai-0.0.33 → videosdk_plugins_assemblyai-0.0.35}/README.md +0 -0
- {videosdk_plugins_assemblyai-0.0.33 → videosdk_plugins_assemblyai-0.0.35}/videosdk/plugins/assemblyai/__init__.py +0 -0
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: videosdk-plugins-assemblyai
|
|
3
|
-
Version: 0.0.
|
|
3
|
+
Version: 0.0.35
|
|
4
4
|
Summary: VideoSDK Agent Framework plugin for AssemblyAI
|
|
5
5
|
Author: videosdk
|
|
6
6
|
License-Expression: Apache-2.0
|
|
@@ -13,7 +13,7 @@ Classifier: Topic :: Multimedia :: Video
|
|
|
13
13
|
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
|
|
14
14
|
Requires-Python: >=3.11
|
|
15
15
|
Requires-Dist: aiohttp
|
|
16
|
-
Requires-Dist: videosdk-agents>=0.0.
|
|
16
|
+
Requires-Dist: videosdk-agents>=0.0.35
|
|
17
17
|
Description-Content-Type: text/markdown
|
|
18
18
|
|
|
19
19
|
# VideoSDK Assembly AI Plugin
|
|
@@ -20,7 +20,7 @@ classifiers = [
|
|
|
20
20
|
"Topic :: Multimedia :: Video",
|
|
21
21
|
"Topic :: Scientific/Engineering :: Artificial Intelligence",
|
|
22
22
|
]
|
|
23
|
-
dependencies = ["videosdk-agents>=0.0.
|
|
23
|
+
dependencies = ["videosdk-agents>=0.0.35","aiohttp"]
|
|
24
24
|
|
|
25
25
|
[tool.hatch.version]
|
|
26
26
|
path = "videosdk/plugins/assemblyai/version.py"
|
|
@@ -28,7 +28,7 @@ class AssemblyAISTT(BaseSTT):
|
|
|
28
28
|
input_sample_rate: int = 48000,
|
|
29
29
|
target_sample_rate: int = 16000,
|
|
30
30
|
format_turns: bool = True,
|
|
31
|
-
|
|
31
|
+
keyterms_prompt: list[str] | None = None,
|
|
32
32
|
end_of_turn_confidence_threshold: float = 0.5,
|
|
33
33
|
min_end_of_turn_silence_when_confident: int = 800,
|
|
34
34
|
max_turn_silence: int = 2000,
|
|
@@ -40,7 +40,7 @@ class AssemblyAISTT(BaseSTT):
|
|
|
40
40
|
input_sample_rate (int): The input sample rate to use for the STT plugin. Defaults to 48000.
|
|
41
41
|
target_sample_rate (int): The target sample rate to use for the STT plugin. Defaults to 16000.
|
|
42
42
|
format_turns (bool): Whether to format turns. Defaults to True.
|
|
43
|
-
|
|
43
|
+
keyterms_prompt (list[str] | None, optional): The word boost list to use for the STT plugin. Defaults to None.
|
|
44
44
|
end_of_turn_confidence_threshold (float): The end of turn confidence threshold to use for the STT plugin. Defaults to 0.5.
|
|
45
45
|
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.
|
|
46
46
|
max_turn_silence (int): The maximum turn silence to use for the STT plugin. Defaults to 2000.
|
|
@@ -59,7 +59,7 @@ class AssemblyAISTT(BaseSTT):
|
|
|
59
59
|
self.input_sample_rate = input_sample_rate
|
|
60
60
|
self.target_sample_rate = target_sample_rate
|
|
61
61
|
self.format_turns = format_turns
|
|
62
|
-
self.
|
|
62
|
+
self.keyterms_prompt = keyterms_prompt or []
|
|
63
63
|
self.end_of_turn_confidence_threshold = end_of_turn_confidence_threshold
|
|
64
64
|
self.min_end_of_turn_silence_when_confident = min_end_of_turn_silence_when_confident
|
|
65
65
|
self.max_turn_silence = max_turn_silence
|
|
@@ -77,8 +77,8 @@ class AssemblyAISTT(BaseSTT):
|
|
|
77
77
|
if self.max_turn_silence != 3000:
|
|
78
78
|
connection_params["max_turn_silence"] = self.max_turn_silence
|
|
79
79
|
|
|
80
|
-
if self.
|
|
81
|
-
connection_params["
|
|
80
|
+
if self.keyterms_prompt:
|
|
81
|
+
connection_params["keyterms_prompt"] = json.dumps(self.keyterms_prompt)
|
|
82
82
|
|
|
83
83
|
self.ws_url = f"wss://streaming.assemblyai.com/v3/ws?{urlencode(connection_params)}"
|
|
84
84
|
logger.info(f"[AssemblyAI] WebSocket URL: {self.ws_url}")
|
|
@@ -162,6 +162,7 @@ class AssemblyAISTT(BaseSTT):
|
|
|
162
162
|
|
|
163
163
|
headers = {
|
|
164
164
|
"Authorization": self.api_key,
|
|
165
|
+
"User-Agent": "AssemblyAI/1.0 (integration=VideoSDK)"
|
|
165
166
|
}
|
|
166
167
|
|
|
167
168
|
try:
|
|
@@ -293,4 +294,4 @@ class AssemblyAISTT(BaseSTT):
|
|
|
293
294
|
|
|
294
295
|
if self._session:
|
|
295
296
|
await self._session.close()
|
|
296
|
-
self._session = None
|
|
297
|
+
self._session = None
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
__version__ = "0.0.35"
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
__version__ = "0.0.33"
|
|
File without changes
|
|
File without changes
|
|
File without changes
|