run_lang2sql 단에서 파라미터 받아올 수 있도록 수정

Author: HWALIMLEE

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]:
     """

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)

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
     )