8000 Add rerank_params in semantic search · video-db/videodb-python@710ec9a · GitHub
[go: up one dir, main page]
Skip to content

Commit 710ec9a

Browse files
ashish-spextashish-spext
committed
Add rerank_params in semantic search
1 parent 60d7019 commit 710ec9a

File tree

4 files changed

+11
-0
lines changed

4 files changed

+11
-0
lines changed

‎videodb/_constants.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@ class SemanticSearchDefaultValues:
3737
result_threshold = 5
3838
score_threshold = 0.2
3939
rerank = False
40+
rerank_param = {}
4041

4142

4243
class Segmenter:

‎videodb/collection.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -389,6 +389,7 @@ def search(
389389
score_threshold: Optional[float] = None,
390390
dynamic_score_percentage: Optional[float] = None,
391391
rerank: bool = SemanticSearchDefaultValues.rerank,
392+
rerank_param: dict = SemanticSearchDefaultValues.rerank_param,
392393
filter: List[Dict[str, Any]] = [],
393394
) -> SearchResult:
394395
"""Search for a query in the collection.
@@ -400,6 +401,7 @@ def search(
400401
:param float score_threshold: Threshold score for the search (optional)
401402
:param float dynamic_score_percentage: Percentage of dynamic score to consider (optional)
402403
:param bool rerank: Rerank search results (optional)
404+
:param dict rerank_param: Parameters for reranking (optional)
403405
:raise SearchError: If the search fails
404406
:return: :class:`SearchResult <SearchResult>` object
405407
:rtype: :class:`videodb.search.SearchResult`
@@ -414,6 +416,7 @@ def search(
414416
score_threshold=score_threshold,
415417
dynamic_score_percentage=dynamic_score_percentage,
416418
rerank=rerank,
419+
rerank_param=rerank_param,
417420
filter=filter,
418421
)
419422

‎videodb/search.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,7 @@ def search_inside_video(
124124
score_threshold: Optional[float] = None,
125125
dynamic_score_percentage: Optional[float] = None,
126126
rerank: bool = SemanticSearchDefaultValues.rerank,
127+
rerank_param: dict = SemanticSearchDefaultValues.rerank_param,
127128
**kwargs,
128129
):
129130
search_data = self._connection.post(
@@ -138,6 +139,7 @@ def search_inside_video(
< 10000 /div>
138139
or SemanticSearchDefaultValues.result_threshold,
139140
"dynamic_score_percentage": dynamic_score_percentage,
140141
"rerank": rerank,
142+
"rerank_param": rerank_param,
141143
**kwargs,
142144
},
143145
)
@@ -153,6 +155,7 @@ def search_inside_collection(
153155
score_threshold: Optional[float] = None,
154156
dynamic_score_percentage: Optional[float] = None,
155157
rerank: bool = SemanticSearchDefaultValues.rerank,
158+
rerank_param: dict = SemanticSearchDefaultValues.rerank_param,
156159
**kwargs,
157160
):
158161
search_data = self._connection.post(
@@ -167,6 +170,7 @@ def search_inside_collection(
167170
or SemanticSearchDefaultValues.result_threshold,
168171
"dynamic_score_percentage": dynamic_score_percentage,
169172
"rerank": rerank,
173+
"rerank_param": rerank_param,
170174
**kwargs,
171175
},
172176
)

‎videodb/video.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@ def search(
7171
score_threshold: Optional[float] = None,
7272
dynamic_score_percentage: Optional[float] = None,
7373
rerank: bool = SemanticSearchDefaultValues.rerank,
74+
rerank_param: dict = SemanticSearchDefaultValues.rerank_param,
7475
filter: List[Dict[str, Any]] = [],
7576
**kwargs,
7677
) -> SearchResult:
@@ -83,6 +84,7 @@ def search(
8384
:param float score_threshold: (optional) Threshold score for the search
8485
:param float dynamic_score_percentage: (optional) Percentage of dynamic score to consider
8586
:param bool rerank: (optional) Rerank search results
87+
:param dict rerank_param: (optional) Parameters for reranking
8688
:raise SearchError: If the search fails
8789
:return: :class:`SearchResult <SearchResult>` object
8890
:rtype: :class:`videodb.search.SearchResult`
@@ -97,6 +99,7 @@ def search(
9799
score_threshold=score_threshold,
98100
dynamic_score_percentage=dynamic_score_percentage,
99101
rerank=rerank,
102+
rerank_param=rerank_param,
100103
filter=filter,
101104
**kwargs,
102105
)

0 commit comments

Comments
 (0)
0