Merge pull request video-db#29 from video-db/ankit/add-search-filter

feat: add search filters

Author: ankit-v2-3

videodb/__about__.py

@@ -1,7 +1,7 @@
 """ About information for videodb sdk"""
 
 
-__version__ = "0.2.6"
+__version__ = "0.2.7"
 __title__ = "videodb"
 __author__ = "videodb"
 __email__ = "[email protected]"

videodb/collection.py

@@ -4,6 +4,8 @@
     Optional,
     Union,
     List,
+    Dict,
+    Any,
 )
 from videodb._upload import (
     upload,
@@ -105,6 +107,7 @@ def search(
         result_threshold: Optional[int] = None,
         score_threshold: Optional[float] = None,
         dynamic_score_percentage: Optional[float] = None,
+        filter: List[Dict[str, Any]] = [],
     ) -> SearchResult:
         search = SearchFactory(self._connection).get_search(search_type)
         return search.search_inside_collection(
@@ -115,6 +118,7 @@ def search(
             result_threshold=result_threshold,
             score_threshold=score_threshold,
             dynamic_score_percentage=dynamic_score_percentage,
+            filter=filter,
         )
 
     def search_title(self, query) -> List[Video]:

videodb/video.py

@@ -1,4 +1,4 @@
-from typing import Optional, Union, List, Dict, Tuple
+from typing import Optional, Union, List, Dict, Tuple, Any
 from videodb._utils._video import play_stream
 from videodb._constants import (
     ApiPath,
@@ -54,6 +54,7 @@ def search(
         result_threshold: Optional[int] = None,
         score_threshold: Optional[float] = None,
         dynamic_score_percentage: Optional[float] = None,
+        filter: List[Dict[str, Any]] = [],
         **kwargs,
     ) -> SearchResult:
         search = SearchFactory(self._connection).get_search(search_type)
@@ -65,6 +66,7 @@ def search(
             result_threshold=result_threshold,
             score_threshold=score_threshold,
             dynamic_score_percentage=dynamic_score_percentage,
+            filter=filter,
             **kwargs,
         )
 
@@ -299,6 +301,7 @@ def index_scenes(
         extraction_type: SceneExtractionType = SceneExtractionType.shot_based,
         extraction_config: Dict = {},
         prompt: Optional[str] = None,
+        metadata: Dict = {},
         model_name: Optional[str] = None,
         model_config: Optional[Dict] = None,
         name: Optional[str] = None,
@@ -311,6 +314,7 @@ def index_scenes(
                 "extraction_type": extraction_type,
                 "extraction_config": extraction_config,
                 "prompt": prompt,
+                "metadata": metadata,
                 "model_name": model_name,
                 "model_config": model_config,
                 "name": name,