Skip to content

Commit e540875

Browse files
authored
Merge pull request #197 from CausalInferenceLab/194-graph-preset-설정반영
run_lang2sql 단에서 파라미터 받아올 수 있도록 수정
2 parents 9ca985c + 871dae8 commit e540875

File tree

3 files changed

+3
-1
lines changed

3 files changed

+3
-1
lines changed

engine/query_executor.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,6 @@ def execute_query(
4949
- "messages": 전체 LLM 응답 메시지 목록
5050
- "searched_tables": 참조된 테이블 목록 등 추가 정보
5151
"""
52-
5352
logger.info("Processing query: %s", query)
5453

5554
# 그래프 선택

interface/app_pages/lang2sql.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -142,5 +142,6 @@
142142
retriever_name=user_retriever,
143143
top_n=user_top_n,
144144
device=device,
145+
use_enriched=use_enriched,
145146
)
146147
display_result(res=res)

interface/core/lang2sql_runner.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ def run_lang2sql(
1616
retriever_name,
1717
top_n,
1818
device,
19+
use_enriched,
1920
):
2021
"""
2122
Lang2SQL 실행 함수.
@@ -40,4 +41,5 @@ def run_lang2sql(
4041
retriever_name=retriever_name,
4142
top_n=top_n,
4243
device=device,
44+
use_enriched_graph=use_enriched,
4345
)

0 commit comments

Comments
 (0)