Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
File size: 1,687 Bytes
19d30fe 7b33394 19d30fe |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
import json
import logging
from db import SanatanDatabase
from metadata import MetadataFilter, MetadataWhereClause
if __name__ == "__main__":
logging.basicConfig()
collection_name = "divya_prabandham"
database = SanatanDatabase()
print("count = ", database.count(collection_name))
while True:
query = input("Search for: ")
if query.strip() == "":
break
# response = database.search_for_literal(
# collection_name=collection_name, literal_to_search_for=query, n_results=1
# )
metadata_where_clause = MetadataWhereClause(
filters=[
MetadataFilter(
metadata_field="prabandham_code",
metadata_search_operator="$eq",
metadata_value="NAT",
),
# MetadataFilter(
# metadata_field="decade",
# metadata_search_operator="$eq",
# metadata_value=4,
# ),
MetadataFilter(
metadata_field="position_in_chapter",
metadata_search_operator="$eq",
metadata_value=5,
),
]
)
response = database.search(
collection_name=collection_name,
query=query,
metadata_where_clause=metadata_where_clause,
n_results=5,
)
print("Matches", json.dumps(response["metadatas"][0], ensure_ascii=False, indent=1))
# print("Document: ")
# print(response["documents"][0][0])
# print("Metadata: ")
# print(response["metadatas"][0][0]) |