fix(python/sdk): fix word_boost for realtime transcriber (#2009)

s0h3yl · mez171 · commit 710d81f25cbd · 2023-09-19T13:56:25.000+01:00
GitOrigin-RevId: dabd8b26960684bd254dfd54f9702fd24aca2897
diff --git a/assemblyai/transcriber.py b/assemblyai/transcriber.py
@@ -1012,7 +1012,7 @@ def connect(
             "sample_rate": self._sample_rate,
         }
         if self._word_boost:
-            params["word_boost"] = self._word_boost
+            params["word_boost"] = json.dumps(self._word_boost)
 
         websocket_base_url = self._client.settings.base_url.replace("https", "wss")
 
diff --git a/tests/unit/test_realtime_transcriber.py b/tests/unit/test_realtime_transcriber.py
@@ -58,7 +58,7 @@ def mocked_websocket_connect(
 
     assert (
         actual_url
-        == f"wss://api.assemblyai.com/v2/realtime/ws?{urlencode(dict(sample_rate=44100, word_boost=['AssemblyAI']))}"
+        == f"wss://api.assemblyai.com/v2/realtime/ws?{urlencode(dict(sample_rate=44100, word_boost=json.dumps(['AssemblyAI'])))}"
     )
     assert actual_additional_headers == {"Authorization": aai.settings.api_key}
     assert actual_open_timeout == 15.0

Original file line number	Diff line number	Diff line change
`@@ -1012,7 +1012,7 @@ def connect(`
`1012`	`1012`	`"sample_rate": self._sample_rate,`
`1013`	`1013`	`}`
`1014`	`1014`	`if self._word_boost:`
`1015`		`- params["word_boost"] = self._word_boost`
	`1015`	`+ params["word_boost"] = json.dumps(self._word_boost)`
`1016`	`1016`
`1017`	`1017`	`websocket_base_url = self._client.settings.base_url.replace("https", "wss")`
`1018`	`1018`
Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,7 @@ def mocked_websocket_connect(`
`58`	`58`
`59`	`59`	`assert (`
`60`	`60`	`actual_url`
`61`		`- == f"wss://api.assemblyai.com/v2/realtime/ws?{urlencode(dict(sample_rate=44100, word_boost=['AssemblyAI']))}"`
	`61`	`+ == f"wss://api.assemblyai.com/v2/realtime/ws?{urlencode(dict(sample_rate=44100, word_boost=json.dumps(['AssemblyAI'])))}"`
`62`	`62`	`)`
`63`	`63`	`assert actual_additional_headers == {"Authorization": aai.settings.api_key}`
`64`	`64`	`assert actual_open_timeout == 15.0`