Skip to content

Commit 89d819a

Browse files
committed
feat: Update Graph entry point to detect_language and add corresponding edge
1 parent f81b927 commit 89d819a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

llm_utils/graph.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -211,9 +211,10 @@ def query_maker_node_with_db_guide(state: QueryMakerState):
211211

212212
# StateGraph 생성 및 구성
213213
builder = StateGraph(QueryMakerState)
214-
builder.set_entry_point(QUERY_REFINER)
214+
builder.set_entry_point(DETECT_LANGUAGE)
215215

216216
# 노드 추가
217+
builder.add_node(DETECT_LANGUAGE, detect_language_regex)
217218
builder.add_node(QUERY_REFINER, query_refiner_node)
218219
builder.add_node(GET_TABLE_INFO, get_table_info_node)
219220
# builder.add_node(QUERY_MAKER, query_maker_node) # query_maker_node_with_db_guide
@@ -222,6 +223,7 @@ def query_maker_node_with_db_guide(state: QueryMakerState):
222223
) # query_maker_node_with_db_guide
223224

224225
# 기본 엣지 설정
226+
builder.add_edge(DETECT_LANGUAGE, QUERY_REFINER)
225227
builder.add_edge(QUERY_REFINER, GET_TABLE_INFO)
226228
builder.add_edge(GET_TABLE_INFO, QUERY_MAKER)
227229

0 commit comments

Comments
 (0)