init

agent/conversation.py

@@ -19,11 +19,13 @@ class ConversationAgent(BaseAgent):
 
         workflow.add_node("analyze_intent", self._analyze_intent)
         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_node("update_context", self._update_context)
 
         workflow.add_edge("analyze_intent", "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", "update_context")
         workflow.add_edge("update_context", END)
 
@@ -61,6 +63,11 @@ class ConversationAgent(BaseAgent):
         state.current_step = "response_generated"
         return state
 
+    def _generate_sql(self, state: AgentState) -> AgentState:
+        """生成 SQL"""
+        from . import nodes
+        return nodes.generate_sql(state, self.model)
+
     def _update_context(self, state: AgentState) -> AgentState:
         """更新对话上下文与历史"""
         for message in state.messages: