init

agent/graph.py

@@ -20,10 +20,12 @@ class BaseAgent:
 
         workflow.add_node("process_input", nodes.process_input)
         workflow.add_node("normalize_input", self._normalize_input)
+        workflow.add_node("generate_sql", self._generate_sql)
         workflow.add_node("generate_response", self._generate_response)
 
         workflow.add_edge("process_input", "normalize_input")
-        workflow.add_edge("normalize_input", "generate_response")
+        workflow.add_edge("normalize_input", "generate_sql")
+        workflow.add_edge("generate_sql", "generate_response")
         workflow.add_edge("generate_response", END)
 
         workflow.set_entry_point("process_input")
@@ -38,6 +40,10 @@ class BaseAgent:
         """规范化用户输入"""
         return nodes.normalize_input(state, self.model)
 
+    def _generate_sql(self, state: AgentState) -> AgentState:
+        """生成 SQL"""
+        return nodes.generate_sql(state, self.model)
+
     def run(self, user_input: str, **kwargs) -> Dict[str, Any]:
         """运行代理并处理用户输入"""
         initial_state = AgentState(