diff --git a/routes/sql_explorer.py b/routes/sql_explorer.py index af6e377..a82e813 100644 --- a/routes/sql_explorer.py +++ b/routes/sql_explorer.py @@ -216,7 +216,12 @@ Return ONLY the SQL query, without any explanation or surrounding text/markdown. if generated_sql.endswith("```"): generated_sql = generated_sql[:-3] - return generated_sql.strip(), None + # Remove leading SQL comment lines + sql_lines = generated_sql.strip().splitlines() + filtered_lines = [line for line in sql_lines if not line.strip().startswith('--')] + final_sql = "\n".join(filtered_lines).strip() + + return final_sql, None except requests.exceptions.RequestException as e: current_app.logger.error(f"Gemini API request error: {e}")