sanatan_ai / tests /test_divya_prabandham_verse_fix.py
vikramvasudevan's picture
Upload folder using huggingface_hub
fac55e5 verified
import json
import logging
from config import SanatanConfig
from db import SanatanDatabase
from metadata import MetadataFilter, MetadataWhereClause
if __name__ == "__main__":
logging.basicConfig()
collection_name = "divya_prabandham_taniyans"
database = SanatanDatabase()
# database.delete_taniyans_in_divya_prabandham()
# database.fix_taniyans_in_divya_prabandham()
config = SanatanConfig()
c = config.get_scripture_by_name("divya_prabandham_taniyans")
database.build_global_index_for_scripture(c)
results = database.get(collection_name,None,n_results=2)
print(results)