From 2d67badd32470559bc582d802a3aba3dce7a7c4b Mon Sep 17 00:00:00 2001 From: Peter Stockings Date: Sat, 5 Apr 2025 21:51:56 +1100 Subject: [PATCH] Remove comments from generated SQL queries --- routes/sql_explorer.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) 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}")