feat: setting graph input by 'user_input' state

Author: nonegom

llm_utils/agent.py

@@ -28,8 +28,6 @@
 }
 main_agent_parser = JsonOutputParser(schema=main_agent_schema)
 
-
-
 def manager_agent(state: QueryMakerState) -> dict:
     """
     가장 처음 시작하는 agent로 질문의 유무를 판단해서 적절한 Agent를 호출합니다.
@@ -44,7 +42,7 @@ def manager_agent(state: QueryMakerState) -> dict:
         state.update({
             "messages": state["messages"] + [response], # 기록용
             "intent": parsed_output.get("intent", "end"), # 분기용
-            "user_input": parsed_output.get("user_input", ""), # SQL 쿼리 변환 대상 질문
+            "user_input": parsed_output.get("user_input",state['messages'][-1].content), # SQL 쿼리 변환 대상 질문
             "intent_reason": parsed_output.get("intent_reason", "") # 분기 이유
         })
         return state

llm_utils/graph.py

@@ -59,7 +59,7 @@ def query_refiner_node(state: QueryMakerState):
 def get_table_info_node(state: QueryMakerState):
     # retriever_name과 top_n을 이용하여 검색 수행
     documents_dict = search_tables(
-        query=state["messages"][0].content,
+        query=state["user_input"],
         retriever_name=state["retriever_name"],
         top_n=state["top_n"],
         device=state["device"],
@@ -74,7 +74,7 @@ def query_maker_node(state: QueryMakerState):
     # sturctured output 사용
     res = query_maker_chain.invoke(
         input={
-            "user_input": [state["messages"][0].content],
+            "user_input": [state["user_input"]],
             "refined_input": [state["refined_input"]],
             "searched_tables": [json.dumps(state["searched_tables"])],
             "user_database_env": [state["user_database_env"]],

llm_utils/state.py

@@ -12,4 +12,8 @@ class QueryMakerState(TypedDict):
     generated_query: str
     retriever_name: str
     top_n: int
-    device: str
+    device: str
+    intent: str
+    intent_reason: str
+    user_input: str
+