실습 1 — State 이해 실습 따라 하기

# ============================================================
# 18차시 | State 이해
# 교안: 18차시_State이해.pptx
# 역할: RAG State 전체 칸과, 질문 하나로 빈 노트를 만드는 함수.
# 입력: make_initial_state(question)
# 출력: 모든 키가 기본값인 딕셔너리. 이 파일을 실행해도 화면 출력은 없다.
#
#   State  = RAGState. 이후 노드가 같은 노트를 나눠 쓴다.
#   Node   = 이 파일에는 없다. make_initial_state 는 노드가 아니라
#            invoke 에 넣기 전 첫 노트를 만드는 함수다.
#   reducer = tried_queries, log.
#            초기값 tried_queries=[question] 은 이미 한 칸이 들어 있는 리스트다.
#            이후 노드가 ["다른 검색어"] 를 돌려주면 뒤에 붙는다.
#            log 는 [] 에서 시작한다.
#
# 그래프는 없다. 첫 노트의 칸만 채운다.
#
#   question ──? query, tried_queries[0]
#   나머지 칸은 빈 문자열, 빈 리스트, 0, False
# ============================================================

from typing import TypedDict, Annotated, List
import operator

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

class RAGState(TypedDict):
    """LangGraph의 모든 노드가 공유하는 State입니다."""

    # --------------------------------------------------------
    # 기본 질문
    # --------------------------------------------------------

    # 사용자가 처음 입력한 질문
    question: str

    # 실제 검색에 사용하는 질문
    query: str

    # --------------------------------------------------------
    # 29차시 - 질문 분류
    # --------------------------------------------------------

    # 질문 종류
    intent: str

    # --------------------------------------------------------
    # Retriever
    # --------------------------------------------------------

    # 검색된 문서
    documents: list

    # 검색된 문서의 유사도 점수
    scores: list

    # 검색 성공 여부
    retrieval_ok: bool

    # 검색 실패 이유
    fail_reason: str

    # --------------------------------------------------------
    # Generator
    # --------------------------------------------------------

    # 생성된 답변
    answer: str

    # 근거 부족 여부
    insufficient: bool

    # 답변에 출처가 포함되었는지 여부
    has_citation: bool

    # 답변 생성 중 발생한 오류
    gen_error: str

    # --------------------------------------------------------
    # Verifier
    # --------------------------------------------------------

    # 검증 결과
    grade: str

    # 검증 결과의 이유
    reason: str

    # --------------------------------------------------------
    # 재생성
    # --------------------------------------------------------

    # 답변 재생성 횟수
    retries: int

    # --------------------------------------------------------
    # 30차시 - Rewrite
    # --------------------------------------------------------

    # 검색어 재작성 횟수
    rewrites: int

    # 지금까지 사용한 검색어를 계속 누적
    # reducer: return {"tried_queries": ["새 검색어"]} 이면 뒤에 붙는다.
    tried_queries: Annotated[List[str], operator.add]

    # --------------------------------------------------------
    # 전체 실행 로그
    # --------------------------------------------------------

    # 각 노드의 실행 기록을 계속 누적
    # reducer: return {"log": ["한 줄"]} 이면 기존 log 뒤에 붙는다.
    log: Annotated[List[str], operator.add]

# ============================================================
# 2. 초기 State 생성
# ============================================================

# 노드가 아님. invoke 에 넘길 첫 공유 노트를 만든다.
#   읽기: 인자 question
#   쓰기: 아래 딕셔너리의 모든 키 (State를 갱신하는 return 이 아니다)
def make_initial_state(question: str) -> dict:
    """사용자 질문을 받아 처음 사용할 State를 만듭니다."""

    return {
        # 기본 질문
        "question": question,
        "query": question,

        # 질문 분류
        "intent": "",

        # Retriever
        "documents": [],
        "scores": [],
        "retrieval_ok": False,
        "fail_reason": "",

        # Generator
        "answer": "",
        "insufficient": False,
        "has_citation": False,
        "gen_error": "",

        # Verifier
        "grade": "",
        "reason": "",

        # 재생성 횟수
        "retries": 0,

        # 검색어 재작성 횟수
        "rewrites": 0,

        # 최초 검색어 기록
        "tried_queries": [question],

        # 실행 로그
        "log": [],
    }