10. 노드를 이어서 돌려보기

# ============================================================
# 19차시 | Node 이해
# 교안: 19차시_Node이해.pptx
# 역할: LangGraph 없이 노드 함수를 순서대로 호출해 같은 흐름을 손으로 잇는다.
# 입력: 질문 문자열
# 출력: 질문, 답변 앞 80자, 각 노드 log
#
#   State  = manual_run 안의 딕셔너리. 그래프가 갱신해 주지 않으므로
#            노드 return 을 state.update 로 직접 합친다.
#   Node   = retrieve_node, generate_node, fallback_node (다른 파일).
#   Edge   = 여기에는 없다. if state["retrieval_ok"] 가 갈림길 역할을 한다.
#   reducer = 그래프 밖이라 log 가 자동으로 이어 붙지 않는다.
#            new_log = 기존 log + 노드 log 로 직접 붙인 뒤 다시 넣는다.
#            (reducer 가 있을 때의 operator.add 와 같은 일)
#
# 흐름
#   START 자리(함수 시작)
#     |
#     v
#   retrieve_node
#     |
#     +-- retrieval_ok 참 → generate_node → 반환
#     |
#     +-- 거짓 → fallback_node → 반환
#
# 테스트 질문
#   "환불은 며칠 이내인가요?"  → 생성 쪽으로 기대
#   "대표이사 이름은?"        → 안내 쪽으로 기대
# ============================================================

import warnings

from retrieve import retrieve_node
from generate import generate_node
from fallback import fallback_node

# 불필요한 경고 메시지를 숨깁니다.
warnings.filterwarnings("ignore")

# ============================================================
# Node들을 직접 연결하여 실행
# ============================================================
# 노드가 아님. 세 노드를 순서와 if 로 직접 호출하는 함수.
#   읽기: 인자 question, 그리고 각 노드가 돌려준 State 칸
#   쓰기: 함수 지역 state 전체 (그래프 return 이 아님)
def manual_run(question: str):
    """그래프 없이 Node들을 직접 연결하여 실행합니다."""

    # 처음 사용할 State를 만듭니다.
    state = {
        "question": question,
        "query": question,
        "retries": 0,
        "log": [],
        "tried_queries": [],
    }

    # --------------------------------------------------------
    # 1. Retriever Node 실행
    # retrieve_node 읽기: query, question / 쓰기: documents, scores, retrieval_ok, log
    # --------------------------------------------------------

    # 질문과 관련된 문서를 검색합니다.
    result = retrieve_node(state)

    # 검색 과정의 로그를 따로 보관합니다.
    # 그래프 reducer 가 없으므로 log 를 직접 이어 붙인다.
    new_log = state["log"] + result["log"]

    # 검색 결과를 State에 추가합니다.
    state.update(result)

    # 기존 로그와 검색 로그를 합칩니다.
    state["log"] = new_log

    # --------------------------------------------------------
    # 2. 검색 성공 여부에 따라 분기
    # 조건부 엣지 대신 if. 참이면 generate, 거짓이면 fallback.
    # --------------------------------------------------------

    # 검색에 성공하면 답변을 생성합니다.
    # generate_node 읽기: documents, question / 쓰기: answer, insufficient, has_citation, log
    if state["retrieval_ok"]:
        result = generate_node(state)

    # 검색에 실패하면 안내 메시지를 만듭니다.
    # fallback_node 읽기: tried_queries / 쓰기: answer, grade, log
    else:
        result = fallback_node(state)

    # --------------------------------------------------------
    # 3. 두 번째 Node의 결과를 State에 추가
    # --------------------------------------------------------

    # 지금까지의 로그와 새로운 로그를 합칩니다.
    new_log = state["log"] + result["log"]

    # Node 실행 결과를 State에 추가합니다.
    state.update(result)

    # 누적된 로그를 다시 State에 저장합니다.
    state["log"] = new_log

    # 최종 State를 반환합니다.
    return state

# ============================================================
# 전체 흐름 테스트
# ============================================================

# 테스트할 질문을 준비합니다.
questions = [
    "환불은 며칠 이내인가요?",
    "대표이사 이름은?",
]

# 질문을 하나씩 실행합니다.
for question in questions:

    # 전체 흐름을 실행합니다.
    result = manual_run(question)

    # 질문과 답변을 출력합니다.
    print("Q:", question)
    print("A:", result["answer"][:80])

    # 어떤 Node가 실행되었는지 로그를 출력합니다.
    for log in result["log"]:
        print("   ", log)

    # 질문별 결과를 구분합니다.
    print("-" * 55)