# -*- coding: utf-8 -*-
"""튜닝 후 테스트 (기본값 사용)"""
from utils.animal_rag import get_animal_rag

# 새로 import해서 변경사항 적용
import importlib
import utils.animal_rag
importlib.reload(utils.animal_rag)

rag = utils.animal_rag.get_animal_rag()

queries = [
    '가이시딘',
    '개시딘 피부염',
    '심장사상충 예방약',
    '강아지 구토약',
]

print("=" * 60)
print("튜닝 후 테스트 (n_results=5, threshold=0.2)")
print("=" * 60)

for q in queries:
    # 기본값 사용 (n_results=5)
    results = rag.search(q)
    print(f'\n🔍 "{q}" - {len(results)}개 결과')
    for r in results:
        print(f"  [{r['score']:.0%}] {r['source'][:30]} - {r['section'][:20]}")