그래프 빌더 기능 추가 #134

- Streamlit에서 LangGraph 워크플로우를 구성하고 세션에 적용하는 페이지 구현.
- 프리셋 및 커스텀 옵션을 통해 노드 시퀀스를 설정하고, QUERY_MAKER 포함 여부에 따라 그래프를 동적으로 생성하도록 수정
- 관련 UI 요소 및 세션 상태 관리 기능 추가.

Author: ehddnr301

interface/graph_builder.py

@@ -0,0 +1,180 @@
+"""
+LangGraph 워크플로우를 Streamlit에서 구성하고 세션에 적용하는 페이지.
+
+기능 개요:
+- 프리셋(기본/확장) 또는 커스텀 토글로 노드 시퀀스를 구성
+- QUERY_MAKER 포함 여부를 토글하여 마지막 노드를 제어
+- 선택이 바뀌면 즉시 컴파일된 그래프를 세션 상태에 반영
+- 현재 적용된 그래프 설정을 확인 가능
+"""
+
+from typing import List
+
+import streamlit as st
+from langgraph.graph import StateGraph, END
+
+from llm_utils.graph_utils.base import (
+    QueryMakerState,
+    GET_TABLE_INFO,
+    PROFILE_EXTRACTION,
+    CONTEXT_ENRICHMENT,
+    QUERY_MAKER,
+    get_table_info_node,
+    profile_extraction_node,
+    context_enrichment_node,
+    query_maker_node,
+)
+
+
+def build_selected_sequence(preset: str, use_profile: bool, use_context: bool) -> List[str]:
+    """
+    프리셋과 커스텀 토글에 따라 실행할 노드 시퀀스를 생성합니다.
+
+    Args:
+        preset (str): "기본" | "확장" | "커스텀" 중 하나
+        use_profile (bool): 커스텀에서 PROFILE_EXTRACTION 포함 여부
+        use_context (bool): 커스텀에서 CONTEXT_ENRICHMENT 포함 여부
+
+    Returns:
+        List[str]: 노드 식별자들의 실행 순서
+    """
+    sequence: List[str] = [GET_TABLE_INFO]
+
+    if preset == "기본":
+        sequence += [QUERY_MAKER]
+    elif preset == "확장":
+        sequence += [PROFILE_EXTRACTION, CONTEXT_ENRICHMENT, QUERY_MAKER]
+    else:
+        if use_profile:
+            sequence.append(PROFILE_EXTRACTION)
+        if use_context:
+            sequence.append(CONTEXT_ENRICHMENT)
+        sequence.append(QUERY_MAKER)
+
+    return sequence
+
+
+def build_state_graph(sequence: List[str]) -> StateGraph:
+    """
+    주어진 시퀀스대로 노드를 추가하고, 인접 노드 간 엣지를 연결한 그래프 빌더를 반환합니다.
+
+    마지막 노드는 항상 END로 연결합니다.
+
+    Args:
+        sequence (List[str]): 실행 순서에 따른 노드 식별자 목록
+
+    Returns:
+        StateGraph: 컴파일 전 그래프 빌더 객체
+    """
+    builder = StateGraph(QueryMakerState)
+    builder.set_entry_point(GET_TABLE_INFO)
+
+    # 노드 등록
+    for node_id in sequence:
+        if node_id == GET_TABLE_INFO:
+            builder.add_node(GET_TABLE_INFO, get_table_info_node)
+        elif node_id == PROFILE_EXTRACTION:
+            builder.add_node(PROFILE_EXTRACTION, profile_extraction_node)
+        elif node_id == CONTEXT_ENRICHMENT:
+            builder.add_node(CONTEXT_ENRICHMENT, context_enrichment_node)
+        elif node_id == QUERY_MAKER:
+            builder.add_node(QUERY_MAKER, query_maker_node)
+
+    # 엣지 연결
+    for i in range(len(sequence) - 1):
+        builder.add_edge(sequence[i], sequence[i + 1])
+
+    # 종료 연결: 마지막 노드가 무엇이든 END로 연결
+    if len(sequence) > 0:
+        builder.add_edge(sequence[-1], END)
+
+    return builder
+
+
+def render_sequence(sequence: List[str]) -> str:
+    """
+    노드 시퀀스를 사람이 읽기 쉬운 문자열로 변환합니다.
+
+    Args:
+        sequence (List[str]): 실행 순서에 따른 노드 식별자 목록
+
+    Returns:
+        str: 예) "GET_TABLE_INFO → PROFILE_EXTRACTION → ..."
+    """
+    label_map = {
+        GET_TABLE_INFO: "GET_TABLE_INFO",
+        PROFILE_EXTRACTION: "PROFILE_EXTRACTION",
+        CONTEXT_ENRICHMENT: "CONTEXT_ENRICHMENT",
+        QUERY_MAKER: "QUERY_MAKER",
+    }
+    return " → ".join(label_map[s] for s in sequence)
+
+
+ 
+
+
+st.title("LangGraph 구성 UI")
+st.caption("기본/확장/커스텀으로 StateGraph를 구성하고 세션에 적용합니다.")
+
+preset = st.radio("프리셋 선택", ("기본", "확장", "커스텀"), horizontal=True)
+
+use_profile = False
+use_context = False
+if preset == "커스텀":
+    st.subheader("커스텀 옵션")
+    use_profile = st.checkbox("PROFILE_EXTRACTION 포함", value=True)
+    use_context = st.checkbox("CONTEXT_ENRICHMENT 포함", value=True)
+    use_query_maker = st.checkbox("QUERY_MAKER 포함", value=True)
+else:
+    # 프리셋에서는 QUERY_MAKER 자동 포함
+    use_query_maker = True
+
+def build_sequence_with_qm(preset: str, use_profile: bool, use_context: bool, use_qm: bool) -> List[str]:
+    """
+    QUERY_MAKER 포함 여부를 반영하여 시퀀스를 생성합니다.
+
+    - use_qm=False면 마지막 노드는 반드시 GET_TABLE_INFO입니다.
+    - use_qm=True면 프리셋/커스텀 로직에 따라 마지막 노드는 QUERY_MAKER가 됩니다.
+
+    Args:
+        preset (str): "기본" | "확장" | "커스텀" 중 하나
+        use_profile (bool): PROFILE_EXTRACTION 포함 여부(커스텀 전용)
+        use_context (bool): CONTEXT_ENRICHMENT 포함 여부(커스텀 전용)
+        use_qm (bool): QUERY_MAKER 포함 여부
+
+    Returns:
+        List[str]: 노드 식별자들의 실행 순서
+    """
+    # QUERY_MAKER가 비활성화되면 마지막 노드는 반드시 GET_TABLE_INFO
+    if not use_qm:
+        return [GET_TABLE_INFO]
+    # 활성화된 경우 프리셋/커스텀 구성에 따라 마지막 노드는 QUERY_MAKER
+    base_seq = build_selected_sequence(preset, use_profile, use_context)
+    return base_seq
+
+sequence = build_sequence_with_qm(preset, use_profile, use_context, use_query_maker)
+
+st.subheader("실행 순서")
+st.write(render_sequence(sequence))
+
+st.subheader("그래프 생성")
+config = {"preset": preset, "use_profile": use_profile, "use_context": use_context, "use_query_maker": use_query_maker}
+
+# 선택이 바뀌면 자동으로 세션 그래프 갱신
+prev_config = st.session_state.get("graph_config")
+if ("graph" not in st.session_state) or (prev_config != config):
+    _builder = build_state_graph(sequence)
+    st.session_state["graph"] = _builder.compile()
+    st.session_state["graph_config"] = config
+    st.info("그래프가 세션에 적용되었습니다.")
+
+# 수동 새로고침 버튼
+if st.button("세션 그래프 새로고침"):
+    _builder = build_state_graph(sequence)
+    st.session_state["graph"] = _builder.compile()
+    st.session_state["graph_config"] = config
+    st.success("세션 그래프가 새로고침되었습니다.")
+
+with st.expander("현재 세션 그래프 설정"):
+    st.json(st.session_state.get("graph_config", {}))
+

