Spaces:
Sleeping
Sleeping
Update duckdb-nsql/eval/prompt_formatters.py
Browse files
duckdb-nsql/eval/prompt_formatters.py
CHANGED
|
@@ -66,6 +66,7 @@ class RajkumarFormatter:
|
|
| 66 |
def format_model_output(cls, output_sql: str, prompt: str) -> str:
|
| 67 |
"""Format model output."""
|
| 68 |
clean_sql = (output_sql
|
|
|
|
| 69 |
.replace('```sql\n', '')
|
| 70 |
.replace('```duckdb\n', '')
|
| 71 |
.replace('```\n', '')
|
|
|
|
| 66 |
def format_model_output(cls, output_sql: str, prompt: str) -> str:
|
| 67 |
"""Format model output."""
|
| 68 |
clean_sql = (output_sql
|
| 69 |
+
.replace('sql\n', '')
|
| 70 |
.replace('```sql\n', '')
|
| 71 |
.replace('```duckdb\n', '')
|
| 72 |
.replace('```\n', '')
|