feat(search-v2): add search config trace support

videodb/collection.py

@@ -462,7 +462,13 @@ def dub_video(
         if dub_data:
             return Video(self._connection, **dub_data)
 
-    def search(self, query: str, *args, **kwargs) -> Union[SearchResponse, SearchResult, RTStreamSearchResult]:
+    def search(
+        self,
+        query: str,
+        *args,
+        config: Optional[Dict[str, Any]] = None,
+        **kwargs,
+    ) -> Union[SearchResponse, SearchResult, RTStreamSearchResult]:
         """Search the collection.
 
         New search is used by default. Calls that use legacy-shaped parameters are
@@ -485,9 +491,13 @@ def search(self, query: str, *args, **kwargs) -> Union[SearchResponse, SearchRes
             "return_fields",
             "include_clip",
             "session_id",
+            "config",
         }
         unsupported_params = {"index_name", "index_names", "index_id", "index_ids"}
 
+        if config is not None:
+            kwargs["config"] = config
+
         if args:
             legacy_arg_names = [
                 "search_type",

videodb/search.py

@@ -194,6 +194,7 @@ def __init__(self, _connection, **kwargs):
         self.session_id = kwargs.get("session_id")
         self.waiting_for = kwargs.get("waiting_for") or "none"
         self.clarification = kwargs.get("clarification")
+        self.trace = kwargs.get("trace")
         raw_results = kwargs.get("results", [])
         if self.response_type in {"shots", "deepsearch"}:
             self.results = SearchResult(_connection, results=raw_results)

videodb/video.py

@@ -84,7 +84,13 @@ def update(self, name: Optional[str] = None) -> None:
         if name is not None:
             self.name = response_data.get("name", name)
 
-    def search(self, query: str, *args, **kwargs) -> Union[SearchResponse, SearchResult]:
+    def search(
+        self,
+        query: str,
+        *args,
+        config: Optional[Dict[str, Any]] = None,
+        **kwargs,
+    ) -> Union[SearchResponse, SearchResult]:
         """Search this video.
 
         New search is used by default. Calls that use legacy-shaped parameters are
@@ -107,9 +113,13 @@ def search(self, query: str, *args, **kwargs) -> Union[SearchResponse, SearchRes
             "return_fields",
             "include_clip",
             "session_id",
+            "config",
         }
         unsupported_params = {"index_name", "index_names", "index_id", "index_ids"}
 
+        if config is not None:
+            kwargs["config"] = config
+
         if args:
             legacy_arg_names = [
                 "search_type",