Remove comments from generated SQL queries
This commit is contained in:
@@ -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}")
|
||||
|
||||
Reference in New Issue
Block a user