83ecf88bd4
## APC 코드 체계 확장 - 기존: 023%만 검색 (~2023년 제품만) - 변경: 02% OR 92% + 13자리 검증 - 02%: 2023년 이전 item_seq (9자리) 기반 APC - 92%: 2024년 이후 item_seq (10자리) 기반 APC - 999% 등 청구프로그램 임의코드는 제외 ## 동물약 챗봇 피부약 추천 개선 - 피부약 2단계 추천 구조 추가 - 1차(치료): 의약품 (개시딘겔, 테르비덤 등) - 2차(보조케어): 의약외품 (스킨카솔 - 회복기 피부보호) - 스킨카솔은 의약외품임을 명시하여 치료제로 오인 방지 ## 기타 - RAG 테스트 스크립트 추가 - 수인약품 API 문서화
22 lines
590 B
Python
22 lines
590 B
Python
# -*- coding: utf-8 -*-
|
|
import importlib
|
|
import utils.animal_rag as ar
|
|
importlib.reload(ar)
|
|
|
|
rag = ar.AnimalDrugRAG()
|
|
rag._init_db()
|
|
|
|
queries = ['구토 멈추는 약', '구토 치료제', '마로피턴트', '세레니아']
|
|
|
|
for q in queries:
|
|
query_emb = rag._get_embedding(q)
|
|
results = rag.table.search(query_emb).limit(5).to_list()
|
|
|
|
print(f'\n🔍 "{q}"')
|
|
for r in results:
|
|
dist = r.get('_distance', 10)
|
|
score = 1 / (1 + dist)
|
|
source = r['source'][:40]
|
|
section = r['section'][:25]
|
|
print(f' [{score:.0%}] {source} - {section}')
|