5. 실습 ① — 노드별 예외 처리 표준화

import functools
import traceback

def safe_node(default_updates=None, name=None):
    # 노드에서 예외가 나도 그래프가 중단되지 않게 감싼다
    def deco(fn):
        node_name = name or fn.__name__

        @functools.wraps(fn)
        def wrapper(state):
            try:
                return fn(state)
            except Exception as e:
                err = f"{type(e).__name__}: {str(e)[:80]}"
                print(f"[ERROR] {node_name}: {err}")
                print(traceback.format_exc()[:400])

                updates = dict(default_updates or {})
                updates.setdefault("node_error", err)
                updates.setdefault("log",
                                   [f"{node_name} 오류: {err}"])
                return updates
        return wrapper
    return deco

from nodes.safe import safe_node

@safe_node({"documents": [], "retrieval_ok": False,
            "fail_reason": "search_error"})
def retriever_node(state):
    ...        # 기존 코드 그대로

@safe_node({"answer": "", "insufficient": True,
            "gen_error": "unknown"})
def generator_node(state):
    ...
@safe_node({"grade": "pass", "reason": "검증 불가 - 통과 처리"})
def verifier_node(state):
    ...

# ==============================================================
# 31차시. PPT 31차시_예외처리설계. 코드 폴더는 src/31
# PPT: 31차시 안전 실행과 운영 로그
# 역할: safe_node. 노드 함수를 실행하고, 예외가 나면 node_error와 log로 바꿔 반환합니다.
# 읽는 키: 호출자가 넘기는 state 전체
# 쓰는 키: 성공 시 노드의 return 그대로. 실패 시 node_error, log
# 그래프:
#   safe_node(fn, state)
#     |-- fn(state) 성공 --> 그 dict
#     +-- 예외 ----------> {node_error, log: ['노드 실행 중 오류 발생']}
# --------------------------------------------------------------
# 파이썬만 아는 분을 위한 말
# 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에 적습니다.
# 그래프 함수 시그니처는 fn(state) -> dict 입니다. safe_node는 그 함수를 한 번 감쌉니다.
# 오류 객체는 print로도 찍고, 문자열로 node_error에 넣습니다.
# 이 파일은 OpenAI와 FAISS를 부르지 않습니다.
# ==============================================================

# ============================================================
# Safe Node
# LangGraph 노드를 안전하게 실행합니다.
# 노드에서 오류가 발생해도 그래프가 중단되지 않도록 합니다.
# ============================================================

def safe_node(fn, state):
    """
    노드 함수를 실행하고,
    오류가 발생하면 오류 정보를 State로 반환합니다.
    """

    try:
        # 원래 노드를 실행합니다.
        result = fn(state)

        # 정상 실행 결과를 그대로 반환합니다.
        return result

    except Exception as e:

        # 발생한 오류를 화면에 출력합니다.
        print("[ERROR]", e)

        # 오류를 State 정보로 바꿔 반환합니다.
        return {
            "node_error": str(e),
            "log": ["노드 실행 중 오류 발생"]
        }