Method query

agents/sql_agent.py:180–235

执行查询，失败时自动纠错并重试（Reflection 循环） 流程：SQL生成 → 执行 → [失败] → 错误反馈给LLM → 重新生成 → 最多重试 max_retries 次 Args: question: 用户问题 max_retries: 最大重试次数（默认3次） Returns: { "sql": 最终执行的SQL, "data": 查询结

(self, question: str, max_retries: int = 3)

            return asyncio.run(coro)
    
    def query(self, question: str, max_retries: int = 3) -> Dict[str, Any]:
        """执行查询，失败时自动纠错并重试（Reflection 循环）
        
        流程：SQL生成 → 执行 → [失败] → 错误反馈给LLM → 重新生成 → 最多重试 max_retries 次
        
        Args:
            question: 用户问题
            max_retries: 最大重试次数（默认3次）
            
        Returns:
            {
                "sql": 最终执行的SQL,
                "data": 查询结果JSON字符串（成功时）,
                "error": 错误信息（成功时为None）,
                "retry_count": 实际重试次数（0表示首次成功）
            }
        """
        result = {
            "sql": None,
            "data": None,
            "error": None,
            "retry_count": 0
        }
        
        try:
            sql = self._generate_sql(question)
            result["sql"] = sql
            
            if not sql:
                result["error"] = "未能生成有效的SQL"
                return result
            
            for attempt in range(max_retries):
                query_result = self._run_async(self._execute_sql_via_mcp(sql))
                result_data = json.loads(query_result)
                
                if isinstance(result_data, dict) and "error" in result_data:
                    error_msg = result_data["error"]
                    
                    if attempt < max_retries - 1:
                        print(f"[SQL纠错] 第{attempt + 1}次执行失败: {error_msg}，正在让LLM自动修复...")
                        sql = self._correct_sql(question, sql, error_msg, attempt + 1)
                        result["sql"] = sql
                        result["retry_count"] = attempt + 1
                    else:
                        result["error"] = f"SQL执行失败（已自动重试{attempt}次）: {error_msg}"
                else:
                    result["data"] = query_result
                    if attempt > 0:
                        print(f"[SQL纠错] 第{attempt}次修复后执行成功")
                    break
                    
        except Exception as e:
            result["error"] = f"查询失败: {str(e)}"
        
        return result