# -*- coding: utf-8 -*-
"""개선된 RAG 테스트"""
import importlib
import utils.animal_rag
importlib.reload(utils.animal_rag)

rag = utils.animal_rag.AnimalDrugRAG()

queries = [
    '가이시딘',
    '개시딘',
    '개시딘 피부염',
    '심장사상충 예방약',
    '강아지 구토약',
    '고양이 귀진드기',
    '넥스가드',
    '후시딘 동물용',
]

print("=" * 70)
print("🎯 개선된 RAG 테스트 (prefix 추가 후)")
print("=" * 70)

for q in queries:
    results = rag.search(q)
    print(f'\n🔍 "{q}" - {len(results)}개 결과')
    for r in results[:3]:  # 상위 3개만
        product = r.get('product_name', '')[:20] if 'product_name' in r else ''
        print(f"  [{r['score']:.0%}] {r['source'][:35]}")
        # 청크 prefix 확인
        text_preview = r['text'][:80].replace('\n', ' ')
        print(f"      → {text_preview}...")