peterstockings/workout
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove comments from generated SQL queries

Browse Source
This commit is contained in:
Peter Stockings
2025-04-05 21:51:56 +11:00
parent 64dda01af6
commit 2d67badd32
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
routes/sql_explorer.py
View File

@@ -216,7 +216,12 @@ Return ONLY the SQL query, without any explanation or surrounding text/markdown.
if generated_sql.endswith("```"): if generated_sql.endswith("```"):
generated_sql = generated_sql[:-3] 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: except requests.exceptions.RequestException as e:
current_app.logger.error(f"Gemini API request error: {e}") current_app.logger.error(f"Gemini API request error: {e}")