diff --git a/backend/director/agents/upload.py b/backend/director/agents/upload.py index 522cc56..eeafad7 100644 --- 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 index ce96205..68d4b18 100644 --- 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 )