Skip to content

Commit 6e65feb

Browse files
committed
Give custom search type support
1 parent 72c07ae commit 6e65feb

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

videodb/_constants.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ class SearchType:
1717
keyword = "keyword"
1818
scene = "scene"
1919
llm = "llm"
20+
custom = "custom"
2021

2122

2223
class IndexType:

videodb/search.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -256,6 +256,7 @@ def search_inside_collection(self, **kwargs):
256256
SearchType.semantic: SemanticSearch,
257257
SearchType.keyword: KeywordSearch,
258258
SearchType.scene: SceneSearch,
259+
SearchType.custom: SemanticSearch,
259260
}
260261

261262

0 commit comments

Comments
 (0)