Merge pull request #28 from video-db/ankit/add-title-search

Ankit/add title search
video-db · Nov 6, 2024 · f8f5ed0 · f8f5ed0
2 parents c67327e + 668988a
commit f8f5ed0
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 1 deletion.
diff --git a/videodb/__about__.py b/videodb/__about__.py
@@ -1,7 +1,7 @@
 """ About information for videodb sdk"""
 
 
-__version__ = "0.2.5"
+__version__ = "0.2.6"
 __title__ = "videodb"
 __author__ = "videodb"
 __email__ = "[email protected]"

diff --git a/videodb/_constants.py b/videodb/_constants.py
@@ -15,6 +15,7 @@ class SearchType:
     semantic = "semantic"
     keyword = "keyword"
     scene = "scene"
+    llm = "llm"
 
 
 class IndexType:
@@ -68,6 +69,7 @@ class ApiPath:
     describe = "describe"
     storage = "storage"
     download = "download"
+    title = "title"
 
 
 class Status:

diff --git a/videodb/collection.py b/videodb/collection.py
@@ -117,6 +117,19 @@ def search(
             dynamic_score_percentage=dynamic_score_percentage,
         )
 
+    def search_title(self, query) -> List[Video]:
+        search_data = self._connection.post(
+            path=f"{ApiPath.collection}/{self.id}/{ApiPath.search}/{ApiPath.title}",
+            data={
+                "query": query,
+                "search_type": SearchType.llm,
+            },
+        )
+        return [
+            {"video": Video(self._connection, **result.get("video"))}
+            for result in search_data
+        ]
+
     def upload(
         self,
         file_path: str = None,