Merge pull request #100 from video-db/fix-upload-agent-params

add collection_id as required param in upload agent
video-db · Dec 6, 2024 · 88b61b0 · 88b61b0
2 parents c685fc9 + 149e817
commit 88b61b0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/backend/director/agents/upload.py b/backend/director/agents/upload.py
@@ -40,7 +40,7 @@
             "description": "Collection ID to upload the content",
         },
     },
-    "required": ["url", "media_type"],
+    "required": ["url", "media_type", "collection_id"],
 }
 
 

diff --git a/backend/director/tools/videodb_tool.py b/backend/director/tools/videodb_tool.py
@@ -178,8 +178,7 @@ def semantic_search(
             video = self.collection.get_video(video_id)
             search_resuls = video.search(query=query, index_type=index_type, **kwargs)
         else:
-            if index_type == IndexType.scene:
-                kwargs.pop("scene_index_id", None)
+            kwargs.pop("scene_index_id", None)
             search_resuls = self.collection.search(
                 query=query, index_type=index_type, **kwargs
             )
-Original file line number
+Diff line change
@@ Expand Up / @@ -40,7 +40,7 @@ @@
                 "description": "Collection ID to upload the content",
             },
         },
-        "required": ["url", "media_type"],
+        "required": ["url", "media_type", "collection_id"],
     }
@@ Expand Down @@