interface/lang2sql.py

@@ -98,6 +98,14 @@ def display_result(
     def should_show(_key: str) -> bool:
         return st.session_state.get(_key, True)
 
+    has_query = bool(res.get("generated_query"))
+    # 섹션 표시 여부를 QUERY_MAKER 출력 유무에 따라 제어
+    show_sql_section = has_query and should_show("show_sql")
+    show_result_desc = has_query and should_show("show_result_description")
+    show_reinterpreted = has_query and should_show("show_question_reinterpreted_by_ai")
+    show_table_section = has_query and should_show("show_table")
+    show_chart_section = has_query and should_show("show_chart")
+
     if should_show("show_token_usage"):
         st.markdown("---")
         token_summary = TokenUtils.get_token_usage_summary(data=res["messages"])
@@ -110,7 +118,7 @@ def should_show(_key: str) -> bool:
         """
         )
 
-    if should_show("show_sql"):
+    if show_sql_section:
         st.markdown("---")
         generated_query = res.get("generated_query")
         if generated_query:
@@ -138,7 +146,7 @@ def should_show(_key: str) -> bool:
                 st.warning("쿼리 텍스트가 문자열이 아닙니다.")
                 st.text(str(query_text))
 
-    if should_show("show_result_description"):
+    if show_result_desc:
         st.markdown("---")
         st.markdown("**결과 설명:**")
         result_message = res["messages"][-1].content
@@ -158,7 +166,7 @@ def should_show(_key: str) -> bool:
             st.warning("결과 메시지가 문자열이 아닙니다.")
             st.text(str(result_message))
 
-    if should_show("show_question_reinterpreted_by_ai"):
+    if show_reinterpreted:
         st.markdown("---")
         st.markdown("**AI가 재해석한 사용자 질문:**")
         try:
@@ -178,7 +186,11 @@ def should_show(_key: str) -> bool:
         st.markdown("**참고한 테이블 목록:**")
         st.write(res.get("searched_tables", []))
 
-    if should_show("show_table"):
+    # QUERY_MAKER가 비활성화된 경우 안내 메시지 출력
+    if not has_query:
+        st.info("QUERY_MAKER 없이 실행되었습니다. 검색된 테이블 정보만 표시합니다.")
+
+    if show_table_section:
         st.markdown("---")
         try:
             sql_raw = (
@@ -195,7 +207,7 @@ def should_show(_key: str) -> bool:
         except Exception as e:
             st.error(f"쿼리 실행 중 오류 발생: {e}")
 
-    if should_show("show_chart"):
+    if show_chart_section:
         st.markdown("---")
         try:
             sql_raw = (

interface/streamlit_app.py

@@ -11,7 +11,11 @@
     "lang2sql": {
         "page": "lang2sql.py",
         "title": "Lang2SQL",
-    }
+    },
+    "graph_builder": {
+        "page": "graph_builder.py",
+        "title": "Graph Builder",
+    },
 }