# -*- coding: utf-8 -*-
"""임계값 없이 raw 검색 결과 확인"""
from utils.animal_rag import get_animal_rag

rag = get_animal_rag()
rag._init_db()

queries = [
    '가이시딘',      # 오타 버전
    '개시딘',        # 정확한 이름
    '개시딘 겔',
    '피부 농피증',
    '후시딘',        # 사람용 약 이름으로 검색
]

print("=" * 70)
print("임계값 제거 후 RAW 검색 결과 (상위 5개)")
print("=" * 70)

for q in queries:
    # 임계값 없이 raw 검색
    query_emb = rag._get_embedding(q)
    results = rag.table.search(query_emb).limit(5).to_list()
    
    print(f'\n🔍 "{q}"')
    for r in results:
        distance = r.get("_distance", 10)
        score = 1 / (1 + distance)
        source = r["source"]
        section = r["section"]
        print(f"  [{score:.1%}] (dist:{distance:.2f}) {source} - {section}")