6. 실습 ③ — 출처 인용 검증하기

import re
def check_citation(answer, n_docs):
    # ① 인용 번호가 하나라도 있는가
    found = re.findall(r"\[(\d+)\]", answer)
    if not found:
        return False, "인용 번호가 전혀 없음"
    # 문자열 번호를 숫자로 바꾸기
    nums = []
    for n in found:
        nums.append(int(n))
    # ② 존재하지 않는 번호를 인용하지 않았는가
    bad = []
    for n in nums:
        if n < 1 or n > n_docs:
            bad.append(n)
    if bad:
        return False, f"존재하지 않는 근거 번호: {bad}"
    return True, f"인용 {len(nums)}건 정상"
# 사용 예
docs = search("환불은 며칠 이내인가요?")
ans  = chain.invoke({"context": build_context(docs),
                     "question": "환불은 며칠 이내인가요?"})
ok, msg = check_citation(ans, len(docs))
print("인용 검증:", "✓" if ok else "✗", msg)