7. 실습 ③ — 무한루프 체험하기

# ============================================================
# 20차시 | Edge 이해
# 교안: 20차시_Edge이해.pptx
# 역할: step 에서 다시 step 으로 가는 엣지가 끝없이 돌고, 상한에서 멈추는 예.
# 입력: count = 0
# 출력: "실행 N회차" 와 중단 예외 이름
#
#   State  = LoopState. 칸은 count 하나.
#   Node   = step. count 를 1 올리고 그 칸만 return 한다.
#   Edge   = START → step, 그리고 step → step. END 로 가는 엣지가 없다.
#   compile / invoke = 앱으로 확정한 뒤 실행한다.
#   recursion_limit = 스텝이 이 횟수를 넘으면 GraphRecursionError.
#            조건부 엣지가 없어도, 자기 자신으로 가는 엣지면 반복된다.
#
# 흐름
#   START → step → step → step → ... (END 없음)
# ============================================================

from typing import TypedDict
from langgraph.graph import StateGraph, START

# ============================================================
# 1. State 정의
# ============================================================

class LoopState(TypedDict):
    # step 노드가 실행된 횟수를 저장합니다.
    count: int

# ============================================================
# 2. Node 정의
# ============================================================

# 노드 step
#   읽기: count
#   쓰기: count
def step(state: LoopState):
    """실행 횟수를 1 증가시키는 노드입니다."""

    # 현재 실행 횟수를 1 증가시킵니다.
    count = state["count"] + 1

    # 현재 몇 번째 실행인지 출력합니다.
    print(f"실행 {count}회차")

    # 변경된 count를 State에 저장합니다.
    return {"count": count}

# ============================================================
# 3. 그래프 생성
# ============================================================

# LoopState를 사용하는 그래프를 만듭니다.
graph = StateGraph(LoopState)

# step 노드를 그래프에 등록합니다.
graph.add_node("step", step)

# ============================================================
# 4. Edge 연결
# ============================================================

# 그래프가 시작되면 step 노드를 실행합니다.
graph.add_edge(START, "step")

# step 실행 후 다시 step으로 이동하여 무한 반복합니다.
graph.add_edge("step", "step")

# ============================================================
# 5. 그래프 컴파일
# ============================================================

# compile: END 가 없는 연결도 앱으로 확정할 수 있다.
app = graph.compile()

# ============================================================
# 6. 그래프 실행
# ============================================================

try:
    # invoke 설정 recursion_limit=10. 이 횟수를 넘으면 예외가 난다.
    app.invoke(
        {"count": 0},
        {"recursion_limit": 10}
    )

except Exception as error:
    # 실행 한도를 넘으면 발생한 오류 종류를 출력합니다.
    print("중단됨:", type(error).__name__)