8. 실습 ④ — 14차시 실패 해결 확인

# ==============================================================
# 30차시. PPT 30차시_질의재작성재검색. 코드 폴더는 src/30
# PPT: 30차시 검색어 재작성
# 역할: 일상 표현 질문으로 graph30을 돌려 재작성 횟수, 최종 query, tried_queries, grade를 출력합니다.
# 읽는 키: question으로 만든 초기 State
# 쓰는 키: 화면 출력
# 그래프:
#   classify --> retrieve --> (실패 시) rewrite --> retrieve --> generate --> verify
# --------------------------------------------------------------
# 파이썬만 아는 분을 위한 말
# State는 노드들이 함께 쓰는 딕셔너리입니다. 함수는 state를 읽고, 바꿀 키만 return 합니다.
# return에 적지 않은 키는 이전 값이 남습니다.
# log와 tried_queries는 리스트라서, 새로 return한 항목이 뒤에 이어 붙습니다.
# Retriever: 검색 담당. 질문과 비슷한 문서 조각만 고릅니다. 답 문장은 만들지 않습니다.
# Generator: 고른 문서를 근거로 답 문장을 만듭니다. 문서가 없으면 안내 문장만 돌려줍니다.
# Verifier: 답이 근거와 맞는지 검사하고 grade를 적습니다.
#   grade=pass     답을 확정하고 그래프를 끝냅니다.
#   grade=retry    같은 문서로 답 문장만 다시 씁니다. retries가 1 늘어납니다.
#   grade=research 문서가 부족합니다. 검색어를 바꾼 뒤 다시 찾습니다. rewrites가 늘어납니다.
#   grade=giveup   더 진행하지 않고 fallback으로 안내 문장만 돌려줍니다.
# fallback: 실패했을 때 답을 지어내지 않고 안내 문장을 반환하는 노드입니다.
# intent: 질문 종류. greeting 인사, calc 계산, scope 문서 밖, document 문서 검색.
# rewrite: 사용자가 친 question은 유지하고, 검색에 쓰는 query만 바꿉니다.
# safe_node: 노드를 try로 감쌉니다. 오류가 나도 그래프는 멈추지 않고 node_error에 적습니다.
# QUESTIONS 세 개는 구어체라 첫 검색이 실패하기 쉬운 예시입니다.
# recursion_limit은 30입니다. rewrite 왕복이 있어도 그래프가 무한히 돌지 않게 막는 상한입니다.
# ==============================================================

# ============================================================
# 30차시 - Rewrite 테스트
# 검색에 실패하기 쉬운 질문을 그래프에 전달합니다.
# Rewrite 후 재검색이 정상적으로 이루어지는지 확인합니다.
# ============================================================

import os
import sys

# ------------------------------------------------------------
# 필요한 모듈의 경로를 설정합니다.
# ------------------------------------------------------------

# 현재 파일이 있는 30차시 폴더입니다.
CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))

# 공통 파일이 있는 src 폴더입니다.
SRC_DIR = os.path.abspath(os.path.join(CURRENT_DIR, ".."))

# src 폴더를 Python 검색 경로에 추가합니다.
sys.path.insert(0, SRC_DIR)

# 이전 차시에서 만든 모듈을 사용할 수 있게 합니다.
for folder in ["18", "19", "24", "25", "26", "29"]:

    path = os.path.join(SRC_DIR, folder)

    sys.path.append(path)

# 현재 30차시 모듈을 가장 먼저 찾도록 합니다.
sys.path.insert(0, CURRENT_DIR)

# ------------------------------------------------------------
# 30차시 그래프와 초기 State 생성 함수를 가져옵니다.
# ------------------------------------------------------------

from graph30 import app
from graph_state2 import make_initial_state

# ============================================================
# 테스트 질문
# 일상적인 표현 때문에 검색에 실패하기 쉬운 질문입니다.
# ============================================================

QUESTIONS = [
    "반품하고 싶은데 언제까지 가능해요?",
    "물건 바꾸려면 택배비 누가 내나요?",
    "돈 돌려받으려면 어떻게 해요?",
]

# ============================================================
# 테스트 실행
# 각 질문을 그래프에 전달하고 Rewrite 결과를 확인합니다.
# ============================================================

for question in QUESTIONS:

    # 질문 구분선을 출력합니다.
    print("\n" + "=" * 60)
    print("Q:", question)
    print("=" * 60)

    # --------------------------------------------------------
    # 1. 질문으로 초기 State를 만듭니다.
    # --------------------------------------------------------

    state = make_initial_state(question)

    # --------------------------------------------------------
    # 2. LangGraph를 실행합니다.
    # --------------------------------------------------------

    result = app.invoke(
        state,
        {
            "recursion_limit": 30
        }
    )

    # --------------------------------------------------------
    # 3. 최종 답변을 출력합니다.
    # --------------------------------------------------------

    print(
        "A:",
        result.get("answer", "")
    )

    # --------------------------------------------------------
    # 4. Rewrite 횟수를 출력합니다.
    # --------------------------------------------------------

    print(
        "재작성:",
        result.get("rewrites", 0),
        "회"
    )

    # --------------------------------------------------------
    # 5. 마지막으로 사용한 검색어를 출력합니다.
    # --------------------------------------------------------

    print(
        "최종 검색어:",
        result.get("query", "")
    )

    # --------------------------------------------------------
    # 6. 지금까지 사용한 검색어를 모두 출력합니다.
    # --------------------------------------------------------

    print(
        "시도한 검색어:",
        result.get("tried_queries", [])
    )

    # --------------------------------------------------------
    # 7. 최종 검증 결과를 출력합니다.
    # --------------------------------------------------------

    print(
        "최종 판정:",
        result.get("grade", "")
    )