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

Ankit/add title search

Author: ankit-v2-3

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]"

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:

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,