Remove comments from generated SQL queries

This commit is contained in:
Peter Stockings
2025-04-05 21:51:56 +11:00
parent 64dda01af6
commit 2d67badd32

View File

@@ -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}")