sanatan_ai / tests /test_yt_search.py
vikramvasudevan's picture
Upload folder using huggingface_hub
6193938 verified
import json
import logging
from db import SanatanDatabase
from metadata import MetadataFilter, MetadataWhereClause
if __name__ == "__main__":
logging.basicConfig()
collection_name = "yt_metadata"
database = SanatanDatabase()
print("count = ", database.count(collection_name))
response = database.fetch_first_match(
collection_name=collection_name,
metadata_where_clause=MetadataWhereClause(filters=[
# MetadataFilter(metadata_field="channel_id", metadata_search_operator="$eq", metadata_value="UCqa48rNanVRKmG4qxl-YmEQ"),
MetadataFilter(metadata_field="channel_title", metadata_search_operator="$eq", metadata_value="sookthi"),
MetadataFilter(metadata_field="video_title", metadata_search_operator="$eq", metadata_value="krishna mangalam")
])
)
print(json.dumps(response, indent=1, ensure_ascii=False))