8. 실습 ④ — 장애 시뮬레이션

# ==============================================================
# 31차시. PPT 31차시_예외처리설계. 코드 폴더는 src/31
# PPT: 31차시 안전 실행과 운영 로그
# 역할: retriever_node 자리에 항상 예외를 내는 함수를 넣어, safe_node가 그래프를 살리는지 봅니다.
# 읽는 키: question
# 쓰는 키: graph31.ask의 반환 중 node_error를 출력
# 그래프:
#   classify --> retrieve(일부러 오류) --> safe_node가 node_error 기록 --> 이후 분기
# --------------------------------------------------------------
# 파이썬만 아는 분을 위한 말
# 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에 적습니다.
# unittest.mock.patch로 graph31.retriever_node 이름만 잠깐 바꿉니다. 원본 파일은 수정하지 않습니다.
# broken_node는 RuntimeError('일부러 발생시킨 오류')를 일으킵니다.
# 여기까지 코드가 진행되면 print('시스템 생존')이 실행됩니다.
# graph31 import가 FAISS와 API 준비를 하므로 이 파일은 실행하지 않았습니다.
# ==============================================================

# ============================================================
# Safe Node 장애 테스트
# Retriever에서 일부러 오류를 발생시킵니다.
# 오류가 발생해도 전체 그래프가 중단되지 않는지 확인합니다.
# ============================================================

from unittest.mock import patch

import graph31

# ============================================================
# 테스트용 오류 노드
# 실행하면 무조건 오류를 발생시킵니다.
# ============================================================

def broken_node(state):

    # 실제 장애 상황을 가정하여 오류를 발생시킵니다.
    raise RuntimeError(
        "일부러 발생시킨 오류"
    )

# ============================================================
# Retriever 장애 테스트
# ============================================================

# graph31의 retriever_node를 broken_node로 잠시 교체합니다.
with patch(
    "graph31.retriever_node",
    broken_node
):

    # 평소처럼 질문을 실행합니다.
    result = graph31.ask(
        "환불은 며칠 이내인가요?"
    )

    # --------------------------------------------------------
    # 실행 결과 확인
    # --------------------------------------------------------

    # 여기까지 실행되면 그래프가 중단되지 않은 것입니다.
    print("시스템 생존")

    # safe_node가 잡은 오류 내용을 출력합니다.
    print(
        "오류:",
        result["node_error"]
    )