31 lines
978 B
Python
31 lines
978 B
Python
import sys
|
|
from app.services.rag_service import retrieve, answer_query
|
|
from app.services.query_intent import extract_max_price, extract_brand_mention, is_count_query
|
|
|
|
def run_test():
|
|
queries = [
|
|
"How many products are there in Cadbury?",
|
|
"Suggest low sugar biscuits",
|
|
"Recommend Paneer under ₹150",
|
|
"Suggest panner less than ₹100",
|
|
]
|
|
|
|
for q in queries:
|
|
print("="*60)
|
|
print(f"QUERY: {q}")
|
|
print(f" Is Count: {is_count_query(q)}")
|
|
print(f" Brand Mention: {extract_brand_mention(q)}")
|
|
print(f" Max Price: {extract_max_price(q)}")
|
|
|
|
prods = retrieve(q)
|
|
print(" Retrieved Products:")
|
|
for p in prods[:5]:
|
|
print(f" - {p.brand} | {p.product_name} | Price: {p.price_range} | Cat: {p.category}")
|
|
|
|
ans = answer_query(q)
|
|
print(f" ANSWER:\n{ans.answer}")
|
|
print("="*60)
|
|
|
|
if __name__ == "__main__":
|
|
run_test()
|