From f7b87840d6f39be61ae9dfaedaba0ca58a0fe075 Mon Sep 17 00:00:00 2001 From: HWALIM Date: Sun, 12 Oct 2025 22:01:44 +0900 Subject: [PATCH 1/4] =?UTF-8?q?graph=5Fconfig=20=EB=B3=80=EC=88=98=20?= =?UTF-8?q?=EA=B0=92=20=EB=B0=9B=EC=95=84=EC=98=AC=20=EC=88=98=20=EC=9E=88?= =?UTF-8?q?=EB=8F=84=EB=A1=9D=20=EC=88=98=EC=A0=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- engine/query_executor.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/engine/query_executor.py b/engine/query_executor.py index 7cba03a..41e22ae 100644 --- a/engine/query_executor.py +++ b/engine/query_executor.py @@ -14,6 +14,8 @@ from utils.llm.graph_utils.enriched_graph import builder as enriched_builder from utils.llm.llm_response_parser import LLMResponseParser +import streamlit as st + logger = logging.getLogger(__name__) @@ -24,7 +26,7 @@ def execute_query( retriever_name: str = "기본", top_n: int = 5, device: str = "cpu", - use_enriched_graph: bool = False, + use_enriched_graph: bool = st.session_state.get('graph_config', {}).get('preset') not in (None, '기본'), session_state: Optional[Union[Dict[str, Any], Any]] = None, ) -> Dict[str, Any]: """ @@ -49,7 +51,6 @@ def execute_query( - "messages": 전체 LLM 응답 메시지 목록 - "searched_tables": 참조된 테이블 목록 등 추가 정보 """ - logger.info("Processing query: %s", query) # 그래프 선택 @@ -73,6 +74,7 @@ def execute_query( # CLI 환경: 매번 새로운 그래프 컴파일 graph = graph_builder.compile() + # 그래프 실행 res = graph.invoke( input={ From ccf36719fdd37a7229deffd889504a9d0daf0265 Mon Sep 17 00:00:00 2001 From: HWALIM Date: Mon, 20 Oct 2025 19:57:54 +0900 Subject: [PATCH 2/4] =?UTF-8?q?run=5Flang2sql=20=EB=8B=A8=EC=97=90?= =?UTF-8?q?=EC=84=9C=20=ED=8C=8C=EB=9D=BC=EB=AF=B8=ED=84=B0=20=EB=B0=9B?= =?UTF-8?q?=EC=95=84=EC=98=AC=20=EC=88=98=20=EC=9E=88=EB=8F=84=EB=A1=9D=20?= =?UTF-8?q?=EC=88=98=EC=A0=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- engine/query_executor.py | 4 +--- interface/app_pages/lang2sql.py | 1 + interface/core/lang2sql_runner.py | 2 ++ 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/engine/query_executor.py b/engine/query_executor.py index 41e22ae..bf8e383 100644 --- a/engine/query_executor.py +++ b/engine/query_executor.py @@ -14,8 +14,6 @@ from utils.llm.graph_utils.enriched_graph import builder as enriched_builder from utils.llm.llm_response_parser import LLMResponseParser -import streamlit as st - logger = logging.getLogger(__name__) @@ -26,7 +24,7 @@ def execute_query( retriever_name: str = "기본", top_n: int = 5, device: str = "cpu", - use_enriched_graph: bool = st.session_state.get('graph_config', {}).get('preset') not in (None, '기본'), + use_enriched_graph: bool = False, session_state: Optional[Union[Dict[str, Any], Any]] = None, ) -> Dict[str, Any]: """ diff --git a/interface/app_pages/lang2sql.py b/interface/app_pages/lang2sql.py index 0c96156..e7eb5f1 100644 --- a/interface/app_pages/lang2sql.py +++ b/interface/app_pages/lang2sql.py @@ -142,5 +142,6 @@ retriever_name=user_retriever, top_n=user_top_n, device=device, + use_enriched=use_enriched ) display_result(res=res) diff --git a/interface/core/lang2sql_runner.py b/interface/core/lang2sql_runner.py index da46cc6..3ee1cec 100644 --- a/interface/core/lang2sql_runner.py +++ b/interface/core/lang2sql_runner.py @@ -16,6 +16,7 @@ def run_lang2sql( retriever_name, top_n, device, + use_enriched ): """ Lang2SQL 실행 함수. @@ -40,4 +41,5 @@ def run_lang2sql( retriever_name=retriever_name, top_n=top_n, device=device, + use_enriched_graph=use_enriched ) From daa8228f44e69f64fc0e9e57162434bc9c423800 Mon Sep 17 00:00:00 2001 From: HWALIM Date: Mon, 20 Oct 2025 20:25:22 +0900 Subject: [PATCH 3/4] format resolve --- interface/app_pages/lang2sql.py | 2 +- interface/core/lang2sql_runner.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/interface/app_pages/lang2sql.py b/interface/app_pages/lang2sql.py index e7eb5f1..10d3466 100644 --- a/interface/app_pages/lang2sql.py +++ b/interface/app_pages/lang2sql.py @@ -142,6 +142,6 @@ retriever_name=user_retriever, top_n=user_top_n, device=device, - use_enriched=use_enriched + use_enriched=use_enriched, ) display_result(res=res) diff --git a/interface/core/lang2sql_runner.py b/interface/core/lang2sql_runner.py index 3ee1cec..f37bd4a 100644 --- a/interface/core/lang2sql_runner.py +++ b/interface/core/lang2sql_runner.py @@ -16,7 +16,7 @@ def run_lang2sql( retriever_name, top_n, device, - use_enriched + use_enriched, ): """ Lang2SQL 실행 함수. @@ -41,5 +41,5 @@ def run_lang2sql( retriever_name=retriever_name, top_n=top_n, device=device, - use_enriched_graph=use_enriched + use_enriched_graph=use_enriched, ) From 871dae8a4b025fed0c9b80538a32d645c31ba56b Mon Sep 17 00:00:00 2001 From: HWALIM Date: Mon, 20 Oct 2025 20:29:19 +0900 Subject: [PATCH 4/4] format resolve --- engine/query_executor.py | 1 - 1 file changed, 1 deletion(-) diff --git a/engine/query_executor.py b/engine/query_executor.py index bf8e383..01f5b7b 100644 --- a/engine/query_executor.py +++ b/engine/query_executor.py @@ -72,7 +72,6 @@ def execute_query( # CLI 환경: 매번 새로운 그래프 컴파일 graph = graph_builder.compile() - # 그래프 실행 res = graph.invoke( input={