1- from typing import Optional , Union , List , Dict , Tuple
1+ from typing import Optional , Union , List , Dict , Tuple , Any
22from videodb ._utils ._video import play_stream
33from videodb ._constants import (
44 ApiPath ,
@@ -54,6 +54,7 @@ def search(
5454 result_threshold : Optional [int ] = None ,
5555 score_threshold : Optional [float ] = None ,
5656 dynamic_score_percentage : Optional [float ] = None ,
57+ filter : List [Dict [str , Any ]] = [],
5758 ** kwargs ,
5859 ) -> SearchResult :
5960 search = SearchFactory (self ._connection ).get_search (search_type )
@@ -65,6 +66,7 @@ def search(
6566 result_threshold = result_threshold ,
6667 score_threshold = score_threshold ,
6768 dynamic_score_percentage = dynamic_score_percentage ,
69+ filter = filter ,
6870 ** kwargs ,
6971 )
7072
@@ -299,6 +301,7 @@ def index_scenes(
299301 extraction_type : SceneExtractionType = SceneExtractionType .shot_based ,
300302 extraction_config : Dict = {},
301303 prompt : Optional [str ] = None ,
304+ metadata : Dict = {},
302305 model_name : Optional [str ] = None ,
303306 model_config : Optional [Dict ] = None ,
304307 name : Optional [str ] = None ,
@@ -311,6 +314,7 @@ def index_scenes(
311314 "extraction_type" : extraction_type ,
312315 "extraction_config" : extraction_config ,
313316 "prompt" : prompt ,
317+ "metadata" : metadata ,
314318 "model_name" : model_name ,
315319 "model_config" : model_config ,
316320 "name" : name ,
0 commit comments