Update duckdb-nsql/eval/prompt_formatters.py
Browse files
duckdb-nsql/eval/prompt_formatters.py
CHANGED
|
@@ -65,7 +65,6 @@ class RajkumarFormatter:
|
|
| 65 |
|
| 66 |
@classmethod
|
| 67 |
def format_model_output(cls, output_sql: str, prompt: str) -> str:
|
| 68 |
-
print(output_sql)
|
| 69 |
pattern = r"```(?:sql|mysql|duckdb)?\n?(.*?)```"
|
| 70 |
match = re.search(pattern, output_sql, re.DOTALL)
|
| 71 |
sql = match.group(1).strip() if match else output_sql.strip()
|
|
@@ -74,7 +73,6 @@ class RajkumarFormatter:
|
|
| 74 |
if not sql:
|
| 75 |
sql = output_sql.strip()
|
| 76 |
|
| 77 |
-
print(sql)
|
| 78 |
# Ensure single trailing semicolon
|
| 79 |
return sql.rstrip(';') + ';'
|
| 80 |
|
|
|
|
| 65 |
|
| 66 |
@classmethod
|
| 67 |
def format_model_output(cls, output_sql: str, prompt: str) -> str:
|
|
|
|
| 68 |
pattern = r"```(?:sql|mysql|duckdb)?\n?(.*?)```"
|
| 69 |
match = re.search(pattern, output_sql, re.DOTALL)
|
| 70 |
sql = match.group(1).strip() if match else output_sql.strip()
|
|
|
|
| 73 |
if not sql:
|
| 74 |
sql = output_sql.strip()
|
| 75 |
|
|
|
|
| 76 |
# Ensure single trailing semicolon
|
| 77 |
return sql.rstrip(';') + ';'
|
| 78 |
|