Merge pull request #17 from video-db/language-support

ankit-v2-3 · web-flow · commit 588f6e819ec5 · 2024-06-06T12:52:12.000+05:30
Language code support added in spoken word index
diff --git a/videodb/video.py b/videodb/video.py
@@ -133,7 +133,12 @@ def get_transcript_text(self, force: bool = False) -> str:
         self._fetch_transcript(force)
         return self.transcript_text
 
-    def index_spoken_words(self, force: bool = False, callback_url: str = None) -> None:
+    def index_spoken_words(
+        self,
+        language_code: Optional[str] = None,
+        force: bool = False,
+        callback_url: str = None,
+    ) -> None:
         """Semantic indexing of spoken words in the video
 
         :raises InvalidRequestError: If the video is already indexed
@@ -144,6 +149,7 @@ def index_spoken_words(self, force: bool = False, callback_url: str = None) -> N
             path=f"{ApiPath.video}/{self.id}/{ApiPath.index}",
             data={
                 "index_type": IndexType.semantic,
+                "language_code": language_code,
                 "force": force,
                 "callback_url": callback_url,
             },
