Sweep: Execute dbt SQL fails if the query starts with a Jinga comment

[sweep-ai]

Description

This pull request addresses an issue where executing dbt SQL statements that begin with a Jinja comment causes failures. It introduces a fix by stripping leading Jinja comments from the SQL statement before execution.

Summary

• Added an import for the re module in dbt_core_integration.py to utilize regular expressions for processing SQL statements.
• Implemented a new step in the execute_sql method to remove Jinja comments from the start of the SQL statement. This ensures that SQL queries starting with Jinja comments are executed without errors.
• The change is made within the execute_sql function of the dbt_core_integration.py file, enhancing its robustness in handling SQL queries with Jinja syntax.

Fixes #1156.

---

💡 To get Sweep to edit this pull request, you can:

• Comment below, and Sweep can edit the entire PR
• Comment on a file, Sweep will only modify the commented file
• Edit the original issue to get Sweep to recreate the PR from scratch

This is an automated message generated by Sweep AI.

[sweep-ai]

Rollback Files For Sweep

• Rollback changes to dbt_core_integration.py

This is an automated message generated by Sweep AI.

[sweep-ai]

Sweep: PR Review

Authors of pull request: @sweep-ai[bot]

dbt_core_integration.py

Added the re module import and implemented a regex substitution to strip leading Jinja comments in the execute_sql method.

Sweep Found These Issues

• The regex substitution to strip leading Jinja comments may inadvertently remove valid SQL content if the regex pattern is not precise enough.

vscode-dbt-power-user/dbt_core_integration.py

Lines 620 to 625 in e73ab7f

	def execute_sql(self, raw_sql: str) -> DbtAdapterExecutionResult:
	"""Execute dbt SQL statement against database"""
	with self.adapter.connection_named("master"):
	# Strip leading Jinja comments
	raw_sql = re.sub(r'^{#.*#}\s*', '', raw_sql, flags=re.MULTILINE)


View Diff

---