feat: add keyward search

Author: ankit-v2-3

videodb/__init__.py

@@ -5,7 +5,7 @@
 
 from typing import Optional
 from videodb._utils._video import play_stream
-from videodb._constants import VIDEO_DB_API, MediaType
+from videodb._constants import VIDEO_DB_API, MediaType, SearchType
 from videodb.client import Connection
 from videodb.exceptions import (
     VideodbError,
@@ -26,6 +26,7 @@
     "SearchError",
     "play_stream",
     "MediaType",
+    "SearchType",
 ]
 
 

videodb/_constants.py

@@ -11,6 +11,7 @@ class MediaType:
 
 class SearchType:
     semantic = "semantic"
+    keyword = "keyword"
 
 
 class IndexType:

videodb/search.py

@@ -148,7 +148,35 @@ def search_inside_collection(
         return SearchResult(self._connection, **search_data)
 
 
-search_type = {SearchType.semantic: SemanticSearch}
+class KeywordSearch(Search):
+    def __init__(self, _connection):
+        self._connection = _connection
+
+    def search_inside_video(
+        self,
+        video_id: str,
+        query: str,
+        result_threshold: Optional[int] = None,
+        score_threshold: Optional[int] = None,
+        dynamic_score_percentage: Optional[int] = None,
+        **kwargs,
+    ):
+        search_data = self._connection.post(
+            path=f"{ApiPath.video}/{video_id}/{ApiPath.search}",
+            data={
+                "type": SearchType.keyword,
+                "query": query,
+                "score_threshold": score_threshold,
+                "result_threshold": result_threshold,
+            },
+        )
+        return SearchResult(self._connection, **search_data)
+
+    def search_inside_collection(**kwargs):
+        raise NotImplementedError("Keyword search will be implemented in the future")
+
+
+search_type = {SearchType.semantic: SemanticSearch, SearchType.keyword: KeywordSearch}
 
 
 class SearchFactory: