# -*- coding: utf-8 -*-
from utils.animal_rag import get_animal_rag

rag = get_animal_rag()

queries = [
    '가이시딘 어떻게 써?',
    '심장사상충 예방약 추천',
    '고양이 구충제',
    '강아지 진통제',
    '귀진드기 약',
    '피부염 치료',
    '구토 멈추는 약',
    '항생제 추천',
    '넥스가드 용법',
    '셀라멕틴 스팟온'
]

print("=" * 60)
print("RAG 검색 품질 테스트")
print("=" * 60)

for q in queries:
    results = rag.search(q, n_results=3)
    print(f'\n🔍 "{q}"')
    if not results:
        print('  ❌ 검색 결과 없음 (score < 0.3)')
    else:
        for r in results:
            print(f"  [{r['score']:.0%}] {r['source']} - {r['section']}")
            # 첫 100자 미리보기
            preview = r['text'][:100].replace('\n', ' ')
            print(f"      → {preview}...")